berget 0.1.0 → 1.1.0

package/index.ts

@@ -1,11 +1,8 @@
 #!/usr/bin/env node
 
 import { program } from 'commander'
-import * as fs from 'fs'
-import * as path from 'path'
-import { createAuthenticatedClient } from './src/client'
-import { handleError } from './src/utils/error-handler'
-import chalk from 'chalk'
+import { registerCommands } from './src/commands'
+import { checkBergetConfig } from './src/utils/config-checker'
 
 // Set version and description
 program
@@ -22,531 +19,10 @@ program
   )
   .version(process.env.npm_package_version || '0.0.1', '-v, --version')
   .option('--local', 'Use local API endpoint (hidden)', false)
+  .option('--debug', 'Enable debug output', false)
 
-// Import services
-import { AuthService } from './src/services/auth-service'
-import { ApiKeyService, ApiKey } from './src/services/api-key-service'
-import { ClusterService, Cluster } from './src/services/cluster-service'
-
-// Auth commands
-program
-  .command('login')
-  .description('Log in to Berget')
-  .action(async () => {
-    const authService = AuthService.getInstance()
-    await authService.login()
-  })
-
-program
-  .command('logout')
-  .description('Log out from Berget')
-  .action(() => {
-    const { clearAuthToken } = require('./src/client')
-    clearAuthToken()
-    console.log(chalk.green('You have been logged out from Berget'))
-  })
-
-program
-  .command('whoami')
-  .description('Show information about the logged in user')
-  .action(async () => {
-    try {
-      const authService = AuthService.getInstance()
-      const profile = await authService.getUserProfile()
-
-      if (profile) {
-        console.log(
-          chalk.bold(`Logged in as: ${profile.name || profile.login}`)
-        )
-        console.log(`Email: ${chalk.cyan(profile.email || 'Not available')}`)
-        console.log(`Role: ${chalk.cyan(profile.role || 'Not available')}`)
-
-        if (profile.company) {
-          console.log(`Company: ${chalk.cyan(profile.company.name)}`)
-        }
-      } else {
-        console.log(
-          chalk.yellow('You are not logged in. Use `berget login` to log in.')
-        )
-      }
-    } catch (error) {
-      handleError('You are not logged in or an error occurred', error)
-    }
-  })
-
-// API Key commands
-const apiKey = program.command('api-key').description('Manage API keys')
-
-apiKey
-  .command('list')
-  .description('List all API keys')
-  .action(async () => {
-    try {
-      const apiKeyService = ApiKeyService.getInstance()
-      const keys = await apiKeyService.listApiKeys()
-
-      if (keys.length === 0) {
-        console.log(
-          chalk.yellow(
-            'No API keys found. Create one with `berget api-key create --name <name>`'
-          )
-        )
-        return
-      }
-
-      console.log(chalk.bold('Your API keys:'))
-      console.log('')
-
-      // Create a table-like format with headers
-      console.log(
-        chalk.dim('ID'.padEnd(10)) +
-          chalk.dim('NAME'.padEnd(25)) +
-          chalk.dim('PREFIX'.padEnd(12)) +
-          chalk.dim('STATUS'.padEnd(12)) +
-          chalk.dim('CREATED'.padEnd(12)) +
-          chalk.dim('LAST USED')
-      )
-
-      console.log(chalk.dim('─'.repeat(85)))
-
-      keys.forEach((key: ApiKey) => {
-        const lastUsed = key.lastUsed ? key.lastUsed.substring(0, 10) : 'Never'
-        const status = key.active
-          ? chalk.green('● Active')
-          : chalk.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.dim(
-          'Use `berget api-key create --name <name>` to create a new API key'
-        )
-      )
-      console.log(
-        chalk.dim('Use `berget api-key delete <id>` to delete an API key')
-      )
-      console.log(
-        chalk.dim('Use `berget api-key rotate <id>` to rotate an API key')
-      )
-    } catch (error) {
-      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(async (options) => {
-    try {
-      if (!options.name) {
-        console.error(chalk.red('Error: --name is required'))
-        console.log('')
-        console.log(
-          'Usage: berget api-key create --name <name> [--description <description>]'
-        )
-        return
-      }
-
-      console.log(chalk.blue('Creating API key...'))
-
-      const apiKeyService = ApiKeyService.getInstance()
-      const result = await apiKeyService.createApiKey({
-        name: options.name,
-        description: options.description,
-      })
-
-      console.log('')
-      console.log(chalk.green('✓ API key created'))
-      console.log('')
-      console.log(chalk.bold('API key details:'))
-      console.log('')
-      console.log(`${chalk.dim('ID:')}          ${result.id}`)
-      console.log(`${chalk.dim('Name:')}        ${result.name}`)
-      if (result.description) {
-        console.log(`${chalk.dim('Description:')} ${result.description}`)
-      }
-      console.log(
-        `${chalk.dim('Created:')}     ${new Date(
-          result.created
-        ).toLocaleString()}`
-      )
-      console.log('')
-      console.log(chalk.bold('API key:'))
-      console.log(chalk.cyan(result.key))
-      console.log('')
-      console.log(
-        chalk.yellow('⚠️  IMPORTANT: Save this API key in a secure location.')
-      )
-      console.log(chalk.yellow('   It will not be displayed again.'))
-
-      console.log('')
-      console.log(
-        chalk.dim(
-          'Use this key in your applications to authenticate with the Berget API.'
-        )
-      )
-    } catch (error) {
-      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(async (id) => {
-    try {
-      console.log(chalk.blue(`Deleting API key ${id}...`))
-
-      const apiKeyService = ApiKeyService.getInstance()
-      await apiKeyService.deleteApiKey(id)
-
-      console.log(chalk.green(`✓ API key ${id} has been deleted`))
-      console.log('')
-      console.log(
-        chalk.dim(
-          'Applications using this key will no longer be able to authenticate.'
-        )
-      )
-      console.log(
-        chalk.dim('Use `berget api-key list` to see your remaining API keys.')
-      )
-    } catch (error) {
-      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(async (id) => {
-    try {
-      console.log(chalk.blue(`Rotating API key ${id}...`))
-      console.log(
-        chalk.dim('This will invalidate the old key and generate a new one.')
-      )
-
-      const apiKeyService = ApiKeyService.getInstance()
-      const result = await apiKeyService.rotateApiKey(id)
-
-      console.log('')
-      console.log(chalk.green('✓ API key rotated'))
-      console.log('')
-      console.log(chalk.bold('New API key details:'))
-      console.log('')
-      console.log(`${chalk.dim('ID:')}          ${result.id}`)
-      console.log(`${chalk.dim('Name:')}        ${result.name}`)
-      if (result.description) {
-        console.log(`${chalk.dim('Description:')} ${result.description}`)
-      }
-      console.log(
-        `${chalk.dim('Created:')}     ${new Date(
-          result.created
-        ).toLocaleString()}`
-      )
-      console.log('')
-      console.log(chalk.bold('New API key:'))
-      console.log(chalk.cyan(result.key))
-      console.log('')
-      console.log(
-        chalk.yellow(
-          '⚠️  IMPORTANT: Update your applications with this new API key.'
-        )
-      )
-      console.log(
-        chalk.yellow(
-          '   The old key has been invalidated and will no longer work.'
-        )
-      )
-      console.log(chalk.yellow('   This new key will not be displayed again.'))
-    } catch (error) {
-      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(async (id, options) => {
-    try {
-      console.log(chalk.blue(`Fetching usage statistics for API key ${id}...`))
-
-      const apiKeyService = ApiKeyService.getInstance()
-      const usage = await apiKeyService.getApiKeyUsage(id)
-
-      console.log('')
-      console.log(
-        chalk.bold(`Usage statistics for API key: ${usage.name} (${id})`)
-      )
-      console.log('')
-
-      // Period information
-      console.log(
-        chalk.dim(`Period: ${usage.period.start} to ${usage.period.end}`)
-      )
-      console.log('')
-
-      // Request statistics
-      console.log(chalk.bold('Request statistics:'))
-      console.log(
-        `Total requests: ${chalk.cyan(usage.requests.total.toLocaleString())}`
-      )
-
-      // Daily breakdown if available
-      if (usage.requests.daily && usage.requests.daily.length > 0) {
-        console.log('')
-        console.log(chalk.bold('Daily breakdown:'))
-        console.log(chalk.dim('─'.repeat(30)))
-        console.log(chalk.dim('DATE'.padEnd(12) + 'REQUESTS'))
-
-        usage.requests.daily.forEach((day: { date: string; count: number }) => {
-          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.bold('Model usage:'))
-        console.log(chalk.dim('─'.repeat(70)))
-        console.log(
-          chalk.dim('MODEL'.padEnd(20)) +
-            chalk.dim('REQUESTS'.padEnd(10)) +
-            chalk.dim('INPUT'.padEnd(12)) +
-            chalk.dim('OUTPUT'.padEnd(12)) +
-            chalk.dim('TOTAL TOKENS')
-        )
-
-        usage.models.forEach(
-          (model: {
-            name: string
-            requests: number
-            tokens: {
-              input: number
-              output: number
-              total: number
-            }
-          }) => {
-            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.dim(
-          'Use these statistics to understand your API usage and optimize your costs.'
-        )
-      )
-    } catch (error) {
-      handleError('Failed to get API key usage', error)
-    }
-  })
-
-// Cluster commands
-const cluster = 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(async () => {
-    try {
-      const clusterService = ClusterService.getInstance()
-      const clusters = await clusterService.listClusters()
-
-      console.log('NAME                   STATUS    NODES    CREATED')
-      clusters.forEach((cluster: Cluster) => {
-        console.log(
-          `${cluster.name.padEnd(22)} ${cluster.status.padEnd(9)} ${String(
-            cluster.nodes
-          ).padEnd(8)} ${cluster.created}`
-        )
-      })
-    } catch (error) {
-      handleError('Failed to list clusters', error)
-    }
-  })
-
-cluster
-  .command('usage')
-  .description('Get usage metrics for a specific cluster')
-  .argument('<clusterId>', 'Cluster ID')
-  .action(async (clusterId) => {
-    try {
-      const clusterService = ClusterService.getInstance()
-      const usage = await clusterService.getClusterUsage(clusterId)
-
-      console.log('Cluster Usage:')
-      console.log(JSON.stringify(usage, null, 2))
-    } catch (error) {
-      handleError('Failed to get cluster usage', error)
-    }
-  })
-
-// Autocomplete command
-program
-  .command('autocomplete')
-  .command('install')
-  .description('Install shell autocompletion')
-  .action(() => {
-    console.log(chalk.green('✓ Berget autocomplete installed in your shell'))
-    console.log(chalk.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
-program
-  .command('token-usage')
-  .description('Get token usage statistics')
-  .option('--model <modelId>', 'Get usage for a specific model')
-  .action(async (options) => {
-    try {
-      const client = createAuthenticatedClient()
-      let response
-
-      if (options.model) {
-        const { data, error } = await 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 } = await 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) {
-      handleError('Failed to get token usage', error)
-    }
-  })
-
-// Add models command
-program
-  .command('models')
-  .description('List available AI models')
-  .option('--id <modelId>', 'Get details for a specific model')
-  .action(async (options) => {
-    try {
-      const client = createAuthenticatedClient()
-      let response
-
-      if (options.id) {
-        const { data, error } = await 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 } = await 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: any) => {
-          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) {
-      handleError('Failed to get models', error)
-    }
-  })
-
-// Add team command
-program
-  .command('team')
-  .description('Manage team members')
-  .action(async () => {
-    try {
-      const client = createAuthenticatedClient()
-      const { data, error } = await client.GET('/v1/users')
-      if (error) throw new Error(JSON.stringify(error))
-
-      console.log('Team Members:')
-      console.log(
-        'NAME                     EMAIL                           ROLE'
-      )
-      data.forEach((user: any) => {
-        console.log(
-          `${user.name.padEnd(24)} ${user.email.padEnd(30)} ${user.role}`
-        )
-      })
-    } catch (error) {
-      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
-    }
-  }
-}
+// Register all commands
+registerCommands(program)
 
 // Check for .bergetconfig if not running a command
 if (process.argv.length <= 2) {

package/package.json

@@ -1,12 +1,16 @@
 {
   "name": "berget",
-  "version": "0.1.0",
+  "version": "1.1.0",
   "main": "dist/index.js",
   "bin": {
     "berget": "dist/index.js"
   },
   "private": false,
   "scripts": {
+    "start": "node --import tsx ./index.ts --local",
+    "login": "node --import tsx ./index.ts --local auth login",
+    "logout": "node --import tsx ./index.ts --local auth logout",
+    "whoami": "node --import tsx ./index.ts --local auth whoami",
     "build": "tsc",
     "prepublishOnly": "npm run build",
     "generate-types": "openapi-typescript https://api.berget.ai/openapi.json -o src/types/api.d.ts"
@@ -16,6 +20,7 @@
   "description": "This is a cli command for interacting with the AI infrastructure provider Berget",
   "devDependencies": {
     "@types/node": "^20.11.20",
+    "tsx": "^4.19.3",
     "typescript": "^5.3.3"
   },
   "dependencies": {