@teamvibe/poller 0.2.1 → 0.2.3

package/dist/cli/doctor.js

@@ -1,17 +1,23 @@
 import * as fs from 'fs';
 import * as path from 'path';
 import { execSync } from 'child_process';
-import { PLIST_PATH, SERVICE_LABEL, TEAMVIBE_DIR, LOGS_DIR } from './plist.js';
+import { PLIST_PATH, SERVICE_LABEL, TEAMVIBE_DIR, LOGS_DIR, isContainerRuntime } from './plist.js';
+function ok(name, message) {
+    return { name, status: 'ok', message };
+}
+function fail(name, message) {
+    return { name, status: 'fail', message };
+}
+function skipped(name, reason) {
+    return { name, status: 'skipped', message: `Skipped (${reason})` };
+}
 function check(name, fn) {
     try {
         const result = fn();
-        if (result === true) {
-            return { name, ok: true, message: 'OK' };
-        }
-        return { name, ok: true, message: result };
+        return ok(name, result === true ? 'OK' : result);
     }
     catch (error) {
-        return { name, ok: false, message: error instanceof Error ? error.message : String(error) };
+        return fail(name, error instanceof Error ? error.message : String(error));
     }
 }
 function checkNodeVersion() {
@@ -24,7 +30,9 @@ function checkNodeVersion() {
         return `${version} (>= v18)`;
     });
 }
-function checkTeamVibeDir() {
+function checkTeamVibeDir(container) {
+    if (container)
+        return skipped('TeamVibe directory', 'container');
     return check('TeamVibe directory', () => {
         if (!fs.existsSync(TEAMVIBE_DIR)) {
             throw new Error(`${TEAMVIBE_DIR} does not exist. Run 'poller install' first.`);
@@ -32,7 +40,16 @@ function checkTeamVibeDir() {
         return TEAMVIBE_DIR;
     });
 }
-function checkEnvFile() {
+function checkEnvFile(container) {
+    if (container) {
+        // In containers, check process.env instead of .env file
+        const hasToken = !!process.env['TEAMVIBE_POLLER_TOKEN'];
+        const hasDirect = !!(process.env['SQS_QUEUE_URL'] && process.env['SESSIONS_TABLE']);
+        if (!hasToken && !hasDirect) {
+            return fail('Environment config', 'Missing auth env vars (TEAMVIBE_POLLER_TOKEN or SQS_QUEUE_URL + SESSIONS_TABLE)');
+        }
+        return ok('Environment config', hasToken ? 'Token mode (env vars)' : 'Direct mode (env vars)');
+    }
     return check('Environment config', () => {
         const envPath = path.join(TEAMVIBE_DIR, '.env');
         if (!fs.existsSync(envPath)) {
@@ -48,10 +65,9 @@ function checkEnvFile() {
     });
 }
 function checkLaunchdPlist() {
+    if (process.platform !== 'darwin')
+        return skipped('Launchd plist', 'not macOS');
     return check('Launchd plist', () => {
-        if (process.platform !== 'darwin') {
-            return 'Skipped (not macOS)';
-        }
         if (!fs.existsSync(PLIST_PATH)) {
             throw new Error(`Plist not found at ${PLIST_PATH}. Run 'poller install'.`);
         }
@@ -66,10 +82,9 @@ function checkLaunchdPlist() {
     });
 }
 function checkServiceRunning() {
+    if (process.platform !== 'darwin')
+        return skipped('Service status', 'not macOS');
     return check('Service status', () => {
-        if (process.platform !== 'darwin') {
-            return 'Skipped (not macOS)';
-        }
         try {
             const output = execSync(`launchctl list | grep ${SERVICE_LABEL}`, { encoding: 'utf-8' });
             if (output.trim()) {
@@ -91,7 +106,9 @@ function checkServiceRunning() {
         }
     });
 }
-function checkLogsDir() {
+function checkLogsDir(container) {
+    if (container)
+        return skipped('Logs directory', 'container uses stdout');
     return check('Logs directory', () => {
         if (!fs.existsSync(LOGS_DIR)) {
             throw new Error(`${LOGS_DIR} does not exist`);
@@ -140,48 +157,55 @@ function checkBrainsDir() {
         return `${brains.length} brain(s) cloned`;
     });
 }
-function checkDiskSpace() {
+function checkDiskSpace(container) {
     return check('Disk space', () => {
-        try {
-            const output = execSync(`df -h "${TEAMVIBE_DIR}" | tail -1`, { encoding: 'utf-8' }).trim();
-            const parts = output.split(/\s+/);
-            const available = parts[3];
-            const capacityStr = parts[4]?.replace('%', '');
-            const capacity = parseInt(capacityStr || '0', 10);
-            if (capacity > 95) {
-                throw new Error(`Disk ${capacity}% full — only ${available} available`);
-            }
-            if (capacity > 85) {
-                return `Warning: ${capacity}% used (${available} available)`;
-            }
-            return `${available} available (${capacity}% used)`;
-        }
-        catch (error) {
-            if (error instanceof Error && error.message.includes('Disk'))
-                throw error;
-            throw new Error('Could not check disk space');
-        }
+        // Use TEAMVIBE_DIR for installed setups, fall back to / for containers or when dir doesn't exist
+        const dfTarget = !container && fs.existsSync(TEAMVIBE_DIR) ? TEAMVIBE_DIR : '/';
+        const output = execSync(`df -h "${dfTarget}" | tail -1`, { encoding: 'utf-8' }).trim();
+        const parts = output.split(/\s+/);
+        const available = parts[3];
+        const capacityStr = parts[4]?.replace('%', '');
+        const capacity = parseInt(capacityStr || '0', 10);
+        if (capacity > 95) {
+            throw new Error(`Disk ${capacity}% full — only ${available} available`);
+        }
+        if (capacity > 85) {
+            return `Warning: ${capacity}% used (${available} available)`;
+        }
+        return `${available} available (${capacity}% used)`;
     });
 }
 export async function doctor() {
-    console.log('TeamVibe Poller - Health Check\n');
+    const container = isContainerRuntime();
+    const mode = container ? 'container' : 'installed';
+    console.log(`TeamVibe Poller - Health Check (${mode})\n`);
     const results = [
         checkNodeVersion(),
-        checkTeamVibeDir(),
-        checkEnvFile(),
+        checkTeamVibeDir(container),
+        checkEnvFile(container),
         checkLaunchdPlist(),
         checkServiceRunning(),
-        checkLogsDir(),
+        checkLogsDir(container),
         checkClaudeCli(),
         checkBrainsDir(),
-        checkDiskSpace(),
+        checkDiskSpace(container),
     ];
     let hasFailure = false;
     for (const result of results) {
-        const icon = result.ok ? '\x1b[32m✓\x1b[0m' : '\x1b[31m✗\x1b[0m';
+        let icon;
+        switch (result.status) {
+            case 'ok':
+                icon = '\x1b[32m✓\x1b[0m';
+                break;
+            case 'skipped':
+                icon = '\x1b[33m–\x1b[0m';
+                break;
+            case 'fail':
+                icon = '\x1b[31m✗\x1b[0m';
+                hasFailure = true;
+                break;
+        }
         console.log(`  ${icon} ${result.name}: ${result.message}`);
-        if (!result.ok)
-            hasFailure = true;
     }
     console.log('');
     if (hasFailure) {

package/dist/cli/plist.d.ts

@@ -2,4 +2,9 @@ export declare const SERVICE_LABEL = "ai.teamvibe.poller";
 export declare const PLIST_PATH: string;
 export declare const TEAMVIBE_DIR: string;
 export declare const LOGS_DIR: string;
+/**
+ * Detect container runtime (Railway, Docker, etc.) where `poller install` was never run.
+ * In containers, env vars are injected directly — no ~/.teamvibe directory exists.
+ */
+export declare function isContainerRuntime(): boolean;
 export declare function generatePlist(nodePath: string, pollerPath: string, envVars?: Record<string, string>): string;

package/dist/cli/plist.js

@@ -1,9 +1,27 @@
+import * as fs from 'fs';
 import * as path from 'path';
 import * as os from 'os';
 export const SERVICE_LABEL = 'ai.teamvibe.poller';
 export const PLIST_PATH = path.join(os.homedir(), 'Library', 'LaunchAgents', `${SERVICE_LABEL}.plist`);
 export const TEAMVIBE_DIR = path.join(os.homedir(), '.teamvibe');
 export const LOGS_DIR = path.join(TEAMVIBE_DIR, 'logs');
+/**
+ * Detect container runtime (Railway, Docker, etc.) where `poller install` was never run.
+ * In containers, env vars are injected directly — no ~/.teamvibe directory exists.
+ */
+export function isContainerRuntime() {
+    // Railway-specific env vars
+    if (process.env['RAILWAY_ENVIRONMENT'] || process.env['RAILWAY_SERVICE_ID'])
+        return true;
+    // Generic container signals
+    if (process.env['CONTAINER'] === 'true' || process.env['DOCKER'] === 'true')
+        return true;
+    // Fallback heuristic: has auth env vars but no ~/.teamvibe directory
+    const hasEnvAuth = !!(process.env['TEAMVIBE_POLLER_TOKEN'] || (process.env['SQS_QUEUE_URL'] && process.env['SESSIONS_TABLE']));
+    if (hasEnvAuth && !fs.existsSync(TEAMVIBE_DIR))
+        return true;
+    return false;
+}
 export function generatePlist(nodePath, pollerPath, envVars) {
     let envSection = '';
     if (envVars && Object.keys(envVars).length > 0) {

package/dist/logger.js

@@ -14,8 +14,8 @@ if (!existsSync(SESSIONS_DIR)) {
 function formatDate(date) {
     return date.toISOString();
 }
-function writeToFile(level, message, ...args) {
-    const timestamp = formatDate(new Date());
+const now = () => formatDate(new Date());
+function writeToFile(level, message, timestamp, ...args) {
     const formattedArgs = args
         .map((arg) => (typeof arg === 'object' ? JSON.stringify(arg) : String(arg)))
         .join(' ');
@@ -35,8 +35,7 @@ export class SessionLogger {
         const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
         this.logFile = join(SESSIONS_DIR, `${timestamp}_${sessionId}.log`);
     }
-    write(level, message) {
-        const timestamp = formatDate(new Date());
+    write(level, message, timestamp) {
         const logLine = `[${timestamp}] [${level}] ${message}\n`;
         try {
             appendFileSync(this.logFile, logLine);
@@ -46,27 +45,30 @@ export class SessionLogger {
         }
     }
     info(message) {
-        console.log(`[${this.sessionId}] ${message}`);
-        this.write('INFO', message);
-        writeToFile('INFO', `[${this.sessionId}] ${message}`);
+        const ts = now();
+        console.log(`[${ts}] [${this.sessionId}] ${message}`);
+        this.write('INFO', message, ts);
+        writeToFile('INFO', `[${this.sessionId}] ${message}`, ts);
     }
     error(message) {
-        console.error(`[${this.sessionId}] ${message}`);
-        this.write('ERROR', message);
-        writeToFile('ERROR', `[${this.sessionId}] ${message}`);
+        const ts = now();
+        console.error(`[${ts}] [${this.sessionId}] ${message}`);
+        this.write('ERROR', message, ts);
+        writeToFile('ERROR', `[${this.sessionId}] ${message}`, ts);
     }
     claude(stream, message) {
+        const ts = now();
         const prefix = `[Claude ${stream}]`;
         if (stream === 'stderr') {
-            console.error(`[${this.sessionId}] ${prefix}`, message);
+            console.error(`[${ts}] [${this.sessionId}] ${prefix}`, message);
         }
         else {
-            console.log(`[${this.sessionId}] ${prefix}`, message);
+            console.log(`[${ts}] [${this.sessionId}] ${prefix}`, message);
         }
-        this.write(`CLAUDE_${stream.toUpperCase()}`, message);
+        this.write(`CLAUDE_${stream.toUpperCase()}`, message, ts);
     }
     logPrompt(prompt) {
-        this.write('PROMPT', `\n${'='.repeat(50)}\n${prompt}\n${'='.repeat(50)}`);
+        this.write('PROMPT', `\n${'='.repeat(50)}\n${prompt}\n${'='.repeat(50)}`, now());
     }
     getLogFile() {
         return this.logFile;
@@ -74,22 +76,26 @@ export class SessionLogger {
 }
 export const logger = {
     info(message, ...args) {
-        console.log(message, ...args);
-        writeToFile('INFO', message, ...args);
+        const ts = now();
+        console.log(`[${ts}] [INFO] ${message}`, ...args);
+        writeToFile('INFO', message, ts, ...args);
     },
     error(message, ...args) {
-        console.error(message, ...args);
-        writeToFile('ERROR', message, ...args);
+        const ts = now();
+        console.error(`[${ts}] [ERROR] ${message}`, ...args);
+        writeToFile('ERROR', message, ts, ...args);
     },
     warn(message, ...args) {
-        console.warn(message, ...args);
-        writeToFile('WARN', message, ...args);
+        const ts = now();
+        console.warn(`[${ts}] [WARN] ${message}`, ...args);
+        writeToFile('WARN', message, ts, ...args);
     },
     debug(message, ...args) {
+        const ts = now();
         if (process.env['DEBUG']) {
-            console.debug(message, ...args);
+            console.debug(`[${ts}] [DEBUG] ${message}`, ...args);
         }
-        writeToFile('DEBUG', message, ...args);
+        writeToFile('DEBUG', message, ts, ...args);
     },
     createSession(sessionId) {
         return new SessionLogger(sessionId);

package/package.json

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

package/dist/control-handler.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/control-handler.test.js

@@ -1,74 +0,0 @@
-import { describe, test, expect } from 'bun:test';
-import { isControlMessage } from './control-handler.js';
-import { classifyUpdate } from './auto-update.js';
-describe('isControlMessage', () => {
-    test('valid restart message', () => {
-        expect(isControlMessage({ source: 'control', action: 'restart' })).toBe(true);
-    });
-    test('valid update message', () => {
-        expect(isControlMessage({ source: 'control', action: 'update' })).toBe(true);
-    });
-    test('valid update with targetVersion', () => {
-        expect(isControlMessage({ source: 'control', action: 'update', targetVersion: '1.2.3' })).toBe(true);
-    });
-    test('rejects non-control source', () => {
-        expect(isControlMessage({ source: 'slack', action: 'restart' })).toBe(false);
-    });
-    test('rejects unknown action', () => {
-        expect(isControlMessage({ source: 'control', action: 'delete' })).toBe(false);
-    });
-    test('rejects null', () => {
-        expect(isControlMessage(null)).toBe(false);
-    });
-    test('rejects undefined', () => {
-        expect(isControlMessage(undefined)).toBe(false);
-    });
-    test('rejects string', () => {
-        expect(isControlMessage('control')).toBe(false);
-    });
-    test('rejects empty object', () => {
-        expect(isControlMessage({})).toBe(false);
-    });
-    test('rejects regular queue message', () => {
-        expect(isControlMessage({
-            id: '123',
-            source: 'cron',
-            thread_id: 'thread1',
-            sender: { id: 'user1', name: 'Test' },
-            text: 'hello',
-        })).toBe(false);
-    });
-});
-describe('classifyUpdate', () => {
-    test('detects patch update', () => {
-        expect(classifyUpdate('1.2.3', '1.2.4')).toBe('patch');
-    });
-    test('detects minor update', () => {
-        expect(classifyUpdate('1.2.3', '1.3.0')).toBe('minor');
-    });
-    test('detects major update', () => {
-        expect(classifyUpdate('1.2.3', '2.0.0')).toBe('major');
-    });
-    test('returns null for same version', () => {
-        expect(classifyUpdate('1.2.3', '1.2.3')).toBeNull();
-    });
-    test('returns null for downgrade (patch)', () => {
-        expect(classifyUpdate('1.2.4', '1.2.3')).toBeNull();
-    });
-    test('returns null for downgrade (minor)', () => {
-        expect(classifyUpdate('1.3.0', '1.2.5')).toBeNull();
-    });
-    test('returns null for downgrade (major)', () => {
-        expect(classifyUpdate('2.0.0', '1.9.9')).toBeNull();
-    });
-    test('handles zero versions', () => {
-        expect(classifyUpdate('0.0.1', '0.0.2')).toBe('patch');
-        expect(classifyUpdate('0.1.0', '0.2.0')).toBe('minor');
-    });
-    test('minor update with different patch', () => {
-        expect(classifyUpdate('1.2.3', '1.3.5')).toBe('minor');
-    });
-    test('major update with different minor/patch', () => {
-        expect(classifyUpdate('1.2.3', '2.1.0')).toBe('major');
-    });
-});