@oddsquad/vite-plugin-lit 0.6.1 → 0.7.0

package/lib/plugins/private-fields.js

@@ -0,0 +1,259 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+import {parseAst} from 'vite';
+import MagicString from 'magic-string';
+import {JS_FILE_RE} from './shared.js';
+const PRIVATE_KEY_HOLDERS = new Set([
+  'PropertyDefinition',
+  'MethodDefinition',
+  'AccessorProperty',
+]);
+const isNode = (v) =>
+  v !== null && typeof v === 'object' && typeof v.type === 'string';
+/** Private names declared directly in a class body. */
+const declaredNames = (cls) => {
+  const body = cls.body;
+  const out = [];
+  for (const member of body?.body ?? []) {
+    if (
+      PRIVATE_KEY_HOLDERS.has(member.type) &&
+      isNode(member.key) &&
+      member.key.type === 'PrivateIdentifier'
+    ) {
+      out.push(member);
+    }
+  }
+  return out;
+};
+/**
+ * Rewrites native `#private` class members into computed keys backed by
+ * `Symbol.for(...)`. The key is stable across re-evaluations of the module,
+ * so instances built by an earlier evaluation of a class and methods copied
+ * from a later one agree on where private state lives. That is what lets HMR
+ * patch a class in place. Returns `null` when there is nothing to rewrite or
+ * the module can't be handled safely.
+ */
+export const rewritePrivateNames = (code, id) => {
+  if (!code.includes('#')) {
+    return null;
+  }
+  const [file] = id.split('?', 1);
+  let ast;
+  try {
+    ast = parseAst(code, {sourceType: 'module'}, file);
+  } catch {
+    return null;
+  }
+  let prefix = '__litPriv';
+  while (code.includes(prefix)) {
+    prefix += '_';
+  }
+  const magic = new MagicString(code);
+  const decls = [];
+  const classKeyCounts = new Map();
+  let symbolCount = 0;
+  let failed = false;
+  const symbolFor = (scope, name) => scope.names.get(name);
+  const resolve = (stack, name) => {
+    for (let i = stack.length - 1; i >= 0; i--) {
+      const sym = symbolFor(stack[i], name);
+      if (sym !== undefined) {
+        return sym;
+      }
+    }
+    failed = true;
+    return '';
+  };
+  const classKeyOf = (cls, hint) => {
+    const cid = cls.id;
+    const base = cid?.name ?? hint ?? 'anonymous';
+    const n = classKeyCounts.get(base) ?? 0;
+    classKeyCounts.set(base, n + 1);
+    return n === 0 ? base : `${base}~${n}`;
+  };
+  const rewriteMemberAccess = (node, sym) => {
+    const prop = node.property;
+    // Find the `.` / `?.` token before the property; the object's own end
+    // can't be used because a parenthesised object ends before its `)`.
+    let i = prop.start - 1;
+    while (i >= 0 && /\s/.test(code[i])) {
+      i--;
+    }
+    if (code[i] !== '.') {
+      failed = true;
+      return;
+    }
+    const optional = code[i - 1] === '?';
+    const from = optional ? i - 1 : i;
+    magic.overwrite(from, prop.end, `${optional ? '?.' : ''}[${sym}]`);
+  };
+  const visit = (node, stack, hint) => {
+    if (failed || node === null || typeof node !== 'object') {
+      return;
+    }
+    if (Array.isArray(node)) {
+      for (const item of node) {
+        visit(item, stack, hint);
+      }
+      return;
+    }
+    if (!isNode(node)) {
+      return;
+    }
+    if (node.type === 'ClassDeclaration' || node.type === 'ClassExpression') {
+      const members = declaredNames(node);
+      const heritage = node.superClass;
+      // Decorators are outside the class body's private-name scope.
+      visit(node.decorators, stack, undefined);
+      visit(heritage, stack, undefined);
+      if (members.length === 0) {
+        visit(node.body, stack, undefined);
+        return;
+      }
+      const classKey = classKeyOf(node, hint);
+      const scope = {names: new Map()};
+      for (const member of members) {
+        const name = member.key.name;
+        if (!scope.names.has(name)) {
+          const ident = `${prefix}${symbolCount++}`;
+          scope.names.set(name, ident);
+          decls.push(
+            `const ${ident} = Symbol.for(${JSON.stringify(`@oddsquad/vite-plugin-lit#private:${file}:${classKey}#${name}`)});`
+          );
+        }
+        if (member.decorators?.length) {
+          // A decorator directly before a computed key would parse as an
+          // index access on the decorator expression.
+          failed = true;
+          return;
+        }
+        const key = member.key;
+        const sym = scope.names.get(name);
+        // A computed key at the start of a member can continue the previous
+        // member's initializer when that one relied on ASI (`a = 1` + newline
+        // + `#b`), so lead with an explicit terminator.
+        let lead = '';
+        if (member.start === key.start) {
+          let j = key.start - 1;
+          while (j >= 0 && /\s/.test(code[j])) {
+            j--;
+          }
+          if (code[j] !== '{' && code[j] !== ';') {
+            lead = ';';
+          }
+        }
+        magic.overwrite(key.start, key.end, `${lead}[${sym}]`);
+      }
+      // Members: keys of non-private members and all values live in scope.
+      const inner = [...stack, scope];
+      const body = node.body;
+      for (const member of body.body ?? []) {
+        for (const k in member) {
+          if (k === 'parent' || (k === 'key' && isPrivateKey(member))) {
+            continue;
+          }
+          visit(member[k], inner, undefined);
+        }
+      }
+      return;
+    }
+    if (node.type === 'MemberExpression' && isNode(node.property)) {
+      if (node.property.type === 'PrivateIdentifier') {
+        const sym = resolve(stack, node.property.name);
+        if (failed) {
+          return;
+        }
+        rewriteMemberAccess(node, sym);
+        visit(node.object, stack, undefined);
+        return;
+      }
+    }
+    if (
+      node.type === 'BinaryExpression' &&
+      isNode(node.left) &&
+      node.left.type === 'PrivateIdentifier'
+    ) {
+      const sym = resolve(stack, node.left.name);
+      if (failed) {
+        return;
+      }
+      magic.overwrite(node.left.start, node.left.end, sym);
+      visit(node.right, stack, undefined);
+      return;
+    }
+    let childHint;
+    for (const key in node) {
+      if (key === 'parent') {
+        continue;
+      }
+      switch (node.type) {
+        case 'VariableDeclarator':
+          childHint =
+            key === 'init' && isNode(node.id) && node.id.type === 'Identifier'
+              ? node.id.name
+              : undefined;
+          break;
+        case 'AssignmentExpression':
+          childHint =
+            key === 'right' &&
+            isNode(node.left) &&
+            node.left.type === 'Identifier'
+              ? node.left.name
+              : undefined;
+          break;
+        case 'ExportDefaultDeclaration':
+          childHint = 'default';
+          break;
+        default:
+          childHint = undefined;
+      }
+      visit(node[key], stack, childHint);
+    }
+  };
+  const isPrivateKey = (member) =>
+    PRIVATE_KEY_HOLDERS.has(member.type) &&
+    isNode(member.key) &&
+    member.key.type === 'PrivateIdentifier';
+  visit(ast, [], undefined);
+  if (failed || decls.length === 0) {
+    return null;
+  }
+  const header = `${decls.join('')}\n`;
+  if (code.startsWith('#!')) {
+    const nl = code.indexOf('\n');
+    if (nl === -1) {
+      return null;
+    }
+    magic.appendLeft(nl + 1, header);
+  } else {
+    magic.prepend(header);
+  }
+  return {code: magic.toString(), map: magic.generateMap({hires: 'boundary'})};
+};
+/**
+ * Dev-only transform that swaps native `#private` class members for stable
+ * symbol keys so HMR's in-place class patching keeps working. See
+ * {@link rewritePrivateNames}. `enabled` is read lazily because options are
+ * re-resolved against the loaded env after the plugin array is built.
+ */
+export const litPrivateFields = (enabled) => ({
+  name: 'lit-private-fields',
+  apply: 'serve',
+  transform(code, id, options) {
+    if (options?.ssr || !enabled()) {
+      return null;
+    }
+    if (id.startsWith('\0') || id.includes('/node_modules/')) {
+      return null;
+    }
+    const [file, query] = id.split('?', 2);
+    if (!JS_FILE_RE.test(file) && !query?.includes('html-proxy')) {
+      return null;
+    }
+    return rewritePrivateNames(code, id);
+  },
+});
+//# sourceMappingURL=private-fields.js.map

package/lib/plugins/private-fields.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"private-fields.js","sourceRoot":"","sources":["../../src/lib/plugins/private-fields.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAC,QAAQ,EAAC,MAAM,MAAM,CAAC;AAC9B,OAAO,WAAW,MAAM,cAAc,CAAC;AACvC,OAAO,EAAC,UAAU,EAAC,MAAM,aAAa,CAAC;AAcvC,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC;IAClC,oBAAoB;IACpB,kBAAkB;IAClB,kBAAkB;CACnB,CAAC,CAAC;AAEH,MAAM,MAAM,GAAG,CAAC,CAAU,EAAa,EAAE,CACvC,CAAC,KAAK,IAAI,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAQ,CAAU,CAAC,IAAI,KAAK,QAAQ,CAAC;AAE9E,uDAAuD;AACvD,MAAM,aAAa,GAAG,CAAC,GAAS,EAAU,EAAE;IAC1C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAmC,CAAC;IACrD,MAAM,GAAG,GAAW,EAAE,CAAC;IACvB,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;QACtC,IACE,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;YAClB,MAAM,CAAC,GAAG,CAAC,IAAI,KAAK,mBAAmB,EACvC,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CACjC,IAAY,EACZ,EAAU,EAC0D,EAAE;IACtE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAChC,IAAI,GAAS,CAAC;IACd,IAAI,CAAC;QACH,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAC,UAAU,EAAE,QAAQ,EAAC,EAAE,IAAI,CAAoB,CAAC;IACxE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,MAAM,GAAG,WAAW,CAAC;IACzB,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,CAAC;IAChB,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC;IACpC,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,MAAM,SAAS,GAAG,CAAC,KAAiB,EAAE,IAAY,EAAsB,EAAE,CACxE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAExB,MAAM,OAAO,GAAG,CAAC,KAAmB,EAAE,IAAY,EAAU,EAAE;QAC5D,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YACtC,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;gBACtB,OAAO,GAAG,CAAC;YACb,CAAC;QACH,CAAC;QACD,MAAM,GAAG,IAAI,CAAC;QACd,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,CAAC,GAAS,EAAE,IAAwB,EAAU,EAAE;QACjE,MAAM,GAAG,GAAG,GAAG,CAAC,EAAwC,CAAC;QACzD,MAAM,IAAI,GAAG,GAAG,EAAE,IAAI,IAAI,IAAI,IAAI,WAAW,CAAC;QAC9C,MAAM,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC;IACzC,CAAC,CAAC;IAEF,MAAM,mBAAmB,GAAG,CAAC,IAAU,EAAE,GAAW,EAAQ,EAAE;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,QAAgB,CAAC;QACnC,sEAAsE;QACtE,oEAAoE;QACpE,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;QACvB,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACpC,CAAC,EAAE,CAAC;QACN,CAAC;QACD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;YACpB,MAAM,GAAG,IAAI,CAAC;YACd,OAAO;QACT,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC;QACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,GAAG,GAAG,CAAC,CAAC;IACrE,CAAC,CAAC;IAEF,MAAM,KAAK,GAAG,CACZ,IAAa,EACb,KAAmB,EACnB,IAAwB,EAClB,EAAE;QACR,IAAI,MAAM,IAAI,IAAI,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxD,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;gBACxB,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC3B,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACxE,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC;YACjC,8DAA8D;YAC9D,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACzC,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YAClC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACxC,MAAM,KAAK,GAAe,EAAC,KAAK,EAAE,IAAI,GAAG,EAAE,EAAC,CAAC;YAC7C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,IAAI,GAAI,MAAM,CAAC,GAAiC,CAAC,IAAI,CAAC;gBAC5D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3B,MAAM,KAAK,GAAG,GAAG,MAAM,GAAG,WAAW,EAAE,EAAE,CAAC;oBAC1C,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;oBAC7B,KAAK,CAAC,IAAI,CACR,SAAS,KAAK,iBAAiB,IAAI,CAAC,SAAS,CAC3C,qCAAqC,IAAI,IAAI,QAAQ,IAAI,IAAI,EAAE,CAChE,IAAI,CACN,CAAC;gBACJ,CAAC;gBACD,IAAK,MAAM,CAAC,UAAoC,EAAE,MAAM,EAAE,CAAC;oBACzD,+DAA+D;oBAC/D,4CAA4C;oBAC5C,MAAM,GAAG,IAAI,CAAC;oBACd,OAAO;gBACT,CAAC;gBACD,MAAM,GAAG,GAAG,MAAM,CAAC,GAAW,CAAC;gBAC/B,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAE,CAAC;gBACnC,oEAAoE;gBACpE,sEAAsE;gBACtE,gDAAgD;gBAChD,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,IAAI,MAAM,CAAC,KAAK,KAAK,GAAG,CAAC,KAAK,EAAE,CAAC;oBAC/B,IAAI,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC;oBACtB,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACpC,CAAC,EAAE,CAAC;oBACN,CAAC;oBACD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;wBACvC,IAAI,GAAG,GAAG,CAAC;oBACb,CAAC;gBACH,CAAC;gBACD,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,GAAG,GAAG,CAAC,CAAC;YACzD,CAAC;YACD,qEAAqE;YACrE,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,EAAE,KAAK,CAAC,CAAC;YAChC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAY,CAAC;YAC/B,KAAK,MAAM,MAAM,IAAK,IAAI,CAAC,IAAe,IAAI,EAAE,EAAE,CAAC;gBACjD,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;oBACvB,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;wBAC5D,SAAS;oBACX,CAAC;oBACD,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9D,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;gBAC/C,MAAM,GAAG,GAAG,OAAO,CACjB,KAAK,EACJ,IAAI,CAAC,QAAsC,CAAC,IAAI,CAClD,CAAC;gBACF,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO;gBACT,CAAC;gBACD,mBAAmB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBAC/B,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;gBACrC,OAAO;YACT,CAAC;QACH,CAAC;QAED,IACE,IAAI,CAAC,IAAI,KAAK,kBAAkB;YAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;YACjB,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,mBAAmB,EACtC,CAAC;YACD,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,EAAG,IAAI,CAAC,IAAkC,CAAC,IAAI,CAAC,CAAC;YAC1E,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YACD,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YACrD,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACpC,OAAO;QACT,CAAC;QAED,IAAI,SAA6B,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YACD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,oBAAoB;oBACvB,SAAS;wBACP,GAAG,KAAK,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,YAAY;4BAChE,CAAC,CAAE,IAAI,CAAC,EAAgC,CAAC,IAAI;4BAC7C,CAAC,CAAC,SAAS,CAAC;oBAChB,MAAM;gBACR,KAAK,sBAAsB;oBACzB,SAAS;wBACP,GAAG,KAAK,OAAO;4BACf,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;4BACjB,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY;4BAC7B,CAAC,CAAE,IAAI,CAAC,IAAkC,CAAC,IAAI;4BAC/C,CAAC,CAAC,SAAS,CAAC;oBAChB,MAAM;gBACR,KAAK,0BAA0B;oBAC7B,SAAS,GAAG,SAAS,CAAC;oBACtB,MAAM;gBACR;oBACE,SAAS,GAAG,SAAS,CAAC;YAC1B,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;QACrC,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,MAAY,EAAW,EAAE,CAC7C,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;QAClB,MAAM,CAAC,GAAG,CAAC,IAAI,KAAK,mBAAmB,CAAC;IAE1C,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;IAC1B,IAAI,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;IACrC,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QAC1B,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,CAAC,UAAU,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC;IACD,OAAO,EAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,KAAK,CAAC,WAAW,CAAC,EAAC,KAAK,EAAE,UAAU,EAAC,CAAC,EAAC,CAAC;AAC/E,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,OAAsB,EAAU,EAAE,CAAC,CAAC;IACnE,IAAI,EAAE,oBAAoB;IAC1B,KAAK,EAAE,OAAO;IACd,SAAS,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO;QACzB,IAAI,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACzD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAC7D,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,mBAAmB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACvC,CAAC;CACF,CAAC,CAAC","sourcesContent":["/**\n * @license\n * Copyright 2026 Google LLC\n * SPDX-License-Identifier: BSD-3-Clause\n */\n\nimport type {Plugin} from 'vite';\nimport {parseAst} from 'vite';\nimport MagicString from 'magic-string';\nimport {JS_FILE_RE} from './shared.js';\n\ntype Node = Record<string, unknown> & {\n  type: string;\n  start: number;\n  end: number;\n};\n\n/** One class body that declares at least one private name. */\ninterface ClassScope {\n  /** Private name (without `#`) to the identifier of its symbol constant. */\n  names: Map<string, string>;\n}\n\nconst PRIVATE_KEY_HOLDERS = new Set([\n  'PropertyDefinition',\n  'MethodDefinition',\n  'AccessorProperty',\n]);\n\nconst isNode = (v: unknown): v is Node =>\n  v !== null && typeof v === 'object' && typeof (v as Node).type === 'string';\n\n/** Private names declared directly in a class body. */\nconst declaredNames = (cls: Node): Node[] => {\n  const body = cls.body as {body?: Node[]} | undefined;\n  const out: Node[] = [];\n  for (const member of body?.body ?? []) {\n    if (\n      PRIVATE_KEY_HOLDERS.has(member.type) &&\n      isNode(member.key) &&\n      member.key.type === 'PrivateIdentifier'\n    ) {\n      out.push(member);\n    }\n  }\n  return out;\n};\n\n/**\n * Rewrites native `#private` class members into computed keys backed by\n * `Symbol.for(...)`. The key is stable across re-evaluations of the module,\n * so instances built by an earlier evaluation of a class and methods copied\n * from a later one agree on where private state lives. That is what lets HMR\n * patch a class in place. Returns `null` when there is nothing to rewrite or\n * the module can't be handled safely.\n */\nexport const rewritePrivateNames = (\n  code: string,\n  id: string\n): {code: string; map: ReturnType<MagicString['generateMap']>} | null => {\n  if (!code.includes('#')) {\n    return null;\n  }\n  const [file] = id.split('?', 1);\n  let ast: Node;\n  try {\n    ast = parseAst(code, {sourceType: 'module'}, file) as unknown as Node;\n  } catch {\n    return null;\n  }\n\n  let prefix = '__litPriv';\n  while (code.includes(prefix)) {\n    prefix += '_';\n  }\n\n  const magic = new MagicString(code);\n  const decls: string[] = [];\n  const classKeyCounts = new Map<string, number>();\n  let symbolCount = 0;\n  let failed = false;\n\n  const symbolFor = (scope: ClassScope, name: string): string | undefined =>\n    scope.names.get(name);\n\n  const resolve = (stack: ClassScope[], name: string): string => {\n    for (let i = stack.length - 1; i >= 0; i--) {\n      const sym = symbolFor(stack[i], name);\n      if (sym !== undefined) {\n        return sym;\n      }\n    }\n    failed = true;\n    return '';\n  };\n\n  const classKeyOf = (cls: Node, hint: string | undefined): string => {\n    const cid = cls.id as {name?: string} | null | undefined;\n    const base = cid?.name ?? hint ?? 'anonymous';\n    const n = classKeyCounts.get(base) ?? 0;\n    classKeyCounts.set(base, n + 1);\n    return n === 0 ? base : `${base}~${n}`;\n  };\n\n  const rewriteMemberAccess = (node: Node, sym: string): void => {\n    const prop = node.property as Node;\n    // Find the `.` / `?.` token before the property; the object's own end\n    // can't be used because a parenthesised object ends before its `)`.\n    let i = prop.start - 1;\n    while (i >= 0 && /\\s/.test(code[i])) {\n      i--;\n    }\n    if (code[i] !== '.') {\n      failed = true;\n      return;\n    }\n    const optional = code[i - 1] === '?';\n    const from = optional ? i - 1 : i;\n    magic.overwrite(from, prop.end, `${optional ? '?.' : ''}[${sym}]`);\n  };\n\n  const visit = (\n    node: unknown,\n    stack: ClassScope[],\n    hint: string | undefined\n  ): void => {\n    if (failed || node === null || typeof node !== 'object') {\n      return;\n    }\n    if (Array.isArray(node)) {\n      for (const item of node) {\n        visit(item, stack, hint);\n      }\n      return;\n    }\n    if (!isNode(node)) {\n      return;\n    }\n\n    if (node.type === 'ClassDeclaration' || node.type === 'ClassExpression') {\n      const members = declaredNames(node);\n      const heritage = node.superClass;\n      // Decorators are outside the class body's private-name scope.\n      visit(node.decorators, stack, undefined);\n      visit(heritage, stack, undefined);\n      if (members.length === 0) {\n        visit(node.body, stack, undefined);\n        return;\n      }\n      const classKey = classKeyOf(node, hint);\n      const scope: ClassScope = {names: new Map()};\n      for (const member of members) {\n        const name = (member.key as unknown as {name: string}).name;\n        if (!scope.names.has(name)) {\n          const ident = `${prefix}${symbolCount++}`;\n          scope.names.set(name, ident);\n          decls.push(\n            `const ${ident} = Symbol.for(${JSON.stringify(\n              `@oddsquad/vite-plugin-lit#private:${file}:${classKey}#${name}`\n            )});`\n          );\n        }\n        if ((member.decorators as unknown[] | undefined)?.length) {\n          // A decorator directly before a computed key would parse as an\n          // index access on the decorator expression.\n          failed = true;\n          return;\n        }\n        const key = member.key as Node;\n        const sym = scope.names.get(name)!;\n        // A computed key at the start of a member can continue the previous\n        // member's initializer when that one relied on ASI (`a = 1` + newline\n        // + `#b`), so lead with an explicit terminator.\n        let lead = '';\n        if (member.start === key.start) {\n          let j = key.start - 1;\n          while (j >= 0 && /\\s/.test(code[j])) {\n            j--;\n          }\n          if (code[j] !== '{' && code[j] !== ';') {\n            lead = ';';\n          }\n        }\n        magic.overwrite(key.start, key.end, `${lead}[${sym}]`);\n      }\n      // Members: keys of non-private members and all values live in scope.\n      const inner = [...stack, scope];\n      const body = node.body as Node;\n      for (const member of (body.body as Node[]) ?? []) {\n        for (const k in member) {\n          if (k === 'parent' || (k === 'key' && isPrivateKey(member))) {\n            continue;\n          }\n          visit(member[k], inner, undefined);\n        }\n      }\n      return;\n    }\n\n    if (node.type === 'MemberExpression' && isNode(node.property)) {\n      if (node.property.type === 'PrivateIdentifier') {\n        const sym = resolve(\n          stack,\n          (node.property as unknown as {name: string}).name\n        );\n        if (failed) {\n          return;\n        }\n        rewriteMemberAccess(node, sym);\n        visit(node.object, stack, undefined);\n        return;\n      }\n    }\n\n    if (\n      node.type === 'BinaryExpression' &&\n      isNode(node.left) &&\n      node.left.type === 'PrivateIdentifier'\n    ) {\n      const sym = resolve(stack, (node.left as unknown as {name: string}).name);\n      if (failed) {\n        return;\n      }\n      magic.overwrite(node.left.start, node.left.end, sym);\n      visit(node.right, stack, undefined);\n      return;\n    }\n\n    let childHint: string | undefined;\n    for (const key in node) {\n      if (key === 'parent') {\n        continue;\n      }\n      switch (node.type) {\n        case 'VariableDeclarator':\n          childHint =\n            key === 'init' && isNode(node.id) && node.id.type === 'Identifier'\n              ? (node.id as unknown as {name: string}).name\n              : undefined;\n          break;\n        case 'AssignmentExpression':\n          childHint =\n            key === 'right' &&\n            isNode(node.left) &&\n            node.left.type === 'Identifier'\n              ? (node.left as unknown as {name: string}).name\n              : undefined;\n          break;\n        case 'ExportDefaultDeclaration':\n          childHint = 'default';\n          break;\n        default:\n          childHint = undefined;\n      }\n      visit(node[key], stack, childHint);\n    }\n  };\n\n  const isPrivateKey = (member: Node): boolean =>\n    PRIVATE_KEY_HOLDERS.has(member.type) &&\n    isNode(member.key) &&\n    member.key.type === 'PrivateIdentifier';\n\n  visit(ast, [], undefined);\n  if (failed || decls.length === 0) {\n    return null;\n  }\n\n  const header = `${decls.join('')}\\n`;\n  if (code.startsWith('#!')) {\n    const nl = code.indexOf('\\n');\n    if (nl === -1) {\n      return null;\n    }\n    magic.appendLeft(nl + 1, header);\n  } else {\n    magic.prepend(header);\n  }\n  return {code: magic.toString(), map: magic.generateMap({hires: 'boundary'})};\n};\n\n/**\n * Dev-only transform that swaps native `#private` class members for stable\n * symbol keys so HMR's in-place class patching keeps working. See\n * {@link rewritePrivateNames}. `enabled` is read lazily because options are\n * re-resolved against the loaded env after the plugin array is built.\n */\nexport const litPrivateFields = (enabled: () => boolean): Plugin => ({\n  name: 'lit-private-fields',\n  apply: 'serve',\n  transform(code, id, options) {\n    if (options?.ssr || !enabled()) {\n      return null;\n    }\n    if (id.startsWith('\\0') || id.includes('/node_modules/')) {\n      return null;\n    }\n    const [file, query] = id.split('?', 2);\n    if (!JS_FILE_RE.test(file) && !query?.includes('html-proxy')) {\n      return null;\n    }\n    return rewritePrivateNames(code, id);\n  },\n});\n"]}

package/lib/runtime/child-state.d.ts

@@ -0,0 +1,100 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+/**
+ * Carries child element state across a template re-instantiation.
+ *
+ * When a hot patch changes a template's text, lit-html sees a new template,
+ * clones fresh DOM and swaps it in: every custom element inside the edited
+ * template is re-created and starts from its defaults. For a short window
+ * after a hot patch this module pairs each element dropped from a root with
+ * the next freshly created element of the same tag connected in that root
+ * (FIFO, i.e. document order), then either copies the old element's state
+ * onto the new one (`'transfer'`) or, where the new element carries no
+ * bindings, puts the old element back in its place (`'reuse'`).
+ *
+ * Dependency-free and duck-typed like `patch.ts`. The lit-html internals it
+ * reads (`_$litPart$`, `_$committedValue`, `_$template`, `_$parts`,
+ * `_$startNode`, `_$endNode`) are looked up by their development names first
+ * and then by the stable minified names lit's prod build gives them
+ * (`_$AH`, `_$AD`, `_$AV`, `_$AA`, `_$AB`). Every access is guarded; when the
+ * part tree can't be read, transfer skips only attribute-driven keys and
+ * reuse falls back to transfer.
+ */
+export type ChildStateMode = 'reset' | 'transfer' | 'reuse';
+/** `Symbol.for` description prefix of the dev-only `#private` rewrite. */
+export declare const PRIVATE_KEY_PREFIX = "@oddsquad/vite-plugin-lit#private:";
+/** Hard cap on how long a transfer window stays open. */
+export declare const WINDOW_CAP_MS = 1000;
+/** Duck-typed lit-html part. */
+export interface PartLike {
+    type: number;
+    element?: unknown;
+    name?: unknown;
+}
+/** Duck-typed element the window works with. */
+export type ElementLike = HTMLElement & {
+    renderRoot?: unknown;
+    hasUpdated?: boolean;
+    updateComplete?: unknown;
+};
+/**
+ * Every lit-html part rendered into the tree `el` lives in: the parts of each
+ * render container (`_$litPart$`) among `el`'s ancestors up to its root.
+ * `undefined` when `el` isn't in a tree or the walk hits a surprise.
+ */
+export declare const partsAround: (el: Node) => PartLike[] | undefined;
+/**
+ * Reactive property keys the new element already got from its template:
+ * keys whose attribute is present on it (static attributes), plus, when the
+ * part tree could be read, keys bound by a property, boolean or attribute
+ * part targeting it.
+ */
+export declare const templateDrivenKeys: (el: ElementLike, parts: readonly PartLike[] | undefined) => Set<PropertyKey>;
+/** Own symbol keys of `el` minted by the dev-only `#private` rewrite. */
+export declare const privateKeys: (el: object) => symbol[];
+/**
+ * Copies `from`'s state onto `to`: reactive properties not driven by `to`'s
+ * template, and the `#private` members the dev rewrite turned into
+ * `Symbol.for` keys. Nothing else — Lit's own bookkeeping lives in string
+ * keys and must stay per-instance.
+ */
+export declare const transferState: (from: ElementLike, to: ElementLike, parts: readonly PartLike[] | undefined) => void;
+/**
+ * Whether `el` (and its light DOM) carries no bindings, so another element
+ * can take its place without lit-html noticing: no part targets it or an
+ * element inside it, no child part's markers or parent lie inside it, and
+ * nothing rendered into it as a container. `false` when `parts` is unknown.
+ */
+export declare const isReuseSafe: (el: ElementLike, parts: readonly PartLike[] | undefined) => boolean;
+/**
+ * Puts `old` back where `fresh` is: `old` takes `fresh`'s attributes and
+ * light-DOM children, then replaces it. `old` keeps its identity, state,
+ * shadow DOM and controllers, and reconnects.
+ */
+export declare const swapBack: (old: ElementLike, fresh: ElementLike) => void;
+export interface ChildStateOptions {
+    /** Read at every `open()`. */
+    mode: () => ChildStateMode;
+    /** Hard cap on a window's lifetime. Defaults to {@link WINDOW_CAP_MS}. */
+    capMs?: number;
+}
+export interface ChildState {
+    /**
+     * From the instrumented `connectedCallback`, before the original runs.
+     * `fresh` is whether the element has never been connected before.
+     */
+    connected(el: ElementLike, fresh: boolean): void;
+    /** From the instrumented `disconnectedCallback`. */
+    disconnected(el: ElementLike): void;
+    /** A hot patch is about to re-render instances: open (or extend) a window. */
+    open(): void;
+    /** Keep the window open until these instances' updates complete. */
+    watch(els: Iterable<ElementLike>): void;
+    /** Whether a window is currently open. */
+    readonly active: boolean;
+}
+export declare const createChildState: (options: ChildStateOptions) => ChildState;
+//# sourceMappingURL=child-state.d.ts.map

package/lib/runtime/child-state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"child-state.d.ts","sourceRoot":"","sources":["../../src/lib/runtime/child-state.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;;;;;;;;;GAmBG;AAEH,MAAM,MAAM,cAAc,GAAG,OAAO,GAAG,UAAU,GAAG,OAAO,CAAC;AAE5D,0EAA0E;AAC1E,eAAO,MAAM,kBAAkB,uCAAuC,CAAC;AAEvE,yDAAyD;AACzD,eAAO,MAAM,aAAa,OAAO,CAAC;AAelC,gCAAgC;AAChC,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED,gDAAgD;AAChD,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG;IACtC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B,CAAC;AAuDF;;;;GAIG;AACH,eAAO,MAAM,WAAW,OAAQ,IAAI,KAAG,QAAQ,EAAE,GAAG,SAkBnD,CAAC;AAiCF;;;;;GAKG;AACH,eAAO,MAAM,kBAAkB,OACzB,WAAW,SACR,SAAS,QAAQ,EAAE,GAAG,SAAS,KACrC,GAAG,CAAC,WAAW,CAyBjB,CAAC;AAEF,yEAAyE;AACzE,eAAO,MAAM,WAAW,OAAQ,MAAM,KAAG,MAAM,EAG5C,CAAC;AAEJ;;;;;GAKG;AACH,eAAO,MAAM,aAAa,SAClB,WAAW,MACb,WAAW,SACR,SAAS,QAAQ,EAAE,GAAG,SAAS,KACrC,IA6BF,CAAC;AAQF;;;;;GAKG;AACH,eAAO,MAAM,WAAW,OAClB,WAAW,SACR,SAAS,QAAQ,EAAE,GAAG,SAAS,KACrC,OA6BF,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,QAAQ,QAAS,WAAW,SAAS,WAAW,KAAG,IAa/D,CAAC;AAwBF,MAAM,WAAW,iBAAiB;IAChC,8BAA8B;IAC9B,IAAI,EAAE,MAAM,cAAc,CAAC;IAC3B,0EAA0E;IAC1E,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,SAAS,CAAC,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC;IACjD,oDAAoD;IACpD,YAAY,CAAC,EAAE,EAAE,WAAW,GAAG,IAAI,CAAC;IACpC,8EAA8E;IAC9E,IAAI,IAAI,IAAI,CAAC;IACb,oEAAoE;IACpE,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC;IACxC,0CAA0C;IAC1C,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;CAC1B;AAED,eAAO,MAAM,gBAAgB,YAAa,iBAAiB,KAAG,UA6M7D,CAAC"}