kramscan 0.1.1 → 0.2.0

package/dist/commands/onboard.js

@@ -1,164 +1,211 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerOnboardCommand = registerOnboardCommand;
-const readline = __importStar(require("readline"));
+const inquirer_1 = __importDefault(require("inquirer"));
+const axios_1 = __importDefault(require("axios"));
+const openai_1 = __importDefault(require("openai"));
 const config_1 = require("../core/config");
 const logger_1 = require("../utils/logger");
-// ─── ANSI Helpers ──────────────────────────────────────────────────
-const c = {
-    reset: "\x1b[0m",
-    bold: "\x1b[1m",
-    dim: "\x1b[2m",
-    cyan: "\x1b[36m",
-    green: "\x1b[32m",
-    yellow: "\x1b[33m",
-    gray: "\x1b[90m",
-    white: "\x1b[37m",
-    brightCyan: "\x1b[96m",
-};
-// ─── Prompt Utilities ──────────────────────────────────────────────
-function ask(rl, question, defaultVal) {
-    const defaultHint = defaultVal ? ` ${c.gray}(${defaultVal})${c.reset}` : "";
-    return new Promise((resolve) => {
-        rl.question(`  ${c.cyan}?${c.reset} ${question}${defaultHint} `, (answer) => {
-            resolve(answer.trim() || defaultVal || "");
-        });
-    });
-}
-function askConfirm(rl, question, defaultVal = true) {
-    const hint = defaultVal ? `${c.gray}(Y/n)${c.reset}` : `${c.gray}(y/N)${c.reset}`;
-    return new Promise((resolve) => {
-        rl.question(`  ${c.cyan}?${c.reset} ${question} ${hint} `, (answer) => {
-            const a = answer.trim().toLowerCase();
-            if (a === "")
-                resolve(defaultVal);
-            else
-                resolve(a === "y" || a === "yes");
-        });
-    });
+const theme_1 = require("../utils/theme");
+function getDefaultModel(provider) {
+    switch (provider) {
+        case "anthropic":
+            return "claude-3-5-sonnet-20241022";
+        case "gemini":
+            return "gemini-2.0-flash";
+        case "openrouter":
+            return "anthropic/claude-3.5-sonnet";
+        case "mistral":
+            return "mistral-large-latest";
+        case "kimi":
+            return "moonshot-v1-8k";
+        case "groq":
+            return "llama-3.1-8b-instant";
+        default:
+            return "gpt-4";
+    }
 }
-function askList(rl, question, choices, defaultVal) {
-    const choicesStr = choices
-        .map((ch, i) => {
-        const isDefault = ch === defaultVal;
-        return `    ${isDefault ? c.brightCyan + "❯" : " "} ${ch}${c.reset}`;
-    })
-        .join("\n");
-    return new Promise((resolve) => {
-        console.log(`  ${c.cyan}?${c.reset} ${question}`);
-        console.log(choicesStr);
-        rl.question(`  ${c.gray}Enter choice:${c.reset} `, (answer) => {
-            const trimmed = answer.trim();
-            if (choices.includes(trimmed)) {
-                resolve(trimmed);
-            }
-            else {
-                resolve(defaultVal || choices[0]);
-            }
-        });
-    });
+function getEnvApiKey(provider) {
+    const envVars = {
+        openai: process.env.OPENAI_API_KEY,
+        anthropic: process.env.ANTHROPIC_API_KEY,
+        gemini: process.env.GEMINI_API_KEY,
+        mistral: process.env.MISTRAL_API_KEY,
+        openrouter: process.env.OPENROUTER_API_KEY,
+        kimi: process.env.KIMI_API_KEY,
+        groq: process.env.GROQ_API_KEY,
+    };
+    return envVars[provider] || "";
 }
-function askPassword(rl, question) {
-    return new Promise((resolve) => {
-        rl.question(`  ${c.cyan}?${c.reset} ${question} ${c.gray}(hidden)${c.reset} `, (answer) => {
-            resolve(answer.trim());
+async function modelExists(provider, apiKey, model) {
+    if (!apiKey || !model) {
+        return true;
+    }
+    if (provider === "gemini") {
+        const resp = await axios_1.default.get("https://generativelanguage.googleapis.com/v1beta/models", { params: { key: apiKey } });
+        const models = resp.data?.models || [];
+        return models.some((m) => {
+            const id = m.name?.startsWith("models/")
+                ? m.name.slice("models/".length)
+                : m.name;
+            return (id === model &&
+                (m.supportedGenerationMethods || []).includes("generateContent"));
         });
-    });
+    }
+    if (provider === "openai" ||
+        provider === "openrouter" ||
+        provider === "kimi" ||
+        provider === "groq") {
+        const baseURL = provider === "openrouter"
+            ? "https://openrouter.ai/api/v1"
+            : provider === "kimi"
+                ? "https://api.moonshot.cn/v1"
+                : provider === "groq"
+                    ? "https://api.groq.com/openai/v1"
+                    : undefined;
+        const client = new openai_1.default(baseURL ? { apiKey, baseURL } : { apiKey });
+        const resp = await client.models.list();
+        return (resp.data || []).some((m) => m.id === model);
+    }
+    return true;
 }
-// ─── Command Registration ─────────────────────────────────────────
 function registerOnboardCommand(program) {
     program
         .command("onboard")
         .description("First-time setup wizard")
         .action(async () => {
-        const store = (0, config_1.getConfigStore)();
-        const rl = readline.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-        });
+        const config = await (0, config_1.getConfig)();
         console.log("");
-        console.log(`  ${c.bold}${c.brightCyan}━━━ KramScan Setup Wizard ━━━${c.reset}`);
-        console.log(`  ${c.gray}Configure your scanning environment${c.reset}`);
+        console.log(theme_1.theme.brand.bold("🚀 KramScan Setup Wizard"));
+        console.log(theme_1.theme.gray("─".repeat(50)));
+        console.log(theme_1.theme.white("  Configure your scanning environment in a few easy steps."));
         console.log("");
-        // AI Configuration
-        const aiEnabled = await askConfirm(rl, "Enable AI analysis?", false);
-        store.set("ai.enabled", aiEnabled);
-        if (aiEnabled) {
-            const aiProvider = await askList(rl, "Select AI provider", [
-                "openai",
-                "anthropic",
-                "gemini",
-                "openrouter",
-                "mistral",
-                "kimi"
-            ], "openai");
-            store.set("ai.provider", aiProvider);
-            const apiKey = await askPassword(rl, "API key (leave blank to skip)");
-            if (apiKey) {
-                store.set("ai.apiKey", apiKey);
+        // Smart detection for API keys
+        const detectedProviders = Object.keys({
+            openai: process.env.OPENAI_API_KEY,
+            anthropic: process.env.ANTHROPIC_API_KEY,
+            gemini: process.env.GEMINI_API_KEY,
+            mistral: process.env.MISTRAL_API_KEY,
+            openrouter: process.env.OPENROUTER_API_KEY,
+            kimi: process.env.KIMI_API_KEY,
+            groq: process.env.GROQ_API_KEY,
+        }).filter(p => !!getEnvApiKey(p));
+        if (detectedProviders.length > 0) {
+            console.log(theme_1.theme.green(`  ✨ Detected API keys in environment for: ${detectedProviders.join(", ")}`));
+            console.log("");
+        }
+        const answers = await inquirer_1.default.prompt([
+            {
+                type: "confirm",
+                name: "aiEnabled",
+                message: "Enable AI analysis?",
+                default: config.ai.enabled,
+            },
+            {
+                type: "list",
+                name: "aiProvider",
+                message: "Select AI provider",
+                choices: [
+                    "openai",
+                    "anthropic",
+                    "gemini",
+                    "openrouter",
+                    "mistral",
+                    "kimi",
+                    "groq",
+                ],
+                default: config.ai.provider,
+                when: (a) => a.aiEnabled,
+            },
+            {
+                type: "password",
+                name: "apiKey",
+                message: "API key (leave blank to keep existing)",
+                default: "",
+                mask: "*",
+                when: (a) => a.aiEnabled,
+            },
+            {
+                type: "input",
+                name: "model",
+                message: "Default AI model",
+                default: (a) => config.ai.defaultModel ||
+                    getDefaultModel((a.aiProvider || config.ai.provider)),
+                when: (a) => a.aiEnabled,
+            },
+            {
+                type: "list",
+                name: "reportFormat",
+                message: "Default report format",
+                choices: ["word", "txt", "json"],
+                default: config.report.defaultFormat,
+            },
+            {
+                type: "confirm",
+                name: "strictScope",
+                message: "Enable strict scope enforcement?",
+                default: config.scan.strictScope,
+            },
+            {
+                type: "number",
+                name: "rateLimit",
+                message: "Requests per second rate limit",
+                default: config.scan.rateLimitPerSecond,
+                validate: (v) => Number.isFinite(v) && v > 0 ? true : "Enter a positive number",
+            },
+        ]);
+        config.ai.enabled = !!answers.aiEnabled;
+        if (config.ai.enabled) {
+            config.ai.provider = answers.aiProvider;
+            if (answers.apiKey) {
+                config.ai.apiKey = answers.apiKey;
+            }
+            const keyForCheck = config.ai.apiKey || getEnvApiKey(config.ai.provider);
+            let chosenModel = String(answers.model || getDefaultModel(config.ai.provider));
+            if (keyForCheck) {
+                try {
+                    const ok = await modelExists(config.ai.provider, keyForCheck, chosenModel);
+                    if (!ok) {
+                        logger_1.logger.warn(`Model '${chosenModel}' is not available for provider '${config.ai.provider}'.`);
+                        logger_1.logger.warn("Tip: run 'kramscan ai models' to see valid models.");
+                        const retry = await inquirer_1.default.prompt([
+                            {
+                                type: "confirm",
+                                name: "retry",
+                                message: "Enter a different model now?",
+                                default: true,
+                            },
+                        ]);
+                        if (retry.retry) {
+                            const modelAns = await inquirer_1.default.prompt([
+                                {
+                                    type: "input",
+                                    name: "model",
+                                    message: "Default AI model",
+                                    default: getDefaultModel(config.ai.provider),
+                                },
+                            ]);
+                            chosenModel = String(modelAns.model || chosenModel);
+                        }
+                    }
+                }
+                catch (error) {
+                    logger_1.logger.warn(`Model preflight check failed: ${error.message}`);
+                }
             }
-            // Provider-specific default models
-            let defaultModel = "gpt-4";
-            if (aiProvider === "anthropic")
-                defaultModel = "claude-3-5-sonnet-20241022";
-            else if (aiProvider === "gemini")
-                defaultModel = "gemini-2.0-flash-exp";
-            else if (aiProvider === "openrouter")
-                defaultModel = "anthropic/claude-3.5-sonnet";
-            else if (aiProvider === "mistral")
-                defaultModel = "mistral-large-latest";
-            else if (aiProvider === "kimi")
-                defaultModel = "moonshot-v1-8k";
-            const model = await ask(rl, "Default AI model", defaultModel);
-            store.set("ai.defaultModel", model);
+            config.ai.defaultModel = chosenModel;
         }
-        // Report Configuration
-        const reportFormat = await askList(rl, "Default report format", ["word", "txt", "json"], "word");
-        store.set("report.defaultFormat", reportFormat);
-        // Scan Configuration
-        const strictScope = await askConfirm(rl, "Enable strict scope enforcement?", true);
-        store.set("scan.strictScope", strictScope);
-        const rateLimitStr = await ask(rl, "Requests per second rate limit", "5");
-        const rateLimit = parseInt(rateLimitStr, 10) || 5;
-        store.set("scan.rateLimitPerSecond", rateLimit);
-        rl.close();
+        config.report.defaultFormat = answers.reportFormat;
+        config.scan.strictScope = !!answers.strictScope;
+        config.scan.rateLimitPerSecond = answers.rateLimit;
+        await (0, config_1.setConfig)(config);
         console.log("");
         logger_1.logger.success("Onboarding complete! Your configuration has been saved.");
-        console.log(`  ${c.gray}Config location: ~/.kramscan/config.json${c.reset}`);
-        console.log(`  ${c.gray}Run ${c.cyan}kramscan${c.gray} to get started.${c.reset}`);
+        console.log("Config location: ~/.kramscan/config.json");
+        console.log("Run 'kramscan' to get started.");
         console.log("");
     });
 }

package/dist/commands/report.js

@@ -5,60 +5,59 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerReportCommand = registerReportCommand;
 const chalk_1 = __importDefault(require("chalk"));
-const logger_1 = require("../utils/logger");
-const config_1 = require("../core/config");
+const docx_1 = require("docx");
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
-const os_1 = __importDefault(require("os"));
-const docx_1 = require("docx");
+const config_1 = require("../core/config");
+const scan_storage_1 = require("../core/scan-storage");
+const ai_client_1 = require("../core/ai-client");
+const logger_1 = require("../utils/logger");
 function registerReportCommand(program) {
     program
         .command("report [scan-file]")
         .description("Generate a professional security report")
         .option("-f, --format <type>", "Report format: word|json|txt")
         .option("-o, --output <file>", "Output filename")
+        .option("--ai-summary", "Generate an AI-powered executive summary")
         .action(async (scanFile, options) => {
         console.log("");
-        console.log(chalk_1.default.bold.cyan("📄 Generating Security Report"));
-        console.log(chalk_1.default.gray("─".repeat(50)));
+        console.log(chalk_1.default.bold.cyan("Generating Security Report"));
+        console.log(chalk_1.default.gray("-".repeat(50)));
         console.log("");
+        let spinner = null;
         try {
-            // Load scan results
-            let filepath;
-            if (scanFile) {
-                filepath = path_1.default.isAbsolute(scanFile)
-                    ? scanFile
-                    : path_1.default.join(process.cwd(), scanFile);
-            }
-            else {
-                // Find latest scan
-                const scanDir = path_1.default.join(os_1.default.homedir(), ".kramscan", "scans");
-                const files = await promises_1.default.readdir(scanDir);
-                const scanFiles = files.filter((f) => f.endsWith(".json"));
-                if (scanFiles.length === 0) {
-                    logger_1.logger.error("No scan results found. Run 'kramscan scan <url>' first.");
-                    process.exit(1);
-                }
-                scanFiles.sort().reverse();
-                filepath = path_1.default.join(scanDir, scanFiles[0]);
-                logger_1.logger.info(`Using latest scan: ${scanFiles[0]}`);
+            const resolved = await (0, scan_storage_1.resolveScanFile)(scanFile);
+            const filepath = resolved.filepath;
+            if (resolved.isLatest) {
+                logger_1.logger.info(`Using latest scan: ${resolved.filename}`);
             }
             const content = await promises_1.default.readFile(filepath, "utf-8");
             const scanResult = JSON.parse(content);
-            // Determine format
             const config = await (0, config_1.getConfig)();
-            const format = options.format || config.report.defaultFormat;
-            const spinner = logger_1.logger.spinner(`Generating ${format.toUpperCase()} report...`);
+            const format = (options.format || config.report.defaultFormat);
+            let aiSummary;
+            if (options.aiSummary) {
+                spinner = logger_1.logger.spinner("Generating AI executive summary...");
+                try {
+                    const aiClient = await (0, ai_client_1.createAIClient)();
+                    aiSummary = await (0, ai_client_1.generateExecutiveSummary)(aiClient, scanResult);
+                    spinner.succeed("AI summary generated!");
+                }
+                catch (err) {
+                    spinner.warn(`AI summary failed: ${err.message}`);
+                }
+            }
+            spinner = logger_1.logger.spinner(`Generating ${format.toUpperCase()} report...`);
             let outputPath;
             switch (format) {
                 case "word":
-                    outputPath = await generateWordReport(scanResult, options.output);
+                    outputPath = await generateWordReport(scanResult, options.output, aiSummary);
                     break;
                 case "json":
-                    outputPath = await generateJsonReport(scanResult, options.output);
+                    outputPath = await generateJsonReport(scanResult, options.output, aiSummary);
                     break;
                 case "txt":
-                    outputPath = await generateTxtReport(scanResult, options.output);
+                    outputPath = await generateTxtReport(scanResult, options.output, aiSummary);
                     break;
                 default:
                     throw new Error(`Unsupported format: ${format}`);
@@ -69,18 +68,20 @@ function registerReportCommand(program) {
             console.log("");
         }
         catch (error) {
+            if (spinner) {
+                spinner.fail("Report generation failed");
+            }
             logger_1.logger.error(error.message);
             process.exit(1);
         }
     });
 }
-async function generateWordReport(scanResult, outputFile) {
+async function generateWordReport(scanResult, outputFile, aiSummary) {
     const doc = new docx_1.Document({
         sections: [
             {
                 properties: {},
                 children: [
-                    // Title
                     new docx_1.Paragraph({
                         text: "Security Assessment Report",
                         heading: docx_1.HeadingLevel.HEADING_1,
@@ -95,7 +96,6 @@ async function generateWordReport(scanResult, outputFile) {
                         alignment: docx_1.AlignmentType.CENTER,
                     }),
                     new docx_1.Paragraph({ text: "" }),
-                    // Executive Summary
                     new docx_1.Paragraph({
                         text: "Executive Summary",
                         heading: docx_1.HeadingLevel.HEADING_2,
@@ -103,45 +103,37 @@ async function generateWordReport(scanResult, outputFile) {
                     new docx_1.Paragraph({
                         children: [
                             new docx_1.TextRun({
-                                text: `This report contains the results of an automated security assessment performed on ${scanResult.target}. `,
-                            }),
-                            new docx_1.TextRun({
-                                text: `A total of ${scanResult.summary.total} vulnerabilities were identified, `,
-                            }),
-                            new docx_1.TextRun({
-                                text: `including ${scanResult.summary.critical} critical and ${scanResult.summary.high} high severity issues.`,
+                                text: aiSummary || `This report contains the results of an automated security assessment performed on ${scanResult.target}. ` +
+                                    `A total of ${scanResult.summary.total} vulnerabilities were identified, ` +
+                                    `including ${scanResult.summary.critical} critical and ${scanResult.summary.high} high severity issues.`,
                             }),
                         ],
                     }),
                     new docx_1.Paragraph({ text: "" }),
-                    // Scan Statistics
                     new docx_1.Paragraph({
                         text: "Scan Statistics",
                         heading: docx_1.HeadingLevel.HEADING_2,
                     }),
-                    new docx_1.Paragraph({ text: `• URLs Crawled: ${scanResult.metadata.crawledUrls}` }),
-                    new docx_1.Paragraph({ text: `• Forms Tested: ${scanResult.metadata.testedForms}` }),
-                    new docx_1.Paragraph({ text: `• Requests Made: ${scanResult.metadata.requestsMade}` }),
+                    new docx_1.Paragraph({ text: `- URLs Crawled: ${scanResult.metadata.crawledUrls}` }),
+                    new docx_1.Paragraph({ text: `- Forms Tested: ${scanResult.metadata.testedForms}` }),
+                    new docx_1.Paragraph({ text: `- Requests Made: ${scanResult.metadata.requestsMade}` }),
                     new docx_1.Paragraph({
-                        text: `• Duration: ${(scanResult.duration / 1000).toFixed(2)} seconds`,
+                        text: `- Duration: ${(scanResult.duration / 1000).toFixed(2)} seconds`,
                     }),
                     new docx_1.Paragraph({ text: "" }),
-                    // Findings
                     new docx_1.Paragraph({
                         text: "Detailed Findings",
                         heading: docx_1.HeadingLevel.HEADING_2,
                     }),
-                    ...scanResult.vulnerabilities.flatMap((vuln, i) => [
+                    ...scanResult.vulnerabilities.flatMap((vuln, index) => [
                         new docx_1.Paragraph({
-                            text: `${i + 1}. ${vuln.title} [${vuln.severity.toUpperCase()}]`,
+                            text: `${index + 1}. ${vuln.title} [${vuln.severity.toUpperCase()}]`,
                             heading: docx_1.HeadingLevel.HEADING_3,
                         }),
                         new docx_1.Paragraph({ text: `URL: ${vuln.url}` }),
                         new docx_1.Paragraph({ text: `Type: ${vuln.type}` }),
                         new docx_1.Paragraph({ text: `Description: ${vuln.description}` }),
-                        ...(vuln.evidence
-                            ? [new docx_1.Paragraph({ text: `Evidence: ${vuln.evidence}` })]
-                            : []),
+                        ...(vuln.evidence ? [new docx_1.Paragraph({ text: `Evidence: ${vuln.evidence}` })] : []),
                         ...(vuln.remediation
                             ? [new docx_1.Paragraph({ text: `Remediation: ${vuln.remediation}` })]
                             : []),
@@ -153,28 +145,23 @@ async function generateWordReport(scanResult, outputFile) {
         ],
     });
     const buffer = await docx_1.Packer.toBuffer(doc);
-    const reportsDir = path_1.default.join(os_1.default.homedir(), ".kramscan", "reports");
-    await promises_1.default.mkdir(reportsDir, { recursive: true });
+    const reportsDir = await (0, scan_storage_1.ensureReportsDirectory)();
     const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
     const filename = outputFile || `report-${timestamp}.docx`;
-    const filepath = path_1.default.isAbsolute(filename)
-        ? filename
-        : path_1.default.join(reportsDir, filename);
+    const filepath = path_1.default.isAbsolute(filename) ? filename : path_1.default.join(reportsDir, filename);
     await promises_1.default.writeFile(filepath, buffer);
     return filepath;
 }
-async function generateJsonReport(scanResult, outputFile) {
-    const reportsDir = path_1.default.join(os_1.default.homedir(), ".kramscan", "reports");
-    await promises_1.default.mkdir(reportsDir, { recursive: true });
+async function generateJsonReport(scanResult, outputFile, aiSummary) {
+    const reportsDir = await (0, scan_storage_1.ensureReportsDirectory)();
     const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
     const filename = outputFile || `report-${timestamp}.json`;
-    const filepath = path_1.default.isAbsolute(filename)
-        ? filename
-        : path_1.default.join(reportsDir, filename);
-    await promises_1.default.writeFile(filepath, JSON.stringify(scanResult, null, 2));
+    const filepath = path_1.default.isAbsolute(filename) ? filename : path_1.default.join(reportsDir, filename);
+    const finalResult = aiSummary ? { ...scanResult, aiSummary } : scanResult;
+    await promises_1.default.writeFile(filepath, JSON.stringify(finalResult, null, 2));
     return filepath;
 }
-async function generateTxtReport(scanResult, outputFile) {
+async function generateTxtReport(scanResult, outputFile, aiSummary) {
     const lines = [];
     lines.push("=".repeat(60));
     lines.push("SECURITY ASSESSMENT REPORT");
@@ -186,7 +173,12 @@ async function generateTxtReport(scanResult, outputFile) {
     lines.push("");
     lines.push("EXECUTIVE SUMMARY");
     lines.push("-".repeat(60));
-    lines.push(`Total Vulnerabilities: ${scanResult.summary.total} (${scanResult.summary.critical} Critical, ${scanResult.summary.high} High, ${scanResult.summary.medium} Medium, ${scanResult.summary.low} Low, ${scanResult.summary.info} Info)`);
+    if (aiSummary) {
+        lines.push(aiSummary);
+    }
+    else {
+        lines.push(`Total Vulnerabilities: ${scanResult.summary.total} (${scanResult.summary.critical} Critical, ${scanResult.summary.high} High, ${scanResult.summary.medium} Medium, ${scanResult.summary.low} Low, ${scanResult.summary.info} Info)`);
+    }
     lines.push("");
     lines.push("SCAN STATISTICS");
     lines.push("-".repeat(60));
@@ -197,29 +189,29 @@ async function generateTxtReport(scanResult, outputFile) {
     lines.push("DETAILED FINDINGS");
     lines.push("-".repeat(60));
     lines.push("");
-    scanResult.vulnerabilities.forEach((vuln, i) => {
-        lines.push(`${i + 1}. ${vuln.title} [${vuln.severity.toUpperCase()}]`);
+    scanResult.vulnerabilities.forEach((vuln, index) => {
+        lines.push(`${index + 1}. ${vuln.title} [${vuln.severity.toUpperCase()}]`);
         lines.push(`   URL: ${vuln.url}`);
         lines.push(`   Type: ${vuln.type}`);
         lines.push(`   Description: ${vuln.description}`);
-        if (vuln.evidence)
+        if (vuln.evidence) {
             lines.push(`   Evidence: ${vuln.evidence}`);
-        if (vuln.remediation)
+        }
+        if (vuln.remediation) {
             lines.push(`   Remediation: ${vuln.remediation}`);
-        if (vuln.cwe)
+        }
+        if (vuln.cwe) {
             lines.push(`   CWE: ${vuln.cwe}`);
+        }
         lines.push("");
     });
     lines.push("=".repeat(60));
     lines.push("End of Report");
     lines.push("=".repeat(60));
-    const reportsDir = path_1.default.join(os_1.default.homedir(), ".kramscan", "reports");
-    await promises_1.default.mkdir(reportsDir, { recursive: true });
+    const reportsDir = await (0, scan_storage_1.ensureReportsDirectory)();
     const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
     const filename = outputFile || `report-${timestamp}.txt`;
-    const filepath = path_1.default.isAbsolute(filename)
-        ? filename
-        : path_1.default.join(reportsDir, filename);
+    const filepath = path_1.default.isAbsolute(filename) ? filename : path_1.default.join(reportsDir, filename);
     await promises_1.default.writeFile(filepath, lines.join("\n"));
     return filepath;
 }