ChatGPT-Next-Web/src-tauri/src/fetch.rs

//
// HTTP request handler module
//

use std::time::Duration;
use std::error::Error;
use std::sync::atomic::{AtomicU32, Ordering};
use std::collections::HashMap;
use reqwest::Client;
use reqwest::header::{HeaderName, HeaderMap};

static REQUEST_COUNTER: AtomicU32 = AtomicU32::new(0);

#[derive(Debug, Clone, serde::Serialize)]
pub struct FetchResponse {
  request_id: u32,
  status: u16,
  status_text: String,
  headers: HashMap<String, String>,
  body: Vec<u8>,
}

#[tauri::command]
pub async fn http_fetch(
  method: String,
  url: String,
  headers: HashMap<String, String>,
  body: Vec<u8>,
) -> Result<FetchResponse, String> {
  
  let request_id = REQUEST_COUNTER.fetch_add(1, Ordering::SeqCst);

  let mut _headers = HeaderMap::new();
  for (key, value) in &headers {
    match key.parse::<HeaderName>() {
      Ok(header_name) => {
        match value.parse() {
          Ok(header_value) => {
            _headers.insert(header_name, header_value);
          }
          Err(err) => {
            return Err(format!("failed to parse header value '{}': {}", value, err));
          }
        }
      }
      Err(err) => {
        return Err(format!("failed to parse header name '{}': {}", key, err));
      }
    }
  }

  // Parse HTTP method
  let method = method.parse::<reqwest::Method>()
    .map_err(|err| format!("failed to parse method: {}", err))?;

  // Create client
  let client = Client::builder()
    .default_headers(_headers)
    .redirect(reqwest::redirect::Policy::limited(3))
    .connect_timeout(Duration::new(10, 0))
    .timeout(Duration::new(30, 0))
    .build()
    .map_err(|err| format!("failed to create client: {}", err))?;

  // Build request
  let mut request = client.request(
    method.clone(),
    url.parse::<reqwest::Url>()
      .map_err(|err| format!("failed to parse url: {}", err))?
  );

  // For request methods that need a body, add the request body
  if method == reqwest::Method::POST 
    || method == reqwest::Method::PUT 
    || method == reqwest::Method::PATCH 
    || method == reqwest::Method::DELETE {
    if !body.is_empty() {
      let body_bytes = bytes::Bytes::from(body);
      request = request.body(body_bytes);
    }
  }

  // Send request
  let response = request.send().await
    .map_err(|err| {
      let error_msg = err.source()
        .map(|e| e.to_string())
        .unwrap_or_else(|| err.to_string());
      format!("request failed: {}", error_msg)
    })?;

  // Get response status and headers
  let status = response.status().as_u16();
  let status_text = response.status().canonical_reason()
    .unwrap_or("Unknown")
    .to_string();

  let mut response_headers = HashMap::new();
  for (name, value) in response.headers() {
    response_headers.insert(
      name.as_str().to_string(),
      std::str::from_utf8(value.as_bytes())
        .unwrap_or("<invalid utf8>")
        .to_string()
    );
  }

  // Read response body
  let response_body = response.bytes().await
    .map_err(|err| format!("failed to read response body: {}", err))?;

  Ok(FetchResponse {
    request_id,
    status,
    status_text,
    headers: response_headers,
    body: response_body.to_vec(),
  })
}

#[tauri::command]
pub async fn http_fetch_text(
  method: String,
  url: String,
  headers: HashMap<String, String>,
  body: String,
) -> Result<String, String> {
  
  // Convert string body to bytes
  let body_bytes = body.into_bytes();
  
  // Call the main fetch method
  let response = http_fetch(method, url, headers, body_bytes).await?;
  
  // Convert response body to string
  let response_text = String::from_utf8(response.body)
    .map_err(|err| format!("failed to convert response to text: {}", err))?;
  
  Ok(response_text)
}

#[tauri::command]
pub async fn http_fetch_json(
  method: String,
  url: String,
  headers: HashMap<String, String>,
  body: serde_json::Value,
) -> Result<serde_json::Value, String> {
  
  // Convert JSON to string and then to bytes
  let body_string = serde_json::to_string(&body)
    .map_err(|err| format!("failed to serialize JSON body: {}", err))?;
  let body_bytes = body_string.into_bytes();
  
  // Ensure the correct Content-Type is set
  let mut json_headers = headers;
  if !json_headers.contains_key("content-type") && !json_headers.contains_key("Content-Type") {
    json_headers.insert("Content-Type".to_string(), "application/json".to_string());
  }
  
  // Call the main fetch method
  let response = http_fetch(method, url, json_headers, body_bytes).await?;
  
  // Parse response body as JSON
  let response_json: serde_json::Value = serde_json::from_slice(&response.body)
    .map_err(|err| format!("failed to parse response as JSON: {}", err))?;
  
  Ok(response_json)
}