@teamvibe/poller 0.1.16 → 0.1.18

package/dist/auth-provider.js

@@ -1,3 +1,4 @@
+import { POLLER_VERSION } from './config.js';
 import { logger } from './logger.js';
 let currentCredentials = null;
 let currentConfig = null;
@@ -5,13 +6,15 @@ let refreshTimer = null;
 let heartbeatTimer = null;
 const REFRESH_INTERVAL_MS = 50 * 60 * 1000; // 50 minutes (before 1hr expiry)
 const HEARTBEAT_INTERVAL_MS = 60 * 1000; // 1 minute
-async function sendHeartbeat(apiUrl, token) {
+async function sendHeartbeat(apiUrl, token, version) {
     try {
         const response = await fetch(`${apiUrl}/heartbeat`, {
             method: 'POST',
             headers: {
                 Authorization: `Bearer ${token}`,
+                'Content-Type': 'application/json',
             },
+            body: JSON.stringify({ version }),
         });
         if (!response.ok) {
             logger.warn(`Heartbeat failed (${response.status})`);
@@ -44,8 +47,8 @@ export async function initAuth(apiUrl, token) {
     logger.info(`   SQS Queue: ${auth.config.sqsQueueUrl}`);
     logger.info(`   Sessions Table: ${auth.config.sessionsTable}`);
     // Send initial heartbeat and schedule periodic heartbeats
-    sendHeartbeat(apiUrl, token);
-    heartbeatTimer = setInterval(() => sendHeartbeat(apiUrl, token), HEARTBEAT_INTERVAL_MS);
+    sendHeartbeat(apiUrl, token, POLLER_VERSION);
+    heartbeatTimer = setInterval(() => sendHeartbeat(apiUrl, token, POLLER_VERSION), HEARTBEAT_INTERVAL_MS);
     // Schedule credential refresh
     refreshTimer = setInterval(async () => {
         try {

package/dist/brain-manager.d.ts

@@ -3,10 +3,7 @@ interface BrainConfig {
     gitRepoUrl: string;
     branch: string;
 }
-/**
- * Get or clone brain repo. Returns the working directory path.
- */
-export declare function getBrainPath(brain?: BrainConfig): Promise<string>;
+export declare function getBrainPath(brain?: BrainConfig, channelId?: string, workspaceId?: string): Promise<string>;
 /**
  * Ensure base brain repo is cloned and up to date.
  * Uses same cooldown pattern as channel brains.
@@ -20,7 +17,7 @@ export declare function getBaseBrainPath(): string;
  * Commit and push any changes in a brain repo.
  * Silently skips if there are no changes or no remote.
  */
-export declare function pushBrainChanges(brainDir: string, brainId: string): Promise<void>;
+export declare function pushBrainChanges(brainDir: string, brainId: string, workspaceId?: string): Promise<void>;
 /**
  * Ensure base paths exist
  */

package/dist/brain-manager.js

@@ -1,6 +1,6 @@
 import { exec } from 'child_process';
 import { promisify } from 'util';
-import { existsSync } from 'fs';
+import { existsSync, mkdirSync } from 'fs';
 import { join } from 'path';
 import { config } from './config.js';
 import { logger } from './logger.js';
@@ -10,9 +10,41 @@ const lastUpdateTimes = new Map();
 /**
  * Get or clone brain repo. Returns the working directory path.
  */
-export async function getBrainPath(brain) {
-    if (!brain?.gitRepoUrl)
+/**
+ * Report a brain sync event (pull/push) to the TeamVibe API.
+ * Fire-and-forget — errors are logged but not thrown.
+ */
+async function reportBrainSync(brainId, workspaceId, event) {
+    if (!config.TEAMVIBE_API_URL || !config.TEAMVIBE_POLLER_TOKEN)
+        return;
+    try {
+        const response = await fetch(`${config.TEAMVIBE_API_URL}/brain-sync`, {
+            method: 'POST',
+            headers: {
+                Authorization: `Bearer ${config.TEAMVIBE_POLLER_TOKEN}`,
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({ brainId, workspaceId, event }),
+        });
+        if (!response.ok) {
+            logger.warn(`Brain sync report failed (${response.status})`);
+        }
+    }
+    catch (error) {
+        logger.warn(`Brain sync report error: ${error instanceof Error ? error.message : error}`);
+    }
+}
+export async function getBrainPath(brain, channelId, workspaceId) {
+    if (!brain?.gitRepoUrl) {
+        if (channelId) {
+            const channelDir = join(config.BRAINS_PATH, `channel_${channelId}`);
+            if (!existsSync(channelDir)) {
+                mkdirSync(channelDir, { recursive: true });
+            }
+            return channelDir;
+        }
         return config.DEFAULT_BRAIN_PATH;
+    }
     const brainDir = join(config.BRAINS_PATH, brain.brainId);
     if (!existsSync(brainDir)) {
         logger.info(`Cloning brain ${brain.brainId} from ${brain.gitRepoUrl} (branch: ${brain.branch})...`);
@@ -30,9 +62,13 @@ export async function getBrainPath(brain) {
             }
         }
         logger.info(`Brain ${brain.brainId} cloned successfully`);
+        if (workspaceId)
+            reportBrainSync(brain.brainId, workspaceId, 'pulled');
     }
     else if (config.BRAIN_AUTO_UPDATE) {
         await updateBrain(brainDir, brain.brainId, brain.branch);
+        if (workspaceId)
+            reportBrainSync(brain.brainId, workspaceId, 'pulled');
     }
     return brainDir;
 }
@@ -85,7 +121,7 @@ export function getBaseBrainPath() {
  * Commit and push any changes in a brain repo.
  * Silently skips if there are no changes or no remote.
  */
-export async function pushBrainChanges(brainDir, brainId) {
+export async function pushBrainChanges(brainDir, brainId, workspaceId) {
     try {
         // Check if this is a git repo with a remote
         try {
@@ -110,6 +146,8 @@ export async function pushBrainChanges(brainDir, brainId) {
         await execAsync('git commit -m "auto: session update"', { cwd: brainDir });
         await execAsync('git push', { cwd: brainDir });
         logger.info(`Brain ${brainId} changes pushed successfully`);
+        if (workspaceId)
+            reportBrainSync(brainId, workspaceId, 'pushed');
     }
     catch (error) {
         const errorMessage = error instanceof Error ? error.message : String(error);
@@ -120,7 +158,6 @@ export async function pushBrainChanges(brainDir, brainId) {
  * Ensure base paths exist
  */
 export async function ensureDirectories() {
-    const { mkdirSync } = await import('fs');
     if (!existsSync(config.BRAINS_PATH)) {
         mkdirSync(config.BRAINS_PATH, { recursive: true });
     }

package/dist/cli/commands.js

@@ -10,7 +10,8 @@ Usage: poller [command]
 
 Commands:
   (no command)    Start the poller (default)
-  install         Install as a macOS launchd service (interactive)
+  install         Install as a macOS launchd service
+                  Flags: --token <token> --claude-token <token> [--api-url <url>] [--max-concurrent <n>]
   uninstall       Remove the launchd service
   update          Update to the latest version and restart
   start           Start the installed service

package/dist/cli/install.js

@@ -17,18 +17,37 @@ function tryGetClaudeSetupToken() {
     }
     return null;
 }
+function parseFlags(args) {
+    const flags = {};
+    for (let i = 0; i < args.length; i++) {
+        const arg = args[i];
+        const next = args[i + 1];
+        if (arg.startsWith('--') && next && !next.startsWith('--')) {
+            flags[arg.slice(2)] = next;
+            i++;
+        }
+    }
+    return flags;
+}
 export async function install() {
     if (process.platform !== 'darwin') {
         console.error('Error: Service installation is only supported on macOS.');
         process.exit(1);
     }
+    const flags = parseFlags(process.argv.slice(3));
+    const nonInteractive = !!(flags['token']);
     console.log('TeamVibe Poller - Service Installation\n');
     // Check if already installed
     if (fs.existsSync(PLIST_PATH)) {
-        const overwrite = await confirm('Service is already installed. Overwrite?', false);
-        if (!overwrite) {
-            console.log('Aborted.');
-            return;
+        if (nonInteractive) {
+            console.log('Service already installed, overwriting...');
+        }
+        else {
+            const overwrite = await confirm('Service is already installed. Overwrite?', false);
+            if (!overwrite) {
+                console.log('Aborted.');
+                return;
+            }
         }
         // Unload existing service
         try {
@@ -41,34 +60,70 @@ export async function install() {
     // Load existing .env values as defaults
     const existingEnv = loadExistingEnv();
     // Collect configuration
-    console.log('Step 1: Poller Token\n');
-    console.log('  Get your token from the TeamVibe dashboard (Pollers > Setup Instructions).');
-    console.log('  If you lost it, use "Regenerate Token" from the poller menu.\n');
-    const token = await prompt('TEAMVIBE_POLLER_TOKEN', existingEnv['TEAMVIBE_POLLER_TOKEN']);
+    let token;
+    if (flags['token']) {
+        token = flags['token'];
+        console.log(`Using provided token: ${token.slice(0, 8)}...`);
+    }
+    else {
+        console.log('Step 1: Poller Token\n');
+        console.log('  Get your token from the TeamVibe dashboard (Pollers > Setup Instructions).');
+        console.log('  If you lost it, use "Regenerate Token" from the poller menu.\n');
+        token = await prompt('TEAMVIBE_POLLER_TOKEN', existingEnv['TEAMVIBE_POLLER_TOKEN']);
+    }
     if (!token) {
         console.error('\nError: TEAMVIBE_POLLER_TOKEN is required.');
         process.exit(1);
     }
-    console.log('\nStep 2: Claude Code Authentication\n');
-    console.log('  The poller needs Claude Code credentials to run AI sessions.');
-    console.log('  Run `claude setup-token` in a terminal to generate a token.\n');
-    let claudeOAuthToken = existingEnv['CLAUDE_CODE_OAUTH_TOKEN'] || '';
-    const autoToken = tryGetClaudeSetupToken();
-    if (autoToken) {
-        console.log('  Auto-detected Claude setup token.');
-        const useAuto = await confirm('  Use the detected token?');
-        if (useAuto) {
-            claudeOAuthToken = autoToken;
+    let claudeOAuthToken = flags['claude-token'] || existingEnv['CLAUDE_CODE_OAUTH_TOKEN'] || '';
+    if (!claudeOAuthToken) {
+        if (nonInteractive) {
+            console.log('\nDetecting Claude Code token...');
+        }
+        else {
+            console.log('\nStep 2: Claude Code Authentication\n');
+            console.log('  The poller needs Claude Code credentials to run AI sessions.');
+            console.log('  Run `claude setup-token` in a terminal to generate a token.\n');
+        }
+        const autoToken = tryGetClaudeSetupToken();
+        if (autoToken) {
+            if (nonInteractive) {
+                claudeOAuthToken = autoToken;
+                console.log('Auto-detected Claude setup token.');
+            }
+            else {
+                console.log('  Auto-detected Claude setup token.');
+                const useAuto = await confirm('  Use the detected token?');
+                if (useAuto) {
+                    claudeOAuthToken = autoToken;
+                }
+            }
+        }
+        if (!claudeOAuthToken && !nonInteractive) {
+            claudeOAuthToken = await prompt('CLAUDE_CODE_OAUTH_TOKEN');
         }
     }
-    if (!claudeOAuthToken) {
-        claudeOAuthToken = await prompt('CLAUDE_CODE_OAUTH_TOKEN');
+    else if (flags['claude-token']) {
+        console.log('Using provided Claude Code token.');
+    }
+    let apiUrl;
+    let maxConcurrent;
+    if (nonInteractive) {
+        apiUrl = flags['api-url'] || existingEnv['TEAMVIBE_API_URL'] || 'https://poller.api.teamvibe.ai';
+        maxConcurrent = flags['max-concurrent'] || existingEnv['MAX_CONCURRENT_SESSIONS'] || '5';
+    }
+    else {
+        console.log('\nStep 3: Optional Settings\n');
+        apiUrl = await prompt('TEAMVIBE_API_URL', existingEnv['TEAMVIBE_API_URL'] || 'https://poller.api.teamvibe.ai');
+        maxConcurrent = await prompt('MAX_CONCURRENT_SESSIONS', existingEnv['MAX_CONCURRENT_SESSIONS'] || '5');
     }
-    console.log('\nStep 3: Optional Settings\n');
-    const apiUrl = await prompt('TEAMVIBE_API_URL', existingEnv['TEAMVIBE_API_URL'] || 'https://poller.api.teamvibe.ai');
-    const maxConcurrent = await prompt('MAX_CONCURRENT_SESSIONS', existingEnv['MAX_CONCURRENT_SESSIONS'] || '5');
     // Resolve claude CLI path
-    console.log('\nStep 4: Detecting paths\n');
+    if (!nonInteractive) {
+        console.log('\nStep 4: Detecting paths\n');
+    }
+    else {
+        console.log('\nDetecting paths...');
+    }
     let claudePath = '';
     try {
         claudePath = execSync('which claude', { encoding: 'utf-8' }).trim();
@@ -77,6 +132,10 @@ export async function install() {
         // Not found
     }
     if (!claudePath) {
+        if (nonInteractive) {
+            console.error('Error: claude CLI not found in PATH.');
+            process.exit(1);
+        }
         claudePath = await prompt('Path to claude CLI (not found in PATH)');
         if (!claudePath) {
             console.error('Error: claude CLI path is required.');
@@ -106,6 +165,9 @@ export async function install() {
         catch {
             console.error('\n  Warning: Could not install globally. The service may not start after reboot.');
             console.error('  Run `npm install -g @teamvibe/poller` manually, then re-run `poller install`.');
+            if (nonInteractive) {
+                return;
+            }
             const cont = await confirm('  Continue anyway?', false);
             if (!cont)
                 return;
@@ -128,7 +190,7 @@ export async function install() {
         envLines.push(`CLAUDE_CODE_OAUTH_TOKEN=${claudeOAuthToken}`);
     }
     let writeEnv = true;
-    if (fs.existsSync(envPath)) {
+    if (fs.existsSync(envPath) && !nonInteractive) {
         writeEnv = await confirm('\n.env file already exists. Overwrite?', false);
         if (!writeEnv) {
             console.log('Keeping existing .env file.');

package/dist/config.d.ts

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+export declare const POLLER_VERSION: string;
 declare const configSchema: z.ZodObject<{
     AWS_REGION: z.ZodDefault<z.ZodString>;
     SQS_QUEUE_URL: z.ZodOptional<z.ZodString>;

package/dist/config.js

@@ -1,5 +1,9 @@
+import { createRequire } from 'module';
 import { z } from 'zod';
 import { config as dotenvConfig } from 'dotenv';
+const require = createRequire(import.meta.url);
+const pkg = require('../package.json');
+export const POLLER_VERSION = pkg.version;
 dotenvConfig();
 const DEFAULT_DATA_DIR = `${process.env['HOME']}/.teamvibe`;
 const configSchema = z.object({

package/dist/poller.js

@@ -82,7 +82,7 @@ async function processMessage(received) {
     // Get brain path for this channel
     let kbPath;
     try {
-        kbPath = await getBrainPath(queueMessage.teamvibe.brain);
+        kbPath = await getBrainPath(queueMessage.teamvibe.brain, queueMessage.response_context.slack?.channel, queueMessage.teamvibe.workspaceId);
     }
     catch (error) {
         const errorMessage = error instanceof Error ? error.message : String(error);
@@ -139,7 +139,7 @@ async function processMessage(received) {
             sessionLog.info('Claude Code completed successfully');
             // Push any changes in the channel brain repo
             if (queueMessage.teamvibe.brain?.brainId) {
-                await pushBrainChanges(kbPath, queueMessage.teamvibe.brain.brainId);
+                await pushBrainChanges(kbPath, queueMessage.teamvibe.brain.brainId, queueMessage.teamvibe.workspaceId);
             }
             if (lockToken) {
                 const lastMessageTs = queueMessage.response_context.slack?.message_ts;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teamvibe/poller",
-  "version": "0.1.16",
+  "version": "0.1.18",
   "type": "module",
   "main": "./dist/index.js",
   "bin": {