@kt3k/tku 1.2.0 → 1.2.2

package/dist/main.js

@@ -163,10 +163,109 @@ function formatTable(result, omitted = 0) {
   lines.push(`${totalDisplay.padStart(maxWidth)}  total (${result.totalFiles} files)`);
   return lines.join("\n");
 }
+function buildTree(files) {
+  const root = {
+    name: ".",
+    tokens: 0,
+    children: []
+  };
+  for (const file of files) {
+    const parts = file.path.split("/");
+    let node = root;
+    for (let i = 0; i < parts.length; i++) {
+      const name = parts[i];
+      const isFile = i === parts.length - 1;
+      let child = node.children.find((c) => c.name === name);
+      if (!child) {
+        child = {
+          name,
+          tokens: 0,
+          children: []
+        };
+        node.children.push(child);
+      }
+      if (isFile) {
+        child.tokens = file.tokens;
+      }
+      node = child;
+    }
+  }
+  function aggregate(node) {
+    if (node.children.length === 0) return node.tokens;
+    node.tokens = node.children.reduce((sum, c) => sum + aggregate(c), 0);
+    return node.tokens;
+  }
+  aggregate(root);
+  function sortChildren(node) {
+    for (const child of node.children) sortChildren(child);
+    const dirs = node.children.filter((c) => c.children.length > 0).sort((a, b) => b.tokens - a.tokens);
+    const leaves = node.children.filter((c) => c.children.length === 0).sort((a, b) => b.tokens - a.tokens);
+    node.children = [
+      ...dirs,
+      ...leaves
+    ];
+  }
+  sortChildren(root);
+  return root;
+}
+function formatTree(result, options = {}) {
+  const tree = buildTree(result.files);
+  const dirsOnly = options.dirs ?? false;
+  const lines = [];
+  const allTokens = [];
+  function collectTokens(node) {
+    if (dirsOnly && node.children.length === 0) return;
+    allTokens.push(node.tokens);
+    for (const child of node.children) collectTokens(child);
+  }
+  collectTokens(tree);
+  const maxWidth = Math.max("tokens".length, ...allTokens.map((t) => formatTokenCount(t).length));
+  function render(node, prefix, isLast, isRoot) {
+    const display = formatTokenCount(node.tokens);
+    const connector = isRoot ? "" : isLast ? "\u2514\u2500\u2500 " : "\u251C\u2500\u2500 ";
+    const name = node.children.length > 0 && !isRoot ? `${node.name}/` : node.name;
+    lines.push(`${display.padStart(maxWidth)}  ${prefix}${connector}${name}`);
+    const visibleChildren = dirsOnly ? node.children.filter((c) => c.children.length > 0) : node.children;
+    const childPrefix = isRoot ? "" : prefix + (isLast ? "    " : "\u2502   ");
+    for (let i = 0; i < visibleChildren.length; i++) {
+      render(visibleChildren[i], childPrefix, i === visibleChildren.length - 1, false);
+    }
+  }
+  lines.push(`${"tokens".padStart(maxWidth)}  path`);
+  render(tree, "", true, true);
+  return lines.join("\n");
+}
+function summarizeByDir(result) {
+  const dirMap = /* @__PURE__ */ new Map();
+  for (const file of result.files) {
+    const parts = file.path.split("/");
+    if (parts.length < 2) continue;
+    for (let i = 1; i < parts.length; i++) {
+      const dir = parts.slice(0, i).join("/");
+      dirMap.set(dir, (dirMap.get(dir) ?? 0) + file.tokens);
+    }
+  }
+  const files = [
+    ...dirMap.entries()
+  ].map(([path, tokens]) => ({
+    path,
+    tokens
+  }));
+  return {
+    ...result,
+    files
+  };
+}
 function formatResult(result, options = {}) {
-  const { json = false, top, sort = "tokens" } = options;
+  const { json = false, top, sort = "tokens", tree = false, dirs = false } = options;
+  if (tree) {
+    return formatTree(result, {
+      dirs
+    });
+  }
+  const effective = dirs ? summarizeByDir(result) : result;
   const sorted = [
-    ...result.files
+    ...effective.files
   ];
   if (sort === "path") {
     sorted.sort((a, b) => a.path.localeCompare(b.path));
@@ -176,7 +275,7 @@ function formatResult(result, options = {}) {
   const filtered = top !== void 0 ? sorted.slice(0, top) : sorted;
   const omitted = sorted.length - filtered.length;
   const adjusted = {
-    ...result,
+    ...effective,
     files: filtered
   };
   return json ? JSON.stringify(adjusted, null, 2) : formatTable(adjusted, omitted);
@@ -193,6 +292,9 @@ Options:
       --json                  Output results as JSON
       --top <n>               Show only the top N files by token count
       --sort <field>          Sort by "tokens" or "path" (default: tokens)
+  -t, --tree                  Display results as a directory tree
+                              (ignores --top, --json, --sort)
+      --dirs                  Summarize by directory (table or tree)
   -h, --help                  Show this help message`);
 }
 async function main() {
@@ -223,6 +325,15 @@ async function main() {
         type: "string",
         default: "tokens"
       },
+      tree: {
+        type: "boolean",
+        short: "t",
+        default: false
+      },
+      dirs: {
+        type: "boolean",
+        default: false
+      },
       help: {
         type: "boolean",
         short: "h",
@@ -266,7 +377,9 @@ async function main() {
     const output = formatResult(result, {
       json: values.json,
       top,
-      sort
+      sort,
+      tree: values.tree,
+      dirs: values.dirs
     });
     console.log(output);
   } catch (e) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kt3k/tku",
-  "version": "1.2.0",
+  "version": "1.2.2",
   "description": "",
   "main": "dist/main.js",
   "bin": {