@arcteninc/core 0.0.145 → 0.0.146

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arcteninc/core",
-  "version": "0.0.145",
+  "version": "0.0.146",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",

package/scripts/cli-init-wizard.ts

@@ -135,7 +135,12 @@ async function discoverContext(): Promise<InitContext> {
     }
     if (!envContent.includes('ARCTEN_API_KEY')) {
       envContent += `\nARCTEN_API_KEY=${apiKeyInput}\n`;
-      envContent += `NEXT_PUBLIC_ARCTEN_API_URL=https://api.arcten.com\n`;
+      // Use correct env prefix based on framework
+      if (framework === 'vite') {
+        envContent += `VITE_ARCTEN_API_URL=https://api.arcten.com\n`;
+      } else {
+        envContent += `NEXT_PUBLIC_ARCTEN_API_URL=https://api.arcten.com\n`;
+      }
       fs.writeFileSync(envPath, envContent.trim() + '\n');
       console.log('  Saved API key to .env.local');
     }
@@ -429,15 +434,100 @@ export async function POST(req: NextRequest) {
     fs.writeFileSync(path.join(tokenDir, 'route.ts'), tokenRouteContent);
     createdFiles.push(`${appDir}/api/arcten/token/route.ts`);
     console.log(`  Created ${appDir}/api/arcten/token/route.ts`);
+  } else if (framework === 'vite') {
+    // Create Vite middleware plugin for token endpoint
+    const vitePluginContent = `// Arcten Token Plugin for Vite Dev Server
+// This provides the /api/arcten/token endpoint during development
+
+export function arctenTokenPlugin() {
+  return {
+    name: 'arcten-token',
+    configureServer(server: any) {
+      server.middlewares.use('/api/arcten/token', async (req: any, res: any, next: any) => {
+        if (req.method !== 'POST') {
+          return next();
+        }
+
+        let body = '';
+        req.on('data', (chunk: any) => body += chunk);
+        req.on('end', async () => {
+          try {
+            const { user } = JSON.parse(body || '{}');
+            const apiKey = process.env.ARCTEN_API_KEY;
+
+            if (!apiKey) {
+              res.statusCode = 500;
+              res.setHeader('Content-Type', 'application/json');
+              res.end(JSON.stringify({ error: 'ARCTEN_API_KEY not configured' }));
+              return;
+            }
+
+            // Dynamic import to avoid bundling issues
+            const { verifyToken } = await import('@arcteninc/core/server');
+            const tokenResponse = await verifyToken({
+              apiKey,
+              user: user || { id: 'anonymous', name: 'User' },
+              apiUrl: process.env.VITE_ARCTEN_API_URL || 'https://api.arcten.com',
+            });
+
+            res.setHeader('Content-Type', 'application/json');
+            res.end(JSON.stringify(tokenResponse));
+          } catch (error: any) {
+            console.error('Token generation error:', error);
+            res.statusCode = 500;
+            res.setHeader('Content-Type', 'application/json');
+            res.end(JSON.stringify({ error: error.message || 'Failed to generate token' }));
+          }
+        });
+      });
+    },
+  };
+}
+`;
+    fs.writeFileSync(path.join(projectRoot, 'vite-plugin-arcten.ts'), vitePluginContent);
+    createdFiles.push('vite-plugin-arcten.ts');
+    console.log('  Created vite-plugin-arcten.ts');
+
+    // Update vite.config.ts to include the plugin
+    const viteConfigPath = path.join(projectRoot, 'vite.config.ts');
+    if (fs.existsSync(viteConfigPath)) {
+      let viteConfig = fs.readFileSync(viteConfigPath, 'utf-8');
+
+      if (!viteConfig.includes('arctenTokenPlugin')) {
+        // Add import at the top
+        if (viteConfig.includes("from 'vite'")) {
+          viteConfig = viteConfig.replace(
+            /from ['"]vite['"]/,
+            `from 'vite'\nimport { arctenTokenPlugin } from './vite-plugin-arcten'`
+          );
+        } else {
+          viteConfig = `import { arctenTokenPlugin } from './vite-plugin-arcten'\n` + viteConfig;
+        }
+
+        // Add to plugins array
+        viteConfig = viteConfig.replace(
+          /plugins:\s*\[/,
+          'plugins: [arctenTokenPlugin(), '
+        );
+
+        fs.writeFileSync(viteConfigPath, viteConfig);
+        createdFiles.push('vite.config.ts (modified)');
+        console.log('  Modified vite.config.ts');
+      }
+    }
   }
 
   // 3. Create wrapper component based on placement
   const toolsImportPath = path.relative(appDirPath, context.toolsFile).replace(/\.ts$/, '').replace(/\\/g, '/');
 
-  if (placement === 'everywhere') {
-    const wrapperContent = `"use client";
+  // Get the correct env variable syntax based on framework
+  const envVarSyntax = framework === 'vite'
+    ? 'import.meta.env.VITE_ARCTEN_API_URL'
+    : 'process.env.NEXT_PUBLIC_ARCTEN_API_URL';
+  const useClientDirective = framework === 'vite' ? '' : '"use client";\n\n';
 
-import { ArctenAgent } from "@arcteninc/core";
+  if (placement === 'everywhere') {
+    const wrapperContent = `${useClientDirective}import { ArctenAgent } from "@arcteninc/core";
 import { toolMetadata } from "../.arcten/tool-metadata";
 import { allTools, safeToolNames } from "../.arcten/arcten.tools";
 
@@ -452,7 +542,7 @@ export function ArctenWrapper({ children }: ArctenWrapperProps) {
         {children}
       </div>
       <ArctenAgent
-        apiBaseUrl={process.env.NEXT_PUBLIC_ARCTEN_API_URL || "https://api.arcten.com"}
+        apiBaseUrl={${envVarSyntax} || "https://api.arcten.com"}
         user={{ id: "user-1", name: "User" }}
         tools={allTools}
         safeToolNames={safeToolNames}
@@ -468,46 +558,74 @@ export default ArctenWrapper;
     createdFiles.push(`${appDir}/ArctenWrapper.tsx`);
     console.log(`  Created ${appDir}/ArctenWrapper.tsx`);
 
-    // 4. Modify layout.tsx to wrap with ArctenWrapper
-    const layoutPath = path.join(appDirPath, 'layout.tsx');
-    if (fs.existsSync(layoutPath)) {
-      let layoutContent = fs.readFileSync(layoutPath, 'utf-8');
-
-      // Add import if not present
-      if (!layoutContent.includes('ArctenWrapper')) {
-        // Add import at the top (after existing imports)
-        const importStatement = `import { ArctenWrapper } from "./ArctenWrapper";\n`;
-        const lastImportIndex = layoutContent.lastIndexOf('import ');
-        if (lastImportIndex !== -1) {
-          const endOfImport = layoutContent.indexOf('\n', lastImportIndex);
-          layoutContent = layoutContent.slice(0, endOfImport + 1) + importStatement + layoutContent.slice(endOfImport + 1);
-        } else {
-          layoutContent = importStatement + layoutContent;
+    if (framework === 'vite') {
+      // 4a. Modify main.tsx for Vite
+      const mainPath = path.join(projectRoot, 'src', 'main.tsx');
+      if (fs.existsSync(mainPath)) {
+        let mainContent = fs.readFileSync(mainPath, 'utf-8');
+
+        if (!mainContent.includes('ArctenWrapper')) {
+          // Add import after existing imports
+          const importStatement = `import { ArctenWrapper } from './ArctenWrapper'\n`;
+          const lastImportIndex = mainContent.lastIndexOf('import ');
+          if (lastImportIndex !== -1) {
+            const endOfImport = mainContent.indexOf('\n', lastImportIndex);
+            mainContent = mainContent.slice(0, endOfImport + 1) + importStatement + mainContent.slice(endOfImport + 1);
+          } else {
+            mainContent = importStatement + mainContent;
+          }
+
+          // Wrap <App /> with <ArctenWrapper>
+          mainContent = mainContent.replace(
+            /<App\s*\/>/,
+            '<ArctenWrapper><App /></ArctenWrapper>'
+          );
+
+          fs.writeFileSync(mainPath, mainContent);
+          createdFiles.push('src/main.tsx (modified)');
+          console.log('  Modified src/main.tsx');
+        }
+      }
+    } else {
+      // 4b. Modify layout.tsx for Next.js
+      const layoutPath = path.join(appDirPath, 'layout.tsx');
+      if (fs.existsSync(layoutPath)) {
+        let layoutContent = fs.readFileSync(layoutPath, 'utf-8');
+
+        // Add import if not present
+        if (!layoutContent.includes('ArctenWrapper')) {
+          // Add import at the top (after existing imports)
+          const importStatement = `import { ArctenWrapper } from "./ArctenWrapper";\n`;
+          const lastImportIndex = layoutContent.lastIndexOf('import ');
+          if (lastImportIndex !== -1) {
+            const endOfImport = layoutContent.indexOf('\n', lastImportIndex);
+            layoutContent = layoutContent.slice(0, endOfImport + 1) + importStatement + layoutContent.slice(endOfImport + 1);
+          } else {
+            layoutContent = importStatement + layoutContent;
+          }
+
+          // Wrap {children} with <ArctenWrapper>
+          layoutContent = layoutContent.replace(
+            /(\{children\})/,
+            '<ArctenWrapper>{children}</ArctenWrapper>'
+          );
+
+          fs.writeFileSync(layoutPath, layoutContent);
+          createdFiles.push(`${appDir}/layout.tsx (modified)`);
+          console.log(`  Modified ${appDir}/layout.tsx`);
         }
-
-        // Wrap {children} with <ArctenWrapper>
-        layoutContent = layoutContent.replace(
-          /(\{children\})/,
-          '<ArctenWrapper>{children}</ArctenWrapper>'
-        );
-
-        fs.writeFileSync(layoutPath, layoutContent);
-        createdFiles.push(`${appDir}/layout.tsx (modified)`);
-        console.log(`  Modified ${appDir}/layout.tsx`);
       }
     }
   } else if (placement === 'specific') {
     // Create standalone component for specific pages
-    const componentContent = `"use client";
-
-import { ArctenAgent } from "@arcteninc/core";
+    const componentContent = `${useClientDirective}import { ArctenAgent } from "@arcteninc/core";
 import { toolMetadata } from "../.arcten/tool-metadata";
 import { allTools, safeToolNames } from "../.arcten/arcten.tools";
 
 export function ArctenChat() {
   return (
     <ArctenAgent
-      apiBaseUrl={process.env.NEXT_PUBLIC_ARCTEN_API_URL || "https://api.arcten.com"}
+      apiBaseUrl={${envVarSyntax} || "https://api.arcten.com"}
       user={{ id: "user-1", name: "User" }}
       tools={allTools}
       safeToolNames={safeToolNames}

package/scripts/cli-sync.ts

@@ -663,17 +663,28 @@ async function main() {
     const scriptDir = path.dirname(new URL(import.meta.url).pathname).replace(/^\/([A-Z]:)/, '$1');
 
     try {
+      // Read config to get tool file path
+      const configPath = path.join(projectRoot, '.arcten', 'config.ts');
+      let toolsArg = '';
+      if (fs.existsSync(configPath)) {
+        const configContent = fs.readFileSync(configPath, 'utf-8');
+        const toolFileMatch = configContent.match(/toolFile:\s*["']([^"']+)["']/);
+        if (toolFileMatch) {
+          toolsArg = ` --tools "${toolFileMatch[1]}"`;
+        }
+      }
+
       // Try running the wrapper script directly first
       const extractScript = path.join(scriptDir, 'cli-extract-types-auto-wrapper.js');
       if (fs.existsSync(extractScript)) {
-        execSync(`node "${extractScript}"`, {
+        execSync(`node "${extractScript}"${toolsArg}`, {
           cwd: projectRoot,
           stdio: options.quiet ? 'pipe' : ['inherit', 'pipe', 'pipe'],
         });
         if (!options.quiet) console.log('  Done');
       } else {
         // Fall back to npx
-        execSync('npx arcten-extract-types', {
+        execSync(`npx arcten-extract-types${toolsArg}`, {
           cwd: projectRoot,
           stdio: options.quiet ? 'pipe' : ['inherit', 'pipe', 'pipe'],
         });