mkctx 3.0.0 → 4.0.1

package/bin/mkctx.js

@@ -1,20 +1,21 @@
 #!/usr/bin/env node
 
-const fs = require('fs');
+'use strict';
+
+const fs   = require('fs');
 const path = require('path');
 
 // ============================================
-// LAZY LOAD DEPENDENCIES (for faster startup)
+// LAZY-LOADED DEPENDENCIES
 // ============================================
 
 let inquirer, chalk, ora;
 
 function loadDependencies() {
-    if (!inquirer) {
-        inquirer = require('inquirer');
-        chalk = require('chalk');
-        ora = require('ora');
-    }
+    if (inquirer) return;
+    inquirer = require('inquirer');
+    chalk    = require('chalk');
+    ora      = require('ora');
 }
 
 // ============================================
@@ -23,29 +24,29 @@ function loadDependencies() {
 
 const CONFIG_FILE = 'mkctx.config.json';
 
-const DEFAULT_PROJECT_CONFIG = {
-    src: ".",
-    ignore: "mkctx.config.json, pnpm-lock.yaml, **/.titan/, mkctx/, node_modules/, .git/, dist/, build/, target/, .next/, out/, .cache, package-lock.json, README.md, *.log, temp/, tmp/, coverage/, .nyc_output, .env, .env.local, .env.development.local, .env.test.local, .env.production.local, npm-debug.log*, yarn-debug.log*, yarn-error.log*, .npm, .yarn-integrity, .parcel-cache, .vuepress/dist, .svelte-kit, **/*.rs.bk, .idea/, .vscode/, .DS_Store, Thumbs.db, *.swp, *.swo, .~lock.*, Cargo.lock, .cargo/registry/, .cargo/git/, .rustup/, *.pdb, *.dSYM/, *.so, *.dll, *.dylib, *.exe, *.lib, *.a, *.o, *.rlib, *.d, *.tmp, *.bak, *.orig, *.rej, *.pyc, *.pyo, *.class, *.jar, *.war, *.ear, *.zip, *.tar.gz, *.rar, *.7z, *.iso, *.img, *.dmg, *.pdf, *.doc, *.docx, *.xls, *.xlsx, *.ppt, *.pptx",
-    output: "./mkctx",
-    first_comment: "/* Project Context */",
-    last_comment: "/* End of Context */"
+const DEFAULT_CONFIG = {
+    src:           '.',
+    ignore:        'mkctx.config.json, pnpm-lock.yaml, **/.titan/, mkctx/, node_modules/, .git/, dist/, build/, target/, .next/, out/, .cache, package-lock.json, *.log, temp/, tmp/, coverage/, .nyc_output, .env, .env.local, .env.development.local, .env.test.local, .env.production.local, npm-debug.log*, yarn-debug.log*, yarn-error.log*, .npm, .yarn-integrity, .parcel-cache, .vuepress/dist, .svelte-kit, **/*.rs.bk, .idea/, .vscode/, .DS_Store, Thumbs.db, *.swp, *.swo, .~lock.*, Cargo.lock, .cargo/registry/, .cargo/git/, .rustup/, *.pdb, *.dSYM/, *.so, *.dll, *.dylib, *.exe, *.lib, *.a, *.o, *.rlib, *.d, *.tmp, *.bak, *.orig, *.rej, *.pyc, *.pyo, *.class, *.jar, *.war, *.ear, *.zip, *.tar.gz, *.rar, *.7z, *.iso, *.img, *.dmg, *.pdf, *.doc, *.docx, *.xls, *.xlsx, *.ppt, *.pptx',
+    output:        './mkctx',
+    first_comment: '/* Project Context */',
+    last_comment:  '/* End of Context */',
 };
 
-// Language mapping for syntax highlighting
+const VALID_FORMATS = ['json', 'md', 'toon', 'xml'];
+
 const LANG_MAP = {
-    js: 'javascript', ts: 'typescript', jsx: 'jsx', tsx: 'tsx',
-    py: 'python', rb: 'ruby', go: 'go', rs: 'rust',
-    java: 'java', kt: 'kotlin', cs: 'csharp', cpp: 'cpp',
-    c: 'c', h: 'c', hpp: 'cpp', php: 'php',
-    sh: 'bash', bash: 'bash', zsh: 'bash', ps1: 'powershell',
-    sql: 'sql', html: 'html', css: 'css', scss: 'scss',
-    sass: 'sass', less: 'less', json: 'json', xml: 'xml',
-    yaml: 'yaml', yml: 'yaml', md: 'markdown', vue: 'vue',
+    js: 'javascript', ts: 'typescript', jsx: 'jsx',    tsx: 'tsx',
+    py: 'python',     rb: 'ruby',       go: 'go',      rs: 'rust',
+    java: 'java',     kt: 'kotlin',     cs: 'csharp',  cpp: 'cpp',
+    c: 'c',           h: 'c',           hpp: 'cpp',    php: 'php',
+    sh: 'bash',       bash: 'bash',     zsh: 'bash',   ps1: 'powershell',
+    sql: 'sql',       html: 'html',     css: 'css',    scss: 'scss',
+    sass: 'sass',     less: 'less',     json: 'json',  xml: 'xml',
+    yaml: 'yaml',     yml: 'yaml',      md: 'markdown', vue: 'vue',
     svelte: 'svelte', dockerfile: 'dockerfile', makefile: 'makefile',
-    toml: 'toml', ini: 'ini', cfg: 'ini', env: 'bash'
+    toml: 'toml',     ini: 'ini',       cfg: 'ini',    env: 'bash',
 };
 
-// Text file extensions
 const TEXT_EXTENSIONS = new Set([
     '.js', '.ts', '.jsx', '.tsx', '.mjs', '.cjs',
     '.py', '.pyw', '.rb', '.rake', '.go', '.rs',
@@ -65,7 +66,7 @@ const TEXT_EXTENSIONS = new Set([
     '.ex', '.exs', '.erl', '.hrl', '.clj', '.cljs', '.cljc',
     '.hs', '.lhs', '.elm', '.pug', '.jade',
     '.ejs', '.hbs', '.handlebars', '.twig', '.blade.php',
-    '.astro', '.prisma', '.sol'
+    '.astro', '.prisma', '.sol',
 ]);
 
 const KNOWN_FILES = new Set([
@@ -73,376 +74,517 @@ const KNOWN_FILES = new Set([
     'procfile', 'vagrantfile', 'jenkinsfile',
     '.gitignore', '.gitattributes', '.editorconfig',
     '.eslintrc', '.prettierrc', '.babelrc',
-    '.env', '.env.example', '.env.local'
+    '.env', '.env.example', '.env.local',
+    'readme.md', 'readme.txt', 'readme',
+    'license', 'license.md', 'license.txt',
 ]);
 
-// ============================================
-// GLOBAL STATE
-// ============================================
+const SYSTEM_IGNORES = [
+    '.git', '.DS_Store', 'Thumbs.db', 'node_modules',
+    '.svn', '.hg', '__pycache__', '.pytest_cache',
+    '.mypy_cache', '.vscode', '.idea',
+];
 
-let generatedContext = null;
-let contextFiles = [];
-let contextStats = {};
+// Flags that take a value (not boolean)
+const VALUE_FLAGS = new Set([
+    'src', 'output', 'format', 'ignore', 'name', 'first-comment', 'last-comment',
+    's', 'o', 'f', 'n',
+]);
+
+// Flags that trigger non-interactive mode
+const NON_INTERACTIVE_FLAGS = new Set([
+    'src', 's', 'output', 'o', 'format', 'f',
+    'ignore', 'name', 'n', 'first-comment', 'last-comment',
+]);
 
 // ============================================
-// CONFIGURATION MANAGEMENT
+// CLI ARGUMENT PARSING
 // ============================================
 
-function loadProjectConfig() {
-    if (fs.existsSync(CONFIG_FILE)) {
-        try {
-            const config = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf-8'));
-            return { ...DEFAULT_PROJECT_CONFIG, ...config };
-        } catch (err) {
-            console.log(chalk.yellow('⚠️  Error parsing config file, using defaults'));
-            return { ...DEFAULT_PROJECT_CONFIG };
+function parseArgs(argv) {
+    const args  = argv.slice(2);
+    const flags = {};
+    let command = null;
+
+    if (args.length > 0 && !args[0].startsWith('-')) {
+        command = args[0];
+    }
+
+    for (let i = 0; i < args.length; i++) {
+        const arg = args[i];
+
+        if (arg.startsWith('--') && arg.includes('=')) {
+            const eq = arg.indexOf('=');
+            flags[arg.slice(2, eq)] = arg.slice(eq + 1);
+            continue;
+        }
+
+        if (arg.startsWith('-') && !arg.startsWith('--') && arg.includes('=')) {
+            const eq = arg.indexOf('=');
+            flags[arg.slice(1, eq)] = arg.slice(eq + 1);
+            continue;
+        }
+
+        if (arg.startsWith('--')) {
+            const key  = arg.slice(2);
+            const next = args[i + 1];
+            if (VALUE_FLAGS.has(key) && next !== undefined && !isFlagToken(next)) {
+                flags[key] = next;
+                i++;
+            } else {
+                flags[key] = true;
+            }
+            continue;
+        }
+
+        if (arg.startsWith('-') && arg.length === 2) {
+            const key  = arg.slice(1);
+            const next = args[i + 1];
+            if (VALUE_FLAGS.has(key) && next !== undefined && !isFlagToken(next)) {
+                flags[key] = next;
+                i++;
+            } else {
+                flags[key] = true;
+            }
         }
     }
-    return null;
+
+    return { command, flags };
+}
+
+function isFlagToken(str) {
+    return /^--[a-zA-Z]/.test(str) || /^-[a-zA-Z]/.test(str);
+}
+
+function isNonInteractiveMode(flags) {
+    return Object.keys(flags).some(f => NON_INTERACTIVE_FLAGS.has(f));
+}
+
+// Resolve short aliases to canonical names
+function resolveFlags(flags) {
+    return {
+        src:          flags.src    || flags.s,
+        output:       flags.output || flags.o,
+        format:       flags.format || flags.f,
+        name:         flags.name   || flags.n,
+        ignore:       flags.ignore,
+        firstComment: flags['first-comment'],
+        lastComment:  flags['last-comment'],
+    };
 }
 
-function hasProjectConfig() {
+// ============================================
+// CONFIG MANAGEMENT
+// ============================================
+
+function configFileExists() {
     return fs.existsSync(CONFIG_FILE);
 }
 
-function createProjectConfig() {
-    loadDependencies();
-    
-    if (!fs.existsSync('mkctx')) {
-        fs.mkdirSync('mkctx', { recursive: true });
+function loadConfig() {
+    if (!configFileExists()) return null;
+    try {
+        const raw = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf-8'));
+        return { ...DEFAULT_CONFIG, ...raw };
+    } catch {
+        console.log(chalk.yellow('⚠️  Error parsing config file, using defaults'));
+        return { ...DEFAULT_CONFIG };
     }
+}
 
-    fs.writeFileSync(CONFIG_FILE, JSON.stringify(DEFAULT_PROJECT_CONFIG, null, 2));
-    updateGitignore();
+function buildConfig(cliFlags = {}) {
+    const base = configFileExists() ? loadConfig() : { ...DEFAULT_CONFIG };
+    if (cliFlags.src)          base.src           = cliFlags.src;
+    if (cliFlags.output)       base.output        = cliFlags.output;
+    if (cliFlags.ignore)       base.ignore        = cliFlags.ignore;
+    if (cliFlags.firstComment) base.first_comment = cliFlags.firstComment;
+    if (cliFlags.lastComment)  base.last_comment  = cliFlags.lastComment;
+    return base;
+}
 
+function createConfigFile() {
+    loadDependencies();
+    fs.mkdirSync('mkctx', { recursive: true });
+    fs.writeFileSync(CONFIG_FILE, JSON.stringify(DEFAULT_CONFIG, null, 2));
+    appendToGitignore();
     console.log(chalk.green('\n✅ Configuration created:'));
     console.log(chalk.white('   - mkctx.config.json'));
     console.log(chalk.white('   - mkctx/ folder'));
     console.log(chalk.white('   - Entry in .gitignore\n'));
 }
 
-function updateGitignore() {
+function appendToGitignore() {
     const gitignorePath = '.gitignore';
-    let content = '';
-
-    if (fs.existsSync(gitignorePath)) {
-        content = fs.readFileSync(gitignorePath, 'utf-8');
+    const current = fs.existsSync(gitignorePath)
+        ? fs.readFileSync(gitignorePath, 'utf-8')
+        : '';
+    if (!current.includes('mkctx/')) {
+        fs.writeFileSync(gitignorePath, current + '\n# mkctx - generated context\nmkctx/\n');
     }
+}
+
+// ============================================
+// FORMAT RESOLUTION
+// ============================================
+
+function resolveFormats(formatArg) {
+    if (!formatArg)            return ['md'];
+    if (formatArg === 'all')   return [...VALID_FORMATS];
+
+    const requested = formatArg.split(',').map(f => f.trim().toLowerCase());
+    const invalid   = requested.filter(f => !VALID_FORMATS.includes(f));
 
-    if (!content.includes('mkctx/')) {
-        const entry = '\n# mkctx - generated context\nmkctx/\n';
-        content += entry;
-        fs.writeFileSync(gitignorePath, content);
+    if (invalid.length > 0) {
+        console.error(`❌ Invalid format(s): ${invalid.join(', ')}. Valid: ${VALID_FORMATS.join(', ')}, all`);
+        process.exit(1);
     }
+
+    return requested;
 }
 
 // ============================================
-// UTILITY FUNCTIONS
+// UTILITY HELPERS
 // ============================================
 
 function formatSize(bytes) {
-    const sizes = ['B', 'KB', 'MB', 'GB'];
+    const units = ['B', 'KB', 'MB', 'GB'];
     if (bytes === 0) return '0 B';
     const i = Math.floor(Math.log(bytes) / Math.log(1024));
-    return `${(bytes / Math.pow(1024, i)).toFixed(2)} ${sizes[i]}`;
+    return `${(bytes / Math.pow(1024, i)).toFixed(2)} ${units[i]}`;
 }
 
 function estimateTokens(text) {
     return Math.ceil(text.length / 4);
 }
 
+function toUnixPath(filePath) {
+    return filePath.replace(/\\/g, '/');
+}
+
 // ============================================
-// FILE OPERATIONS
+// FILE FILTERING
 // ============================================
 
 function isTextFile(filename) {
-    const ext = path.extname(filename).toLowerCase();
+    const ext      = path.extname(filename).toLowerCase();
     const basename = path.basename(filename).toLowerCase();
-
-    if (KNOWN_FILES.has(basename)) return true;
-    if (ext && TEXT_EXTENSIONS.has(ext)) return true;
-
-    return false;
+    return KNOWN_FILES.has(basename) || (!!ext && TEXT_EXTENSIONS.has(ext));
 }
 
 function getLanguage(filename) {
-    const ext = path.extname(filename).slice(1).toLowerCase();
+    const ext      = path.extname(filename).slice(1).toLowerCase();
     const basename = path.basename(filename).toLowerCase();
-
-    if (basename === 'dockerfile') return 'dockerfile';
-    if (basename === 'makefile') return 'makefile';
-    if (basename.startsWith('.env')) return 'bash';
-
+    if (basename === 'dockerfile')      return 'dockerfile';
+    if (basename === 'makefile')        return 'makefile';
+    if (basename.startsWith('.env'))    return 'bash';
     return LANG_MAP[ext] || ext || 'text';
 }
 
 function parseIgnorePatterns(ignoreString) {
     if (!ignoreString) return [];
-    return ignoreString
-        .split(',')
-        .map(p => p.trim())
-        .filter(Boolean);
+    return ignoreString.split(',').map(p => p.trim()).filter(Boolean);
 }
 
-function matchWildcard(pattern, filename) {
-    const regexPattern = pattern
-        .replace(/\./g, '\\.')
-        .replace(/\*\*/g, '.*')
-        .replace(/\*/g, '[^/]*');
-    const regex = new RegExp(`^${regexPattern}$`, 'i');
-    return regex.test(filename);
+function matchWildcard(pattern, subject) {
+    const regex = new RegExp(
+        '^' + pattern
+            .replace(/\./g, '\\.')
+            .replace(/\*\*/g, '.*')
+            .replace(/\*/g, '[^/]*') + '$',
+        'i'
+    );
+    return regex.test(subject);
 }
 
 function shouldIgnore(fullPath, name, relativePath, patterns) {
-    const systemIgnores = [
-        '.git', '.DS_Store', 'Thumbs.db', 'node_modules',
-        '.svn', '.hg', '__pycache__', '.pytest_cache',
-        '.mypy_cache', '.vscode', '.idea'
-    ];
-
-    for (const ignore of systemIgnores) {
-        if (fullPath.includes(path.sep + ignore + path.sep) ||
-            fullPath.includes(ignore + path.sep) ||
-            name === ignore) {
-            return true;
-        }
-    }
+    const normFull     = toUnixPath(fullPath);
+    const normRelative = toUnixPath(relativePath);
+
+    const inSystemIgnore = SYSTEM_IGNORES.some(ig =>
+        normFull.includes(`/${ig}/`) ||
+        normFull.includes(`/${ig}`)  ||
+        normFull.endsWith(`/${ig}`)  ||
+        name === ig
+    );
+    if (inSystemIgnore) return true;
 
-    for (const pattern of patterns) {
+    return patterns.some(pattern => {
         if (pattern.includes('*')) {
-            if (matchWildcard(pattern, name)) return true;
-            if (matchWildcard(pattern, relativePath)) return true;
+            return matchWildcard(pattern, name) || matchWildcard(pattern, normRelative);
         }
-
         if (pattern.endsWith('/')) {
             const dir = pattern.slice(0, -1);
-            if (fullPath.includes(path.sep + dir + path.sep) ||
-                fullPath.includes(dir + path.sep) ||
-                name === dir) {
-                return true;
-            }
-        }
-
-        if (relativePath === pattern || name === pattern) {
-            return true;
+            return normFull.includes(`/${dir}/`) ||
+                   normFull.endsWith(`/${dir}`)  ||
+                   name === dir;
         }
-    }
-
-    return false;
+        return normRelative === pattern || name === pattern;
+    });
 }
 
-function getFiles(srcPath, config) {
-    const files = [];
+// ============================================
+// FILE SCANNING
+// ============================================
+
+function scanFiles(srcPath, config) {
     const ignorePatterns = parseIgnorePatterns(config.ignore);
+    const files          = [];
+    const stats          = { files: 0, totalSize: 0, totalLines: 0, filesByExt: {} };
 
     function walk(dir) {
         if (!fs.existsSync(dir)) return;
 
         let entries;
-        try {
-            entries = fs.readdirSync(dir, { withFileTypes: true });
-        } catch (err) {
-            return;
-        }
+        try { entries = fs.readdirSync(dir, { withFileTypes: true }); }
+        catch { return; }
 
         for (const entry of entries) {
-            const fullPath = path.join(dir, entry.name);
+            const fullPath     = path.join(dir, entry.name);
             const relativePath = path.relative(srcPath, fullPath);
 
-            if (shouldIgnore(fullPath, entry.name, relativePath, ignorePatterns)) {
-                continue;
-            }
-
-            if (entry.isDirectory()) {
-                walk(fullPath);
-            } else if (entry.isFile() && isTextFile(entry.name)) {
-                files.push({
-                    fullPath,
-                    relativePath,
-                    name: entry.name,
-                    ext: path.extname(entry.name).slice(1).toLowerCase()
-                });
-            }
-        }
-    }
-
-    walk(srcPath);
-    return files.sort((a, b) => a.relativePath.localeCompare(b.relativePath));
-}
-
-function buildContextContent(files, config, srcPath) {
-    let content = '';
-    let totalSize = 0;
-    let totalLines = 0;
-    const filesByExt = {};
-
-    if (config.first_comment) {
-        content += config.first_comment + '\n\n';
-    }
-
-    content += '## Project Structure\n\n```\n';
-    const dirs = new Set();
-    files.forEach(f => {
-        const dir = path.dirname(f.relativePath);
-        if (dir !== '.') dirs.add(dir);
-    });
-    dirs.forEach(d => content += `📁 ${d}/\n`);
-    content += `\n${files.length} files total\n\`\`\`\n\n`;
-
-    content += '## Source Files\n\n';
+            if (shouldIgnore(fullPath, entry.name, relativePath, ignorePatterns)) continue;
 
-    for (const file of files) {
-        let fileContent;
-        try {
-            fileContent = fs.readFileSync(file.fullPath, 'utf-8');
-        } catch (err) {
-            console.log(chalk.yellow(`⚠️  Could not read: ${file.relativePath}`));
-            continue;
-        }
+            if (entry.isDirectory()) { walk(fullPath); continue; }
+            if (!entry.isFile() || !isTextFile(entry.name)) continue;
 
-        const lang = getLanguage(file.name);
-        const lines = fileContent.split('\n').length;
+            let content;
+            try { content = fs.readFileSync(fullPath, 'utf-8'); }
+            catch { continue; }
 
-        totalSize += Buffer.byteLength(fileContent, 'utf-8');
-        totalLines += lines;
-        filesByExt[file.ext || 'other'] = (filesByExt[file.ext || 'other'] || 0) + 1;
+            const ext      = path.extname(entry.name).slice(1).toLowerCase() || null;
+            const lines    = content.split('\n').length;
+            const size     = Buffer.byteLength(content, 'utf-8');
+            const language = getLanguage(entry.name);
 
-        content += `### ${file.relativePath}\n\n`;
-        content += '```' + lang + '\n';
-        content += fileContent;
+            stats.totalSize  += size;
+            stats.totalLines += lines;
+            stats.filesByExt[ext || 'other'] = (stats.filesByExt[ext || 'other'] || 0) + 1;
 
-        if (!fileContent.endsWith('\n')) {
-            content += '\n';
+            files.push({ path: toUnixPath(relativePath), name: entry.name, extension: ext, language, lines, size, content });
         }
-
-        content += '```\n\n';
     }
 
-    if (config.last_comment) {
-        content += config.last_comment;
-    }
-
-    contextStats = {
-        files: files.length,
-        totalSize,
-        totalLines,
-        filesByExt,
-        estimatedTokens: estimateTokens(content)
-    };
+    walk(srcPath);
+    files.sort((a, b) => a.path.localeCompare(b.path));
+    stats.files = files.length;
 
-    return content;
+    return { files, stats };
 }
 
 // ============================================
-// CONTEXT GENERATION
+// OUTPUT FORMATTERS
 // ============================================
 
-async function generateContextDynamic() {
-    loadDependencies();
+function toJson(files) {
+    return JSON.stringify(files, null, 2);
+}
 
-    const { srcPath } = await inquirer.prompt([
-        {
-            type: 'input',
-            name: 'srcPath',
-            message: 'Enter the source path to analyze:',
-            default: '.',
-            validate: (input) => {
-                if (!fs.existsSync(input)) {
-                    return `Path does not exist: ${input}`;
-                }
-                return true;
-            }
-        }
-    ]);
+function toMarkdown(files, config) {
+    const dirs = [...new Set(
+        files.map(f => path.dirname(f.path)).filter(d => d !== '.')
+    )].sort();
+
+    const structure = dirs.map(d => `📁 ${d}/`).join('\n');
+
+    const sources = files.map(file => {
+        const body = file.content.endsWith('\n') ? file.content : file.content + '\n';
+        return `### ${file.path}\n\n\`\`\`${file.language}\n${body}\`\`\`\n`;
+    }).join('\n');
+
+    return [
+        config.first_comment,
+        '',
+        '## Project Structure',
+        '',
+        '```',
+        structure,
+        '',
+        `${files.length} files total`,
+        '```',
+        '',
+        '## Source Files',
+        '',
+        sources,
+        config.last_comment,
+    ].join('\n');
+}
 
-    const config = { ...DEFAULT_PROJECT_CONFIG, src: srcPath };
-    return generateContextFromConfig(config, srcPath);
+function toToon(files, stats) {
+    const header = [
+        '# Project Context',
+        `# Generated: ${new Date().toISOString()}`,
+        `# Files: ${files.length}`,
+        `# Lines: ${stats.totalLines}`,
+        `# Size: ${stats.totalSize} bytes`,
+        '',
+        `files[${files.length}]{path,name,extension,language,lines,size}:`,
+        ...files.map(f =>
+            `  ${escapeToon(f.path)},${escapeToon(f.name)},${f.extension || ''},${f.language},${f.lines},${f.size}`
+        ),
+        '',
+    ].join('\n');
+
+    const bodies = files.map((f, i) => [
+        '---',
+        `[${i}] ${f.path}`,
+        `language: ${f.language}`,
+        'content:',
+        ...f.content.split('\n').map(l => `  ${l}`),
+    ].join('\n')).join('\n');
+
+    return header + bodies;
 }
 
-async function generateContextFromConfigFile() {
-    loadDependencies();
-    
-    const config = loadProjectConfig();
-    if (!config) {
-        console.log(chalk.yellow('\n⚠️  No config file found.'));
-        return null;
-    }
+function escapeToon(value) {
+    if (value == null) return '';
+    const str         = String(value);
+    const needsQuotes = str.includes(',') || str.includes('\n') || str.includes('"')
+        || str.startsWith(' ') || str.endsWith(' ');
+    return needsQuotes
+        ? `"${str.replace(/"/g, '""').replace(/\n/g, '\\n')}"`
+        : str;
+}
 
-    return generateContextFromConfig(config, config.src);
+function toXml(files) {
+    const fileEntries = files.map(f => [
+        '  <file>',
+        `    <path>${escapeXml(f.path)}</path>`,
+        `    <name>${escapeXml(f.name)}</name>`,
+        `    <extension>${escapeXml(f.extension || '')}</extension>`,
+        `    <language>${escapeXml(f.language)}</language>`,
+        `    <lines>${f.lines}</lines>`,
+        `    <size>${f.size}</size>`,
+        `    <content><![CDATA[\n${f.content}${f.content.endsWith('\n') ? '' : '\n'}]]></content>`,
+        '  </file>',
+    ].join('\n')).join('\n');
+
+    return `<?xml version="1.0" encoding="UTF-8"?>\n<context>\n${fileEntries}\n</context>\n`;
 }
 
-function generateContextFromConfig(config, srcPath) {
-    const spinner = ora(`Scanning ${srcPath}...`).start();
+function escapeXml(str) {
+    if (str == null) return '';
+    return String(str)
+        .replace(/&/g, '&amp;')
+        .replace(/</g, '&lt;')
+        .replace(/>/g, '&gt;')
+        .replace(/"/g, '&quot;')
+        .replace(/'/g, '&apos;');
+}
 
-    if (!fs.existsSync(srcPath)) {
-        spinner.fail(`Source path does not exist: ${srcPath}`);
-        return null;
+function renderFormat(format, files, stats, config) {
+    switch (format) {
+        case 'json': return { content: toJson(files),             ext: 'json' };
+        case 'md':   return { content: toMarkdown(files, config), ext: 'md'   };
+        case 'toon': return { content: toToon(files, stats),      ext: 'toon' };
+        case 'xml':  return { content: toXml(files),              ext: 'xml'  };
     }
+}
+
+// ============================================
+// CONTEXT GENERATION
+// ============================================
 
-    contextFiles = getFiles(srcPath, config);
+function generateContext(config) {
+    const spinner = ora(`Scanning ${config.src}...`).start();
 
-    if (contextFiles.length === 0) {
-        spinner.fail(`No files found in: ${srcPath}`);
+    if (!fs.existsSync(config.src)) {
+        spinner.fail(`Source path does not exist: ${config.src}`);
         return null;
     }
 
-    spinner.text = `Building context from ${contextFiles.length} files...`;
-    generatedContext = buildContextContent(contextFiles, config, srcPath);
+    const { files, stats } = scanFiles(config.src, config);
 
-    spinner.succeed(`Context generated: ${chalk.yellow(contextFiles.length)} files, ${chalk.yellow(formatSize(contextStats.totalSize))}`);
+    if (files.length === 0) {
+        spinner.fail(`No files found in: ${config.src}`);
+        return null;
+    }
 
-    return {
-        content: generatedContext,
-        files: contextFiles,
-        stats: contextStats,
-        config
-    };
+    spinner.succeed(`Context built: ${chalk.yellow(files.length)} files, ${chalk.yellow(formatSize(stats.totalSize))}`);
+    return { files, stats, config };
+}
+
+function printSummary(stats) {
+    console.log(chalk.cyan('\n📊 Context Summary:'));
+    console.log(chalk.white(`   Files: ${stats.files}`));
+    console.log(chalk.white(`   Lines: ${stats.totalLines.toLocaleString()}`));
+    console.log(chalk.white(`   Size:  ${formatSize(stats.totalSize)}`));
 }
 
 // ============================================
 // SAVE CONTEXT
 // ============================================
 
-async function saveContext(result) {
+async function saveContext(result, formats, fileName) {
     loadDependencies();
 
-    let outputPath;
-
-    if (result.config.output) {
-        outputPath = result.config.output;
-    } else {
-        const { savePath } = await inquirer.prompt([
-            {
-                type: 'input',
-                name: 'savePath',
-                message: 'Enter output directory:',
-                default: './mkctx'
-            }
-        ]);
-        outputPath = savePath;
+    if (!fileName) {
+        const answer = await inquirer.prompt([{
+            type: 'input', name: 'fileName',
+            message: 'Enter a name for the output files:',
+            default: 'context',
+        }]);
+        fileName = answer.fileName;
     }
 
-    if (!fs.existsSync(outputPath)) {
-        fs.mkdirSync(outputPath, { recursive: true });
-    }
+    const outputDir = result.config.output || './mkctx';
+    fs.mkdirSync(outputDir, { recursive: true });
 
-    const outputFile = path.join(outputPath, 'context.md');
-    fs.writeFileSync(outputFile, result.content);
+    const savedFiles = [];
 
-    console.log(chalk.green(`\n✅ Context saved to: ${chalk.yellow(outputFile)}`));
-    console.log(chalk.gray(`   ${result.stats.files} files, ${formatSize(result.stats.totalSize)}\n`));
+    for (const format of formats) {
+        const { content, ext } = renderFormat(format, result.files, result.stats, result.config);
+        const outputPath = path.join(outputDir, `${fileName}.${ext}`);
+        fs.writeFileSync(outputPath, content);
+        savedFiles.push({
+            format,
+            file:   outputPath,
+            size:   Buffer.byteLength(content, 'utf-8'),
+            tokens: estimateTokens(content),
+        });
+    }
 
-    return outputFile;
+    console.log(chalk.green('\n✅ Context saved:\n'));
+    for (const { format, file, size, tokens } of savedFiles) {
+        console.log(chalk.white(`   ${chalk.cyan(format.toUpperCase().padEnd(4))} → ${chalk.yellow(file)}`));
+        console.log(chalk.gray(`         ${formatSize(size)} | ~${tokens.toLocaleString()} tokens\n`));
+    }
+
+    return savedFiles;
 }
 
 // ============================================
-// MAIN MENU
+// INTERACTIVE PROMPTS
 // ============================================
 
-async function showMainMenu() {
-    loadDependencies();
+async function promptSrcPath() {
+    const { srcPath } = await inquirer.prompt([{
+        type: 'input', name: 'srcPath',
+        message: 'Enter the source path to analyze:',
+        default: '.',
+        validate: input => fs.existsSync(input) || `Path does not exist: ${input}`,
+    }]);
+    return srcPath;
+}
+
+async function promptFormat() {
+    const { format } = await inquirer.prompt([{
+        type: 'list', name: 'format',
+        message: 'Select output format:',
+        default: 'all',
+        choices: [
+            { name: chalk.magenta('📦 All formats (MD, JSON, TOON, XML)'), value: 'all'  },
+            new inquirer.Separator(),
+            { name: chalk.blue('📝 Markdown (.md)'),                       value: 'md'   },
+            { name: chalk.green('🔧 JSON (.json) - Simple array'),          value: 'json' },
+            { name: chalk.yellow('🎒 TOON (.toon) - Token-optimized'),      value: 'toon' },
+            { name: chalk.red('📄 XML (.xml)'),                             value: 'xml'  },
+        ],
+    }]);
+    return resolveFormats(format);
+}
 
-    const hasConfig = hasProjectConfig();
+async function promptMainMenu() {
+    const hasConfig = configFileExists();
 
     console.log(chalk.cyan('\n╔════════════════════════════════════════╗'));
     console.log(chalk.cyan('║') + chalk.cyan.bold('     📄 mkctx - Make Context            ') + chalk.cyan('║'));
@@ -451,155 +593,96 @@ async function showMainMenu() {
     const choices = [];
 
     if (hasConfig) {
-        choices.push({
-            name: chalk.green('📁 Generate from config file'),
-            value: 'from-config'
-        });
+        choices.push({ name: chalk.green('📁 Generate from config file'), value: 'from-config' });
     }
 
     choices.push(
-        {
-            name: chalk.blue('🔍 Generate dynamically (choose path)'),
-            value: 'dynamic'
-        },
+        { name: chalk.blue('🔍 Generate dynamically (choose path)'), value: 'dynamic' },
         new inquirer.Separator(),
         {
-            name: hasConfig 
-                ? chalk.gray('⚙️  View configuration') 
-                : chalk.yellow('⚙️  Create configuration file'),
-            value: 'config'
+            name:  hasConfig ? chalk.gray('⚙️  View configuration') : chalk.yellow('⚙️  Create configuration file'),
+            value: 'config',
         },
         new inquirer.Separator(),
-        {
-            name: chalk.red('❌ Exit'),
-            value: 'exit'
-        }
+        { name: chalk.red('❌ Exit'), value: 'exit' },
     );
 
-    const { action } = await inquirer.prompt([
-        {
-            type: 'list',
-            name: 'action',
-            message: 'What would you like to do?',
-            choices
-        }
-    ]);
-
-    return action;
-}
-
-async function showPostGenerationMenu(result) {
-    loadDependencies();
-
-    console.log(chalk.cyan('\n📊 Context Summary:'));
-    console.log(chalk.white(`   Files: ${result.stats.files}`));
-    console.log(chalk.white(`   Lines: ${result.stats.totalLines.toLocaleString()}`));
-    console.log(chalk.white(`   Size: ${formatSize(result.stats.totalSize)}`));
-    console.log(chalk.white(`   Est. tokens: ~${result.stats.estimatedTokens.toLocaleString()}`));
-
-    const { action } = await inquirer.prompt([
-        {
-            type: 'list',
-            name: 'action',
-            message: 'What would you like to do with this context?',
-            choices: [
-                {
-                    name: chalk.blue('💾 Save context to file'),
-                    value: 'save'
-                },
-                new inquirer.Separator(),
-                {
-                    name: chalk.gray('🔙 Back to main menu'),
-                    value: 'back'
-                },
-                {
-                    name: chalk.red('❌ Exit'),
-                    value: 'exit'
-                }
-            ]
-        }
-    ]);
+    const { action } = await inquirer.prompt([{
+        type: 'list', name: 'action',
+        message: 'What would you like to do?',
+        choices,
+    }]);
 
     return action;
 }
 
 // ============================================
-// MAIN APPLICATION
+// EXECUTION FLOWS
 // ============================================
 
-async function main() {
-    const args = process.argv.slice(2);
+async function runNonInteractive(rawFlags) {
+    loadDependencies();
 
-    if (args.includes('--help') || args.includes('-h') || args[0] === 'help') {
-        showHelp();
-        return;
-    }
+    const cli     = resolveFlags(rawFlags);
+    const config  = buildConfig(cli);
+    const formats = resolveFormats(cli.format);
+    const fileName = cli.name || 'context';
 
-    if (args.includes('--version') || args.includes('-v') || args[0] === 'version') {
-        showVersion();
-        return;
+    if (!fs.existsSync(config.src)) {
+        console.error(`❌ Source path does not exist: ${config.src}`);
+        process.exit(1);
     }
 
-    if (args[0] === 'config') {
-        loadDependencies();
-        createProjectConfig();
-        return;
-    }
+    const result = generateContext(config);
+    if (!result) process.exit(1);
+
+    printSummary(result.stats);
+    await saveContext(result, formats, fileName);
+    console.log(chalk.yellow('👋 Done!\n'));
+}
 
+async function runInteractive() {
     loadDependencies();
 
     let running = true;
 
     while (running) {
-        const action = await showMainMenu();
+        const action = await promptMainMenu();
 
         switch (action) {
             case 'from-config': {
-                const result = await generateContextFromConfigFile();
-                if (result) {
-                    let postAction = await showPostGenerationMenu(result);
-                    
-                    while (postAction !== 'back' && postAction !== 'exit') {
-                        if (postAction === 'save') {
-                            await saveContext(result);
-                            postAction = await showPostGenerationMenu(result);
-                        }
-                    }
-                    
-                    if (postAction === 'exit') {
-                        running = false;
-                    }
-                }
+                const config = loadConfig();
+                if (!config) { console.log(chalk.yellow('\n⚠️  No config file found.')); break; }
+                const result = generateContext(config);
+                if (!result) break;
+                printSummary(result.stats);
+                const formats = await promptFormat();
+                await saveContext(result, formats);
+                console.log(chalk.yellow('👋 Done!\n'));
+                running = false;
                 break;
             }
 
             case 'dynamic': {
-                const result = await generateContextDynamic();
-                if (result) {
-                    let postAction = await showPostGenerationMenu(result);
-                    
-                    while (postAction !== 'back' && postAction !== 'exit') {
-                        if (postAction === 'save') {
-                            await saveContext(result);
-                            postAction = await showPostGenerationMenu(result);
-                        }
-                    }
-                    
-                    if (postAction === 'exit') {
-                        running = false;
-                    }
-                }
+                const srcPath = await promptSrcPath();
+                const config  = { ...DEFAULT_CONFIG, src: srcPath };
+                const result  = generateContext(config);
+                if (!result) break;
+                printSummary(result.stats);
+                const formats = await promptFormat();
+                await saveContext(result, formats);
+                console.log(chalk.yellow('👋 Done!\n'));
+                running = false;
                 break;
             }
 
             case 'config':
-                if (hasProjectConfig()) {
+                if (configFileExists()) {
                     console.log(chalk.cyan('\n📄 Current configuration:\n'));
-                    const config = loadProjectConfig();
-                    console.log(JSON.stringify(config, null, 2));
+                    console.log(JSON.stringify(loadConfig(), null, 2));
                     console.log(chalk.gray(`\n   Edit ${CONFIG_FILE} to modify settings.\n`));
                 } else {
-                    createProjectConfig();
+                    createConfigFile();
                 }
                 break;
 
@@ -611,9 +694,12 @@ async function main() {
     }
 }
 
+// ============================================
+// HELP & VERSION
+// ============================================
+
 function showHelp() {
     loadDependencies();
-    
     console.log(chalk.cyan(`
 ╔════════════════════════════════════════════════════════════╗
 ║  📄 mkctx - Make Context for AI Code Analysis              ║
@@ -622,17 +708,34 @@ function showHelp() {
  ${chalk.white('Generate context files from your codebase for AI analysis.')}
 
  ${chalk.yellow('Usage:')}
-  mkctx                    Interactive mode (recommended)
-  mkctx config             Create configuration file
-  mkctx help               Show this help message
-  mkctx version            Show version
-
- ${chalk.yellow('Configuration (mkctx.config.json):')}
-  src            Source directory to scan (default: ".")
-  ignore         Comma-separated patterns to ignore
-  output         Output directory (default: "./mkctx")
-  first_comment  Comment at the beginning of context
-  last_comment   Comment at the end of context
+  mkctx                         Interactive mode
+  mkctx config                  Create configuration file
+  mkctx help / --help           Show this help message
+  mkctx version / --version     Show version
+
+ ${chalk.yellow('Non-interactive flags (skip all prompts):')}
+  --src      <path>             Source directory              (default: .)
+  --output   <path>             Output directory              (default: ./mkctx)
+  --format   <fmt>              md, json, toon, xml, all, or comma-separated
+  --name     <filename>         Output file base name         (default: context)
+  --ignore   <patterns>         Comma-separated ignore patterns
+  --first-comment <text>        Override first comment header
+  --last-comment  <text>        Override last comment footer
+
+ ${chalk.yellow('Short aliases:')}
+  -s  --src     -o  --output     -f  --format     -n  --name
+
+ ${chalk.yellow('Examples:')}
+  mkctx --src ./src
+  mkctx --src . --format all --name my-project --output ./docs
+  mkctx --src ./app --format md,json --ignore "*.test.ts,__tests__/"
+  mkctx -s ./src -f toon -n snapshot
+
+ ${chalk.yellow('Output Formats:')}
+  ${chalk.green('JSON')}    Simple array of file objects
+  ${chalk.blue('MD')}      Markdown with code blocks
+  ${chalk.yellow('TOON')}    Token-Oriented Object Notation (LLM optimized)
+  ${chalk.red('XML')}     XML with CDATA sections
 
  ${chalk.gray('More info: https://github.com/pnkkzero/mkctx')}
 `));
@@ -643,18 +746,30 @@ function showVersion() {
         const pkg = require('./package.json');
         console.log(`mkctx v${pkg.version}`);
     } catch {
-        console.log('mkctx v3.0.0');
+        console.log('mkctx v4.0.0');
     }
 }
 
 // ============================================
-// RUN
+// ENTRY POINT
 // ============================================
 
-main().catch((err) => {
-    console.error(`\n❌ Error: ${err.message}`);
-    if (process.env.DEBUG) {
-        console.error(err.stack);
+async function main() {
+    const { command, flags } = parseArgs(process.argv);
+
+    if (flags.help || flags.h || command === 'help')       return showHelp();
+    if (flags.version || flags.v || command === 'version') return showVersion();
+    if (command === 'config') { loadDependencies(); return createConfigFile(); }
+
+    if (isNonInteractiveMode(flags)) {
+        await runNonInteractive(flags);
+    } else {
+        await runInteractive();
     }
+}
+
+main().catch(err => {
+    console.error(`\n❌ Error: ${err.message}`);
+    if (process.env.DEBUG) console.error(err.stack);
     process.exit(1);
 });