berget 0.1.0 → 1.1.0

package/README.md

@@ -0,0 +1,92 @@
+# Berget CLI
+
+A command-line interface for interacting with the Berget AI infrastructure.
+
+## Installation
+
+```bash
+npm install -g berget
+```
+
+## Authentication
+
+The CLI uses OAuth-based authentication with automatic token refresh.
+
+### Login
+
+```bash
+berget auth login
+```
+
+This will open a browser window to authenticate with Berget. After successful authentication, your access token and refresh token will be stored securely in `~/.berget/auth.json`.
+
+### Token Refresh
+
+The CLI automatically handles token refresh when:
+
+1. The access token is about to expire (within 10 minutes of expiration)
+2. A request returns a 401 Unauthorized error
+
+The refresh mechanism uses the stored refresh token to obtain a new access token without requiring you to log in again. If the refresh token itself is expired or invalid, you'll be prompted to log in again.
+
+## Development
+
+### Setup
+
+Clone the repository and install dependencies:
+
+```bash
+git clone https://github.com/berget/cli.git
+cd cli
+npm install
+```
+
+### Testing Locally
+
+Use the `start` script to test the CLI locally with the `--local` flag:
+
+```bash
+npm start -- <command> [options]
+```
+
+For example:
+
+```bash
+# Test login
+npm start -- auth login
+
+# Test whoami
+npm start -- auth whoami
+
+# Test with debug output
+npm start -- auth whoami --debug
+```
+
+The `--debug` flag provides detailed information about token refresh attempts and API responses.
+
+### Testing Token Refresh
+
+To test the token refresh mechanism:
+
+1. Log in with `npm start -- auth login`
+2. Make a request that requires authentication, like `npm start -- auth whoami`
+3. To force a token refresh, you can:
+   - Wait until the token is close to expiration
+   - Manually edit `~/.berget/auth.json` and set `expires_at` to a past timestamp
+   - Use the `--debug` flag to see the token refresh process in action
+
+## Commands
+
+- `auth login` - Log in to Berget
+- `auth logout` - Log out from Berget
+- `auth whoami` - Show current user information
+- `api-keys list` - List API keys
+- `api-keys create` - Create a new API key
+- `models list` - List available AI models
+- `chat run` - Start a chat session with an AI model
+
+For a complete list of commands, run:
+
+```bash
+berget --help
+```

package/dist/index.js

@@ -1,47 +1,9 @@
 #!/usr/bin/env node
 "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 (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 const commander_1 = require("commander");
-const fs = __importStar(require("fs"));
-const path = __importStar(require("path"));
-const client_1 = require("./src/client");
-const error_handler_1 = require("./src/utils/error-handler");
-const chalk_1 = __importDefault(require("chalk"));
+const commands_1 = require("./src/commands");
+const config_checker_1 = require("./src/utils/config-checker");
 // Set version and description
 commander_1.program
     .name('berget')
@@ -54,406 +16,12 @@ commander_1.program
                  __/ |                      
                 |___/   AI on European terms`)
     .version(process.env.npm_package_version || '0.0.1', '-v, --version')
-    .option('--local', 'Use local API endpoint (hidden)', false);
-// Import services
-const auth_service_1 = require("./src/services/auth-service");
-const api_key_service_1 = require("./src/services/api-key-service");
-const cluster_service_1 = require("./src/services/cluster-service");
-// Auth commands
-commander_1.program
-    .command('login')
-    .description('Log in to Berget')
-    .action(() => __awaiter(void 0, void 0, void 0, function* () {
-    const authService = auth_service_1.AuthService.getInstance();
-    yield authService.login();
-}));
-commander_1.program
-    .command('logout')
-    .description('Log out from Berget')
-    .action(() => {
-    const { clearAuthToken } = require('./src/client');
-    clearAuthToken();
-    console.log(chalk_1.default.green('You have been logged out from Berget'));
-});
-commander_1.program
-    .command('whoami')
-    .description('Show information about the logged in user')
-    .action(() => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const authService = auth_service_1.AuthService.getInstance();
-        const profile = yield authService.getUserProfile();
-        if (profile) {
-            console.log(chalk_1.default.bold(`Logged in as: ${profile.name || profile.login}`));
-            console.log(`Email: ${chalk_1.default.cyan(profile.email || 'Not available')}`);
-            console.log(`Role: ${chalk_1.default.cyan(profile.role || 'Not available')}`);
-            if (profile.company) {
-                console.log(`Company: ${chalk_1.default.cyan(profile.company.name)}`);
-            }
-        }
-        else {
-            console.log(chalk_1.default.yellow('You are not logged in. Use `berget login` to log in.'));
-        }
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('You are not logged in or an error occurred', error);
-    }
-}));
-// API Key commands
-const apiKey = commander_1.program.command('api-key').description('Manage API keys');
-apiKey
-    .command('list')
-    .description('List all API keys')
-    .action(() => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-        const keys = yield apiKeyService.listApiKeys();
-        if (keys.length === 0) {
-            console.log(chalk_1.default.yellow('No API keys found. Create one with `berget api-key create --name <name>`'));
-            return;
-        }
-        console.log(chalk_1.default.bold('Your API keys:'));
-        console.log('');
-        // Create a table-like format with headers
-        console.log(chalk_1.default.dim('ID'.padEnd(10)) +
-            chalk_1.default.dim('NAME'.padEnd(25)) +
-            chalk_1.default.dim('PREFIX'.padEnd(12)) +
-            chalk_1.default.dim('STATUS'.padEnd(12)) +
-            chalk_1.default.dim('CREATED'.padEnd(12)) +
-            chalk_1.default.dim('LAST USED'));
-        console.log(chalk_1.default.dim('─'.repeat(85)));
-        keys.forEach((key) => {
-            const lastUsed = key.lastUsed ? key.lastUsed.substring(0, 10) : 'Never';
-            const status = key.active
-                ? chalk_1.default.green('● Active')
-                : chalk_1.default.red('● Inactive');
-            console.log(String(key.id).padEnd(10) +
-                key.name.padEnd(25) +
-                key.prefix.padEnd(12) +
-                status.padEnd(12) +
-                key.created.substring(0, 10).padEnd(12) +
-                lastUsed);
-        });
-        console.log('');
-        console.log(chalk_1.default.dim('Use `berget api-key create --name <name>` to create a new API key'));
-        console.log(chalk_1.default.dim('Use `berget api-key delete <id>` to delete an API key'));
-        console.log(chalk_1.default.dim('Use `berget api-key rotate <id>` to rotate an API key'));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to list API keys', error);
-    }
-}));
-apiKey
-    .command('create')
-    .description('Create a new API key')
-    .option('--name <name>', 'Name of the API key')
-    .option('--description <description>', 'Description of the API key')
-    .action((options) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        if (!options.name) {
-            console.error(chalk_1.default.red('Error: --name is required'));
-            console.log('');
-            console.log('Usage: berget api-key create --name <name> [--description <description>]');
-            return;
-        }
-        console.log(chalk_1.default.blue('Creating API key...'));
-        const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-        const result = yield apiKeyService.createApiKey({
-            name: options.name,
-            description: options.description,
-        });
-        console.log('');
-        console.log(chalk_1.default.green('✓ API key created'));
-        console.log('');
-        console.log(chalk_1.default.bold('API key details:'));
-        console.log('');
-        console.log(`${chalk_1.default.dim('ID:')}          ${result.id}`);
-        console.log(`${chalk_1.default.dim('Name:')}        ${result.name}`);
-        if (result.description) {
-            console.log(`${chalk_1.default.dim('Description:')} ${result.description}`);
-        }
-        console.log(`${chalk_1.default.dim('Created:')}     ${new Date(result.created).toLocaleString()}`);
-        console.log('');
-        console.log(chalk_1.default.bold('API key:'));
-        console.log(chalk_1.default.cyan(result.key));
-        console.log('');
-        console.log(chalk_1.default.yellow('⚠️  IMPORTANT: Save this API key in a secure location.'));
-        console.log(chalk_1.default.yellow('   It will not be displayed again.'));
-        console.log('');
-        console.log(chalk_1.default.dim('Use this key in your applications to authenticate with the Berget API.'));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to create API key', error);
-    }
-}));
-apiKey
-    .command('delete')
-    .description('Delete an API key')
-    .argument('<id>', 'ID of the API key to delete')
-    .action((id) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        console.log(chalk_1.default.blue(`Deleting API key ${id}...`));
-        const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-        yield apiKeyService.deleteApiKey(id);
-        console.log(chalk_1.default.green(`✓ API key ${id} has been deleted`));
-        console.log('');
-        console.log(chalk_1.default.dim('Applications using this key will no longer be able to authenticate.'));
-        console.log(chalk_1.default.dim('Use `berget api-key list` to see your remaining API keys.'));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to delete API key', error);
-    }
-}));
-apiKey
-    .command('rotate')
-    .description('Rotate an API key (creates a new one and invalidates the old one)')
-    .argument('<id>', 'ID of the API key to rotate')
-    .action((id) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        console.log(chalk_1.default.blue(`Rotating API key ${id}...`));
-        console.log(chalk_1.default.dim('This will invalidate the old key and generate a new one.'));
-        const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-        const result = yield apiKeyService.rotateApiKey(id);
-        console.log('');
-        console.log(chalk_1.default.green('✓ API key rotated'));
-        console.log('');
-        console.log(chalk_1.default.bold('New API key details:'));
-        console.log('');
-        console.log(`${chalk_1.default.dim('ID:')}          ${result.id}`);
-        console.log(`${chalk_1.default.dim('Name:')}        ${result.name}`);
-        if (result.description) {
-            console.log(`${chalk_1.default.dim('Description:')} ${result.description}`);
-        }
-        console.log(`${chalk_1.default.dim('Created:')}     ${new Date(result.created).toLocaleString()}`);
-        console.log('');
-        console.log(chalk_1.default.bold('New API key:'));
-        console.log(chalk_1.default.cyan(result.key));
-        console.log('');
-        console.log(chalk_1.default.yellow('⚠️  IMPORTANT: Update your applications with this new API key.'));
-        console.log(chalk_1.default.yellow('   The old key has been invalidated and will no longer work.'));
-        console.log(chalk_1.default.yellow('   This new key will not be displayed again.'));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to rotate API key', error);
-    }
-}));
-apiKey
-    .command('usage')
-    .description('Show usage statistics for an API key')
-    .argument('<id>', 'ID of the API key')
-    .option('--start <date>', 'Start date (YYYY-MM-DD)')
-    .option('--end <date>', 'End date (YYYY-MM-DD)')
-    .action((id, options) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        console.log(chalk_1.default.blue(`Fetching usage statistics for API key ${id}...`));
-        const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-        const usage = yield apiKeyService.getApiKeyUsage(id);
-        console.log('');
-        console.log(chalk_1.default.bold(`Usage statistics for API key: ${usage.name} (${id})`));
-        console.log('');
-        // Period information
-        console.log(chalk_1.default.dim(`Period: ${usage.period.start} to ${usage.period.end}`));
-        console.log('');
-        // Request statistics
-        console.log(chalk_1.default.bold('Request statistics:'));
-        console.log(`Total requests: ${chalk_1.default.cyan(usage.requests.total.toLocaleString())}`);
-        // Daily breakdown if available
-        if (usage.requests.daily && usage.requests.daily.length > 0) {
-            console.log('');
-            console.log(chalk_1.default.bold('Daily breakdown:'));
-            console.log(chalk_1.default.dim('─'.repeat(30)));
-            console.log(chalk_1.default.dim('DATE'.padEnd(12) + 'REQUESTS'));
-            usage.requests.daily.forEach((day) => {
-                console.log(`${day.date.padEnd(12)}${day.count.toLocaleString()}`);
-            });
-        }
-        // Model usage if available
-        if (usage.models && usage.models.length > 0) {
-            console.log('');
-            console.log(chalk_1.default.bold('Model usage:'));
-            console.log(chalk_1.default.dim('─'.repeat(70)));
-            console.log(chalk_1.default.dim('MODEL'.padEnd(20)) +
-                chalk_1.default.dim('REQUESTS'.padEnd(10)) +
-                chalk_1.default.dim('INPUT'.padEnd(12)) +
-                chalk_1.default.dim('OUTPUT'.padEnd(12)) +
-                chalk_1.default.dim('TOTAL TOKENS'));
-            usage.models.forEach((model) => {
-                console.log(model.name.padEnd(20) +
-                    model.requests.toString().padEnd(10) +
-                    model.tokens.input.toLocaleString().padEnd(12) +
-                    model.tokens.output.toLocaleString().padEnd(12) +
-                    model.tokens.total.toLocaleString());
-            });
-        }
-        console.log('');
-        console.log(chalk_1.default.dim('Use these statistics to understand your API usage and optimize your costs.'));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to get API key usage', error);
-    }
-}));
-// Cluster commands
-const cluster = commander_1.program.command('cluster').description('Manage Berget clusters');
-// Removed cluster create command as it's not available in the API
-cluster
-    .command('list')
-    .description('List all Berget clusters')
-    .action(() => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const clusterService = cluster_service_1.ClusterService.getInstance();
-        const clusters = yield clusterService.listClusters();
-        console.log('NAME                   STATUS    NODES    CREATED');
-        clusters.forEach((cluster) => {
-            console.log(`${cluster.name.padEnd(22)} ${cluster.status.padEnd(9)} ${String(cluster.nodes).padEnd(8)} ${cluster.created}`);
-        });
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to list clusters', error);
-    }
-}));
-cluster
-    .command('usage')
-    .description('Get usage metrics for a specific cluster')
-    .argument('<clusterId>', 'Cluster ID')
-    .action((clusterId) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const clusterService = cluster_service_1.ClusterService.getInstance();
-        const usage = yield clusterService.getClusterUsage(clusterId);
-        console.log('Cluster Usage:');
-        console.log(JSON.stringify(usage, null, 2));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to get cluster usage', error);
-    }
-}));
-// Autocomplete command
-commander_1.program
-    .command('autocomplete')
-    .command('install')
-    .description('Install shell autocompletion')
-    .action(() => {
-    console.log(chalk_1.default.green('✓ Berget autocomplete installed in your shell'));
-    console.log(chalk_1.default.green('✓ Shell completion for kubectl also installed'));
-    console.log('');
-    console.log('Restart your shell or run:');
-    console.log('  source ~/.bashrc');
-});
-// Removed flux commands as they're not available in the API
-// Removed collaborator commands as they're not available in the API
-// Removed helm commands as they're not available in the API
-// Removed kubernetes-like commands as they're not available in the API
-// Add token usage command
-commander_1.program
-    .command('token-usage')
-    .description('Get token usage statistics')
-    .option('--model <modelId>', 'Get usage for a specific model')
-    .action((options) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const client = (0, client_1.createAuthenticatedClient)();
-        let response;
-        if (options.model) {
-            const { data, error } = yield client.GET('/v1/usage/tokens/{modelId}', {
-                params: { path: { modelId: options.model } },
-            });
-            if (error)
-                throw new Error(JSON.stringify(error));
-            response = data;
-        }
-        else {
-            const { data, error } = yield client.GET('/v1/usage/tokens');
-            if (error)
-                throw new Error(JSON.stringify(error));
-            response = data;
-        }
-        console.log('Token Usage:');
-        console.log(JSON.stringify(response, null, 2));
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to get token usage', error);
-    }
-}));
-// Add models command
-commander_1.program
-    .command('models')
-    .description('List available AI models')
-    .option('--id <modelId>', 'Get details for a specific model')
-    .action((options) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const client = (0, client_1.createAuthenticatedClient)();
-        let response;
-        if (options.id) {
-            const { data, error } = yield client.GET('/v1/models/{modelId}', {
-                params: { path: { modelId: options.id } },
-            });
-            if (error)
-                throw new Error(JSON.stringify(error));
-            response = data;
-            console.log('Model Details:');
-            console.log(JSON.stringify(response, null, 2));
-        }
-        else {
-            const { data, error } = yield client.GET('/v1/models');
-            if (error)
-                throw new Error(JSON.stringify(error));
-            response = data;
-            console.log('Available Models:');
-            console.log('ID                      OWNED BY                  CAPABILITIES');
-            response.data.forEach((model) => {
-                const capabilities = [];
-                if (model.capabilities.vision)
-                    capabilities.push('vision');
-                if (model.capabilities.function_calling)
-                    capabilities.push('function_calling');
-                if (model.capabilities.json_mode)
-                    capabilities.push('json_mode');
-                console.log(`${model.id.padEnd(24)} ${model.owned_by.padEnd(25)} ${capabilities.join(', ')}`);
-            });
-        }
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to get models', error);
-    }
-}));
-// Add team command
-commander_1.program
-    .command('team')
-    .description('Manage team members')
-    .action(() => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const client = (0, client_1.createAuthenticatedClient)();
-        const { data, error } = yield client.GET('/v1/users');
-        if (error)
-            throw new Error(JSON.stringify(error));
-        console.log('Team Members:');
-        console.log('NAME                     EMAIL                           ROLE');
-        data.forEach((user) => {
-            console.log(`${user.name.padEnd(24)} ${user.email.padEnd(30)} ${user.role}`);
-        });
-    }
-    catch (error) {
-        (0, error_handler_1.handleError)('Failed to list team members', error);
-    }
-}));
-// Auto-detect .bergetconfig and switch clusters
-const checkBergetConfig = () => {
-    const configPath = path.join(process.cwd(), '.bergetconfig');
-    if (fs.existsSync(configPath)) {
-        try {
-            const config = fs.readFileSync(configPath, 'utf8');
-            const match = config.match(/cluster:\s*(.+)/);
-            if (match && match[1]) {
-                const clusterName = match[1].trim();
-                console.log(`🔄 Berget: Switched to cluster "${clusterName}"`);
-                console.log('✓ kubectl config updated');
-                console.log('');
-            }
-        }
-        catch (error) {
-            // Silently ignore errors reading config
-        }
-    }
-};
+    .option('--local', 'Use local API endpoint (hidden)', false)
+    .option('--debug', 'Enable debug output', false);
+// Register all commands
+(0, commands_1.registerCommands)(commander_1.program);
 // Check for .bergetconfig if not running a command
 if (process.argv.length <= 2) {
-    checkBergetConfig();
+    (0, config_checker_1.checkBergetConfig)();
 }
 commander_1.program.parse(process.argv);