@supaku/agentfactory-server 0.1.0

package/dist/src/redis.js

@@ -0,0 +1,343 @@
+import Redis from 'ioredis';
+const log = {
+    info: (msg, data) => console.log(`[redis] ${msg}`, data ? JSON.stringify(data) : ''),
+    warn: (msg, data) => console.warn(`[redis] ${msg}`, data ? JSON.stringify(data) : ''),
+    error: (msg, data) => console.error(`[redis] ${msg}`, data ? JSON.stringify(data) : ''),
+    debug: (_msg, _data) => { },
+};
+let _redis = null;
+/**
+ * Parse Redis URL using the modern URL class to avoid deprecated url.parse()
+ * Supports redis:// and rediss:// (TLS) protocols
+ */
+function parseRedisUrl(redisUrl) {
+    const url = new URL(redisUrl);
+    const options = {};
+    if (url.hostname) {
+        options.host = url.hostname;
+    }
+    if (url.port) {
+        options.port = parseInt(url.port, 10);
+    }
+    if (url.username && url.username !== 'default') {
+        options.username = decodeURIComponent(url.username);
+    }
+    if (url.password) {
+        options.password = decodeURIComponent(url.password);
+    }
+    // Database number from path (e.g., redis://host/0)
+    if (url.pathname && url.pathname.length > 1) {
+        const db = parseInt(url.pathname.slice(1), 10);
+        if (!isNaN(db)) {
+            options.db = db;
+        }
+    }
+    // Enable TLS for rediss:// protocol
+    if (url.protocol === 'rediss:') {
+        options.tls = {};
+    }
+    // Parse query parameters (e.g., ?family=6)
+    for (const [key, value] of url.searchParams) {
+        if (key === 'family') {
+            const family = parseInt(value, 10);
+            if (!isNaN(family)) {
+                options.family = family;
+            }
+        }
+    }
+    return options;
+}
+/**
+ * Check if Redis is configured via REDIS_URL
+ */
+export function isRedisConfigured() {
+    return !!process.env.REDIS_URL;
+}
+/**
+ * Get the shared Redis client instance
+ * Lazily initialized to avoid errors during build
+ */
+export function getRedisClient() {
+    if (!_redis) {
+        const redisUrl = process.env.REDIS_URL;
+        if (!redisUrl) {
+            throw new Error('REDIS_URL not set - Redis operations will fail');
+        }
+        const urlOptions = parseRedisUrl(redisUrl);
+        _redis = new Redis({
+            ...urlOptions,
+            maxRetriesPerRequest: 3,
+            lazyConnect: true,
+        });
+        _redis.on('error', (err) => {
+            log.error('Redis connection error', { error: err });
+        });
+        _redis.on('connect', () => {
+            log.info('Redis connected');
+        });
+    }
+    return _redis;
+}
+/**
+ * Disconnect Redis client (for graceful shutdown)
+ */
+export async function disconnectRedis() {
+    if (_redis) {
+        await _redis.quit();
+        _redis = null;
+        log.info('Redis disconnected');
+    }
+}
+/**
+ * Set a value with optional TTL (seconds)
+ */
+export async function redisSet(key, value, ttlSeconds) {
+    const redis = getRedisClient();
+    const serialized = JSON.stringify(value);
+    if (ttlSeconds) {
+        await redis.setex(key, ttlSeconds, serialized);
+    }
+    else {
+        await redis.set(key, serialized);
+    }
+}
+/**
+ * Get a typed value
+ */
+export async function redisGet(key) {
+    const redis = getRedisClient();
+    const value = await redis.get(key);
+    if (value === null) {
+        return null;
+    }
+    return JSON.parse(value);
+}
+/**
+ * Delete a key
+ * @returns number of keys deleted (0 or 1)
+ */
+export async function redisDel(key) {
+    const redis = getRedisClient();
+    return redis.del(key);
+}
+/**
+ * Check if a key exists
+ */
+export async function redisExists(key) {
+    const redis = getRedisClient();
+    const result = await redis.exists(key);
+    return result === 1;
+}
+/**
+ * Get keys matching a pattern
+ */
+export async function redisKeys(pattern) {
+    const redis = getRedisClient();
+    return redis.keys(pattern);
+}
+// ============================================
+// List Operations (for work queue)
+// ============================================
+/**
+ * Push value to the right of a list (RPUSH)
+ * @returns length of list after push
+ */
+export async function redisRPush(key, value) {
+    const redis = getRedisClient();
+    return redis.rpush(key, value);
+}
+/**
+ * Pop value from the left of a list (LPOP)
+ * @returns the popped value or null if list is empty
+ */
+export async function redisLPop(key) {
+    const redis = getRedisClient();
+    return redis.lpop(key);
+}
+/**
+ * Get a range of elements from a list (LRANGE)
+ * @param start - Start index (0-based, inclusive)
+ * @param stop - Stop index (inclusive, -1 for end)
+ */
+export async function redisLRange(key, start, stop) {
+    const redis = getRedisClient();
+    return redis.lrange(key, start, stop);
+}
+/**
+ * Get the length of a list (LLEN)
+ */
+export async function redisLLen(key) {
+    const redis = getRedisClient();
+    return redis.llen(key);
+}
+/**
+ * Remove elements from a list (LREM)
+ * @param count - Number of occurrences to remove (0 = all)
+ * @returns number of elements removed
+ */
+export async function redisLRem(key, count, value) {
+    const redis = getRedisClient();
+    return redis.lrem(key, count, value);
+}
+// ============================================
+// Set Operations (for worker sessions)
+// ============================================
+/**
+ * Add member to a set (SADD)
+ * @returns number of elements added (0 if already exists)
+ */
+export async function redisSAdd(key, member) {
+    const redis = getRedisClient();
+    return redis.sadd(key, member);
+}
+/**
+ * Remove member from a set (SREM)
+ * @returns number of elements removed
+ */
+export async function redisSRem(key, member) {
+    const redis = getRedisClient();
+    return redis.srem(key, member);
+}
+/**
+ * Get all members of a set (SMEMBERS)
+ */
+export async function redisSMembers(key) {
+    const redis = getRedisClient();
+    return redis.smembers(key);
+}
+/**
+ * Get the number of members in a set (SCARD)
+ */
+export async function redisSCard(key) {
+    const redis = getRedisClient();
+    return redis.scard(key);
+}
+// ============================================
+// Atomic Operations
+// ============================================
+/**
+ * Set a value only if key does not exist (SETNX)
+ * @returns true if key was set, false if it already existed
+ */
+export async function redisSetNX(key, value, ttlSeconds) {
+    const redis = getRedisClient();
+    if (ttlSeconds) {
+        // Use SET with NX and EX options for atomic set-if-not-exists with TTL
+        const result = await redis.set(key, value, 'EX', ttlSeconds, 'NX');
+        return result === 'OK';
+    }
+    else {
+        const result = await redis.setnx(key, value);
+        return result === 1;
+    }
+}
+/**
+ * Set TTL on an existing key (EXPIRE)
+ * @returns true if TTL was set, false if key doesn't exist
+ */
+export async function redisExpire(key, ttlSeconds) {
+    const redis = getRedisClient();
+    const result = await redis.expire(key, ttlSeconds);
+    return result === 1;
+}
+// ============================================
+// Sorted Set Operations (for priority queue)
+// ============================================
+/**
+ * Add member to a sorted set with score (ZADD)
+ * @returns number of elements added (0 if already exists, updates score)
+ */
+export async function redisZAdd(key, score, member) {
+    const redis = getRedisClient();
+    return redis.zadd(key, score, member);
+}
+/**
+ * Remove member from a sorted set (ZREM)
+ * @returns number of elements removed
+ */
+export async function redisZRem(key, member) {
+    const redis = getRedisClient();
+    return redis.zrem(key, member);
+}
+/**
+ * Get members from sorted set by score range (ZRANGEBYSCORE)
+ * Returns members with lowest scores first (highest priority)
+ * @param min - Minimum score (use '-inf' for no minimum)
+ * @param max - Maximum score (use '+inf' for no maximum)
+ * @param limit - Maximum number of results
+ */
+export async function redisZRangeByScore(key, min, max, limit) {
+    const redis = getRedisClient();
+    if (limit !== undefined) {
+        return redis.zrangebyscore(key, min, max, 'LIMIT', 0, limit);
+    }
+    return redis.zrangebyscore(key, min, max);
+}
+/**
+ * Get the number of members in a sorted set (ZCARD)
+ */
+export async function redisZCard(key) {
+    const redis = getRedisClient();
+    return redis.zcard(key);
+}
+/**
+ * Pop the member with the lowest score (ZPOPMIN)
+ * @returns [member, score] or null if set is empty
+ */
+export async function redisZPopMin(key) {
+    const redis = getRedisClient();
+    const result = await redis.zpopmin(key);
+    if (result && result.length >= 2) {
+        return { member: result[0], score: parseFloat(result[1]) };
+    }
+    return null;
+}
+// ============================================
+// Hash Operations (for work item lookup)
+// ============================================
+/**
+ * Set a field in a hash (HSET)
+ * @returns 1 if field is new, 0 if field existed
+ */
+export async function redisHSet(key, field, value) {
+    const redis = getRedisClient();
+    return redis.hset(key, field, value);
+}
+/**
+ * Get a field from a hash (HGET)
+ */
+export async function redisHGet(key, field) {
+    const redis = getRedisClient();
+    return redis.hget(key, field);
+}
+/**
+ * Delete a field from a hash (HDEL)
+ * @returns number of fields removed
+ */
+export async function redisHDel(key, field) {
+    const redis = getRedisClient();
+    return redis.hdel(key, field);
+}
+/**
+ * Get multiple fields from a hash (HMGET)
+ */
+export async function redisHMGet(key, fields) {
+    const redis = getRedisClient();
+    if (fields.length === 0)
+        return [];
+    return redis.hmget(key, ...fields);
+}
+/**
+ * Get all fields and values from a hash (HGETALL)
+ */
+export async function redisHGetAll(key) {
+    const redis = getRedisClient();
+    return redis.hgetall(key);
+}
+/**
+ * Get the number of fields in a hash (HLEN)
+ */
+export async function redisHLen(key) {
+    const redis = getRedisClient();
+    return redis.hlen(key);
+}

package/dist/src/session-storage.d.ts

@@ -0,0 +1,144 @@
+import type { AgentWorkType } from './types';
+/**
+ * Agent session status
+ * - pending: Queued, waiting for a worker to claim
+ * - claimed: Worker has claimed but not yet started
+ * - running: Agent is actively processing
+ * - finalizing: Agent work done, cleanup in progress (worktree removal, orchestrator teardown)
+ * - completed: Agent finished successfully (all cleanup done)
+ * - failed: Agent encountered an error
+ * - stopped: Agent was stopped by user
+ */
+export type AgentSessionStatus = 'pending' | 'claimed' | 'running' | 'finalizing' | 'completed' | 'failed' | 'stopped';
+/**
+ * Agent session state stored in Redis for distributed access
+ */
+export interface AgentSessionState {
+    /** Linear session ID (from webhook) */
+    linearSessionId: string;
+    /** Linear issue ID */
+    issueId: string;
+    /** Issue identifier (e.g., SUP-123) */
+    issueIdentifier?: string;
+    /** Claude CLI session ID for resuming with --resume */
+    claudeSessionId: string | null;
+    /** Git worktree path */
+    worktreePath: string;
+    /** Current agent status */
+    status: AgentSessionStatus;
+    /** Unix timestamp when session was created */
+    createdAt: number;
+    /** Unix timestamp of last update */
+    updatedAt: number;
+    /** Worker ID handling this session (null if pending) */
+    workerId?: string | null;
+    /** Unix timestamp when added to work queue */
+    queuedAt?: number | null;
+    /** Unix timestamp when claimed by worker */
+    claimedAt?: number | null;
+    /** Priority in queue (1-5, lower is higher priority) */
+    priority?: number;
+    /** Prompt context for the session */
+    promptContext?: string;
+    /** Linear organization ID for OAuth token lookup */
+    organizationId?: string;
+    /** Type of work: research, development, inflight, qa, acceptance, refinement (defaults to 'development') */
+    workType?: AgentWorkType;
+    /** Linear Agent ID handling this session */
+    agentId?: string;
+    /** Total cost in USD for this session */
+    totalCostUsd?: number;
+    /** Total input tokens consumed */
+    inputTokens?: number;
+    /** Total output tokens consumed */
+    outputTokens?: number;
+}
+/**
+ * Store agent session state in Redis
+ *
+ * @param linearSessionId - The Linear session ID from webhook
+ * @param state - The session state to store
+ */
+export declare function storeSessionState(linearSessionId: string, state: Omit<AgentSessionState, 'linearSessionId' | 'createdAt' | 'updatedAt'>): Promise<AgentSessionState>;
+/**
+ * Retrieve agent session state from Redis
+ *
+ * @param linearSessionId - The Linear session ID
+ * @returns The session state or null if not found
+ */
+export declare function getSessionState(linearSessionId: string): Promise<AgentSessionState | null>;
+/**
+ * Update the Claude session ID for a session
+ * Called when the Claude init event is received with the session ID
+ *
+ * @param linearSessionId - The Linear session ID
+ * @param claudeSessionId - The Claude CLI session ID
+ */
+export declare function updateClaudeSessionId(linearSessionId: string, claudeSessionId: string): Promise<boolean>;
+/**
+ * Update session status
+ *
+ * @param linearSessionId - The Linear session ID
+ * @param status - The new status
+ */
+export declare function updateSessionStatus(linearSessionId: string, status: AgentSessionState['status']): Promise<boolean>;
+/**
+ * Reset a session for re-queuing after orphan cleanup
+ * Clears workerId and resets status to pending so a new worker can claim it
+ *
+ * @param linearSessionId - The Linear session ID
+ */
+export declare function resetSessionForRequeue(linearSessionId: string): Promise<boolean>;
+/**
+ * Delete session state from KV
+ *
+ * @param linearSessionId - The Linear session ID
+ * @returns Whether the deletion was successful
+ */
+export declare function deleteSessionState(linearSessionId: string): Promise<boolean>;
+/**
+ * Get session state by issue ID
+ * Useful when we have the issue but not the session ID
+ *
+ * @param issueId - The Linear issue ID
+ * @returns The most recent session state for this issue or null
+ */
+export declare function getSessionStateByIssue(issueId: string): Promise<AgentSessionState | null>;
+/**
+ * Mark a session as claimed by a worker
+ *
+ * @param linearSessionId - The Linear session ID
+ * @param workerId - The worker claiming the session
+ */
+export declare function claimSession(linearSessionId: string, workerId: string): Promise<boolean>;
+/**
+ * Update session with worker info when work starts
+ *
+ * @param linearSessionId - The Linear session ID
+ * @param workerId - The worker processing the session
+ * @param worktreePath - Path to the git worktree
+ */
+export declare function startSession(linearSessionId: string, workerId: string, worktreePath: string): Promise<boolean>;
+/**
+ * Get all sessions from Redis
+ * For dashboard display
+ */
+export declare function getAllSessions(): Promise<AgentSessionState[]>;
+/**
+ * Get sessions by status
+ */
+export declare function getSessionsByStatus(status: AgentSessionStatus | AgentSessionStatus[]): Promise<AgentSessionState[]>;
+/**
+ * Transfer session ownership to a new worker
+ * Used when a worker re-registers after disconnection and gets a new ID
+ *
+ * @param linearSessionId - The Linear session ID
+ * @param newWorkerId - The new worker ID to assign
+ * @param oldWorkerId - The previous worker ID (for validation)
+ * @returns Whether the transfer was successful
+ */
+export declare function transferSessionOwnership(linearSessionId: string, newWorkerId: string, oldWorkerId: string): Promise<{
+    transferred: boolean;
+    reason?: string;
+}>;
+//# sourceMappingURL=session-storage.d.ts.map

package/dist/src/session-storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-storage.d.ts","sourceRoot":"","sources":["../../src/session-storage.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,SAAS,CAAA;AAS5C;;;;;;;;;GASG;AACH,MAAM,MAAM,kBAAkB,GAC1B,SAAS,GACT,SAAS,GACT,SAAS,GACT,YAAY,GACZ,WAAW,GACX,QAAQ,GACR,SAAS,CAAA;AAEb;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,uCAAuC;IACvC,eAAe,EAAE,MAAM,CAAA;IACvB,sBAAsB;IACtB,OAAO,EAAE,MAAM,CAAA;IACf,uCAAuC;IACvC,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,uDAAuD;IACvD,eAAe,EAAE,MAAM,GAAG,IAAI,CAAA;IAC9B,wBAAwB;IACxB,YAAY,EAAE,MAAM,CAAA;IACpB,2BAA2B;IAC3B,MAAM,EAAE,kBAAkB,CAAA;IAC1B,8CAA8C;IAC9C,SAAS,EAAE,MAAM,CAAA;IACjB,oCAAoC;IACpC,SAAS,EAAE,MAAM,CAAA;IAGjB,wDAAwD;IACxD,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IACxB,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IACxB,4CAA4C;IAC5C,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IACzB,wDAAwD;IACxD,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,qCAAqC;IACrC,aAAa,CAAC,EAAE,MAAM,CAAA;IAGtB,oDAAoD;IACpD,cAAc,CAAC,EAAE,MAAM,CAAA;IAGvB,4GAA4G;IAC5G,QAAQ,CAAC,EAAE,aAAa,CAAA;IAGxB,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAA;IAGhB,yCAAyC;IACzC,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,kCAAkC;IAClC,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,mCAAmC;IACnC,YAAY,CAAC,EAAE,MAAM,CAAA;CACtB;AAoBD;;;;;GAKG;AACH,wBAAsB,iBAAiB,CACrC,eAAe,EAAE,MAAM,EACvB,KAAK,EAAE,IAAI,CAAC,iBAAiB,EAAE,iBAAiB,GAAG,WAAW,GAAG,WAAW,CAAC,GAC5E,OAAO,CAAC,iBAAiB,CAAC,CAmC5B;AAED;;;;;GAKG;AACH,wBAAsB,eAAe,CACnC,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAkBnC;AAED;;;;;;GAMG;AACH,wBAAsB,qBAAqB,CACzC,eAAe,EAAE,MAAM,EACvB,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,OAAO,CAAC,CA0BlB;AAED;;;;;GAKG;AACH,wBAAsB,mBAAmB,CACvC,eAAe,EAAE,MAAM,EACvB,MAAM,EAAE,iBAAiB,CAAC,QAAQ,CAAC,GAClC,OAAO,CAAC,OAAO,CAAC,CA0BlB;AAED;;;;;GAKG;AACH,wBAAsB,sBAAsB,CAC1C,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,OAAO,CAAC,CA+BlB;AAED;;;;;GAKG;AACH,wBAAsB,kBAAkB,CAAC,eAAe,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAWlF;AAED;;;;;;GAMG;AACH,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAiBnC;AAMD;;;;;GAKG;AACH,wBAAsB,YAAY,CAChC,eAAe,EAAE,MAAM,EACvB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,OAAO,CAAC,CAmClB;AAED;;;;;;GAMG;AACH,wBAAsB,YAAY,CAChC,eAAe,EAAE,MAAM,EACvB,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,OAAO,CAAC,CA2BlB;AAED;;;GAGG;AACH,wBAAsB,cAAc,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAwBnE;AAED;;GAEG;AACH,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,kBAAkB,GAAG,kBAAkB,EAAE,GAChD,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAI9B;AAED;;;;;;;;GAQG;AACH,wBAAsB,wBAAwB,CAC5C,eAAe,EAAE,MAAM,EACvB,WAAW,EAAE,MAAM,EACnB,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC;IAAE,WAAW,EAAE,OAAO,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAyCpD"}