tree-fs 0.1.2 → 0.1.3

package/README.md

@@ -131,6 +131,14 @@ project
 Known files without extensions are correctly identified as files.
 *   `Dockerfile`, `Makefile`, `LICENSE`, `Procfile`, `.gitignore`, `Jenkinsfile`
 
+### 6. Indicators & Comments
+We strip out common markers used to highlight specific files in documentation.
+```text
+project
+├── src/  <-- Working directory
+├── utils.js // Deprecated
+└── .env # Do not commit
+
 ## 📦 CI/CD Integration
 
 You can use `tree-fs` to scaffold environments in GitHub Actions or pipelines.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tree-fs",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Generate file system structures from text-based directory trees. The standard receiver for AI-generated project scaffolding.",
   "bin": {
     "tree-fs": "bin/tree-fs.js"

package/src/normaliser.js

@@ -16,9 +16,22 @@ function normaliseLines(input) {
       const match = normalizedRaw.match(STRIP_REGEX)
       const prefixLength = match ? match[0].length : 0
 
-      // 3. Strip comments (anything after ' #')
-      const commentIndex = normalizedRaw.indexOf(" #")
-      let cleaned = commentIndex !== -1 ? normalizedRaw.substring(0, commentIndex) : normalizedRaw
+      // 3. Strip comments
+      // Look for multiple comment styles. We use "space + marker" to avoid false positives.
+      const commentMarkers = [" #", " <--", " //"]
+      let splitIndex = -1
+
+      for (const marker of commentMarkers) {
+        const idx = normalizedRaw.indexOf(marker)
+        if (idx !== -1) {
+          // Keep the earliest marker found
+          if (splitIndex === -1 || idx < splitIndex) {
+            splitIndex = idx
+          }
+        }
+      }
+
+      let cleaned = splitIndex !== -1 ? normalizedRaw.substring(0, splitIndex) : normalizedRaw
 
       // 4. Check for explicit trailing slash
       const endsWithSlash = cleaned.trim().endsWith("/")