wgpu/cts_runner/src/main.rs

#[cfg(not(target_arch = "wasm32"))]
mod native {
    use std::{
        env, fmt,
        io::{Read, Write},
        rc::Rc,
    };

    use deno_core::anyhow::anyhow;
    use deno_core::error::AnyError;
    use deno_core::op;
    use deno_core::resolve_url_or_path;
    use deno_core::serde_json::json;
    use deno_core::v8;
    use deno_core::JsRuntime;
    use deno_core::RuntimeOptions;
    use deno_core::ZeroCopyBuf;
    use deno_web::BlobStore;
    use termcolor::Ansi;
    use termcolor::Color::Red;
    use termcolor::ColorSpec;
    use termcolor::WriteColor;

    pub async fn run() -> Result<(), AnyError> {
        let mut args_iter = env::args();
        let _ = args_iter.next();
        let url = args_iter
            .next()
            .ok_or_else(|| anyhow!("missing specifier in first command line argument"))?;
        let specifier = resolve_url_or_path(&url, &env::current_dir()?)?;

        let options = RuntimeOptions {
            module_loader: Some(Rc::new(deno_core::FsModuleLoader)),
            get_error_class_fn: Some(&get_error_class_name),
            extensions: vec![
                deno_webidl::deno_webidl::init_ops_and_esm(),
                deno_console::deno_console::init_ops_and_esm(),
                deno_url::deno_url::init_ops_and_esm(),
                deno_web::deno_web::init_ops_and_esm::<Permissions>(BlobStore::default(), None),
                deno_webgpu::deno_webgpu::init_ops_and_esm(true),
                cts_runner::init_ops_and_esm(),
            ],
            ..Default::default()
        };
        let mut isolate = JsRuntime::new(options);
        let args = args_iter.collect::<Vec<String>>();
        let cfg = json!({"args": args, "cwd": env::current_dir().unwrap().to_string_lossy() });

        {
            let context = isolate.global_context();
            let scope = &mut isolate.handle_scope();
            let context_local = v8::Local::new(scope, context);
            let global_obj = context_local.global(scope);
            let bootstrap_str = v8::String::new(scope, "bootstrap").unwrap();
            let bootstrap_fn = global_obj.get(scope, bootstrap_str.into()).unwrap();
            let bootstrap_fn = v8::Local::<v8::Function>::try_from(bootstrap_fn).unwrap();

            let options_v8 = deno_core::serde_v8::to_v8(scope, cfg).unwrap();
            let bootstrap_fn = v8::Local::new(scope, bootstrap_fn);
            let undefined = v8::undefined(scope);
            bootstrap_fn
                .call(scope, undefined.into(), &[options_v8])
                .unwrap();
        }

        isolate.op_state().borrow_mut().put(Permissions {});

        let mod_id = isolate.load_main_module(&specifier, None).await?;
        let mod_rx = isolate.mod_evaluate(mod_id);

        let rx = tokio::spawn(async move {
            match mod_rx.await {
                Ok(err @ Err(_)) => err,
                _ => Ok(()),
            }
        });

        isolate.run_event_loop(false).await?;
        rx.await.unwrap()?;

        Ok(())
    }

    deno_core::extension!(
        cts_runner,
        deps = [deno_webidl, deno_web],
        ops = [op_exit, op_read_file_sync, op_write_file_sync],
        esm_entry_point = "ext:cts_runner/bootstrap.js",
        esm = ["bootstrap.js"],
    );

    #[op]
    fn op_exit(code: i32) -> Result<(), AnyError> {
        std::process::exit(code)
    }

    #[op]
    fn op_read_file_sync(path: String) -> Result<ZeroCopyBuf, AnyError> {
        let path = std::path::Path::new(&path);
        let mut file = std::fs::File::open(path)?;
        let mut buf = Vec::new();
        file.read_to_end(&mut buf)?;
        Ok(ZeroCopyBuf::from(buf))
    }

    #[op]
    fn op_write_file_sync(path: String, buf: ZeroCopyBuf) -> Result<(), AnyError> {
        let path = std::path::Path::new(&path);
        let mut file = std::fs::File::create(path)?;
        file.write_all(&buf)?;
        Ok(())
    }

    fn get_error_class_name(e: &AnyError) -> &'static str {
        deno_core::error::get_custom_error_class(e)
            .or_else(|| deno_webgpu::error::get_error_class_name(e))
            .unwrap_or_else(|| {
                panic!(
                    "Error '{}' contains boxed error of unsupported type:{}",
                    e,
                    e.chain()
                        .map(|e| format!("\n  {:?}", e))
                        .collect::<String>()
                );
            })
    }

    pub fn unwrap_or_exit<T>(result: Result<T, AnyError>) -> T {
        match result {
            Ok(value) => value,
            Err(error) => {
                eprintln!("{}: {:?}", red_bold("error"), error);
                std::process::exit(1);
            }
        }
    }

    fn style<S: AsRef<str>>(s: S, colorspec: ColorSpec) -> impl fmt::Display {
        let mut v = Vec::new();
        let mut ansi_writer = Ansi::new(&mut v);
        ansi_writer.set_color(&colorspec).unwrap();
        ansi_writer.write_all(s.as_ref().as_bytes()).unwrap();
        ansi_writer.reset().unwrap();
        String::from_utf8_lossy(&v).into_owned()
    }

    fn red_bold<S: AsRef<str>>(s: S) -> impl fmt::Display {
        let mut style_spec = ColorSpec::new();
        style_spec.set_fg(Some(Red)).set_bold(true);
        style(s, style_spec)
    }

    // NOP permissions
    struct Permissions;

    impl deno_web::TimersPermission for Permissions {
        fn allow_hrtime(&mut self) -> bool {
            false
        }

        fn check_unstable(&self, _state: &deno_core::OpState, _api_name: &'static str) {}
    }
}

#[cfg(not(target_arch = "wasm32"))]
#[tokio::main(flavor = "current_thread")]
async fn main() {
    native::unwrap_or_exit(native::run().await)
}

#[cfg(target_arch = "wasm32")]
fn main() {
    panic!("This is a native only module. It can't be run on the web!");
}