folder-generator 1.3.3 → 1.3.5

package/bin/cli.js

@@ -1,39 +1,56 @@
 #!/usr/bin/env node
-
 const fs = require('fs');
 const path = require('path');
 const chalk = require('chalk');
 const { parseStructure, createFiles } = require('../lib/generator');
+const yargs = require('yargs/yargs');
+const { hideBin } = require('yargs/helpers');
 
-function normalizeInput(input) {
-  return input
-    .replace(/\r\n/g, '\n')  // Normalize line endings
-    .replace(/\\/g, '/')     // Normalize path separators
-    .replace(/\t/g, '    '); // Convert tabs to spaces
-}
-
-function main() {
-  let input = process.argv.slice(2).join(' ');
+async function main() {
+  const argv = yargs(hideBin(process.argv))
+    .option('input', {
+      alias: 'i',
+      type: 'string',
+      description: 'Folder structure text'
+    })
+    .option('file', {
+      alias: 'f',
+      type: 'string',
+      description: 'Path to file containing folder structure'
+    })
+    .help()
+    .alias('help', 'h')
+    .argv;
 
-  // If no arguments, check if piped input exists
-  if (!input && !process.stdin.isTTY) {
-    input = fs.readFileSync(0, 'utf-8');
-  }
+  let input = '';
 
-  if (!input) {
+  if (argv.file) {
+    try {
+      input = fs.readFileSync(argv.file, 'utf8');
+    } catch (err) {
+      console.error(chalk.red(`Error reading file: ${argv.file}`));
+      process.exit(1);
+    }
+  } else if (argv.input) {
+    input = argv.input;
+  } else if (!process.stdin.isTTY) {
+    // Read from stdin if piped
+    const chunks = [];
+    for await (const chunk of process.stdin) chunks.push(chunk);
+    input = Buffer.concat(chunks).toString('utf8');
+  } else {
     console.error(chalk.red('Error: No input provided'));
-    console.log(chalk.yellow('Usage: foldgen "your/structure"'));
-    console.log(chalk.yellow('   or: echo "structure" | foldgen'));
+    console.log(chalk.yellow('Usage:'));
+    console.log(chalk.yellow('  foldgen --input "your/structure"'));
+    console.log(chalk.yellow('  foldgen --file structure.txt'));
+    console.log(chalk.yellow('  echo "structure" | foldgen'));
     process.exit(1);
   }
 
   try {
-    const normalized = normalizeInput(input);
     console.log(chalk.yellow('\nParsing structure...'));
-    
-    const structure = parseStructure(normalized);
+    const structure = parseStructure(input);
     createFiles(structure);
-    
     console.log(chalk.bold.green('\n✔ Folder structure created successfully!'));
   } catch (error) {
     console.error(chalk.red('\n✖ Error creating structure:'), error.message);

package/lib/generator.js

@@ -8,11 +8,13 @@ function parseStructure(text) {
   let stack = [{ node: root, depth: -1 }];
 
   for (const line of lines) {
-    // Calculate depth based on leading characters
+    // Calculate depth based on leading whitespace
     const depth = (line.match(/^[ │├└─]*/)[0].length) / 4;
     const name = line.replace(/^[ │├└─]*/, '').trim();
 
-    // Find the correct parent
+    if (!name) continue;
+
+    // Find correct parent
     while (depth <= stack[stack.length - 1].depth) {
       stack.pop();
     }
@@ -32,7 +34,7 @@ function createFiles(node, currentPath = '.') {
   const fullPath = path.join(currentPath, node.name);
 
   if (node.name.includes('.')) {
-    // It's a file - ensure directory exists first
+    // It's a file
     fs.mkdirSync(path.dirname(fullPath), { recursive: true });
     fs.writeFileSync(fullPath, '');
     console.log(chalk.green(`✓ Created file: ${fullPath}`));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "folder-generator",
-  "version": "1.3.3",
+  "version": "1.3.5",
   "description": "CLI tool to generate folder structures from text input",
   "bin": {
     "foldgen": "./bin/cli.js"