lakestream 0.0.3

Portable file-utility for object-stores
Documentation 
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

use std::collections::HashMap;
use std::error::Error;

use js_sys::{ArrayBuffer, Uint8Array};
use log::info;
use wasm_bindgen::JsCast;
use wasm_bindgen_futures::JsFuture;
use web_sys::{Headers, Request, RequestInit, RequestMode, Response, Window};

use crate::LakestreamError;

pub async fn http_get_request_with_headers(
    url: &str,
    headers: &HashMap<String, String>,
) -> Result<(String, u16, HashMap<String, String>), LakestreamError> {
    info!("http_get_request_with_headers: {}", url);
    // TODO: implement response headers -- for now forward to http_get_request
    // Call the http_get_request function
    let (response_body, response_status) =
        http_get_request(url, headers).await?;

    // Add the headers to the returned result
    Ok((response_body, response_status, HashMap::new()))
}

pub async fn http_get_request(
    url: &str,
    headers: &HashMap<String, String>,
) -> Result<(String, u16), LakestreamError> {
    info!("http_get_request: {}", url);
    let window = web_sys::window().ok_or("No window available")?;
    let mut request_init = RequestInit::new();
    request_init.method("GET");
    request_init.mode(RequestMode::Cors);

    let headers_map = Headers::new().unwrap();
    for (key, value) in headers.iter() {
        headers_map.set(key, value).unwrap();
    }
    request_init.headers(&headers_map);

    let request = Request::new_with_str_and_init(url, &request_init).unwrap();
    let response_js = JsFuture::from(window.fetch_with_request(&request))
        .await
        .unwrap();
    let response: web_sys::Response = response_js.dyn_into().unwrap();

    let status = response.status();
    if status >= 200 && status < 300 {
        let body_js = JsFuture::from(response.text().unwrap()).await.unwrap();
        let body: String = body_js
            .as_string()
            .ok_or("Failed to convert response body to String")?;
        Ok((body, status))
    } else {
        let body_js = JsFuture::from(
            response
                .array_buffer()
                .map_err(|e| LakestreamError::Js(e))?,
        )
        .await
        .map_err(|e| LakestreamError::Js(e))?;
        let body: js_sys::ArrayBuffer =
            body_js.dyn_into().map_err(|e| LakestreamError::Js(e))?;

        let uint8_array = js_sys::Uint8Array::new(&body);
        let vec = uint8_array.to_vec();
        let body = String::from_utf8_lossy(&vec);
        let error_message = format!("Error: {} - {}", status, body);
        Err(LakestreamError::String(error_message))
    }
}