UNPKG

@light-merlin-dark/tok

Version:

Fast token estimation and cost calculation for enterprise LLMs with CLI and MCP support

148 lines (143 loc) 5.35 kB
"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; }; })(); Object.defineProperty(exports, "__esModule", { value: true }); exports.parseArgs = parseArgs; exports.route = route; const common_1 = require("./common"); const errors_1 = require("./shared/errors"); function parseArgs(args) { const command = args[0] || 'help'; const commandArgs = []; const options = {}; for (let i = 1; i < args.length; i++) { const arg = args[i]; if (arg.startsWith('--')) { // Long option const [key, value] = arg.slice(2).split('='); const camelKey = key.replace(/-([a-z])/g, (_, letter) => letter.toUpperCase()); if (value !== undefined) { options[camelKey] = value; } else if (i + 1 < args.length && !args[i + 1].startsWith('-')) { options[camelKey] = args[++i]; } else { options[camelKey] = true; } } else if (arg.startsWith('-') && arg.length > 1) { // Short option const flags = arg.slice(1); for (const flag of flags) { options[flag] = true; } } else { // Regular argument commandArgs.push(arg); } } return { command, commandArgs, options }; } async function route(args, ctx) { try { // Handle global --help before parsing if (args.length === 0 || args.includes('--help') || args.includes('-h') || args[0] === 'help') { const { commands } = await Promise.resolve().then(() => __importStar(require('./commands'))); showHelp(Object.keys(commands)); return { success: true }; } const { command, commandArgs, options } = parseArgs(args); // Lazy load commands const { commands } = await Promise.resolve().then(() => __importStar(require('./commands'))); const handler = commands[command]; if (!handler) { common_1.logger.error(`Unknown command: ${command}`); common_1.logger.info(`Run 'tok help' for available commands`); return { success: false, message: `Unknown command: ${command}` }; } // Handle command-specific help if (options.help || options.h) { if (handler.help) { console.log(handler.help); } else { common_1.logger.info(`No detailed help available for '${command}'`); if (handler.description) { common_1.logger.info(`Description: ${handler.description}`); } } return { success: true }; } // Execute command return await handler.execute(commandArgs, options, ctx); } catch (error) { if (error instanceof errors_1.CommandError) { common_1.logger.error(error.userMessage); if (ctx.verbose && error.details) { common_1.logger.debug(JSON.stringify(error.details, null, 2), true); } return { success: false, message: error.userMessage, data: { code: error.code, details: error.details } }; } // Unknown error common_1.logger.error(`An unexpected error occurred: ${error}`); return { success: false, message: error instanceof Error ? error.message : String(error) }; } } function showHelp(commandNames) { console.log(` tok - Fast token estimation and cost calculation for LLMs Usage: tok [command] [options] Commands: ${commandNames.map(cmd => ` ${cmd.padEnd(15)} Use --help with command for details`).join('\n')} Global Options: -h, --help Show help -v, --verbose Enable verbose output Examples: tok estimate "Hello world" tok price list tok track summary Run 'tok [command] --help' for command-specific help. `); } //# sourceMappingURL=router.js.map