osagent 0.1.38 → 0.1.40

package/dist/src/ui/commands/doctorCommand.js

@@ -0,0 +1,681 @@
+/**
+ * @license
+ * Copyright 2025 OSAgent OC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { CommandKind } from './types.js';
+import { t } from '../../i18n/index.js';
+import { execSync } from 'child_process';
+import { existsSync, readdirSync, mkdirSync, writeFileSync, readFileSync } from 'fs';
+import { homedir, platform, release, cpus, totalmem, freemem } from 'os';
+import { join } from 'path';
+import { colors, colorize, statusIcon, createBox, divider, createTree, keyValue, sectionHeader, icons, formatBytes, box, } from '../utils/terminalUI.js';
+// ============================================================================
+// SYSTEM INFO
+// ============================================================================
+function getSystemInfo() {
+    let npmVersion = 'unknown';
+    try {
+        npmVersion = execSync('npm --version 2>/dev/null', { encoding: 'utf-8' }).trim();
+    }
+    catch {
+        // ignore
+    }
+    const cpuInfo = cpus();
+    const cpuModel = cpuInfo.length > 0 ? cpuInfo[0].model : 'unknown';
+    return {
+        os: `${platform()} ${release()}`,
+        arch: process.arch,
+        nodeVersion: process.version,
+        npmVersion,
+        memory: `${formatBytes(freemem())} free / ${formatBytes(totalmem())} total`,
+        cpu: `${cpuModel} (${cpuInfo.length} cores)`,
+    };
+}
+// ============================================================================
+// VERSION CHECKS
+// ============================================================================
+function getCurrentVersion() {
+    try {
+        // First try to get from package.json in the install location
+        const pkg = execSync('npm list -g osagent --json 2>/dev/null', { encoding: 'utf-8' });
+        const parsed = JSON.parse(pkg);
+        return parsed.dependencies?.osagent?.version || 'unknown';
+    }
+    catch {
+        // Fallback to checking npm view
+        try {
+            return execSync('npm view osagent-cli version 2>/dev/null', { encoding: 'utf-8' }).trim();
+        }
+        catch {
+            return 'unknown';
+        }
+    }
+}
+function getLatestVersion() {
+    try {
+        const result = execSync('npm view osagent-cli version 2>/dev/null', { encoding: 'utf-8' });
+        return result.trim();
+    }
+    catch {
+        return 'unknown';
+    }
+}
+// ============================================================================
+// HEALTH CHECKS
+// ============================================================================
+function checkNodeVersion() {
+    const version = process.version;
+    const major = parseInt(version.slice(1).split('.')[0], 10);
+    if (major >= 20) {
+        return { name: 'Node.js Runtime', status: 'ok', message: `${version}` };
+    }
+    else if (major >= 18) {
+        return {
+            name: 'Node.js Runtime',
+            status: 'warning',
+            message: `${version} (upgrade recommended)`,
+            fix: 'Install Node.js 20+ for best performance',
+        };
+    }
+    return {
+        name: 'Node.js Runtime',
+        status: 'error',
+        message: `${version} (unsupported)`,
+        fix: 'Install Node.js 20+ from https://nodejs.org',
+    };
+}
+function checkOllamaInstalled() {
+    try {
+        const result = execSync('ollama --version 2>&1', { encoding: 'utf-8', timeout: 5000 });
+        const version = result.match(/ollama version ([\d.]+)/)?.[1] || result.trim();
+        return { name: 'Ollama CLI', status: 'ok', message: `v${version}` };
+    }
+    catch {
+        return {
+            name: 'Ollama CLI',
+            status: 'warning',
+            message: 'not installed',
+            fix: 'Install from https://ollama.com for local models',
+            autoFix: async () => {
+                if (platform() === 'darwin') {
+                    try {
+                        execSync('brew install ollama', { encoding: 'utf-8', stdio: 'pipe' });
+                        return true;
+                    }
+                    catch {
+                        return false;
+                    }
+                }
+                return false;
+            },
+        };
+    }
+}
+function checkOllamaRunning() {
+    try {
+        execSync('curl -s --max-time 2 http://localhost:11434/api/tags', { encoding: 'utf-8' });
+        return { name: 'Ollama Server', status: 'ok', message: 'running on :11434' };
+    }
+    catch {
+        return {
+            name: 'Ollama Server',
+            status: 'warning',
+            message: 'not running',
+            fix: 'Run: ollama serve',
+            autoFix: async () => {
+                try {
+                    // Start ollama in background using spawn
+                    const { spawn } = await import('child_process');
+                    const child = spawn('ollama', ['serve'], {
+                        detached: true,
+                        stdio: 'ignore',
+                    });
+                    child.unref();
+                    // Give it a moment to start
+                    await new Promise(resolve => setTimeout(resolve, 2000));
+                    // Check if it started
+                    try {
+                        execSync('curl -s --max-time 2 http://localhost:11434/api/tags', { encoding: 'utf-8' });
+                        return true;
+                    }
+                    catch {
+                        return false;
+                    }
+                }
+                catch {
+                    return false;
+                }
+            },
+        };
+    }
+}
+function checkGitInstalled() {
+    try {
+        const result = execSync('git --version 2>&1', { encoding: 'utf-8', timeout: 5000 });
+        const version = result.match(/git version ([\d.]+)/)?.[1] || 'installed';
+        return { name: 'Git', status: 'ok', message: `v${version}` };
+    }
+    catch {
+        return {
+            name: 'Git',
+            status: 'warning',
+            message: 'not installed',
+            fix: 'Install Git for version control features',
+        };
+    }
+}
+function checkAPIKeys() {
+    const keys = {
+        OLLAMA_API_KEY: !!process.env['OLLAMA_API_KEY'],
+        GROQ_API_KEY: !!process.env['GROQ_API_KEY'],
+        OPENAI_API_KEY: !!process.env['OPENAI_API_KEY'],
+    };
+    const found = Object.entries(keys).filter(([_, v]) => v).map(([k]) => k);
+    if (found.length > 0) {
+        return {
+            name: 'API Keys',
+            status: 'ok',
+            message: found.map(k => k.replace('_API_KEY', '')).join(', '),
+        };
+    }
+    return {
+        name: 'API Keys',
+        status: 'warning',
+        message: 'none configured',
+        fix: 'Set OLLAMA_API_KEY, GROQ_API_KEY, or OPENAI_API_KEY',
+        details: [
+            'export OLLAMA_API_KEY="your-key"',
+            'Or configure in ~/.osagent/settings.json',
+        ],
+    };
+}
+function checkSettings() {
+    const settingsPath = join(homedir(), '.osagent', 'settings.json');
+    if (existsSync(settingsPath)) {
+        try {
+            const content = readFileSync(settingsPath, 'utf-8');
+            const settings = JSON.parse(content);
+            const keys = Object.keys(settings);
+            return {
+                name: 'Settings File',
+                status: 'ok',
+                message: `${keys.length} settings configured`,
+                details: keys.slice(0, 5),
+            };
+        }
+        catch {
+            return {
+                name: 'Settings File',
+                status: 'error',
+                message: 'invalid JSON',
+                fix: 'Fix JSON syntax or delete ~/.osagent/settings.json',
+            };
+        }
+    }
+    return {
+        name: 'Settings File',
+        status: 'ok',
+        message: 'using defaults',
+    };
+}
+function checkSystemPrompt() {
+    const promptPath = join(homedir(), '.osagent', 'system.md');
+    if (existsSync(promptPath)) {
+        try {
+            const content = readFileSync(promptPath, 'utf-8');
+            const lines = content.split('\n').length;
+            return {
+                name: 'System Prompt',
+                status: 'ok',
+                message: `custom (${lines} lines)`,
+            };
+        }
+        catch {
+            return {
+                name: 'System Prompt',
+                status: 'warning',
+                message: 'unreadable',
+            };
+        }
+    }
+    return { name: 'System Prompt', status: 'ok', message: 'default' };
+}
+function checkCustomAgents() {
+    const agentsDir = join(homedir(), '.osagent', 'agents');
+    if (existsSync(agentsDir)) {
+        try {
+            const agents = readdirSync(agentsDir).filter(f => f.endsWith('.json') || f.endsWith('.yaml') || f.endsWith('.yml'));
+            if (agents.length > 0) {
+                return {
+                    name: 'Custom Agents',
+                    status: 'ok',
+                    message: `${agents.length} agents`,
+                    details: agents.slice(0, 5),
+                };
+            }
+        }
+        catch {
+            // ignore
+        }
+    }
+    return { name: 'Custom Agents', status: 'ok', message: 'none' };
+}
+function checkCustomCommands() {
+    const commandsDir = join(homedir(), '.osagent', 'commands');
+    if (existsSync(commandsDir)) {
+        try {
+            const commands = readdirSync(commandsDir).filter(f => f.endsWith('.js') || f.endsWith('.ts') || f.endsWith('.toml'));
+            if (commands.length > 0) {
+                return {
+                    name: 'Custom Commands',
+                    status: 'ok',
+                    message: `${commands.length} commands`,
+                    details: commands.slice(0, 5),
+                };
+            }
+        }
+        catch {
+            // ignore
+        }
+    }
+    return { name: 'Custom Commands', status: 'ok', message: 'none' };
+}
+function checkMCPServers() {
+    const mcpPath = join(homedir(), '.osagent', 'mcp.json');
+    if (existsSync(mcpPath)) {
+        try {
+            const content = readFileSync(mcpPath, 'utf-8');
+            const mcp = JSON.parse(content);
+            const servers = Object.keys(mcp.mcpServers || {});
+            if (servers.length > 0) {
+                return {
+                    name: 'MCP Servers',
+                    status: 'ok',
+                    message: `${servers.length} configured`,
+                    details: servers.slice(0, 5),
+                };
+            }
+        }
+        catch {
+            return {
+                name: 'MCP Servers',
+                status: 'warning',
+                message: 'invalid config',
+                fix: 'Fix ~/.osagent/mcp.json syntax',
+            };
+        }
+    }
+    return { name: 'MCP Servers', status: 'ok', message: 'none' };
+}
+function checkDiskSpace() {
+    try {
+        if (platform() === 'darwin' || platform() === 'linux') {
+            const result = execSync('df -h ~ | tail -1', { encoding: 'utf-8' });
+            const parts = result.trim().split(/\s+/);
+            const available = parts[3];
+            const usePercent = parseInt(parts[4], 10);
+            if (usePercent > 90) {
+                return {
+                    name: 'Disk Space',
+                    status: 'warning',
+                    message: `${available} free (${usePercent}% used)`,
+                    fix: 'Free up disk space',
+                };
+            }
+            return {
+                name: 'Disk Space',
+                status: 'ok',
+                message: `${available} free`,
+            };
+        }
+    }
+    catch {
+        // ignore
+    }
+    return { name: 'Disk Space', status: 'ok', message: 'unknown' };
+}
+// ============================================================================
+// FORMATTERS
+// ============================================================================
+function formatReport(checks, systemInfo, currentVersion, latestVersion, autoFix = false) {
+    const lines = [];
+    const width = 60;
+    // Header
+    lines.push('');
+    const headerBox = createBox('OS Agent Doctor', width);
+    lines.push(headerBox.top);
+    lines.push(headerBox.line(centerPad(`${icons.doctor} System Health Report`, width - 4)));
+    lines.push(headerBox.bottom);
+    lines.push('');
+    // Version info
+    const updateAvailable = currentVersion !== 'unknown' &&
+        latestVersion !== 'unknown' &&
+        currentVersion !== latestVersion;
+    lines.push(sectionHeader('Version', icons.package));
+    lines.push(divider(width));
+    lines.push(keyValue('Installed', currentVersion));
+    if (updateAvailable) {
+        lines.push(keyValue('Available', colorize(latestVersion, colors.green) + ' ' + colorize('(update available)', colors.yellow)));
+    }
+    else {
+        lines.push(keyValue('Status', colorize('up to date', colors.green)));
+    }
+    lines.push('');
+    // System info
+    lines.push(sectionHeader('System', icons.terminal));
+    lines.push(divider(width));
+    lines.push(keyValue('OS', systemInfo.os));
+    lines.push(keyValue('Architecture', systemInfo.arch));
+    lines.push(keyValue('Node.js', systemInfo.nodeVersion));
+    lines.push(keyValue('npm', systemInfo.npmVersion));
+    lines.push(keyValue('Memory', systemInfo.memory));
+    lines.push('');
+    // Health checks
+    lines.push(sectionHeader('Health Checks', icons.shield));
+    lines.push(divider(width));
+    let errorCount = 0;
+    let warningCount = 0;
+    const fixes = [];
+    for (const check of checks) {
+        const icon = statusIcon(check.status);
+        const statusColor = check.status === 'ok' ? colors.green :
+            check.status === 'warning' ? colors.yellow : colors.red;
+        lines.push(`${icon} ${colorize(check.name.padEnd(20), colors.bold)} ${colorize(check.message, statusColor)}`);
+        if (check.details && check.details.length > 0) {
+            for (const detail of check.details) {
+                lines.push(`   ${colorize(box.pipe, colors.gray)} ${colorize(detail, colors.gray)}`);
+            }
+        }
+        if (check.status === 'error')
+            errorCount++;
+        if (check.status === 'warning')
+            warningCount++;
+        if (check.fix) {
+            fixes.push(`${check.name}: ${check.fix}`);
+        }
+    }
+    lines.push('');
+    // Summary
+    lines.push(sectionHeader('Summary', icons.sparkle));
+    lines.push(divider(width));
+    const total = checks.length;
+    const passed = total - errorCount - warningCount;
+    if (errorCount === 0 && warningCount === 0) {
+        lines.push(colorize(`${icons.success} All ${total} checks passed!`, colors.green, colors.bold));
+    }
+    else {
+        lines.push(`${colorize(`${passed}`, colors.green)} passed, ` +
+            `${colorize(`${warningCount}`, colors.yellow)} warnings, ` +
+            `${colorize(`${errorCount}`, colors.red)} errors`);
+    }
+    lines.push('');
+    // Recommended fixes
+    if (fixes.length > 0) {
+        lines.push(sectionHeader('Recommended Actions', icons.wrench));
+        lines.push(divider(width));
+        fixes.forEach((fix, i) => {
+            lines.push(`  ${colorize(`${i + 1}.`, colors.cyan)} ${fix}`);
+        });
+        lines.push('');
+        if (!autoFix) {
+            lines.push(colorize('Run /doctor fix to auto-fix issues', colors.gray, colors.italic));
+        }
+    }
+    // Quick actions
+    if (updateAvailable) {
+        lines.push('');
+        lines.push(colorize(`${icons.update} Update available! Run: /doctor update`, colors.cyan));
+    }
+    lines.push('');
+    lines.push(sectionHeader('Quick Commands', icons.lightning));
+    lines.push(divider(width));
+    lines.push(`  ${colorize('/doctor update', colors.cyan)}  - Update to latest version`);
+    lines.push(`  ${colorize('/doctor fix', colors.cyan)}     - Auto-fix issues`);
+    lines.push(`  ${colorize('/doctor init', colors.cyan)}    - Initialize config directory`);
+    lines.push('');
+    return lines;
+}
+function centerPad(text, width) {
+    const stripped = text.replace(/\x1b\[[0-9;]*m/g, '');
+    const padding = Math.max(0, width - stripped.length);
+    const leftPad = Math.floor(padding / 2);
+    return ' '.repeat(leftPad) + text;
+}
+// ============================================================================
+// COMMAND DEFINITION
+// ============================================================================
+export const doctorCommand = {
+    name: 'doctor',
+    get description() {
+        return t('Check system health, run diagnostics, and auto-fix issues');
+    },
+    kind: CommandKind.BUILT_IN,
+    subCommands: [
+        {
+            name: 'update',
+            get description() {
+                return t('Update osagent to the latest version');
+            },
+            kind: CommandKind.BUILT_IN,
+            action: async () => {
+                const lines = [];
+                lines.push('');
+                lines.push(sectionHeader('Updating OS Agent...', icons.update));
+                lines.push(divider(50));
+                try {
+                    lines.push(`${spinnerChar()} Checking current version...`);
+                    const oldVersion = getCurrentVersion();
+                    lines.push(`${spinnerChar()} Installing latest version...`);
+                    execSync('npm install -g osagent@latest 2>&1', { encoding: 'utf-8', stdio: 'pipe' });
+                    const newVersion = getLatestVersion();
+                    lines.push('');
+                    lines.push(colorize(`${icons.success} Updated successfully!`, colors.green, colors.bold));
+                    lines.push('');
+                    lines.push(keyValue('Previous', oldVersion));
+                    lines.push(keyValue('Current', colorize(newVersion, colors.green)));
+                    lines.push('');
+                    lines.push(colorize('Restart osagent to use the new version.', colors.yellow));
+                    return {
+                        type: 'message',
+                        messageType: 'info',
+                        content: lines.join('\n'),
+                    };
+                }
+                catch (error) {
+                    lines.push('');
+                    lines.push(colorize(`${icons.error} Update failed`, colors.red, colors.bold));
+                    lines.push('');
+                    lines.push('Try manually:');
+                    lines.push(colorize('  sudo npm install -g osagent@latest', colors.cyan));
+                    return {
+                        type: 'message',
+                        messageType: 'error',
+                        content: lines.join('\n'),
+                    };
+                }
+            },
+        },
+        {
+            name: 'fix',
+            get description() {
+                return t('Automatically fix issues where possible');
+            },
+            kind: CommandKind.BUILT_IN,
+            action: async () => {
+                const lines = [];
+                lines.push('');
+                lines.push(sectionHeader('Auto-fixing issues...', icons.wrench));
+                lines.push(divider(50));
+                const checks = [
+                    checkNodeVersion(),
+                    checkOllamaInstalled(),
+                    checkOllamaRunning(),
+                    checkGitInstalled(),
+                    checkSettings(),
+                    checkAPIKeys(),
+                ];
+                let fixed = 0;
+                let failed = 0;
+                for (const check of checks) {
+                    if (check.status !== 'ok' && check.autoFix) {
+                        lines.push(`${spinnerChar()} Fixing: ${check.name}...`);
+                        try {
+                            const success = await check.autoFix();
+                            if (success) {
+                                lines.push(`  ${statusIcon('ok')} ${check.name} fixed`);
+                                fixed++;
+                            }
+                            else {
+                                lines.push(`  ${statusIcon('warning')} ${check.name} - manual fix needed`);
+                                if (check.fix) {
+                                    lines.push(`    ${colorize(icons.arrow, colors.gray)} ${check.fix}`);
+                                }
+                                failed++;
+                            }
+                        }
+                        catch {
+                            lines.push(`  ${statusIcon('error')} ${check.name} - fix failed`);
+                            failed++;
+                        }
+                    }
+                    else if (check.status !== 'ok' && check.fix) {
+                        lines.push(`${statusIcon('warning')} ${check.name} - manual fix needed`);
+                        lines.push(`  ${colorize(icons.arrow, colors.gray)} ${check.fix}`);
+                    }
+                }
+                // Initialize directory structure if missing
+                const baseDir = join(homedir(), '.osagent');
+                const dirs = ['agents', 'commands', 'prompts', 'memory'];
+                for (const dir of dirs) {
+                    const path = join(baseDir, dir);
+                    if (!existsSync(path)) {
+                        mkdirSync(path, { recursive: true });
+                        lines.push(`${statusIcon('ok')} Created ${dir}/ directory`);
+                        fixed++;
+                    }
+                }
+                lines.push('');
+                lines.push(divider(50));
+                if (fixed > 0) {
+                    lines.push(colorize(`${icons.success} ${fixed} issues fixed`, colors.green));
+                }
+                if (failed > 0) {
+                    lines.push(colorize(`${icons.warning} ${failed} issues need manual attention`, colors.yellow));
+                }
+                if (fixed === 0 && failed === 0) {
+                    lines.push(colorize(`${icons.success} No issues to fix!`, colors.green));
+                }
+                return {
+                    type: 'message',
+                    messageType: 'info',
+                    content: lines.join('\n'),
+                };
+            },
+        },
+        {
+            name: 'init',
+            get description() {
+                return t('Initialize ~/.osagent directory with recommended structure');
+            },
+            kind: CommandKind.BUILT_IN,
+            action: async () => {
+                const lines = [];
+                lines.push('');
+                lines.push(sectionHeader('Initializing OS Agent...', icons.gear));
+                lines.push(divider(50));
+                const baseDir = join(homedir(), '.osagent');
+                const dirs = ['agents', 'commands', 'prompts', 'memory', 'skills'];
+                try {
+                    for (const dir of dirs) {
+                        const path = join(baseDir, dir);
+                        mkdirSync(path, { recursive: true });
+                        lines.push(`${statusIcon('ok')} Created ${colorize(dir + '/', colors.cyan)}`);
+                    }
+                    // Create example agent
+                    const exampleAgent = {
+                        name: 'example-agent',
+                        description: 'An example custom agent',
+                        systemPrompt: 'You are a helpful assistant specialized in...',
+                        model: 'qwen3-coder:480b-cloud',
+                        tier: 'specialist',
+                    };
+                    const agentPath = join(baseDir, 'agents', 'example-agent.json');
+                    if (!existsSync(agentPath)) {
+                        writeFileSync(agentPath, JSON.stringify(exampleAgent, null, 2));
+                        lines.push(`${statusIcon('ok')} Created ${colorize('example-agent.json', colors.cyan)}`);
+                    }
+                    // Create example skill
+                    const exampleSkill = {
+                        name: 'example-skill',
+                        description: 'An example custom skill',
+                        triggers: [{ type: 'keyword', patterns: ['example', 'demo'] }],
+                        steps: [
+                            { name: 'step1', prompt: 'First step...' },
+                            { name: 'step2', prompt: 'Second step...', dependsOn: ['step1'] },
+                        ],
+                    };
+                    const skillPath = join(baseDir, 'skills', 'example-skill.json');
+                    if (!existsSync(skillPath)) {
+                        writeFileSync(skillPath, JSON.stringify(exampleSkill, null, 2));
+                        lines.push(`${statusIcon('ok')} Created ${colorize('example-skill.json', colors.cyan)}`);
+                    }
+                    lines.push('');
+                    lines.push(divider(50));
+                    lines.push(colorize(`${icons.success} Initialization complete!`, colors.green, colors.bold));
+                    lines.push('');
+                    lines.push(sectionHeader('Directory Structure', icons.folder));
+                    lines.push(createTree([
+                        { label: colorize('~/.osagent/', colors.cyan), children: [
+                                { label: colorize('agents/', colors.blue) + ' - Custom AI agents' },
+                                { label: colorize('commands/', colors.blue) + ' - Custom slash commands' },
+                                { label: colorize('prompts/', colors.blue) + ' - Custom prompts' },
+                                { label: colorize('skills/', colors.blue) + ' - Custom skills/workflows' },
+                                { label: colorize('memory/', colors.blue) + ' - Memory storage' },
+                            ] },
+                    ]).join('\n'));
+                    return {
+                        type: 'message',
+                        messageType: 'info',
+                        content: lines.join('\n'),
+                    };
+                }
+                catch (error) {
+                    return {
+                        type: 'message',
+                        messageType: 'error',
+                        content: `${icons.error} Failed to initialize: ${error}`,
+                    };
+                }
+            },
+        },
+    ],
+    action: async () => {
+        const checks = [
+            checkNodeVersion(),
+            checkOllamaInstalled(),
+            checkOllamaRunning(),
+            checkGitInstalled(),
+            checkDiskSpace(),
+            checkAPIKeys(),
+            checkSettings(),
+            checkSystemPrompt(),
+            checkCustomAgents(),
+            checkCustomCommands(),
+            checkMCPServers(),
+        ];
+        const systemInfo = getSystemInfo();
+        const currentVersion = getCurrentVersion();
+        const latestVersion = getLatestVersion();
+        const report = formatReport(checks, systemInfo, currentVersion, latestVersion);
+        return {
+            type: 'message',
+            messageType: 'info',
+            content: report.join('\n'),
+        };
+    },
+};
+function spinnerChar() {
+    return colorize('>', colors.cyan);
+}
+//# sourceMappingURL=doctorCommand.js.map