zyzz 0.0.6 → 0.0.8

package/src/node/internal/Snapshot.ts

@@ -0,0 +1,79 @@
+/** Retains unchanged source text and syntax within an adapter instance. @module */
+import * as Fs from 'node:fs'
+import * as Syntax from '../../compiler/internal/Syntax.js'
+
+/** Creates a source cache; resolution and directory discovery remain owned by the host. */
+export function create() {
+  const files = new Map<string, { source: string; version: string }>()
+  const syntax = new Map<
+    string,
+    { parsed: ReturnType<typeof Syntax.parse>; source: string }
+  >()
+
+  function parse(options: Syntax.parse.Options) {
+    let entry = syntax.get(options.moduleId)
+    if (entry?.source !== options.source) {
+      entry = { parsed: Syntax.parse(options), source: options.source }
+      syntax.set(options.moduleId, entry)
+    }
+    return entry.parsed
+  }
+
+  return {
+    clear() {
+      files.clear()
+      syntax.clear()
+    },
+    delete(file: string, moduleId: string) {
+      files.delete(file)
+      syntax.delete(moduleId)
+    },
+    parse,
+    programs(modules: Readonly<Record<string, string>>) {
+      return new Map(
+        Object.entries(modules).map(([moduleId, source]) => [
+          moduleId,
+          parse({ moduleId, source }),
+        ]),
+      )
+    },
+    async read(file: string) {
+      try {
+        const stat = await Fs.promises.stat(file, { bigint: true })
+        const version = `${stat.dev}:${stat.ino}:${stat.size}:${stat.mtimeNs}:${stat.ctimeNs}`
+        let entry = files.get(file)
+        if (entry?.version !== version) {
+          entry = { source: await Fs.promises.readFile(file, 'utf8'), version }
+          files.set(file, entry)
+        }
+        return entry.source
+      } catch (error) {
+        files.delete(file)
+        throw error
+      }
+    },
+    readSync(file: string) {
+      try {
+        const stat = Fs.statSync(file, { bigint: true })
+        const version = `${stat.dev}:${stat.ino}:${stat.size}:${stat.mtimeNs}:${stat.ctimeNs}`
+        let entry = files.get(file)
+        if (entry?.version !== version) {
+          entry = { source: Fs.readFileSync(file, 'utf8'), version }
+          files.set(file, entry)
+        }
+        return entry.source
+      } catch (error) {
+        files.delete(file)
+        throw error
+      }
+    },
+    retain(
+      paths: ReadonlySet<string>,
+      modules: Readonly<Record<string, string>>,
+    ) {
+      for (const file of files.keys()) if (!paths.has(file)) files.delete(file)
+      for (const id of syntax.keys())
+        if (!Object.hasOwn(modules, id)) syntax.delete(id)
+    },
+  }
+}

package/src/unplugin/index.ts

@@ -4,7 +4,8 @@ import * as Crypto from 'node:crypto'
 import * as Fs from 'node:fs/promises'
 import * as Path from 'node:path'
 import * as Reset from '../node/Reset.js'
-import * as Parser from 'oxc-parser'
+import * as Snapshot from '../node/internal/Snapshot.js'
+import * as Syntax from '../compiler/internal/Syntax.js'
 import { createUnplugin } from 'unplugin'
 import type { UnpluginBuildContext } from 'unplugin'
 import * as Graph from '../compiler/Graph.js'
@@ -30,6 +31,7 @@ const portable = createUnplugin<Options | undefined, false>(
   (options = {}, meta) => {
     const root = Path.resolve(options.root ?? process.cwd())
     const compiler = Graph.create()
+    const snapshot = Snapshot.create()
     let graph: Graph.compile.ReturnType | undefined
     let pending: Promise<void> | undefined
     let files = new Set<string>()
@@ -138,7 +140,7 @@ const portable = createUnplugin<Options | undefined, false>(
                 sourcePattern.test(file) &&
                 !excludedPattern.test(file)
               ) {
-                const source = await Fs.readFile(file, 'utf8')
+                const source = await snapshot.read(file)
 
                 files.add(file)
                 sources.set(file, source)
@@ -205,7 +207,7 @@ const portable = createUnplugin<Options | undefined, false>(
 
             files.add(sidecar)
             try {
-              contracts[file] = await Fs.readFile(sidecar, 'utf8')
+              contracts[file] = await snapshot.read(sidecar)
             } catch (error) {
               if (
                 !required &&
@@ -235,8 +237,9 @@ const portable = createUnplugin<Options | undefined, false>(
           await scan(root)
           for (const [file, source] of sources) {
             const links = (imports[identity(file)] = Object.create(null))
-            for (const node of Parser.parseSync(file, source, {
-              sourceType: 'module',
+            for (const node of snapshot.parse({
+              moduleId: identity(file),
+              source,
             }).program.body) {
               if (
                 (node.type !== 'ImportDeclaration' &&
@@ -282,6 +285,7 @@ const portable = createUnplugin<Options | undefined, false>(
           }
 
           const result = compiler.compile({
+            [Syntax.cache]: snapshot.programs(modules),
             compiler: options.compiler,
             reset: options.reset ? Reset.read() : undefined,
             contracts,
@@ -500,6 +504,7 @@ const portable = createUnplugin<Options | undefined, false>(
               Catalogs.read,
             ),
           ).join('\n')
+          snapshot.retain(files, modules)
           graph = result
           if (meta.framework === 'webpack') emit(this)
           if (meta.framework === 'rollup')

package/src/variants.ts

@@ -51,6 +51,7 @@ type CheckedChoice<
       values: values,
     ) => CheckedStyles<body, tokens, layers, mappings> &
       Binding.Checked<body> &
+      Binding.TokenSlots<body, values, tokens, mappings> &
       (string extends keyof values
         ? never
         : values extends Record<string, string | number> &

package/src/vite/index.ts

@@ -14,6 +14,8 @@ import * as Reset from '../node/Reset.js'
 import { isBuiltin } from 'node:module'
 import * as Parser from 'oxc-parser'
 import * as Walker from 'oxc-walker'
+import * as Snapshot from '../node/internal/Snapshot.js'
+import * as Syntax from '../compiler/internal/Syntax.js'
 import * as Scope from '../compiler/internal/Scope.js'
 import type { Environment, Plugin, Rollup, ViteDevServer } from 'vite'
 import * as Graph from '../compiler/Graph.js'
@@ -40,6 +42,10 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
   if (native && options.compiler === false)
     throw new Error('Native builds require source compilation.')
 
+  const snapshots = new WeakMap<
+    Environment,
+    ReturnType<typeof Snapshot.create>
+  >()
   const states = new WeakMap<Environment, Map<string, Entry>>()
   const discoveries = new WeakMap<Environment, Promise<Map<string, string>>>()
   const contributionFiles = new WeakMap<Environment, Set<string>>()
@@ -51,6 +57,15 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
   const sourceEntrypoints = new Set<string>()
   let root: string
 
+  function snapshot(environment: Environment) {
+    let value = snapshots.get(environment)
+    if (!value) {
+      value = Snapshot.create()
+      snapshots.set(environment, value)
+    }
+    return value
+  }
+
   function entries(environment: Environment) {
     let state = states.get(environment)
 
@@ -273,7 +288,10 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
     file: string,
     event: string,
   ) {
-    if (event === 'delete') entries(environment).delete(file)
+    if (event === 'delete') {
+      entries(environment).delete(file)
+      snapshot(environment).delete(file, sourceId(file))
+    }
 
     const pending = discoveries.get(environment)
     if (!pending || !eager(file)) return
@@ -330,7 +348,7 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
       host.watch(file)
 
       const id = sourceId(file)
-      const text = source ?? (await Fs.readFile(file, 'utf8'))
+      const text = source ?? (await snapshot(entry.environment).read(file))
 
       modules[id] = text
 
@@ -338,8 +356,9 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
 
       imports[id] = resolutions
 
-      const parsed = Parser.parseSync('source.tsx', text, {
-        sourceType: 'module',
+      const parsed = snapshot(entry.environment).parse({
+        moduleId: id,
+        source: text,
       })
 
       for (const node of parsed.program.body) {
@@ -642,6 +661,7 @@ export function zyzz(options: zyzz.Options = {}): Plugin {
       )
 
     const result = entry.compiler.compile({
+      [Syntax.cache]: snapshot(entry.environment).programs(modules),
       compiler: options.compiler,
       reset: options.reset ? Reset.read() : undefined,
       native,