@jk2908/cxx 0.2.6 → 0.3.1

package/dist/vite/index.js

@@ -1,8 +1,12 @@
+import { createRequire } from "node:module";
+var __require = /* @__PURE__ */ createRequire(import.meta.url);
+
 // src/build.ts
 import { createHash } from "node:crypto";
 import fs from "node:fs/promises";
 import path from "node:path";
 import { transform } from "lightningcss";
+import MagicString from "magic-string";
 
 // src/config.ts
 var NAME = "cxx";
@@ -52,31 +56,73 @@ function capitalise(str) {
 function pascalise(str) {
   return str.split(/[ _-]+/).map(capitalise).join("");
 }
-function debounce(fn, wait) {
-  let timeoutId = null;
-  return (...args) => {
-    if (timeoutId) {
-      clearTimeout(timeoutId);
+function createBatcher(wait, flush) {
+  const pending = new Map;
+  let timer = null;
+  let running = false;
+  async function drain() {
+    if (running)
+      return;
+    running = true;
+    try {
+      while (pending.size > 0) {
+        const batch = new Map(pending);
+        pending.clear();
+        await flush(batch);
+      }
+    } finally {
+      running = false;
     }
-    timeoutId = setTimeout(() => {
-      fn.apply(null, args);
+  }
+  function schedule() {
+    if (timer)
+      clearTimeout(timer);
+    timer = setTimeout(() => {
+      timer = null;
+      drain();
     }, wait);
+  }
+  return {
+    add(key, value) {
+      pending.set(key, value);
+      schedule();
+    },
+    flush() {
+      if (timer) {
+        clearTimeout(timer);
+        timer = null;
+      }
+      return drain();
+    }
   };
 }
 
 // src/build.ts
+var decoder = new TextDecoder;
 var collectCache = new LruCache;
 var fileCache = new LruCache;
+var writeChain = Promise.resolve();
 function isENOENT(err) {
   return err instanceof Error && "code" in err && err.code === "ENOENT";
 }
+async function writeFileAtomic(filePath, content) {
+  const tmp = `${filePath}.${process.pid}.${Math.random().toString(36).slice(2)}.tmp`;
+  await fs.writeFile(tmp, content);
+  await fs.rename(tmp, filePath);
+}
+function enqueueWrite(task) {
+  const run = writeChain.then(task, task);
+  writeChain = run.then(() => {
+    return;
+  }, () => {
+    return;
+  });
+  return run;
+}
 async function maybeWrite(filePath, content) {
   const cached = fileCache.get(filePath);
-  if (cached !== content) {
-    await fs.writeFile(filePath, content);
-    fileCache.set(filePath, content);
-    return true;
-  }
+  if (cached === content)
+    return false;
   try {
     if (await fs.readFile(filePath, "utf-8") === content) {
       fileCache.set(filePath, content);
@@ -86,11 +132,12 @@ async function maybeWrite(filePath, content) {
     if (!isENOENT(err))
       throw err;
   }
-  await fs.writeFile(filePath, content);
+  await writeFileAtomic(filePath, content);
   fileCache.set(filePath, content);
   return true;
 }
-var regex = /(?:const|var|let)\s*\[(\w+|)\s*,?\s*(\w+|)\s*,?\s*(\w+|)\s*\]\s*=\s*cxx(?:\s*`([\s\S]*?)`|\s*\.\s*tag(?:\s*<[^>]+>)?\s*\(\s*(['"])(.*?)\5\s*\)\s*`([\s\S]*?)`)/gm;
+var regex = /((?:export\s+)?)(?:const|var|let)\s*\[(\w*)\s*,?\s*(\w*)\s*,?\s*(\w*)\s*\]\s*=\s*cxx(?:\s*`([\s\S]*?)`|\s*\.\s*tag(?:\s*<[^>]+>)?\s*\(\s*(['"])(.*?)\6\s*\)\s*`([\s\S]*?)`)/gm;
+var IDENTIFIER = /^[A-Za-z_$][A-Za-z0-9_$]*$/;
 function hashCss(code) {
   return `cxx-${createHash("sha256").update(code).digest("base64url").slice(0, 12)}`;
 }
@@ -101,12 +148,26 @@ class DuplicateTagError extends Error {
     super(`Tag ${tag} has already been seen. Tags must be unique across your application`);
   }
 }
+
+class InvalidTagError extends Error {
+  name = "InvalidTagError";
+  constructor(tag) {
+    super(`Tag "${tag}" does not map to a valid TypeScript identifier. ` + `Use letters, numbers, spaces, "_" or "-"`);
+  }
+}
 function collect(source, id, config) {
-  const tags = new Map;
+  const configKey = JSON.stringify(config);
   const cached = collectCache.get(id);
-  if (cached?.source === source)
+  if (cached?.source === source && cached.config === configKey)
     return cached;
-  const template = source.replace(regex, (_match, varOne, varTwo, varThree, plainCss, _quote, tag, taggedCss) => {
+  const tags = new Map;
+  const magic = new MagicString(source);
+  let matched = false;
+  regex.lastIndex = 0;
+  let match;
+  while (match = regex.exec(source)) {
+    matched = true;
+    const [, exp, varOne, varTwo, varThree, plainCss, , tag, taggedCss] = match;
     const css = plainCss ?? taggedCss;
     const { code, exports = {} } = transform({
       minify: true,
@@ -119,49 +180,112 @@ function collect(source, id, config) {
     if (tag) {
       const suffix = typeof config.typeSuffix === "string" ? config.typeSuffix : config.typeSuffix === false ? "" : "Classes";
       const tagWithSuffix = pascalise(`${tag}${suffix}`);
+      if (!IDENTIFIER.test(tagWithSuffix))
+        throw new InvalidTagError(tag);
       if (tags.has(tagWithSuffix))
         throw new DuplicateTagError(tagWithSuffix);
       tags.set(tagWithSuffix, Object.keys(styles));
     }
-    return [
-      varOne && `const ${varOne} = \`${code}\``,
-      varTwo && `const ${varTwo} = ${JSON.stringify(styles)}`,
-      varThree && `const ${varThree} = \`${href}\``
-    ].filter(Boolean).join(`
+    const statements = [];
+    if (varOne) {
+      statements.push(`${exp}const ${varOne} = ${JSON.stringify(decoder.decode(code))}`);
+    }
+    if (varTwo) {
+      statements.push(`${exp}const ${varTwo} = ${JSON.stringify(styles)}`);
+    }
+    if (varThree) {
+      statements.push(`${exp}const ${varThree} = ${JSON.stringify(href)}`);
+    }
+    magic.overwrite(match.index, match.index + match[0].length, statements.join(`
+`));
+  }
+  const template = magic.toString();
+  const map = matched ? magic.generateMap({ source: id, includeContent: true, hires: true }) : null;
+  const result = { source, config: configKey, template, tags, map };
+  collectCache.set(id, result);
+  return result;
+}
+function renderTypes(tags) {
+  return [
+    AUTOGEN_MSG,
+    "",
+    `import '${PKG_NAME}'`,
+    "",
+    `declare module '${PKG_NAME}' {`,
+    tagsToType(tags),
+    "}",
+    ""
+  ].join(`
 `);
-  });
-  collectCache.set(id, { source, template, tags });
-  return {
-    template,
-    tags
-  };
 }
-async function writeTypes(tags) {
+async function writeRenderedTypes(tags) {
   const filePath = path.join(GENERATED_DIR, "index.d.ts");
   if (!tags.size) {
-    try {
-      await fs.rm(filePath, { force: true });
-      fileCache.delete(filePath);
-    } catch (err) {
-      if (!isENOENT(err))
-        throw err;
-    }
+    await fs.rm(filePath, { force: true });
+    fileCache.delete(filePath);
     return;
   }
   await fs.mkdir(GENERATED_DIR, { recursive: true });
-  await maybeWrite(filePath, `${AUTOGEN_MSG}
-
-			import '${PKG_NAME}'
-
-			declare module '${PKG_NAME}' {
-				${tagsToType(tags)}
-			}
-		`);
+  await maybeWrite(filePath, renderTypes(tags));
+}
+function writeTypes(tags) {
+  return enqueueWrite(() => writeRenderedTypes(tags));
+}
+var PARTIAL_DIR = path.join(GENERATED_DIR, "tags");
+function partialPath(filePath) {
+  return path.join(PARTIAL_DIR, `${createHash("sha1").update(filePath).digest("hex")}.json`);
+}
+async function readPartials() {
+  let entries;
+  try {
+    entries = await fs.readdir(PARTIAL_DIR);
+  } catch (err) {
+    if (isENOENT(err))
+      return new Map;
+    throw err;
+  }
+  const tags = new Map;
+  for (const entry of entries) {
+    if (!entry.endsWith(".json"))
+      continue;
+    const partial = path.join(PARTIAL_DIR, entry);
+    let parsed;
+    try {
+      parsed = JSON.parse(await fs.readFile(partial, "utf-8"));
+    } catch {
+      continue;
+    }
+    try {
+      await fs.access(parsed.file);
+    } catch {
+      await fs.rm(partial, { force: true });
+      continue;
+    }
+    for (const [name, classes] of Object.entries(parsed.tags)) {
+      if (tags.has(name))
+        throw new DuplicateTagError(name);
+      tags.set(name, classes);
+    }
+  }
+  return tags;
+}
+function writeTypesForFile(filePath, tags) {
+  return enqueueWrite(async () => {
+    await fs.mkdir(PARTIAL_DIR, { recursive: true });
+    const partial = partialPath(filePath);
+    if (!tags.size) {
+      await fs.rm(partial, { force: true });
+    } else {
+      await writeFileAtomic(partial, JSON.stringify({ file: filePath, tags: Object.fromEntries(tags) }));
+    }
+    await writeRenderedTypes(await readPartials());
+  });
 }
 function tagsToType(tags) {
-  return [...tags].map(([tag, classes]) => `
-		export type ${tag} = ${classes.length > 0 ? classes.toSorted((a, b) => a.length - b.length).map((c) => `'${c}'`).join(" | ") : "never"}
-	`).join(`
+  return [...tags].map(([tag, classes]) => {
+    const union = classes.length > 0 ? classes.toSorted((a, b) => a.length - b.length).map((c) => `'${c}'`).join(" | ") : "never";
+    return `	export type ${tag} = ${union}`;
+  }).join(`
 `);
 }
 function flattenTags(tagsByFile) {
@@ -189,20 +313,14 @@ var LEVELS = {
   error: 3,
   fatal: 4
 };
+var defaultLevel = process.env["NODE_ENV"] === "production" ? "error" : "info";
 
 class Logger {
   name;
-  static #defaultLevel = "info";
   #level;
   constructor(level, name) {
     this.name = name;
-    this.#level = level;
-  }
-  static set defaultLevel(level) {
-    Logger.#defaultLevel = level;
-  }
-  static get defaultLevel() {
-    return Logger.#defaultLevel;
+    this.#level = level ?? defaultLevel;
   }
   static toError(err) {
     return err instanceof Error ? err : new Error(String(err), { cause: err });
@@ -225,7 +343,7 @@ ${err.stack}` : "");
     this.#level = level;
   }
   get level() {
-    return this.#level ?? Logger.#defaultLevel;
+    return this.#level;
   }
   log(level, message, error) {
     if (LEVELS[level] < LEVELS[this.level])
@@ -277,11 +395,8 @@ function cxx(pluginConfig = {}) {
   const buildContext = {
     tagsByFile: new Map
   };
-  const logger = new Logger(pluginConfig?.logger?.level ?? false ? "error" : "debug");
-  const watchRoot = normaliseWatchPath(realpathSync.native(process.cwd()));
-  const watchedRoots = [`${watchRoot}/`];
-  let rebuildRunning = false;
-  let rebuildQueued = false;
+  const logger = new Logger;
+  let watchRoot = normaliseWatchPath(process.cwd());
   function resolveWatchFile(filePath) {
     const absolutePath = path2.resolve(watchRoot, filePath);
     const parentPath = path2.dirname(absolutePath);
@@ -294,72 +409,115 @@ function cxx(pluginConfig = {}) {
   }
   function watchedFile(filePath) {
     const resolvedPath = resolveWatchFile(filePath);
-    return watchedRoots.some((root) => resolvedPath.startsWith(root)) && !resolvedPath.includes(`/${GENERATED_DIR}/`) && ACCEPTED_EXTENSIONS.has(path2.extname(resolvedPath));
+    if (resolvedPath.startsWith(`${watchRoot}/`) && !resolvedPath.includes(`/${GENERATED_DIR}/`) && ACCEPTED_EXTENSIONS.has(path2.extname(resolvedPath))) {
+      return resolvedPath;
+    }
+    return null;
   }
-  const rebuild = debounce((event, filePath) => {
-    function queue() {
-      (async () => {
-        if (rebuildRunning) {
-          rebuildQueued = true;
-          return;
-        }
-        rebuildRunning = true;
-        do {
-          rebuildQueued = false;
-          try {
-            const hadTags = buildContext.tagsByFile.has(file);
-            let hasTags = false;
-            if (event === "unlink") {
-              buildContext.tagsByFile.delete(file);
-            } else {
-              const source = await fs2.readFile(file, "utf-8");
-              const { tags } = collect(source, file, pluginConfig);
-              hasTags = tags.size > 0;
-              if (hasTags) {
-                buildContext.tagsByFile.set(file, tags);
-              } else {
-                buildContext.tagsByFile.delete(file);
-              }
-            }
-            if (!hadTags && !hasTags)
-              continue;
-            await writeTypes(flattenTags(buildContext.tagsByFile));
-            logger.info(`updated types from ${file}`);
-          } catch (err) {
-            if (isENOENT(err)) {
-              if (!buildContext.tagsByFile.has(file))
-                continue;
-              buildContext.tagsByFile.delete(file);
-              await writeTypes(flattenTags(buildContext.tagsByFile));
-              logger.info(`updated types from ${file}`);
-              continue;
-            }
-            logger.error(`failed to update types from ${file}`, err);
-          }
-        } while (rebuildQueued);
-        rebuildRunning = false;
-      })();
+  async function onEvent(event, file) {
+    const hadTags = buildContext.tagsByFile.has(file);
+    if (event === "unlink") {
+      if (!hadTags)
+        return;
+      buildContext.tagsByFile.delete(file);
+      await writeTypes(flattenTags(buildContext.tagsByFile));
+      logger.info(`updated types from ${file}`);
+      return;
     }
-    if (!watchedFile(filePath))
+    let source;
+    try {
+      source = await fs2.readFile(file, "utf-8");
+    } catch (err) {
+      if (!isENOENT(err))
+        throw err;
+      if (!hadTags)
+        return;
+      buildContext.tagsByFile.delete(file);
+      await writeTypes(flattenTags(buildContext.tagsByFile));
+      logger.info(`updated types from ${file}`);
       return;
-    const file = resolveWatchFile(filePath);
-    queue();
-  }, 75);
+    }
+    const { tags } = collect(source, file, pluginConfig);
+    const hasTags = tags.size > 0;
+    if (hasTags) {
+      buildContext.tagsByFile.set(file, tags);
+    } else {
+      buildContext.tagsByFile.delete(file);
+    }
+    if (!hadTags && !hasTags)
+      return;
+    await writeTypes(flattenTags(buildContext.tagsByFile));
+    logger.info(`updated types from ${file}`);
+  }
+  const queue = createBatcher(75, async (batch) => {
+    for (const [file, event] of batch) {
+      try {
+        await onEvent(event, file);
+      } catch (err) {
+        logger.error(`failed to update types from ${file}`, err);
+      }
+    }
+  });
   return {
     name: "cxx",
     enforce: "pre",
+    config() {
+      const ignore = pluginConfig.watch?.ignore;
+      if (!ignore?.length)
+        return null;
+      return {
+        server: {
+          watch: {
+            ignored: ignore
+          }
+        }
+      };
+    },
+    configResolved(config) {
+      const root = pluginConfig.watch?.root ?? config.root;
+      try {
+        watchRoot = normaliseWatchPath(realpathSync.native(root));
+      } catch {
+        watchRoot = normaliseWatchPath(root);
+      }
+    },
     transform(code, id) {
-      const { template, tags } = collect(code, id, pluginConfig);
-      if (tags.size)
-        buildContext.tagsByFile.set(id, tags);
+      if (id.startsWith("\x00") || id.includes("/node_modules/"))
+        return null;
+      if (!code.includes("cxx"))
+        return null;
+      const cleanId = id.split("?")[0];
+      if (!ACCEPTED_EXTENSIONS.has(path2.extname(cleanId)))
+        return null;
+      const { template, tags, map } = collect(code, cleanId, pluginConfig);
+      if (template === code)
+        return null;
+      const resolved = resolveWatchFile(cleanId);
+      if (tags.size) {
+        buildContext.tagsByFile.set(resolved, tags);
+      } else {
+        buildContext.tagsByFile.delete(resolved);
+      }
       return {
         code: template,
-        map: null
+        map
       };
     },
     configureServer(server) {
       logger.info("Watching for changes...");
-      server.watcher.on("add", (p) => rebuild("add", p)).on("change", (p) => rebuild("change", p)).on("unlink", (p) => rebuild("unlink", p));
+      server.watcher.on("add", (p) => {
+        const file = watchedFile(p);
+        if (file)
+          queue.add(file, "add");
+      }).on("change", (p) => {
+        const file = watchedFile(p);
+        if (file)
+          queue.add(file, "change");
+      }).on("unlink", (p) => {
+        const file = watchedFile(p);
+        if (file)
+          queue.add(file, "unlink");
+      });
     },
     async writeBundle() {
       await writeTypes(flattenTags(buildContext.tagsByFile));
@@ -370,4 +528,4 @@ export {
   cxx as default
 };
 
-//# debugId=99F5EDE4479120ED64756E2164756E21
+//# debugId=8BACD1ACA42BF59D64756E2164756E21