123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
#!/usr/bin/env -S deno run --unstable --allow-read --allow-write --allow-run// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
import { join, ROOT_PATH, walk } from "./util.js";
// const COMMIT = "c00e471274b6c21acda89b4b13d41742c0285d71"; // Release 12const COMMIT = "c4aa3eaed020a640fec06b48f0a5ea93490d41bb"; // tip of PR (needs merge)const REPO = "kvark/wgpu";const V_WGPU = "0.12";const TARGET_DIR = join(ROOT_PATH, "ext", "webgpu");
async function bash(subcmd, opts = {}) {  const p = Deno.run({ ...opts, cmd: ["bash", "-c", subcmd] });
  // Exit process on failure  const { success, code } = await p.status();  if (!success) {    Deno.exit(code);  }  // Cleanup  p.close();}
async function clearTargetDir() {  await bash(`rm -r ${TARGET_DIR}/*`);}
async function checkoutUpstream() {  // Path of deno_webgpu inside the TAR  const tarPrefix = `${REPO.replace("/", "-")}-${    COMMIT.slice(0, 7)  }/deno_webgpu/`;  const cmd =    `curl -L https://api.github.com/repos/${REPO}/tarball/${COMMIT} | tar -C '${TARGET_DIR}' -xzvf - --strip=2 '${tarPrefix}'`;  // console.log(cmd);  await bash(cmd);}
async function denoCoreVersion() {  const coreCargo = join(ROOT_PATH, "core", "Cargo.toml");  const contents = await Deno.readTextFile(coreCargo);  return contents.match(/^version = "(\d+\.\d+\.\d+)"$/m)[1];}
async function denoWebgpuVersion() {  const coreCargo = join(ROOT_PATH, "runtime", "Cargo.toml");  const contents = await Deno.readTextFile(coreCargo);  return contents.match(    /^deno_webgpu = { version = "(\d+\.\d+\.\d+)", path = "..\/ext\/webgpu" }$/m,  )[1];}
async function patchFile(path, patcher) {  const data = await Deno.readTextFile(path);  const patched = patcher(data);  await Deno.writeTextFile(path, patched);}
async function patchCargo() {  const vDenoCore = await denoCoreVersion();  const vDenoWebgpu = await denoWebgpuVersion();  await patchFile(    join(TARGET_DIR, "Cargo.toml"),    (data) =>      data        .replace(/^version = .*/m, `version = "${vDenoWebgpu}"`)        .replace(`edition = "2018"`, `edition = "2021"`)        .replace(          /^deno_core \= .*$/gm,          `deno_core = { version = "${vDenoCore}", path = "../../core" }`,        )        .replace(          /^wgpu-core \= .*$/gm,          `wgpu-core = { version = "${V_WGPU}", features = ["trace", "replay", "serde"] }`,        )        .replace(          /^wgpu-types \= .*$/gm,          `wgpu-types = { version = "${V_WGPU}", features = ["trace", "replay", "serde"] }`,        ),    // .replace(    //   /^wgpu-core \= .*$/gm,    //   `wgpu-core = { git = "https://github.com/${REPO}", rev = "${COMMIT}", features = ["trace", "replay", "serde"] }`,    // )    // .replace(    //   /^wgpu-types \= .*$/gm,    //   `wgpu-types = { git = "https://github.com/${REPO}", rev = "${COMMIT}", features = ["trace", "replay", "serde"] }`,    // )  );}
async function patchSrcLib() {  await patchFile(    join(TARGET_DIR, "src", "lib.rs"),    (data) =>      data.replace(`prefix "deno:deno_webgpu",`, `prefix "deno:ext/webgpu",`),  );}
async function patchCopyrights() {  const walker = walk(TARGET_DIR, { includeDirs: false });  for await (const entry of walker) {    await patchFile(      entry.path,      (data) =>        data.replace(/^\/\/ Copyright 2018-2021/, "// Copyright 2018-2022"),    );  }}
async function main() {  await clearTargetDir();  await checkoutUpstream();  await patchCargo();  await patchSrcLib();  await patchCopyrights();  await bash(join(ROOT_PATH, "tools", "format.js"));}
await main();