@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
JavaScript
;
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