jbai-cli 1.6.0 → 1.8.0

package/bin/jbai-opencode.js

@@ -5,159 +5,155 @@ const fs = require('fs');
 const path = require('path');
 const os = require('os');
 const config = require('../lib/config');
-
-const token = config.getToken();
-if (!token) {
-  console.error('❌ No token found. Run: jbai token set');
-  process.exit(1);
-}
-
-if (config.isTokenExpired(token)) {
-  console.error('⚠️  Token expired. Run: jbai token refresh');
-  process.exit(1);
-}
-
-const endpoints = config.getEndpoints();
-const environment = config.getEnvironment();
-let args = process.argv.slice(2);
-const handoffConfig = stripHandoffFlag(args);
-args = handoffConfig.args;
-
-// Check for super mode (--super, --yolo, -s)
-const superFlags = ['--super', '--yolo', '-s'];
-const superMode = args.some(a => superFlags.includes(a));
-args = args.filter(a => !superFlags.includes(a));
-
-// Setup OpenCode config with JetBrains provider
-const configDir = path.join(os.homedir(), '.config', 'opencode');
-const configFile = path.join(configDir, 'opencode.json');
-
-if (!fs.existsSync(configDir)) {
-  fs.mkdirSync(configDir, { recursive: true });
-}
-
-// Create or update OpenCode config with JetBrains provider
-const providerName = environment === 'staging' ? 'jbai-staging' : 'jbai';
-let opencodeConfig = {};
-
-if (fs.existsSync(configFile)) {
-  try {
-    opencodeConfig = JSON.parse(fs.readFileSync(configFile, 'utf-8'));
-  } catch {
-    opencodeConfig = {};
+const { ensureToken } = require('../lib/ensure-token');
+
+(async () => {
+  const token = await ensureToken();
+  const endpoints = config.getEndpoints();
+  const environment = config.getEnvironment();
+  let args = process.argv.slice(2);
+  const handoffConfig = stripHandoffFlag(args);
+  args = handoffConfig.args;
+
+  // Check for super mode (--super, --yolo, -s)
+  const superFlags = ['--super', '--yolo', '-s'];
+  const superMode = args.some(a => superFlags.includes(a));
+  args = args.filter(a => !superFlags.includes(a));
+
+  // Setup OpenCode config with JetBrains provider
+  const configDir = path.join(os.homedir(), '.config', 'opencode');
+  const configFile = path.join(configDir, 'opencode.json');
+
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
   }
-}
-
-// Ensure provider section exists
-if (!opencodeConfig.provider) {
-  opencodeConfig.provider = {};
-}
-
-// Environment variable name for the token
-const envVarName = environment === 'staging' ? 'GRAZIE_STAGING_TOKEN' : 'GRAZIE_API_TOKEN';
-
-// Provider names for OpenAI and Anthropic
-const anthropicProviderName = environment === 'staging' ? 'jbai-anthropic-staging' : 'jbai-anthropic';
-
-// Add/update JetBrains OpenAI provider with custom header (using env var reference)
-// Use OpenAI SDK to support max_completion_tokens for GPT-5.x
-opencodeConfig.provider[providerName] = {
-  npm: '@ai-sdk/openai',
-  name: `JetBrains AI OpenAI (${environment})`,
-  options: {
-    baseURL: endpoints.openai,
-    apiKey: `{env:${envVarName}}`,
-    headers: {
-      'Grazie-Authenticate-JWT': `{env:${envVarName}}`
+
+  // Create or update OpenCode config with JetBrains provider
+  const providerName = environment === 'staging' ? 'jbai-staging' : 'jbai';
+  let opencodeConfig = {};
+
+  if (fs.existsSync(configFile)) {
+    try {
+      opencodeConfig = JSON.parse(fs.readFileSync(configFile, 'utf-8'));
+    } catch {
+      opencodeConfig = {};
     }
-  },
-  models: {}
-};
-
-// Add OpenAI models
-// OpenCode requires an output limit in the config.
-// For O-series models we keep a larger context window but still set output.
-config.MODELS.openai.available.forEach(model => {
-  const isOSeries = /^o[1-9]/.test(model);
-  opencodeConfig.provider[providerName].models[model] = {
-    name: model,
-    limit: isOSeries
-      ? { context: 200000, output: 8192 }
-      : { context: 128000, output: 8192 }
+  }
+
+  // Ensure provider section exists
+  if (!opencodeConfig.provider) {
+    opencodeConfig.provider = {};
+  }
+
+  // Environment variable name for the token
+  const envVarName = environment === 'staging' ? 'GRAZIE_STAGING_TOKEN' : 'GRAZIE_API_TOKEN';
+
+  // Provider names for OpenAI and Anthropic
+  const anthropicProviderName = environment === 'staging' ? 'jbai-anthropic-staging' : 'jbai-anthropic';
+
+  // Add/update JetBrains OpenAI provider with custom header (using env var reference)
+  // Use OpenAI SDK to support max_completion_tokens for GPT-5.x
+  opencodeConfig.provider[providerName] = {
+    npm: '@ai-sdk/openai',
+    name: `JetBrains AI OpenAI (${environment})`,
+    options: {
+      baseURL: endpoints.openai,
+      apiKey: `{env:${envVarName}}`,
+      headers: {
+        'Grazie-Authenticate-JWT': `{env:${envVarName}}`
+      }
+    },
+    models: {}
   };
-});
-
-// Add JetBrains Anthropic provider for Claude models
-opencodeConfig.provider[anthropicProviderName] = {
-  npm: '@ai-sdk/anthropic',
-  name: `JetBrains AI Anthropic (${environment})`,
-  options: {
-    baseURL: endpoints.anthropic,
-    apiKey: `{env:${envVarName}}`,
-    headers: {
-      'Grazie-Authenticate-JWT': `{env:${envVarName}}`
-    }
-  },
-  models: {}
-};
-
-// Add Claude models to Anthropic provider
-config.MODELS.claude.available.forEach(model => {
-  opencodeConfig.provider[anthropicProviderName].models[model] = {
-    name: model,
-    limit: { context: 200000, output: 8192 }
+
+  // Add OpenAI models
+  // OpenCode requires an output limit in the config.
+  // For O-series models we keep a larger context window but still set output.
+  config.MODELS.openai.available.forEach(model => {
+    const isOSeries = /^o[1-9]/.test(model);
+    opencodeConfig.provider[providerName].models[model] = {
+      name: model,
+      limit: isOSeries
+        ? { context: 200000, output: 8192 }
+        : { context: 128000, output: 8192 }
+    };
+  });
+
+  // Add JetBrains Anthropic provider for Claude models
+  opencodeConfig.provider[anthropicProviderName] = {
+    npm: '@ai-sdk/anthropic',
+    name: `JetBrains AI Anthropic (${environment})`,
+    options: {
+      baseURL: endpoints.anthropic,
+      apiKey: `{env:${envVarName}}`,
+      headers: {
+        'Grazie-Authenticate-JWT': `{env:${envVarName}}`
+      }
+    },
+    models: {}
   };
-});
-
-// Write config
-fs.writeFileSync(configFile, JSON.stringify(opencodeConfig, null, 2));
-
-// Check if model specified
-const hasModel = args.includes('--model') || args.includes('-m');
-let finalArgs = [];
-
-if (!hasModel) {
-  // Use provider/model format for OpenCode
-  finalArgs.push('--model', `${providerName}/${config.MODELS.openai.default}`);
-}
-
-// Note: OpenCode run is already non-interactive, no super mode needed
-if (superMode) {
-  console.log('ℹ️  OpenCode run is already non-interactive (no super mode flag needed)');
-}
-
-finalArgs.push(...args);
-
-// Set the token in environment variable for OpenCode config to reference
-const childEnv = {
-  ...process.env,
-  [envVarName]: token
-};
-
-const child = runWithHandoff({
-  command: 'opencode',
-  args: finalArgs,
-  env: childEnv,
-  toolName: 'jbai-opencode',
-  handoffDefaults: {
-    enabled: !handoffConfig.disabled,
-    grazieToken: token,
-    grazieEnvironment: environment === 'production' ? 'PRODUCTION' : 'STAGING',
-    grazieModel: config.MODELS.claude.default,
-    cwd: process.cwd(),
-  },
-});
-
-if (child && typeof child.on === 'function') {
-  child.on('error', (err) => {
-    if (err.code === 'ENOENT') {
-      const tool = config.TOOLS.opencode;
-      console.error(`❌ ${tool.name} not found.\n`);
-      console.error(`Install with: ${tool.install}`);
-      console.error(`Or run: jbai install opencode`);
-    } else {
-      console.error(`Error: ${err.message}`);
-    }
-    process.exit(1);
+
+  // Add Claude models to Anthropic provider
+  config.MODELS.claude.available.forEach(model => {
+    opencodeConfig.provider[anthropicProviderName].models[model] = {
+      name: model,
+      limit: { context: 200000, output: 8192 }
+    };
   });
-}
+
+  // NOTE: Gemini models are NOT available via Grazie OpenAI-compatible proxy.
+  // The /user/v5/llm/google/v1/vertex endpoint only works with native Google API
+  // format (used by jbai-gemini), not OpenAI chat/completions format.
+  // Use `jbai gemini` instead for Gemini models.
+
+  // Enable max permissions (yolo mode) by default
+  opencodeConfig.yolo = true;
+
+  // Write config
+  fs.writeFileSync(configFile, JSON.stringify(opencodeConfig, null, 2));
+
+  // Check if model specified
+  const hasModel = args.includes('--model') || args.includes('-m');
+  let finalArgs = ['--yolo'];
+
+  if (!hasModel) {
+    // Use provider/model format for OpenCode
+    finalArgs.push('--model', `${providerName}/${config.MODELS.openai.default}`);
+  }
+
+  finalArgs.push(...args);
+
+  // Set the token in environment variable for OpenCode config to reference
+  const childEnv = {
+    ...process.env,
+    [envVarName]: token
+  };
+
+  const child = runWithHandoff({
+    command: 'opencode',
+    args: finalArgs,
+    env: childEnv,
+    toolName: 'jbai-opencode',
+    handoffDefaults: {
+      enabled: !handoffConfig.disabled,
+      grazieToken: token,
+      grazieEnvironment: environment === 'production' ? 'PRODUCTION' : 'STAGING',
+      grazieModel: config.MODELS.claude.default,
+      cwd: process.cwd(),
+    },
+  });
+
+  if (child && typeof child.on === 'function') {
+    child.on('error', (err) => {
+      if (err.code === 'ENOENT') {
+        const tool = config.TOOLS.opencode;
+        console.error(`❌ ${tool.name} not found.\n`);
+        console.error(`Install with: ${tool.install}`);
+        console.error(`Or run: jbai install opencode`);
+      } else {
+        console.error(`Error: ${err.message}`);
+      }
+      process.exit(1);
+    });
+  }
+})();