roast2d_internal 0.4.0

use anyhow::{Result, bail};
use js_sys::{JSON, Uint8Array};
use std::path::{Path, PathBuf};
use wasm_bindgen::{JsCast, JsValue};
use wasm_bindgen_futures::JsFuture;
use web_sys::{Response, WorkerGlobalScope};

use super::AssetReader;

fn js_value_to_err(context: &str) -> impl FnOnce(JsValue) -> std::io::Error + '_ {
    move |value| {
        let message = match JSON::stringify(&value) {
            Ok(js_str) => format!("Failed to {context}: {js_str}"),
            Err(_) => {
                format!("Failed to {context} and also failed to stringify the JSValue of the error")
            }
        };

        std::io::Error::new(std::io::ErrorKind::Other, message)
    }
}

/// Simplified web asset reader <https://github.com/bevyengine/bevy/blob/main/crates/bevy_asset/src/io/wasm.rs#L27>
pub(crate) struct WebAssetReader {
    root_path: PathBuf,
}

impl AssetReader for WebAssetReader {
    fn new<P: AsRef<Path>>(path: P) -> Self {
        Self {
            root_path: path.as_ref().to_owned(),
        }
    }

    fn get_full_path(&self, path: &str) -> PathBuf {
        self.root_path.join(path)
    }

    async fn read<'a>(&'a self, path: &'a str) -> Result<Vec<u8>> {
        let fullpath = self.get_full_path(path);
        let path = fullpath.to_str().unwrap();
        // Detect the active JS global scope and forward to the correct fetch implementation.
        let global = js_sys::global();
        let promise = if let Some(window) = web_sys::window() {
            window.fetch_with_str(path)
        } else if let Ok(worker) = global.clone().dyn_into::<WorkerGlobalScope>() {
            worker.fetch_with_str(path)
        } else {
            let error = std::io::Error::new(
                std::io::ErrorKind::Other,
                "Unsupported JavaScript global context",
            );
            bail!(error);
        };
        let resp_value = JsFuture::from(promise)
            .await
            .map_err(js_value_to_err("fetch path"))?;
        let resp = resp_value
            .dyn_into::<Response>()
            .map_err(js_value_to_err("convert fetch to Response"))?;
        match resp.status() {
            200 => {
                let data = JsFuture::from(resp.array_buffer().unwrap()).await.unwrap();
                let bytes = Uint8Array::new(&data).to_vec();
                Ok(bytes)
            }
            404 => bail!("Not Found {path}"),
            status => bail!("HTTP Error {status}"),
        }
    }
}