Extract Tokens

1#!/usr/bin/env node
2import path from "node:path";
3import { docsDir, normalizeUrl, parseArgs, readJson, resolveOutputDir, writeJson } from "./lib/common.mjs";
4
5async function main() {
6  const args = parseArgs();
7  const url = normalizeUrl(args._[0] || args.url).toString();
8  const outputDir = resolveOutputDir(url, args.out);
9  const tokens = await extractTokens({ outputDir });
10  await writeJson(path.join(docsDir(outputDir), "tokens.json"), tokens);
11  if (args.json) process.stdout.write(`${JSON.stringify(tokens, null, 2)}\n`);
12}
13
14export async function extractTokens({ outputDir }) {
15  const sourceMapPath = path.join(docsDir(outputDir), "source-map.json");
16  let sourceMap = { pages: [] };
17  try {
18    sourceMap = await readJson(sourceMapPath);
19  } catch {
20    // Allow standalone use before source-map exists.
21  }
22  const colors = new Set();
23  const fonts = new Set();
24  const radii = new Set();
25  const shadows = new Set();
26  const fontSizes = new Set();
27
28  for (const page of sourceMap.pages || []) {
29    for (const color of page.tokens?.colors || []) colors.add(color);
30    for (const font of page.tokens?.fontFamilies || []) fonts.add(font);
31    for (const radius of page.tokens?.radii || []) radii.add(radius);
32    for (const shadow of page.tokens?.shadows || []) shadows.add(shadow);
33    for (const item of page.tokens?.computed || []) {
34      if (item.color && item.color !== "rgba(0, 0, 0, 0)") colors.add(item.color);
35      if (item.backgroundColor && item.backgroundColor !== "rgba(0, 0, 0, 0)") colors.add(item.backgroundColor);
36      if (item.fontFamily) fonts.add(item.fontFamily);
37      if (item.fontSize) fontSizes.add(item.fontSize);
38      if (item.borderRadius && item.borderRadius !== "0px") radii.add(item.borderRadius);
39      if (item.boxShadow && item.boxShadow !== "none") shadows.add(item.boxShadow);
40    }
41  }
42
43  return {
44    colors: Array.from(colors).slice(0, 32),
45    fontFamilies: Array.from(fonts).slice(0, 16),
46    fontSizes: Array.from(fontSizes).slice(0, 16),
47    radii: Array.from(radii).slice(0, 16),
48    shadows: Array.from(shadows).slice(0, 16),
49    recommendedTailwindTokens: {
50      colors: "Map primary, secondary, background, foreground, muted, accent, border from extracted colors.",
51      fonts: "Map sans/display/body from extracted font families and local downloaded fonts.",
52      spacing: "Infer from section rhythm; start with Tailwind defaults and add container widths.",
53    },
54    generatedAt: new Date().toISOString(),
55  };
56}
57
58if (import.meta.url === `file://${process.argv[1]}`) {
59  main().catch((error) => {
60    console.error(error.stack || error.message);
61    process.exit(1);
62  });
63}