ai-extension-preview 0.1.12 → 0.1.13

package/dist/index.js

@@ -138,6 +138,21 @@ const WORK_DIR = path.join(HOME_DIR, '.ai-extension-preview', options.job || 'de
         await ctx.actions.runAction('core:log', { level: 'error', message: 'Initial check failed. Could not verify job or download extension.' });
         process.exit(1);
     }
+    // Wait for Extension files (Manifest)
+    const manifestPath = path.join(WORK_DIR, 'dist', 'manifest.json');
+    let attempts = 0;
+    const maxAttempts = 60; // 2 minutes
+    console.log('[DEBUG] Waiting for extension files...');
+    while (!fs.existsSync(manifestPath) && attempts < maxAttempts) {
+        await new Promise(r => setTimeout(r, 2000));
+        attempts++;
+        if (attempts % 5 === 0)
+            console.log(`Waiting for extension generation... (${attempts * 2}s)`);
+    }
+    if (!fs.existsSync(manifestPath)) {
+        await ctx.actions.runAction('core:log', { level: 'error', message: 'Timed out waiting for extension files. Status check succeeded but files are missing.' });
+        process.exit(1);
+    }
     // Launch Browser
     await ctx.actions.runAction('browser:start', {});
     // Keep process alive

package/dist/plugins/DownloaderPlugin.js

@@ -34,7 +34,7 @@ export const DownloaderPlugin = {
             return axios.create({
                 baseURL: config.host,
                 headers: {
-                    'Authorization': token ? `Bearer ${token}` : undefined,
+                    'X-Preview-Token': token,
                     'X-User-Id': userId
                 },
                 httpsAgent: new https.Agent({
@@ -199,34 +199,7 @@ console.log('[Hot Reload] Active for Job:', CURRENT_JOB_ID);
                 }
             }
         });
-        // Start Polling (Loop)
-        void ctx.actions.runAction('core:log', { level: 'info', message: 'Starting polling loop (Interval: 2000ms)' });
-        // Listen for browser failure to stop polling
-        ctx.events.on('browser:launch-failed', () => {
-            if (checkInterval) {
-                clearInterval(checkInterval);
-                checkInterval = undefined;
-                ctx.actions.runAction('core:log', { level: 'warn', message: 'Polling stopped due to browser launch failure.' });
-                // Update status happens in UI
-            }
-        });
-        checkInterval = setInterval(async () => {
-            try {
-                // Use actions for main log (UI Plugin captures this)
-                // console.error('[DownloaderPlugin] Tick - Checking Status...'); // REMOVE (Outside UI)
-                // Silent polling for CLI mode
-                // await ctx.actions.runAction('core:log', { level: 'info', message: '[DEBUG] Polling...' });
-                await ctx.actions.runAction('downloader:check', null);
-            }
-            catch (err) {
-                await ctx.actions.runAction('core:log', { level: 'error', message: `Poll Error: ${err.message}` });
-            }
-        }, 2000);
-    },
-    dispose(ctx) {
-        if (checkInterval) {
-            clearInterval(checkInterval);
-            checkInterval = undefined;
-        }
+        // Polling removed in favor of push-based updates (POST /refresh)
+        ctx.actions.runAction('core:log', { level: 'info', message: 'Ready. Waiting for update signals...' });
     }
 };

package/dist/plugins/ServerPlugin.js

@@ -22,6 +22,7 @@ export const ServerPlugin = {
             res.setHeader('Access-Control-Allow-Origin', '*');
             res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
             res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+            res.setHeader('Access-Control-Allow-Private-Network', 'true');
             if (req.method === 'OPTIONS') {
                 res.writeHead(204);
                 res.end();
@@ -36,6 +37,17 @@ export const ServerPlugin = {
                     port: allocatedPort
                 }));
             }
+            else if (req.url === '/refresh' && req.method === 'POST') {
+                // Trigger manual check
+                ctx.actions.runAction('core:log', { level: 'info', message: '[API] Refresh request received' });
+                ctx.actions.runAction('downloader:check', null).then((result) => {
+                    ctx.actions.runAction('core:log', { level: 'info', message: `[API] Check result: ${result}` });
+                }).catch((err) => {
+                    ctx.actions.runAction('core:log', { level: 'error', message: `[API] Check failed: ${err.message}` });
+                });
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({ success: true }));
+            }
             else if (req.url === '/disconnect' && req.method === 'POST') {
                 // Trigger browser stop
                 ctx.actions.runAction('core:log', { level: 'info', message: '[API] Disconnect request received' });

package/dist/plugins/browser/BrowserManagerPlugin.js

@@ -49,12 +49,14 @@ export const BrowserManagerPlugin = {
                 stagingDir: STAGING_DIR
             });
         };
+        let isInitialized = false;
         // Action: Start Browser (Orchestrator)
         ctx.actions.registerAction({
             id: 'browser:start',
             handler: async () => {
                 await syncToStaging();
                 await launchBrowser();
+                isInitialized = true;
                 return true;
             }
         });
@@ -69,8 +71,10 @@ export const BrowserManagerPlugin = {
         });
         // Event: Update detected
         ctx.events.on('downloader:updated', async () => {
-            await ctx.actions.runAction('core:log', { level: 'info', message: 'Update detected. Syncing to staging...' });
-            await ctx.actions.runAction('browser:start', {});
+            if (isInitialized) {
+                await ctx.actions.runAction('core:log', { level: 'info', message: 'Update detected. Syncing to staging...' });
+                await ctx.actions.runAction('browser:start', {});
+            }
         });
         // Event: Browser closed (from launcher)
         ctx.events.on('browser:closed', async (data) => {

package/dist/plugins/browser/NativeLauncherPlugin.js

@@ -36,8 +36,8 @@ export const NativeLauncherPlugin = {
                 await ctx.actions.runAction('core:log', { level: 'info', message: `Native Launch Executable: ${executable} ` });
                 await ctx.actions.runAction('core:log', { level: 'info', message: `Native Launch Target: ${safeDist} ` });
                 const cleanArgs = [
-                    `--load - extension=${safeDist} `,
-                    `--user - data - dir=${safeProfile} `,
+                    `--load-extension=${safeDist}`,
+                    `--user-data-dir=${safeProfile}`,
                     '--no-first-run',
                     '--no-default-browser-check',
                     '--disable-gpu',

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "ai-extension-preview",
-    "version": "0.1.12",
+    "version": "0.1.13",
     "description": "Local preview tool for AI Extension Builder",
     "type": "module",
     "bin": {
@@ -21,6 +21,7 @@
     "license": "MIT",
     "scripts": {
         "build": "shx rm -rf dist && tsc -b && shx chmod +x dist/index.js",
+        "prepublishOnly": "npm run build",
         "start": "tsx src/index.ts",
         "dev": "tsx watch src/index.ts",
         "preview": "node dist/index.js"
@@ -50,4 +51,4 @@
         "typescript": "^5.7.2",
         "vitest": "^4.0.16"
     }
-}
+}