flawless_utils/
server.rs

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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
use flawless::workflow::WorkflowTypeAlias;
use reqwest::{
    multipart::{Form, Part},
    Client,
};
use serde::{Deserialize, Serialize};

use crate::WorkflowRef;

/// Represents an instance of a flawless server.
#[derive(Debug, Clone, Deserialize)]
pub struct Server {
    pub(crate) url: String,
    pub(crate) auth_token: Option<String>,
}

impl Server {
    /// Creates a new server.
    pub fn new<A: ToString>(url: A, auth_token: Option<&str>) -> Self {
        Server { url: url.to_string(), auth_token: auth_token.map(|t| t.to_string()) }
    }

    /// Deploys a flawless module.
    pub async fn deploy(&self, module: &'static [u8]) -> Result<DeployedModule, String> {
        let deploy_url = format!("{}/api/module/deploy", self.url);

        let client = Client::new();
        let module = Part::stream(module).mime_str("application/wasm").unwrap();
        let form = Form::new().part("module", module);

        let mut request = client.post(deploy_url).multipart(form);

        if let Some(auth_token) = &self.auth_token {
            request = request.header("Authorization", &format!("Bearer {}", auth_token));
        }

        match request.send().await {
            Ok(response) => {
                let response = response.text().await.map_err(|_err| format!("Unable to read response"))?;
                let response: DeployResponse = serde_json::from_str(&response)
                    .map_err(|err| format!("Failed to parse flawless response as json. {err}"))?;
                match response {
                    DeployResponse::Ok(mut deployed_module) => {
                        deployed_module.server = Some(self.clone());
                        Ok(deployed_module)
                    }
                    DeployResponse::Error(err) => Err(err),
                }
            }
            Err(err) => Err(format!("Failed to send request to `{}`. {err}", self.url)),
        }
    }
}

#[derive(Debug, Deserialize)]
pub enum DeployResponse {
    #[serde(rename = "ok")]
    Ok(DeployedModule),
    #[serde(rename = "error")]
    Error(String),
}

#[derive(Debug, Deserialize)]
pub struct DeployedModule {
    pub(crate) server: Option<Server>,
    name: String,
    version: String,
}

impl DeployedModule {
    /// Start workflow using a `WorkflowTypeAlias`.
    ///
    /// The `#[workflow]` macro will automatically create a type alias for a workflow function. The type alias
    /// will have the same name and visibility as the workflow function.
    ///
    /// # Example
    ///
    /// ```rust,ignore
    /// let workflow = module.start<workflow_function>(arg).await.unwrap();
    /// ```
    pub async fn start<W>(&self, input: W::InputArg) -> Result<WorkflowRef, String>
    where
        W: WorkflowTypeAlias,
    {
        self.start_by_name(W::NAME, input).await
    }

    /// Start workflow by name.
    pub async fn start_by_name<T>(&self, name: &str, input: T) -> Result<WorkflowRef, String>
    where
        T: Serialize,
    {
        let start_url = format!("{}/api/workflow/start", self.server.as_ref().unwrap().url);
        let input = serde_json::to_string(&input).map_err(|err| err.to_string())?;
        let workflow_start = WorkflowStart {
            module: self.name.to_string(),
            version: self.version.to_string(),
            workflow: name.to_string(),
            input,
        };

        let client = Client::new();
        let mut request = client.post(start_url).json(&workflow_start);

        if let Some(auth_token) = &self.server.as_ref().unwrap().auth_token {
            request = request.header("Authorization", &format!("Bearer {}", auth_token));
        }

        match request.send().await {
            Ok(response) => {
                let response = response.text().await.map_err(|_err| format!("Unable to read response"))?;
                let response: WorkflowStartResponse = serde_json::from_str(&response)
                    .map_err(|err| format!("Failed to parse flawless response as json. {err}"))?;
                match response {
                    WorkflowStartResponse::Ok(mut workflow) => {
                        workflow.server = self.server.clone();
                        Ok(workflow)
                    }
                    WorkflowStartResponse::Error(err) => Err(err),
                }
            }
            Err(err) => {
                Err(format!("Failed to send request to `{}`. {err}", self.server.as_ref().unwrap().url))
            }
        }
    }
}

#[derive(Debug, Serialize)]
struct WorkflowStart<T> {
    module: String,
    version: String,
    workflow: String,
    input: T,
}

#[derive(Debug, Deserialize)]
enum WorkflowStartResponse {
    #[serde(rename = "ok")]
    Ok(WorkflowRef),
    #[serde(rename = "error")]
    Error(String),
}