1 files changed, 77 insertions, 0 deletions
diff --git a/op_crates/webgpu/shader.rs b/op_crates/webgpu/shader.rs
new file mode 100644
index 000000000..25adf9b95
--- /dev/null
+++ b/op_crates/webgpu/shader.rs
@@ -0,0 +1,77 @@
+// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.
+
+use deno_core::error::bad_resource_id;
+use deno_core::error::AnyError;
+use deno_core::serde_json::json;
+use deno_core::serde_json::Value;
+use deno_core::ZeroCopyBuf;
+use deno_core::{OpState, Resource};
+use serde::Deserialize;
+use std::borrow::Cow;
+
+use super::error::WebGPUError;
+
+pub(crate) struct WebGPUShaderModule(pub(crate) wgpu_core::id::ShaderModuleId);
+impl Resource for WebGPUShaderModule {
+  fn name(&self) -> Cow<str> {
+    "webGPUShaderModule".into()
+  }
+}
+
+#[derive(Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct CreateShaderModuleArgs {
+  device_rid: u32,
+  label: Option<String>,
+  code: Option<String>,
+  _source_map: Option<()>, // not yet implemented
+}
+
+pub fn op_webgpu_create_shader_module(
+  state: &mut OpState,
+  args: CreateShaderModuleArgs,
+  zero_copy: &mut [ZeroCopyBuf],
+) -> Result<Value, AnyError> {
+  let instance = state.borrow::<super::Instance>();
+  let device_resource = state
+    .resource_table
+    .get::<super::WebGPUDevice>(args.device_rid)
+    .ok_or_else(bad_resource_id)?;
+  let device = device_resource.0;
+
+  let source = match args.code {
+    Some(code) => {
+      wgpu_core::pipeline::ShaderModuleSource::Wgsl(Cow::from(code))
+    }
+    None => wgpu_core::pipeline::ShaderModuleSource::SpirV(Cow::from(unsafe {
+      let (prefix, data, suffix) = zero_copy[0].align_to::<u32>();
+      assert!(prefix.is_empty());
+      assert!(suffix.is_empty());
+      data
+    })),
+  };
+
+  let mut flags = wgpu_types::ShaderFlags::default();
+  flags.set(wgpu_types::ShaderFlags::VALIDATION, true);
+  #[cfg(all(target_os = "macos", target_arch = "x86_64"))]
+  flags.set(wgpu_types::ShaderFlags::EXPERIMENTAL_TRANSLATION, true);
+
+  let descriptor = wgpu_core::pipeline::ShaderModuleDescriptor {
+    label: args.label.map(Cow::from),
+    flags,
+  };
+
+  let (shader_module, maybe_err) = gfx_select!(device => instance.device_create_shader_module(
+    device,
+    &descriptor,
+    source,
+    std::marker::PhantomData
+  ));
+
+  let rid = state.resource_table.add(WebGPUShaderModule(shader_module));
+
+  Ok(json!({
+    "rid": rid,
+    "err": maybe_err.map(WebGPUError::from)
+  }))
+}