refactor parser.ts for improved clarity and functionality

Author: mayconrfreitas

src/parser.ts

@@ -1,7 +1,6 @@
 import path from "node:path";
 import { PlanItem, ParseOptions } from "./types.js";
 import {
-  computeDepth,
   hasExtension,
   isBareKnownFile,
   looksLikeDirToken,
@@ -25,13 +24,13 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
     .map((l) => l.replace(/\s+$/, ""))
     .filter((l) => l.trim().length > 0);
 
-  // inferRoot: primeira linha é algo como "my-project/" (apenas a barra de fechamento)
+  // infer base root from a first-line token like "project/"
   let baseRoot = rootDir;
   if (opts.inferRoot && lines.length > 0) {
-    const firstContent = sanitizeContent(lines[0]);
-    if (firstContent.endsWith("/")) {
-      const token = firstContent.replace(/\/+$/, "");
-      if (!token.includes("/")) {
+    const first = sanitizeContent(lines[0]);
+    if (first.endsWith("/")) {
+      const token = first.replace(/\/+$/, "");
+      if (token && !token.includes("/")) {
         baseRoot = safeJoin(rootDir, token);
         lines = lines.slice(1);
       }
@@ -40,10 +39,12 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
 
   const plan: PlanItem[] = [];
   const seen = new Set<string>();
+
+  // directory stack by depth
   const stack: string[] = [];
   stack[0] = baseRoot;
 
-  // garanta que o diretório baseRoot apareça no plano
+  // ensure baseRoot is part of the plan
   addDir(baseRoot);
 
   for (const raw of lines) {
@@ -52,14 +53,14 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
     const content0 = sanitizeContent(unclean);
     if (!content0) continue;
 
-    // caminho absoluto relativo ao baseRoot: "/foo/bar"
+    // "/foo/bar" relative to baseRoot
     if (content0.startsWith("/")) {
       const rel = content0.replace(/^\/+/, "");
       addAny(rel, 0);
       continue;
     }
 
-    // possui separadores
+    // token with slashes
     if (content0.includes("/")) {
       const isDir = content0.endsWith("/");
       const rel = isDir ? content0.slice(0, -1) : content0;
@@ -77,7 +78,7 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
       continue;
     }
 
-    // token "nu" sem barra
+    // bare token
     if (looksLikeDirToken(content0)) {
       const parent = stack[depth] ?? baseRoot;
       const abs = safeJoin(parent, content0.replace(/\/$/, ""));
@@ -96,14 +97,35 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
   function sanitizeContent(s: string): string {
     let out = s.trim();
     out = stripBullets(out);
-    out = out.replace(/^(?:[│├└]?[─-])\s*/, "");
+    // drop a leading branch marker like "├─ " or "└─ " or "- "
+    out = out.replace(/^(?:[│├└]\s*)?[─-]\s*/u, "");
     out = stripTrailingComments(out);
     out = out.trim();
     out = out.replace(/\/\s+$/, "/");
     out = out.replace(/\s+\((?:opcional|optional)[^)]*\)\s*$/i, "");
     return out;
   }
 
+  function computeDepth(prefix: string): number {
+    // remove the trailing branch marker (├─ or └─) so it doesn't count as indent
+    let p = prefix.replace(/[├└]─\s*$/u, "");
+    // if box pipes exist, depth equals number of pipes
+    const pipes = (p.match(/│/g) || []).length;
+    if (pipes > 0) return pipes;
+    // otherwise, count groups of two spaces
+    p = p.replace(/[├└─]/g, " ").replace(/\t/g, "    ");
+    let groups = 0, acc = 0;
+    for (const ch of p) {
+      if (ch === " ") {
+        acc++;
+        if (acc === 2) { groups++; acc = 0; }
+      } else {
+        acc = 0;
+      }
+    }
+    return groups;
+  }
+
   function addAny(rel: string, depth: number) {
     const isDir =
       rel.endsWith("/") ||
@@ -121,6 +143,21 @@ export function parseToPlan(input: string, opts: ParseOptions): ParseResult {
   }
 
   function addDir(abs: string) {
+    // include all ancestor directories under baseRoot
+    const rel = path.relative(baseRoot, abs);
+    if (rel && rel !== ".") {
+      const parts = rel.split(path.sep).filter(Boolean);
+      let cur = baseRoot;
+      for (const seg of parts) {
+        cur = safeJoin(cur, seg);
+        pushDir(cur);
+      }
+    } else {
+      pushDir(baseRoot);
+    }
+  }
+
+  function pushDir(abs: string) {
     const key = `d:${abs}`;
     if (!seen.has(key)) {
       seen.add(key);