omnikey-cli 1.0.39 → 1.0.41

package/dist/index.js

@@ -12,6 +12,7 @@ const showConfig_1 = require("./showConfig");
 const setConfig_1 = require("./setConfig");
 const grantBrowserAccess_1 = require("./grantBrowserAccess");
 const scheduleJob_1 = require("./scheduleJob");
+const mcpServer_1 = require("./mcpServer");
 const program = new commander_1.Command();
 program
     .name('omnikey')
@@ -95,23 +96,62 @@ program
     .action(async () => {
     await (0, grantBrowserAccess_1.reopenBrowserDebugProfile)();
 });
-const scheduleCmd = program
-    .command('schedule')
-    .description('Manage scheduled prompt jobs');
+const scheduleCmd = program.command('schedule').description('Manage scheduled prompt jobs');
 scheduleCmd
     .command('add')
     .description('Add a new scheduled job')
-    .action(async () => { await (0, scheduleJob_1.scheduleAdd)(); });
+    .action(async () => {
+    await (0, scheduleJob_1.scheduleAdd)();
+});
 scheduleCmd
     .command('list')
     .description('List all scheduled jobs')
-    .action(async () => { await (0, scheduleJob_1.scheduleList)(); });
+    .action(async () => {
+    await (0, scheduleJob_1.scheduleList)();
+});
 scheduleCmd
     .command('remove')
     .description('Remove a scheduled job')
-    .action(async () => { await (0, scheduleJob_1.scheduleRemove)(); });
+    .action(async () => {
+    await (0, scheduleJob_1.scheduleRemove)();
+});
 scheduleCmd
     .command('run-now <id>')
     .description('Immediately run a scheduled job by ID')
-    .action(async (id) => { await (0, scheduleJob_1.scheduleRunNow)(id); });
+    .action(async (id) => {
+    await (0, scheduleJob_1.scheduleRunNow)(id);
+});
+const mcpCmd = program
+    .command('mcp')
+    .description('Manage MCP (Model Context Protocol) servers available to the agent');
+mcpCmd
+    .command('add')
+    .description('Install a new MCP server')
+    .action(async () => {
+    await (0, mcpServer_1.mcpAdd)();
+});
+mcpCmd
+    .command('list')
+    .description('List installed MCP servers')
+    .action(async () => {
+    await (0, mcpServer_1.mcpList)();
+});
+mcpCmd
+    .command('remove')
+    .description('Remove an installed MCP server')
+    .action(async () => {
+    await (0, mcpServer_1.mcpRemove)();
+});
+mcpCmd
+    .command('toggle <id>')
+    .description('Toggle an MCP server enabled/disabled by ID')
+    .action(async (id) => {
+    await (0, mcpServer_1.mcpToggle)(id);
+});
+mcpCmd
+    .command('update <id>')
+    .description('Update an existing MCP server by ID')
+    .action(async (id) => {
+    await (0, mcpServer_1.mcpUpdate)(id);
+});
 program.parseAsync(process.argv);

package/dist/mcpServer.js

@@ -0,0 +1,334 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mcpAdd = mcpAdd;
+exports.mcpList = mcpList;
+exports.mcpRemove = mcpRemove;
+exports.mcpToggle = mcpToggle;
+exports.mcpUpdate = mcpUpdate;
+const axios_1 = __importDefault(require("axios"));
+const inquirer_1 = __importDefault(require("inquirer"));
+const utils_1 = require("./utils");
+async function getJwt() {
+    const config = (0, utils_1.readConfig)();
+    const port = (0, utils_1.getPort)();
+    const licenseKey = config.OMNIKEY_LICENSE_KEY || '';
+    const baseUrl = `http://localhost:${port}`;
+    const res = await axios_1.default.post(`${baseUrl}/api/subscription/activate`, { licenseKey }, { timeout: 10000 });
+    return res.data.token;
+}
+function getBaseUrl() {
+    return `http://localhost:${(0, utils_1.getPort)()}`;
+}
+async function authHeaders() {
+    let token;
+    try {
+        token = await getJwt();
+    }
+    catch (err) {
+        throw new Error(`Authentication failed — make sure the OmniKey backend is running and your license key is configured.\nCause: ${err?.message ?? String(err)}`);
+    }
+    return { Authorization: `Bearer ${token}` };
+}
+function parseLines(input) {
+    return input
+        .split('\n')
+        .map((l) => l.trim())
+        .filter((l) => l.length > 0);
+}
+function parseKeyValueLines(input) {
+    const out = {};
+    for (const line of parseLines(input)) {
+        const idx = line.indexOf('=');
+        if (idx <= 0)
+            continue;
+        const key = line.slice(0, idx).trim();
+        const value = line.slice(idx + 1).trim();
+        if (key)
+            out[key] = value;
+    }
+    return out;
+}
+async function promptTransportFields(transport, defaults) {
+    if (transport === 'stdio') {
+        const ans = await inquirer_1.default.prompt([
+            {
+                type: 'input',
+                name: 'command',
+                message: 'Command (executable path or name):',
+                default: defaults?.command ?? '',
+                validate: (v) => v.trim().length > 0 || 'Command is required for stdio transport',
+            },
+            {
+                type: 'editor',
+                name: 'args',
+                message: 'Args (one per line):',
+                default: (defaults?.args ?? []).join('\n'),
+            },
+            {
+                type: 'editor',
+                name: 'env',
+                message: 'Environment variables (one KEY=VALUE per line):',
+                default: formatKVForEditor(defaults?.env),
+            },
+        ]);
+        return {
+            command: ans.command.trim(),
+            args: parseLines(ans.args),
+            env: parseKeyValueLines(ans.env),
+            url: null,
+            headers: {},
+        };
+    }
+    const ans = await inquirer_1.default.prompt([
+        {
+            type: 'input',
+            name: 'url',
+            message: `URL for ${transport} transport:`,
+            default: defaults?.url ?? '',
+            validate: (v) => v.trim().length > 0 || 'URL is required',
+        },
+        {
+            type: 'editor',
+            name: 'headers',
+            message: 'Headers (one KEY=VALUE per line):',
+            default: formatKVForEditor(defaults?.headers),
+        },
+    ]);
+    return {
+        url: ans.url.trim(),
+        headers: parseKeyValueLines(ans.headers),
+        command: null,
+        args: [],
+        env: {},
+    };
+}
+function formatKVForEditor(dict) {
+    if (!dict)
+        return '';
+    return Object.entries(dict)
+        .map(([k, v]) => `${k}=${v}`)
+        .join('\n');
+}
+async function mcpAdd() {
+    const baseAnswers = await inquirer_1.default.prompt([
+        {
+            type: 'input',
+            name: 'name',
+            message: 'Name (unique, e.g. "github"):',
+            validate: (v) => v.trim().length > 0 || 'Name is required',
+        },
+        {
+            type: 'input',
+            name: 'description',
+            message: 'Description (optional):',
+        },
+        {
+            type: 'list',
+            name: 'transport',
+            message: 'Transport:',
+            choices: [
+                { name: 'stdio (local process)', value: 'stdio' },
+                { name: 'http', value: 'http' },
+                { name: 'sse', value: 'sse' },
+            ],
+            default: 'stdio',
+        },
+        {
+            type: 'confirm',
+            name: 'isEnabled',
+            message: 'Enabled?',
+            default: true,
+        },
+    ]);
+    const transportFields = await promptTransportFields(baseAnswers.transport);
+    try {
+        const headers = await authHeaders();
+        const res = await axios_1.default.post(`${getBaseUrl()}/api/mcp-servers`, {
+            name: baseAnswers.name.trim(),
+            description: baseAnswers.description.trim() || null,
+            transport: baseAnswers.transport,
+            isEnabled: baseAnswers.isEnabled,
+            ...transportFields,
+        }, { headers, timeout: 15000 });
+        console.log('\nMCP server created:');
+        printServer(res.data);
+    }
+    catch (err) {
+        const msg = err.response?.data?.error ?? err.message;
+        console.error(`Error creating MCP server: ${msg}`);
+    }
+}
+async function mcpList() {
+    try {
+        const headers = await authHeaders();
+        const res = await axios_1.default.get(`${getBaseUrl()}/api/mcp-servers`, {
+            headers,
+            timeout: 10000,
+        });
+        const { servers } = res.data;
+        if (servers.length === 0) {
+            console.log('No MCP servers installed.');
+            return;
+        }
+        console.log('\nMCP Servers:');
+        console.log('─'.repeat(90));
+        console.log(padRight('ID', 28) +
+            padRight('Name', 22) +
+            padRight('Transport', 12) +
+            padRight('Enabled', 10) +
+            'Endpoint');
+        console.log('─'.repeat(90));
+        for (const s of servers) {
+            const endpoint = s.transport === 'stdio' ? (s.command ?? '—') : (s.url ?? '—');
+            console.log(padRight(s.id.slice(0, 26), 28) +
+                padRight(s.name.slice(0, 20), 22) +
+                padRight(s.transport, 12) +
+                padRight(s.isEnabled ? 'yes' : 'no', 10) +
+                endpoint.slice(0, 40));
+        }
+        console.log('─'.repeat(90));
+    }
+    catch (err) {
+        const msg = err.response?.data?.error ?? err.message;
+        console.error(`Error fetching MCP servers: ${msg}`);
+    }
+}
+async function pickServer(action) {
+    const headers = await authHeaders();
+    const res = await axios_1.default.get(`${getBaseUrl()}/api/mcp-servers`, {
+        headers,
+        timeout: 10000,
+    });
+    const { servers } = res.data;
+    if (servers.length === 0) {
+        console.log(`No MCP servers to ${action}.`);
+        return null;
+    }
+    const { id } = await inquirer_1.default.prompt([
+        {
+            type: 'list',
+            name: 'id',
+            message: `Select an MCP server to ${action}:`,
+            choices: servers.map((s) => ({
+                name: `${s.name} [${s.transport}] ${s.isEnabled ? '✓' : '✗'}`,
+                value: s.id,
+            })),
+        },
+    ]);
+    return servers.find((s) => s.id === id) ?? null;
+}
+async function mcpRemove() {
+    try {
+        const server = await pickServer('remove');
+        if (!server)
+            return;
+        const { confirm } = await inquirer_1.default.prompt([
+            {
+                type: 'confirm',
+                name: 'confirm',
+                message: `Delete MCP server "${server.name}"?`,
+                default: false,
+            },
+        ]);
+        if (!confirm) {
+            console.log('Aborted.');
+            return;
+        }
+        const headers = await authHeaders();
+        await axios_1.default.delete(`${getBaseUrl()}/api/mcp-servers/${server.id}`, {
+            headers,
+            timeout: 10000,
+        });
+        console.log('MCP server removed.');
+    }
+    catch (err) {
+        const msg = err.response?.data?.error ?? err.message;
+        console.error(`Error removing MCP server: ${msg}`);
+    }
+}
+async function mcpToggle(id) {
+    try {
+        const headers = await authHeaders();
+        const current = await axios_1.default.get(`${getBaseUrl()}/api/mcp-servers/${id}`, {
+            headers,
+            timeout: 10000,
+        });
+        const newState = !current.data.isEnabled;
+        const res = await axios_1.default.patch(`${getBaseUrl()}/api/mcp-servers/${id}`, { isEnabled: newState }, { headers, timeout: 10000 });
+        console.log(`MCP server ${res.data.name} is now ${res.data.isEnabled ? 'enabled' : 'disabled'}.`);
+    }
+    catch (err) {
+        const msg = err.response?.data?.error ?? err.message;
+        console.error(`Error toggling MCP server: ${msg}`);
+    }
+}
+async function mcpUpdate(id) {
+    try {
+        const headers = await authHeaders();
+        const current = (await axios_1.default.get(`${getBaseUrl()}/api/mcp-servers/${id}`, {
+            headers,
+            timeout: 10000,
+        })).data;
+        const baseAnswers = await inquirer_1.default.prompt([
+            {
+                type: 'input',
+                name: 'name',
+                message: 'Name:',
+                default: current.name,
+            },
+            {
+                type: 'input',
+                name: 'description',
+                message: 'Description:',
+                default: current.description ?? '',
+            },
+            {
+                type: 'list',
+                name: 'transport',
+                message: 'Transport:',
+                choices: ['stdio', 'http', 'sse'],
+                default: current.transport,
+            },
+            {
+                type: 'confirm',
+                name: 'isEnabled',
+                message: 'Enabled?',
+                default: current.isEnabled,
+            },
+        ]);
+        const transportFields = await promptTransportFields(baseAnswers.transport, current);
+        const res = await axios_1.default.patch(`${getBaseUrl()}/api/mcp-servers/${id}`, {
+            name: baseAnswers.name.trim(),
+            description: baseAnswers.description.trim() || null,
+            transport: baseAnswers.transport,
+            isEnabled: baseAnswers.isEnabled,
+            ...transportFields,
+        }, { headers, timeout: 15000 });
+        console.log('\nMCP server updated:');
+        printServer(res.data);
+    }
+    catch (err) {
+        const msg = err.response?.data?.error ?? err.message;
+        console.error(`Error updating MCP server: ${msg}`);
+    }
+}
+function printServer(s) {
+    console.log(`  ID:        ${s.id}`);
+    console.log(`  Name:      ${s.name}`);
+    console.log(`  Transport: ${s.transport}`);
+    console.log(`  Enabled:   ${s.isEnabled}`);
+    if (s.transport === 'stdio') {
+        console.log(`  Command:   ${s.command ?? '—'}`);
+        if (s.args.length > 0)
+            console.log(`  Args:      ${s.args.join(' ')}`);
+    }
+    else {
+        console.log(`  URL:       ${s.url ?? '—'}`);
+    }
+}
+function padRight(str, width) {
+    return str.length >= width ? str.slice(0, width) : str + ' '.repeat(width - str.length);
+}

package/package.json

@@ -4,7 +4,7 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "version": "1.0.39",
+  "version": "1.0.41",
   "description": "CLI for onboarding users to Omnikey AI and configuring OPENAI_API_KEY. Use Yarn for install/build.",
   "engines": {
     "node": ">=14.0.0",
@@ -30,6 +30,7 @@
     "@anthropic-ai/sdk": "^0.96.0",
     "@google-cloud/storage": "^7.19.0",
     "@google/genai": "^1.46.0",
+    "@modelcontextprotocol/sdk": "^1.29.0",
     "axios": "^1.13.5",
     "commander": "^11.0.0",
     "cors": "^2.8.5",

package/src/index.ts

@@ -11,6 +11,7 @@ import { showConfig } from './showConfig';
 import { setConfig } from './setConfig';
 import { grantBrowserAccess, reopenBrowserDebugProfile } from './grantBrowserAccess';
 import { scheduleAdd, scheduleList, scheduleRemove, scheduleRunNow } from './scheduleJob';
+import { mcpAdd, mcpList, mcpRemove, mcpToggle, mcpUpdate } from './mcpServer';
 
 const program = new Command();
 
@@ -110,28 +111,73 @@ program
     await reopenBrowserDebugProfile();
   });
 
-const scheduleCmd = program
-  .command('schedule')
-  .description('Manage scheduled prompt jobs');
+const scheduleCmd = program.command('schedule').description('Manage scheduled prompt jobs');
 
 scheduleCmd
   .command('add')
   .description('Add a new scheduled job')
-  .action(async () => { await scheduleAdd(); });
+  .action(async () => {
+    await scheduleAdd();
+  });
 
 scheduleCmd
   .command('list')
   .description('List all scheduled jobs')
-  .action(async () => { await scheduleList(); });
+  .action(async () => {
+    await scheduleList();
+  });
 
 scheduleCmd
   .command('remove')
   .description('Remove a scheduled job')
-  .action(async () => { await scheduleRemove(); });
+  .action(async () => {
+    await scheduleRemove();
+  });
 
 scheduleCmd
   .command('run-now <id>')
   .description('Immediately run a scheduled job by ID')
-  .action(async (id: string) => { await scheduleRunNow(id); });
+  .action(async (id: string) => {
+    await scheduleRunNow(id);
+  });
+
+const mcpCmd = program
+  .command('mcp')
+  .description('Manage MCP (Model Context Protocol) servers available to the agent');
+
+mcpCmd
+  .command('add')
+  .description('Install a new MCP server')
+  .action(async () => {
+    await mcpAdd();
+  });
+
+mcpCmd
+  .command('list')
+  .description('List installed MCP servers')
+  .action(async () => {
+    await mcpList();
+  });
+
+mcpCmd
+  .command('remove')
+  .description('Remove an installed MCP server')
+  .action(async () => {
+    await mcpRemove();
+  });
+
+mcpCmd
+  .command('toggle <id>')
+  .description('Toggle an MCP server enabled/disabled by ID')
+  .action(async (id: string) => {
+    await mcpToggle(id);
+  });
+
+mcpCmd
+  .command('update <id>')
+  .description('Update an existing MCP server by ID')
+  .action(async (id: string) => {
+    await mcpUpdate(id);
+  });
 
 program.parseAsync(process.argv);