1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use crate::{registry, GQLInputValue, GQLType, Value};
use std::borrow::Cow;

/// Upload file type
///
/// Reference: https://github.com/jaydenseric/graphql-multipart-request-spec
pub struct Upload {
    pub filename: String,
    pub content_type: Option<String>,
    pub content: Vec<u8>,
}

impl<'a> GQLType for Upload {
    fn type_name() -> Cow<'static, str> {
        Cow::Borrowed("Upload")
    }

    fn create_type_info(registry: &mut registry::Registry) -> String {
        registry.create_type::<Self, _>(|_| registry::Type::Scalar {
            name: Self::type_name().to_string(),
            description: None,
            is_valid: |value| match value {
                Value::String(s) => s.starts_with("file:"),
                _ => false,
            },
        })
    }
}

impl<'a> GQLInputValue for Upload {
    fn parse(value: &Value) -> Option<Self> {
        if let Value::String(s) = value {
            if s.starts_with("file:") {
                let s = &s[5..];
                if let Some(idx) = s.find("|") {
                    let name_and_type = &s[..idx];
                    let content = &s[idx + 1..];
                    if let Some(type_idx) = name_and_type.find(":") {
                        let name = &name_and_type[..type_idx];
                        let mime_type = &name_and_type[type_idx + 1..];
                        return Some(Self {
                            filename: name.to_string(),
                            content_type: Some(mime_type.to_string()),
                            content: content.as_bytes().to_vec(),
                        });
                    } else {
                        return Some(Self {
                            filename: name_and_type.to_string(),
                            content_type: None,
                            content: content.as_bytes().to_vec(),
                        });
                    }
                }
            }
        }
        None
    }
}