@weasel-js/forge 1.15.2 → 1.16.1

package/dist/vite/index.js

@@ -1,7 +1,7 @@
-import { CAPTURE_PAGES, DEFAULT_TITLE, ISSUES_PATH, CAPTURE_PATH, issueLabel, parsePayload, messageOf } from '../chunk-MRYXJTDO.js';
-import { byComponent } from '../chunk-RVJHUY7W.js';
+import { CAPTURE_PAGES, DEFAULT_TITLE, ISSUES_PATH, CAPTURE_PATH, issueLabel, parsePayload, messageOf } from '../chunk-7YVJS4QS.js';
+import { byComponent } from '../chunk-ZGELUE2H.js';
 import { isGallery } from '../chunk-VUPN73JP.js';
-import { PROP_ARG_TYPES_EXPORT, storyNameFromExport, storyId } from '../chunk-5WS67HMU.js';
+import { PROP_ARG_TYPES_EXPORT, API_EXPORT, storyNameFromExport, storyId } from '../chunk-PN4IFL5E.js';
 import { readFileSync, mkdtempSync, existsSync, mkdirSync, writeFileSync, globSync, statSync, watch, rmSync } from 'fs';
 import { resolve, relative, sep, isAbsolute, join, dirname, basename, matchesGlob, posix } from 'path';
 import { parse } from '@babel/parser';
@@ -55,6 +55,68 @@ function hoistPages(bundle, emit) {
     emit({ type: "asset", fileName: `${name}.html`, source: up(asset.source) });
   }
 }
+
+// src/vite/componentOf.ts
+var titleStandIn = (title) => title.split("/").pop()?.replace(/\s+/g, "");
+function componentOf(tsm, checker, source, fallback) {
+  const module = checker.getSymbolAtLocation(source);
+  const exported = module && checker.tryGetMemberInModuleExports("default", module);
+  if (!exported) return null;
+  const meta = checker.getTypeOfSymbolAtLocation(resolved(tsm, checker, exported), source);
+  let named = meta.getProperty("component");
+  let symbol = named && assigned(tsm, checker, resolved(tsm, checker, exported));
+  const local = !symbol || symbol.getDeclarations()?.some((declaration2) => declaration2.getSourceFile() === source);
+  const imported = local && fallback ? importNamed(tsm, checker, source, fallback) : void 0;
+  if (imported) {
+    named = void 0;
+    symbol = imported;
+  }
+  if (!named && !symbol) return null;
+  const type = checker.getNonNullableType(checker.getTypeOfSymbolAtLocation(symbol ?? named, source));
+  const declaration = symbol?.getDeclarations()?.[0];
+  for (const part of type.isUnion() ? type.types : [type]) {
+    const [call] = part.getCallSignatures();
+    const [param] = call?.getParameters() ?? [];
+    if (call) return { props: param ? checker.getTypeOfSymbolAtLocation(param, source) : null, declaration };
+    const instance = part.getConstructSignatures()[0]?.getReturnType();
+    const held = instance?.getProperty("props");
+    if (held) return { props: checker.getTypeOfSymbolAtLocation(held, source), declaration };
+  }
+  return null;
+}
+var resolved = (tsm, checker, symbol) => symbol.flags & tsm.SymbolFlags.Alias ? checker.getAliasedSymbol(symbol) : symbol;
+function assigned(tsm, checker, meta) {
+  const declaration = meta.getDeclarations()?.[0];
+  let node = declaration && tsm.isVariableDeclaration(declaration) ? declaration.initializer : declaration && tsm.isExportAssignment(declaration) ? declaration.expression : void 0;
+  while (node && !tsm.isObjectLiteralExpression(node)) {
+    if (tsm.isCallExpression(node)) node = node.arguments[0];
+    else if (tsm.isParenthesizedExpression(node) || tsm.isAsExpression(node) || tsm.isSatisfiesExpression(node)) {
+      node = node.expression;
+    } else node = void 0;
+  }
+  const property = node?.properties.find((p) => p.name && tsm.isIdentifier(p.name) && p.name.text === "component");
+  let value;
+  if (property && tsm.isShorthandPropertyAssignment(property)) value = checker.getShorthandAssignmentValueSymbol(property);
+  else if (property && tsm.isPropertyAssignment(property)) value = checker.getSymbolAtLocation(property.initializer);
+  return value && resolved(tsm, checker, value);
+}
+function importNamed(tsm, checker, source, name) {
+  for (const statement of source.statements) {
+    if (!tsm.isImportDeclaration(statement)) continue;
+    const clause = statement.importClause;
+    const bindings = clause?.namedBindings;
+    const names = [
+      ...clause?.name ? [clause.name] : [],
+      ...bindings && tsm.isNamedImports(bindings) ? bindings.elements.map((element) => element.name) : []
+    ];
+    const found = names.find((id) => id.text === name);
+    const symbol = found && checker.getSymbolAtLocation(found);
+    if (symbol) return resolved(tsm, checker, symbol);
+  }
+  return void 0;
+}
+
+// src/vite/depGraph.ts
 var keyOf = (file, local) => `${file}#${local}`;
 var fileOfKey = (key) => key.slice(0, key.lastIndexOf("#"));
 var SCRIPT = /\.[cm]?[jt]sx?$/;
@@ -130,10 +192,10 @@ function createDepGraph({ root, read }) {
         const target = await resolveFrom(imp.spec, importer);
         return target ? declaring(target, imp.imported, depth) : null;
       };
-      const componentOf = async (entry) => {
+      const componentOf2 = async (entry) => {
         const mod = info(entry.file);
         if (!mod) return null;
-        const byTitle = entry.title.split("/").pop()?.replace(/\s+/g, "");
+        const byTitle = titleStandIn(entry.title);
         for (const local of [entry.componentName, byTitle]) {
           const imp = local ? mod.imports.get(local) : void 0;
           const key = imp ? await importKey(entry.file, imp) : null;
@@ -153,7 +215,7 @@ function createDepGraph({ root, read }) {
       for (const [title, list] of titles) {
         let key = null;
         for (const file of new Set(list.map((e) => e.file))) {
-          key = await componentOf(list.find((e) => e.file === file));
+          key = await componentOf2(list.find((e) => e.file === file));
           if (key) break;
         }
         components.set(title, key);
@@ -929,9 +991,9 @@ function createWrapperResolver(read) {
 }
 
 // src/vite/indexStory.ts
-function indexStory(code, file, source, resolved) {
-  const foreign = resolved ? foreignFrom(resolved.foreign) : void 0;
-  return indexFile(code, file, autoTitle(file, source.root, source.stories), resolved?.wrappers, foreign);
+function indexStory(code, file, source, resolved2) {
+  const foreign = resolved2 ? foreignFrom(resolved2.foreign) : void 0;
+  return indexFile(code, file, autoTitle(file, source.root, source.stories), resolved2?.wrappers, foreign);
 }
 async function indexStoryResolved(code, file, source, resolve6, decide = createWrapperResolver((path) => readFileSync(path, "utf8"))) {
   const resolution = await decide(file, foreignCallees(code, file), resolve6);
@@ -949,9 +1011,139 @@ function ownEntries(realms) {
   }
   return [];
 }
+
+// src/vite/apiTable.ts
+var RESERVED = /* @__PURE__ */ new Set(["ref", "key"]);
+var IN_PACKAGES = /[\\/]node_modules[\\/]/;
+function apiTable(reading, declaration, props) {
+  const { checker } = reading;
+  const sources = /* @__PURE__ */ new Set();
+  const record2 = (symbol) => {
+    for (const file of filesOf(symbol)) if (!IN_PACKAGES.test(file)) sources.add(file);
+  };
+  const defaults = declaration ? destructuredDefaults(reading.tsm, declaration) : /* @__PURE__ */ new Map();
+  const listed = [];
+  const inherited = /* @__PURE__ */ new Map();
+  let ref = null;
+  for (const prop2 of props ? checker.getPropertiesOfType(checker.getApparentType(props)) : []) {
+    const name = prop2.getName();
+    if (name === "ref") ref = refOf(reading, prop2, record2);
+    if (RESERVED.has(name)) continue;
+    const files = filesOf(prop2);
+    if (files.length > 0 && files.every(reading.skipped)) {
+      const from = containerName(reading.tsm, prop2) ?? packageOf2(files[0]);
+      inherited.set(from, (inherited.get(from) ?? 0) + 1);
+      continue;
+    }
+    listed.push(propEntry(reading, prop2, defaults.get(name)));
+    record2(prop2);
+  }
+  return {
+    props: listed,
+    inherited: [...inherited].map(([from, count]) => ({ from, count })),
+    ref,
+    sources: [...sources]
+  };
+}
+var filesOf = (symbol) => (symbol.getDeclarations() ?? []).map((declaration) => declaration.getSourceFile().fileName);
+var docOf2 = ({ tsm, checker }, symbol) => tsm.displayPartsToString(symbol.getDocumentationComment(checker)).replace(/\s+/g, " ").trim();
+function tagText({ tsm, checker }, symbol, names) {
+  const tag = symbol.getJsDocTags(checker).find((t) => names.includes(t.name));
+  return tag ? tsm.displayPartsToString(tag.text).trim() : void 0;
+}
+function propEntry(reading, prop2, destructured) {
+  const entry = {
+    name: prop2.getName(),
+    type: writtenType(reading, prop2),
+    required: !(prop2.flags & reading.tsm.SymbolFlags.Optional)
+  };
+  const fallback = destructured ?? tagText(reading, prop2, ["default", "defaultValue"]);
+  if (fallback) entry.default = fallback;
+  const description = docOf2(reading, prop2);
+  if (description) entry.description = description;
+  const deprecated = tagText(reading, prop2, ["deprecated"]);
+  if (deprecated !== void 0) entry.deprecated = deprecated;
+  return entry;
+}
+function writtenType({ tsm, checker, at }, symbol) {
+  const type = checker.getNonNullableType(checker.getTypeOfSymbolAtLocation(symbol, at));
+  const [declaration, ...others] = symbol.getDeclarations() ?? [];
+  const node = declaration && others.length === 0 && tsm.isPropertySignature(declaration) ? declaration.type : void 0;
+  if (node && checker.getNonNullableType(checker.getTypeFromTypeNode(node)) === type) return node.getText().replace(/\s+/g, " ");
+  return checker.typeToString(type, void 0, tsm.TypeFormatFlags.NoTruncation);
+}
+function packageOf2(file) {
+  const parts = file.split(/[\\/]/);
+  const at = parts.lastIndexOf("node_modules");
+  if (at < 0 || at + 1 >= parts.length) return parts.at(-1) ?? file;
+  const name = parts[at + 1];
+  return name.startsWith("@") && parts[at + 2] ? `${name}/${parts[at + 2]}` : name;
+}
+function containerName(tsm, prop2) {
+  const parent = prop2.getDeclarations()?.[0]?.parent;
+  if (!parent) return void 0;
+  if (tsm.isInterfaceDeclaration(parent) || tsm.isClassDeclaration(parent)) return parent.name?.text;
+  if (tsm.isTypeLiteralNode(parent) && tsm.isTypeAliasDeclaration(parent.parent)) return parent.parent.name.text;
+  return void 0;
+}
+function refOf(reading, prop2, record2) {
+  const { tsm, checker, at } = reading;
+  const type = checker.getNonNullableType(checker.getTypeOfSymbolAtLocation(prop2, at));
+  for (const part of type.isUnion() ? type.types : [type]) {
+    const current = part.getProperty("current");
+    if (!current) continue;
+    const held = checker.getNonNullableType(checker.getTypeOfSymbolAtLocation(current, at));
+    const name = checker.typeToString(held, void 0, tsm.TypeFormatFlags.NoTruncation);
+    const declared = (held.getSymbol() ?? held.aliasSymbol)?.getDeclarations() ?? [];
+    if (declared.length > 0 && declared.every((d) => reading.platform(d.getSourceFile().fileName))) {
+      return { type: name, members: [] };
+    }
+    const members = checker.getPropertiesOfType(held).map((member) => {
+      record2(member);
+      return memberEntry(reading, member);
+    });
+    return { type: name, members };
+  }
+  return null;
+}
+function memberEntry(reading, member) {
+  const { tsm, checker, at } = reading;
+  const type = checker.getTypeOfSymbolAtLocation(member, at);
+  const [call] = type.getCallSignatures();
+  const declarations = member.getDeclarations() ?? [];
+  const entry = {
+    name: member.getName(),
+    kind: call ? "method" : "property",
+    signature: call ? checker.signatureToString(call, void 0, tsm.TypeFormatFlags.WriteArrowStyleSignature | tsm.TypeFormatFlags.NoTruncation) : checker.typeToString(type, void 0, tsm.TypeFormatFlags.NoTruncation),
+    readonly: declarations.some((d) => (tsm.getCombinedModifierFlags(d) & tsm.ModifierFlags.Readonly) !== 0)
+  };
+  const description = docOf2(reading, member);
+  if (description) entry.description = description;
+  return entry;
+}
+function destructuredDefaults(tsm, declaration) {
+  const defaults = /* @__PURE__ */ new Map();
+  let node = tsm.isVariableDeclaration(declaration) ? declaration.initializer : declaration;
+  while (node && (tsm.isCallExpression(node) || tsm.isParenthesizedExpression(node) || tsm.isAsExpression(node))) {
+    node = tsm.isCallExpression(node) ? node.arguments[0] : node.expression;
+  }
+  if (!node || !(tsm.isArrowFunction(node) || tsm.isFunctionExpression(node) || tsm.isFunctionDeclaration(node))) {
+    return defaults;
+  }
+  const pattern = node.parameters[0]?.name;
+  if (!pattern || !tsm.isObjectBindingPattern(pattern)) return defaults;
+  for (const element of pattern.elements) {
+    const key = element.propertyName ?? element.name;
+    if (element.initializer && tsm.isIdentifier(key)) defaults.set(key.text, element.initializer.getText());
+  }
+  return defaults;
+}
+
+// src/vite/propTypes.ts
 var PLATFORM = /[\\/]node_modules[\\/](@types[\\/]react(-dom)?|typescript)[\\/]/;
 async function createPropReader(root, stories, { skip } = {}) {
-  const skipped = (file) => PLATFORM.test(file) || skip?.(file) === true;
+  const platform = (file) => PLATFORM.test(file);
+  const skipped = (file) => platform(file) || skip?.(file) === true;
   let tsm;
   try {
     tsm = (await import('typescript')).default;
@@ -990,38 +1182,29 @@ async function createPropReader(root, stories, { skip } = {}) {
     getDirectories: sys.getDirectories
   };
   const service = tsm.createLanguageService(host, tsm.createDocumentRegistry());
+  const sourceOf = (file) => {
+    const program = service.getProgram();
+    const source = program?.getSourceFile(file);
+    return program && source ? { checker: program.getTypeChecker(), source } : null;
+  };
   return {
     read(file) {
-      const program = service.getProgram();
-      const source = program?.getSourceFile(file);
-      if (!program || !source) return null;
-      const checker = program.getTypeChecker();
-      const props = componentProps(tsm, checker, source);
+      const found = sourceOf(file);
+      if (!found) return null;
+      const { checker, source } = found;
+      const props = componentOf(tsm, checker, source)?.props;
       return props ? propTable(tsm, checker, props, source, skipped) : null;
+    },
+    api(file, fallback) {
+      const found = sourceOf(file);
+      if (!found) return null;
+      const { checker, source } = found;
+      const component = componentOf(tsm, checker, source, fallback);
+      if (!component) return null;
+      return apiTable({ tsm, checker, at: source, skipped, platform }, component.declaration, component.props);
     }
   };
 }
-function componentProps(tsm, checker, source) {
-  const module = checker.getSymbolAtLocation(source);
-  const exported = module && checker.tryGetMemberInModuleExports("default", module);
-  if (!exported) return null;
-  const meta = checker.getTypeOfSymbolAtLocation(
-    exported.flags & tsm.SymbolFlags.Alias ? checker.getAliasedSymbol(exported) : exported,
-    source
-  );
-  const named = meta.getProperty("component");
-  if (!named) return null;
-  const component = checker.getNonNullableType(checker.getTypeOfSymbolAtLocation(named, source));
-  for (const part of component.isUnion() ? component.types : [component]) {
-    const [call] = part.getCallSignatures();
-    const [param] = call?.getParameters() ?? [];
-    if (call) return param ? checker.getTypeOfSymbolAtLocation(param, source) : null;
-    const instance = part.getConstructSignatures()[0]?.getReturnType();
-    const held = instance?.getProperty("props");
-    if (held) return checker.getTypeOfSymbolAtLocation(held, source);
-  }
-  return null;
-}
 function propTable(tsm, checker, props, at, skipped) {
   const argTypes = {};
   const reading = { tsm, checker, at, skipped, sources: /* @__PURE__ */ new Set() };
@@ -1041,9 +1224,9 @@ function propTable(tsm, checker, props, at, skipped) {
   return { argTypes, sources: [...reading.sources] };
 }
 var declaredIn = (symbol) => (symbol.getDeclarations() ?? []).map((declaration) => declaration.getSourceFile().fileName);
-var IN_PACKAGES = /[\\/]node_modules[\\/]/;
+var IN_PACKAGES2 = /[\\/]node_modules[\\/]/;
 function record({ sources }, files) {
-  for (const file of files) if (!IN_PACKAGES.test(file)) sources.add(file);
+  for (const file of files) if (!IN_PACKAGES2.test(file)) sources.add(file);
 }
 function controlFor(reading, type, written) {
   const { TypeFlags } = reading.tsm;
@@ -1099,11 +1282,11 @@ function shapeOf(reading, type, depth) {
   if (whole.getCallSignatures().length > 0 || whole.getConstructSignatures().length > 0) return null;
   if (checker.getIndexInfosOfType(whole).length > 0) return null;
   const declared = (whole.getSymbol() ?? whole.aliasSymbol)?.getDeclarations() ?? [];
-  if (declared.some((declaration) => IN_PACKAGES.test(declaration.getSourceFile().fileName))) return null;
+  if (declared.some((declaration) => IN_PACKAGES2.test(declaration.getSourceFile().fileName))) return null;
   const fields = {};
   for (const prop2 of checker.getPropertiesOfType(whole)) {
     const files = declaredIn(prop2);
-    if (files.some((file) => IN_PACKAGES.test(file) || reading.skipped(file))) return null;
+    if (files.some((file) => IN_PACKAGES2.test(file) || reading.skipped(file))) return null;
     const field = shapeOf(reading, checker.getTypeOfSymbolAtLocation(prop2, reading.at), depth + 1);
     if (field) fields[prop2.getName()] = field;
     else if (!(prop2.flags & tsm.SymbolFlags.Optional)) return null;
@@ -1144,7 +1327,38 @@ function writtenOrder(tsm, checker, prop2) {
 }
 
 // src/vite/propControls.ts
-function propControls(stories, options = {}) {
+var errorText = (err) => err instanceof Error ? err.message : String(err);
+function storyExports(reader, file, fallback, { controls, api }, warn) {
+  let code = "";
+  const sources = [];
+  if (controls) {
+    try {
+      const table = reader.read(file);
+      if (table && Object.keys(table.argTypes).length > 0) {
+        code += `
+export const ${PROP_ARG_TYPES_EXPORT} = ${JSON.stringify(table.argTypes)};
+`;
+        sources.push(...table.sources);
+      }
+    } catch (err) {
+      warn(`could not read the props for ${file}: ${errorText(err)}`);
+    }
+  }
+  if (api) {
+    try {
+      const read = reader.api(file, fallback);
+      const table = read && { props: read.props, inherited: read.inherited, ref: read.ref };
+      code += `
+export const ${API_EXPORT} = ${JSON.stringify(table)};
+`;
+      if (read) sources.push(...read.sources);
+    } catch (err) {
+      warn(`could not read the API for ${file}: ${errorText(err)}`);
+    }
+  }
+  return { code, sources };
+}
+function propControls(stories, fallbackOf, readings, options = {}) {
   let root = process.cwd();
   let reader = null;
   return {
@@ -1156,17 +1370,11 @@ function propControls(stories, options = {}) {
     async transform(code, id) {
       if (!stories().includes(id)) return void 0;
       const props = await (reader ??= createPropReader(root, stories, options));
-      let table = null;
-      try {
-        table = props?.read(id) ?? null;
-      } catch (err) {
-        this.warn(`could not read the props for ${id}: ${err instanceof Error ? err.message : String(err)}`);
-      }
-      if (!table || Object.keys(table.argTypes).length === 0) return void 0;
-      for (const source of table.sources) this.addWatchFile(source);
-      return { code: `${code}
-export const ${PROP_ARG_TYPES_EXPORT} = ${JSON.stringify(table.argTypes)};
-`, map: null };
+      if (!props) return void 0;
+      const added = storyExports(props, id, fallbackOf(id), readings, (message) => this.warn(message));
+      if (!added.code) return void 0;
+      for (const source of new Set(added.sources)) this.addWatchFile(source);
+      return { code: code + added.code, map: null };
     }
   };
 }
@@ -1981,12 +2189,19 @@ function watchSources(server, { files, depGraph, reindex, redeps, logError }) {
     });
   }
 }
-function companions(options, storyFiles) {
-  const { storybookShims: shims, propControls: controls } = options;
+function companions(options, files) {
+  const { storybookShims: shims, propControls: controls, api } = options;
+  const storyFiles = () => [...files().keys()];
+  const fallbackOf = (file) => {
+    const title = files().get(file)?.[0]?.title;
+    return title === void 0 ? void 0 : titleStandIn(title);
+  };
+  const readings = { controls: controls !== false, api: api !== false };
+  const narrowed = typeof controls === "object" ? controls : void 0;
   return [
     watchRecovery(),
     ...shims === false ? [] : [storybookShims()],
-    ...controls === false ? [] : [propControls(storyFiles, controls === true ? void 0 : controls)]
+    ...readings.controls || readings.api ? [propControls(storyFiles, fallbackOf, readings, narrowed)] : []
   ];
 }
 function forge(options) {
@@ -2127,7 +2342,7 @@ function forge(options) {
     return true;
   }
   return [
-    ...companions(options, () => [...files().keys()]),
+    ...companions(options, files),
     {
       name: "weaselforge",
       config(config, { command }) {