Risuai 0.6.3 first commit

src-tauri/src/main.rs

@@ -0,0 +1,81 @@
+// Prevents additional console window on Windows in release, DO NOT REMOVE!!
+#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
+
+
+#[tauri::command]
+fn greet(name: &str) -> String {
+    format!("Hello, {}! You've been greeted from Rust!", name)
+}
+
+use serde_json::Value;
+use reqwest::header::{HeaderMap, HeaderName, HeaderValue};
+use base64::{engine::general_purpose, Engine as _};
+use std::time::Duration;
+
+
+#[tauri::command]
+async fn native_request(url: String, body: String, header: String, method:String) -> String {
+    let headers_json: Value = match serde_json::from_str(&header) {
+        Ok(h) => h,
+        Err(e) => return format!(r#"{{"success":false,"body":"{}"}}"#, e.to_string()),
+    };
+
+    let mut headers = HeaderMap::new();
+    let method = method.to_string();
+    if let Some(obj) = headers_json.as_object() {
+        for (key, value) in obj {
+            let header_name = match HeaderName::from_bytes(key.as_bytes()) {
+                Ok(name) => name,
+                Err(e) => return format!(r#"{{"success":false,"body":"{}"}}"#, e.to_string()),
+            };
+            let header_value = match HeaderValue::from_str(value.as_str().unwrap_or("")) {
+                Ok(value) => value,
+                Err(e) => return format!(r#"{{"success":false,"body":"{}"}}"#, e.to_string()),
+            };
+            headers.insert(header_name, header_value);
+        }
+    } else {
+        return format!(r#"{{"success":false,"body":"Invalid header JSON"}}"#);
+    }
+
+    let client = reqwest::Client::new();
+    let response:Result<reqwest::Response, reqwest::Error>;
+
+    if method == "POST" {
+        response = client
+        .post(&url)
+        .headers(headers)
+        .timeout(Duration::from_secs(120))
+        .body(body)
+        .send()
+        .await;
+    }
+    else{
+        response = client
+        .get(&url)
+        .headers(headers)
+        .timeout(Duration::from_secs(120))
+        .send()
+        .await;
+    }
+
+    match response {
+        Ok(resp) => {
+            let bytes = match resp.bytes().await {
+                Ok(b) => b,
+                Err(e) => return format!(r#"{{"success":false,"body":"{}"}}"#, e.to_string()),
+            };
+            let encoded = general_purpose::STANDARD.encode(&bytes);
+
+            format!(r#"{{"success":true,"body":"{}"}}"#, encoded)
+        }
+        Err(e) => format!(r#"{{"success":false,"body":"{}"}}"#, e.to_string()),
+    }
+}
+
+fn main() {
+    tauri::Builder::default()
+        .invoke_handler(tauri::generate_handler![greet, native_request])
+        .run(tauri::generate_context!())
+        .expect("error while running tauri application");
+}