omuchat_tauri_plugin_server/
lib.rs

1// Copyright 2019-2021 Tauri Programme within The Commons Conservancy
2// SPDX-License-Identifier: Apache-2.0
3// SPDX-License-Identifier: MIT
4
5use std::collections::HashMap;
6
7use http::Uri;
8use tauri::{
9    plugin::{Builder as PluginBuilder, TauriPlugin},
10    Runtime,
11};
12use tiny_http::{Header, Response as HttpResponse, Server};
13
14pub struct Request {
15    url: String,
16}
17
18impl Request {
19    pub fn url(&self) -> &str {
20        &self.url
21    }
22}
23
24pub struct Response {
25    headers: HashMap<String, String>,
26}
27
28impl Response {
29    pub fn add_header<H: Into<String>, V: Into<String>>(&mut self, header: H, value: V) {
30        self.headers.insert(header.into(), value.into());
31    }
32}
33
34type OnRequest = Option<Box<dyn Fn(&Request, &mut Response) + Send + Sync>>;
35
36pub struct Builder {
37    port: u16,
38    on_request: OnRequest,
39}
40
41impl Builder {
42    pub fn new(port: u16) -> Self {
43        Self {
44            port,
45            on_request: None,
46        }
47    }
48
49    pub fn on_request<F: Fn(&Request, &mut Response) + Send + Sync + 'static>(
50        mut self,
51        f: F,
52    ) -> Self {
53        self.on_request.replace(Box::new(f));
54        self
55    }
56
57    pub fn build<R: Runtime>(mut self) -> TauriPlugin<R> {
58        let port = self.port;
59        let on_request = self.on_request.take();
60
61        PluginBuilder::new("server")
62            .setup(move |app| {
63                let asset_resolver = app.asset_resolver();
64                std::thread::spawn(move || {
65                    let server =
66                        Server::http(&format!("0.0.0.0:{port}")).expect("Unable to spawn server");
67                    for req in server.incoming_requests() {
68                        let path = req
69                            .url()
70                            .parse::<Uri>()
71                            .map(|uri| uri.path().into())
72                            .unwrap_or_else(|_| req.url().into());
73
74                        #[allow(unused_mut)]
75                        if let Some(mut asset) = asset_resolver.get(path) {
76                            let request = Request {
77                                url: req.url().into(),
78                            };
79                            let mut response = Response {
80                                headers: Default::default(),
81                            };
82
83                            response.add_header("Content-Type", asset.mime_type);
84                            if let Some(csp) = asset.csp_header {
85                                response
86                                    .headers
87                                    .insert("Content-Security-Policy".into(), csp);
88                            }
89
90                            if let Some(on_request) = &on_request {
91                                on_request(&request, &mut response);
92                            }
93
94                            #[cfg(target_os = "linux")]
95                            if let Some(response_csp) =
96                                response.headers.get("Content-Security-Policy")
97                            {
98                                let html = String::from_utf8_lossy(&asset.bytes);
99                                let body =
100                                    html.replacen(tauri::utils::html::CSP_TOKEN, response_csp, 1);
101                                asset.bytes = body.as_bytes().to_vec();
102                            }
103
104                            let mut resp = HttpResponse::from_data(asset.bytes);
105                            for (header, value) in response.headers {
106                                if let Ok(h) = Header::from_bytes(header.as_bytes(), value) {
107                                    resp.add_header(h);
108                                }
109                            }
110                            req.respond(resp).expect("unable to setup response");
111                        }
112                    }
113                });
114                Ok(())
115            })
116            .build()
117    }
118}
omuchat_tauri_plugin_server/lib.rs

omuchat_tauri_plugin_server/
lib.rs
