pi-harness-runtime 0.5.0 → 0.6.3-beta.13

package/packages/session/dist/policy.js

@@ -0,0 +1,291 @@
+/**
+ * Session Manager - Policy Engine
+ *
+ * Session policy enforcement and budget tracking.
+ */
+export class RateLimiter {
+    entries = new Map();
+    requestsPerMinute;
+    constructor(requestsPerMinute) {
+        this.requestsPerMinute = requestsPerMinute;
+    }
+    /**
+     * Check if request is allowed
+     */
+    canMakeRequest(key) {
+        const now = Date.now();
+        const entry = this.entries.get(key);
+        if (!entry || now >= entry.resetAt) {
+            return true;
+        }
+        return entry.count < this.requestsPerMinute;
+    }
+    /**
+     * Record a request
+     */
+    recordRequest(key) {
+        const now = Date.now();
+        const entry = this.entries.get(key);
+        if (!entry || now >= entry.resetAt) {
+            // Start new window
+            this.entries.set(key, {
+                count: 1,
+                resetAt: now + 60000, // 1 minute
+            });
+        }
+        else {
+            entry.count++;
+        }
+    }
+    /**
+     * Get remaining requests
+     */
+    getRemaining(key) {
+        const entry = this.entries.get(key);
+        if (!entry || Date.now() >= entry.resetAt) {
+            return this.requestsPerMinute;
+        }
+        return Math.max(0, this.requestsPerMinute - entry.count);
+    }
+    /**
+     * Clear expired entries
+     */
+    clearExpired() {
+        const now = Date.now();
+        for (const [key, entry] of this.entries) {
+            if (now >= entry.resetAt) {
+                this.entries.delete(key);
+            }
+        }
+    }
+}
+export class BudgetTracker {
+    entries = new Map();
+    /**
+     * Set budget for a key
+     */
+    setBudget(key, limit, resetAtMs) {
+        this.entries.set(key, {
+            spent: 0,
+            limit,
+            resetAt: Date.now() + resetAtMs,
+        });
+    }
+    /**
+     * Check if budget allows spending
+     */
+    canSpend(key, amount) {
+        const entry = this.entries.get(key);
+        if (!entry)
+            return true;
+        // Check if reset time passed
+        if (Date.now() >= entry.resetAt) {
+            return true;
+        }
+        return entry.spent + amount <= entry.limit;
+    }
+    /**
+     * Record spending
+     */
+    recordSpend(key, amount) {
+        const entry = this.entries.get(key);
+        if (!entry)
+            return;
+        // Check if reset time passed
+        if (Date.now() >= entry.resetAt) {
+            entry.spent = amount;
+            entry.resetAt = Date.now() + (entry.resetAt - Date.now());
+        }
+        else {
+            entry.spent += amount;
+        }
+    }
+    /**
+     * Get remaining budget
+     */
+    getRemaining(key) {
+        const entry = this.entries.get(key);
+        if (!entry)
+            return null;
+        if (Date.now() >= entry.resetAt) {
+            return entry.limit;
+        }
+        return Math.max(0, entry.limit - entry.spent);
+    }
+    /**
+     * Get reset time
+     */
+    getResetAt(key) {
+        const entry = this.entries.get(key);
+        if (!entry)
+            return null;
+        return entry.resetAt;
+    }
+}
+// ─── Policy Engine ─────────────────────────────────────────────────────────
+export class PolicyEngine {
+    config;
+    rateLimiter;
+    budgetTracker;
+    sessionStates = new Map();
+    constructor(config = {}) {
+        this.config = {
+            maxRequestsPerMinute: config.maxRequestsPerMinute ?? 60,
+            maxTokensPerDay: config.maxTokensPerDay ?? 1000000,
+            maxCostPerSession: config.maxCostPerSession ?? 100,
+            maxConcurrentSessions: config.maxConcurrentSessions ?? 10,
+            sessionBudget: config.sessionBudget ?? 50,
+        };
+        this.rateLimiter = new RateLimiter(this.config.maxRequestsPerMinute);
+        this.budgetTracker = new BudgetTracker();
+    }
+    /**
+     * Get policy state for a session
+     */
+    getPolicyState(sessionId) {
+        let state = this.sessionStates.get(sessionId);
+        if (!state) {
+            state = {
+                rateLimitRemaining: this.config.maxRequestsPerMinute ?? 60,
+                suspended: false,
+            };
+            this.sessionStates.set(sessionId, state);
+        }
+        return state;
+    }
+    /**
+     * Check if action is allowed
+     */
+    canProceed(sessionId, action) {
+        const state = this.getPolicyState(sessionId);
+        // Check if suspended
+        if (state.suspended) {
+            return false;
+        }
+        // Check rate limit
+        if (!this.rateLimiter.canMakeRequest(sessionId)) {
+            return false;
+        }
+        // Check budget
+        if (state.budgetRemaining !== undefined && state.budgetRemaining <= 0) {
+            return false;
+        }
+        return true;
+    }
+    /**
+     * Record an action
+     */
+    recordAction(sessionId, action) {
+        // Record rate limit
+        this.rateLimiter.recordRequest(sessionId);
+        // Update state
+        const state = this.getPolicyState(sessionId);
+        state.rateLimitRemaining = this.rateLimiter.getRemaining(sessionId);
+        // Update rate limit reset time
+        const resetAt = Date.now() + 60000;
+        state.rateLimitResetAt = new Date(resetAt).toISOString();
+    }
+    /**
+     * Record token usage
+     */
+    recordTokenUsage(sessionId, usage) {
+        const state = this.getPolicyState(sessionId);
+        if (this.config.sessionBudget) {
+            // Convert cost to cents for tracking
+            const costInCents = Math.round(usage.totalCost * 100);
+            this.budgetTracker.recordSpend(sessionId, costInCents);
+            state.budgetRemaining = this.budgetTracker.getRemaining(sessionId)
+                ? this.budgetTracker.getRemaining(sessionId) / 100
+                : undefined;
+            const resetAt = this.budgetTracker.getResetAt(sessionId);
+            if (resetAt) {
+                state.budgetResetAt = new Date(resetAt).toISOString();
+            }
+        }
+    }
+    /**
+     * Suspend a session
+     */
+    suspend(sessionId, reason) {
+        const state = this.getPolicyState(sessionId);
+        state.suspended = true;
+        state.suspensionReason = reason;
+    }
+    /**
+     * Resume a session
+     */
+    resume(sessionId) {
+        const state = this.getPolicyState(sessionId);
+        state.suspended = false;
+        state.suspensionReason = undefined;
+    }
+    /**
+     * Reset session policy state
+     */
+    resetState(sessionId) {
+        this.sessionStates.delete(sessionId);
+    }
+    /**
+     * Set session budget
+     */
+    setSessionBudget(sessionId, budget, periodMs) {
+        const budgetInCents = Math.round(budget * 100);
+        this.budgetTracker.setBudget(sessionId, budgetInCents, periodMs);
+        const state = this.getPolicyState(sessionId);
+        state.budgetRemaining = budget;
+        state.budgetResetAt = new Date(Date.now() + periodMs).toISOString();
+    }
+    /**
+     * Get violation type
+     */
+    getViolationType(sessionId) {
+        const state = this.getPolicyState(sessionId);
+        if (state.suspended) {
+            return `suspended: ${state.suspensionReason}`;
+        }
+        if (state.rateLimitRemaining <= 0) {
+            return "rate_limit_exceeded";
+        }
+        if (state.budgetRemaining !== undefined && state.budgetRemaining <= 0) {
+            return "budget_exceeded";
+        }
+        return null;
+    }
+    /**
+     * Check if session exceeds cost limit
+     */
+    exceedsCostLimit(sessionId, additionalCost) {
+        const state = this.getPolicyState(sessionId);
+        if (this.config.maxCostPerSession && state.budgetRemaining !== undefined) {
+            return state.budgetRemaining - additionalCost < 0;
+        }
+        return false;
+    }
+    /**
+     * Get rate limit info
+     */
+    getRateLimitInfo(sessionId) {
+        const remaining = this.rateLimiter.getRemaining(sessionId);
+        if (remaining === this.config.maxRequestsPerMinute) {
+            return null;
+        }
+        return {
+            remaining,
+            resetAt: new Date(Date.now() + 60000).toISOString(),
+        };
+    }
+    /**
+     * Clean up expired state
+     */
+    cleanup() {
+        this.rateLimiter.clearExpired();
+    }
+}
+// ─── Factory Function ────────────────────────────────────────────────────
+/**
+ * Create a policy engine
+ */
+export function createPolicyEngine(config) {
+    return new PolicyEngine(config);
+}
+//# sourceMappingURL=policy.js.map

package/packages/session/dist/policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy.js","sourceRoot":"","sources":["../src/policy.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,MAAM,OAAO,WAAW;IACf,OAAO,GAAgC,IAAI,GAAG,EAAE,CAAC;IACxC,iBAAiB,CAAS;IAE3C,YAAY,iBAAyB;QACpC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,GAAW;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEpC,IAAI,CAAC,KAAK,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC;QACb,CAAC;QAED,OAAO,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,GAAW;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEpC,IAAI,CAAC,KAAK,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACpC,mBAAmB;YACnB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;gBACrB,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,GAAG,GAAG,KAAK,EAAE,WAAW;aACjC,CAAC,CAAC;QACJ,CAAC;aAAM,CAAC;YACP,KAAK,CAAC,KAAK,EAAE,CAAC;QACf,CAAC;IACF,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,GAAW;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC,iBAAiB,CAAC;QAC/B,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,YAAY;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;IACF,CAAC;CACD;AAUD,MAAM,OAAO,aAAa;IACjB,OAAO,GAA6B,IAAI,GAAG,EAAE,CAAC;IAEtD;;OAEG;IACH,SAAS,CAAC,GAAW,EAAE,KAAa,EAAE,SAAiB;QACtD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACrB,KAAK,EAAE,CAAC;YACR,KAAK;YACL,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SAC/B,CAAC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,GAAW,EAAE,MAAc;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,6BAA6B;QAC7B,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC;QACb,CAAC;QAED,OAAO,KAAK,CAAC,KAAK,GAAG,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,GAAW,EAAE,MAAc;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,6BAA6B;QAC7B,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACjC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC;YACrB,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC3D,CAAC;aAAM,CAAC;YACP,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC;QACvB,CAAC;IACF,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,GAAW;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC,KAAK,CAAC;QACpB,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,GAAW;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,OAAO,KAAK,CAAC,OAAO,CAAC;IACtB,CAAC;CACD;AAED,8EAA8E;AAE9E,MAAM,OAAO,YAAY;IACP,MAAM,CAAqB;IAC3B,WAAW,CAAc;IACzB,aAAa,CAAgB;IACtC,aAAa,GAA6B,IAAI,GAAG,EAAE,CAAC;IAE5D,YAAY,SAA6B,EAAE;QAC1C,IAAI,CAAC,MAAM,GAAG;YACb,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,EAAE;YACvD,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,OAAO;YAClD,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,GAAG;YAClD,qBAAqB,EAAE,MAAM,CAAC,qBAAqB,IAAI,EAAE;YACzD,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,EAAE;SACzC,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAqB,CAAC,CAAC;QACtE,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAiB;QAC/B,IAAI,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;YACZ,KAAK,GAAG;gBACP,kBAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,EAAE;gBAC1D,SAAS,EAAE,KAAK;aAChB,CAAC;YACF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,KAAK,CAAC;IACd,CAAC;IAED;;OAEG;IACH,UAAU,CACT,SAAiB,EACjB,MAAkD;QAElD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAE7C,qBAAqB;QACrB,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC;QACd,CAAC;QAED,mBAAmB;QACnB,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;YACjD,OAAO,KAAK,CAAC;QACd,CAAC;QAED,eAAe;QACf,IAAI,KAAK,CAAC,eAAe,KAAK,SAAS,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC;YACvE,OAAO,KAAK,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,SAAiB,EAAE,MAA+B;QAC9D,oBAAoB;QACpB,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAE1C,eAAe;QACf,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC7C,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAEpE,+BAA+B;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QACnC,KAAK,CAAC,gBAAgB,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB,EAAE,KAAiB;QACpD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC/B,qCAAqC;YACrC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;YACtD,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAEvD,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC;gBACjE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAE,GAAG,GAAG;gBACnD,CAAC,CAAC,SAAS,CAAC;YAEb,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACzD,IAAI,OAAO,EAAE,CAAC;gBACb,KAAK,CAAC,aAAa,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YACvD,CAAC;QACF,CAAC;IACF,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,SAAiB,EAAE,MAAc;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC7C,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,KAAK,CAAC,gBAAgB,GAAG,MAAM,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,SAAiB;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC7C,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC;QACxB,KAAK,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB;QAC3B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB,EAAE,MAAc,EAAE,QAAgB;QACnE,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;QAC/C,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;QAEjE,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC7C,KAAK,CAAC,eAAe,GAAG,MAAM,CAAC;QAC/B,KAAK,CAAC,aAAa,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACrB,OAAO,cAAc,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC;QAED,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;YACnC,OAAO,qBAAqB,CAAC;QAC9B,CAAC;QAED,IAAI,KAAK,CAAC,eAAe,KAAK,SAAS,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC;YACvE,OAAO,iBAAiB,CAAC;QAC1B,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,SAAiB,EAAE,cAAsB;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAE7C,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,KAAK,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YAC1E,OAAO,KAAK,CAAC,eAAe,GAAG,cAAc,GAAG,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,KAAK,CAAC;IACd,CAAC;IAED;;OAEG;IACH,gBAAgB,CACf,SAAiB;QAEjB,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACpD,OAAO,IAAI,CAAC;QACb,CAAC;QAED,OAAO;YACN,SAAS;YACT,OAAO,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC,WAAW,EAAE;SACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACN,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;IACjC,CAAC;CACD;AAED,4EAA4E;AAE5E;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAA2B;IAC7D,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC"}

package/packages/session/dist/store.d.ts

@@ -0,0 +1,108 @@
+/**
+ * Session Manager - Store
+ *
+ * Session persistence to disk with indexing.
+ */
+import type { Session, SessionContext, SessionStoreConfig, SessionStatus } from "./types.js";
+export declare class SessionStore {
+    private readonly rootDir;
+    private readonly autoSave;
+    private readonly autoSaveIntervalMs;
+    private readonly maxSessions?;
+    /**
+     * Get max sessions limit
+     */
+    getMaxSessions(): number | undefined;
+    private indices;
+    private autoSaveTimers;
+    constructor(config: SessionStoreConfig);
+    /**
+     * Ensure directory exists
+     */
+    private ensureDir;
+    /**
+     * Get session directory
+     */
+    private sessionDir;
+    /**
+     * Get session file path
+     */
+    private sessionPath;
+    /**
+     * Get index file path
+     */
+    private indexPath;
+    /**
+     * Load or create index
+     */
+    private loadIndex;
+    /**
+     * Save index
+     */
+    private saveIndex;
+    /**
+     * Save session context to disk
+     */
+    private saveToDisk;
+    /**
+     * Load session context from disk
+     */
+    private loadFromDisk;
+    /**
+     * Create or update session index entry
+     */
+    private updateIndexEntry;
+    /**
+     * Initialize the store
+     */
+    initialize(): Promise<void>;
+    /**
+     * Save session context
+     */
+    save(context: SessionContext): Promise<void>;
+    /**
+     * Schedule auto-save for a session
+     */
+    private scheduleAutoSave;
+    /**
+     * Load session context
+     */
+    load(sessionId: string): Promise<SessionContext | null>;
+    /**
+     * Delete session
+     */
+    delete(sessionId: string): Promise<void>;
+    /**
+     * List sessions for a user
+     */
+    listByUser(userId: string): Promise<Session[]>;
+    /**
+     * List all sessions
+     */
+    listAll(): Promise<Session[]>;
+    /**
+     * Get session summary without loading full context
+     */
+    getSummary(sessionId: string): Promise<Session | null>;
+    /**
+     * Check if session exists
+     */
+    exists(sessionId: string): Promise<boolean>;
+    /**
+     * Get session count
+     */
+    count(): Promise<number>;
+    /**
+     * Get sessions by status
+     */
+    listByStatus(status: SessionStatus): Promise<Session[]>;
+    /**
+     * Close store and cleanup
+     */
+    close(): Promise<void>;
+}
+/**
+ * Create a session store
+ */
+export declare function createSessionStore(config: SessionStoreConfig): SessionStore;
+//# sourceMappingURL=store.d.ts.map

package/packages/session/dist/store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../src/store.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAWH,OAAO,KAAK,EACX,OAAO,EACP,cAAc,EACd,kBAAkB,EAClB,aAAa,EACb,MAAM,YAAY,CAAC;AAoBpB,qBAAa,YAAY;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAU;IACnC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAS;IAC5C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAS;IAEtC;;OAEG;IACH,cAAc,IAAI,MAAM,GAAG,SAAS;IAGpC,OAAO,CAAC,OAAO,CAA6C;IAC5D,OAAO,CAAC,cAAc,CAA0C;gBAEpD,MAAM,EAAE,kBAAkB;IAOtC;;OAEG;YACW,SAAS;IAIvB;;OAEG;IACH,OAAO,CAAC,UAAU;IAIlB;;OAEG;IACH,OAAO,CAAC,WAAW;IAInB;;OAEG;IACH,OAAO,CAAC,SAAS;IAIjB;;OAEG;YACW,SAAS;IAWvB;;OAEG;YACW,SAAS;IAUvB;;OAEG;YACW,UAAU;IAQxB;;OAEG;YACW,YAAY;IAY1B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAaxB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAKjC;;OAEG;IACG,IAAI,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAYlD;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAmBxB;;OAEG;IACG,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IAI7D;;OAEG;IACG,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB9C;;OAEG;IACG,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAgBpD;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAcnC;;OAEG;IACG,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;IAe5D;;OAEG;IACG,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIjD;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC;IAI9B;;OAEG;IACG,YAAY,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAgB7D;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAU5B;AAID;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,kBAAkB,GAAG,YAAY,CAE3E"}

package/packages/session/dist/store.js

@@ -0,0 +1,274 @@
+/**
+ * Session Manager - Store
+ *
+ * Session persistence to disk with indexing.
+ */
+import { readdir, readFile, writeFile, mkdir, rename, unlink, } from "node:fs/promises";
+import { dirname, join } from "node:path";
+// ─── Session Store ────────────────────────────────────────────────────────
+export class SessionStore {
+    rootDir;
+    autoSave;
+    autoSaveIntervalMs;
+    maxSessions;
+    /**
+     * Get max sessions limit
+     */
+    getMaxSessions() {
+        return this.maxSessions;
+    }
+    indices = new Map();
+    autoSaveTimers = new Map();
+    constructor(config) {
+        this.rootDir = config.rootDir;
+        this.autoSave = config.autoSave;
+        this.autoSaveIntervalMs = config.autoSaveIntervalMs;
+        this.maxSessions = config.maxSessions;
+    }
+    /**
+     * Ensure directory exists
+     */
+    async ensureDir(dir) {
+        await mkdir(dir, { recursive: true });
+    }
+    /**
+     * Get session directory
+     */
+    sessionDir(sessionId) {
+        return join(this.rootDir, "sessions", sessionId.slice(0, 2), sessionId);
+    }
+    /**
+     * Get session file path
+     */
+    sessionPath(sessionId) {
+        return join(this.sessionDir(sessionId), "session.json");
+    }
+    /**
+     * Get index file path
+     */
+    indexPath() {
+        return join(this.rootDir, "sessions", "index.json");
+    }
+    /**
+     * Load or create index
+     */
+    async loadIndex() {
+        const path = this.indexPath();
+        try {
+            const content = await readFile(path, "utf-8");
+            const entries = JSON.parse(content);
+            return new Map(entries.map((e) => [e.id, e]));
+        }
+        catch {
+            return new Map();
+        }
+    }
+    /**
+     * Save index
+     */
+    async saveIndex() {
+        const entries = Array.from(this.indices.values());
+        await this.ensureDir(dirname(this.indexPath()));
+        await writeFile(this.indexPath(), JSON.stringify(entries, null, 2), "utf-8");
+    }
+    /**
+     * Save session context to disk
+     */
+    async saveToDisk(context) {
+        const path = this.sessionPath(context.id);
+        await this.ensureDir(dirname(path));
+        const tmp = `${path}.tmp`;
+        await writeFile(tmp, JSON.stringify(context, null, 2), "utf-8");
+        await rename(tmp, path);
+    }
+    /**
+     * Load session context from disk
+     */
+    async loadFromDisk(sessionId) {
+        const path = this.sessionPath(sessionId);
+        try {
+            const content = await readFile(path, "utf-8");
+            return JSON.parse(content);
+        }
+        catch {
+            return null;
+        }
+    }
+    /**
+     * Create or update session index entry
+     */
+    updateIndexEntry(context) {
+        this.indices.set(context.id, {
+            id: context.id,
+            userId: context.userId,
+            status: context.status,
+            createdAt: context.createdAt,
+            updatedAt: context.updatedAt,
+            expiresAt: context.expiresAt,
+            messageCount: context.messages.length,
+            totalTokens: context.tokenUsage.totalTokens,
+        });
+    }
+    /**
+     * Initialize the store
+     */
+    async initialize() {
+        await this.ensureDir(this.rootDir);
+        this.indices = await this.loadIndex();
+    }
+    /**
+     * Save session context
+     */
+    async save(context) {
+        this.updateIndexEntry(context);
+        await this.saveToDisk(context);
+        if (this.autoSave) {
+            this.scheduleAutoSave(context.id);
+        }
+        // Save index periodically
+        await this.saveIndex();
+    }
+    /**
+     * Schedule auto-save for a session
+     */
+    scheduleAutoSave(sessionId) {
+        // Clear existing timer
+        const existing = this.autoSaveTimers.get(sessionId);
+        if (existing) {
+            clearTimeout(existing);
+        }
+        // Schedule new save
+        const timer = setTimeout(async () => {
+            this.autoSaveTimers.delete(sessionId);
+            const context = await this.loadFromDisk(sessionId);
+            if (context) {
+                await this.saveToDisk(context);
+            }
+        }, this.autoSaveIntervalMs);
+        this.autoSaveTimers.set(sessionId, timer);
+    }
+    /**
+     * Load session context
+     */
+    async load(sessionId) {
+        return this.loadFromDisk(sessionId);
+    }
+    /**
+     * Delete session
+     */
+    async delete(sessionId) {
+        // Cancel auto-save
+        const timer = this.autoSaveTimers.get(sessionId);
+        if (timer) {
+            clearTimeout(timer);
+            this.autoSaveTimers.delete(sessionId);
+        }
+        // Remove from index
+        this.indices.delete(sessionId);
+        // Delete files
+        const dir = this.sessionDir(sessionId);
+        try {
+            const files = await readdir(dir);
+            await Promise.all(files.map((f) => unlink(join(dir, f))));
+        }
+        catch {
+            // Ignore if directory doesn't exist
+        }
+        // Update index
+        await this.saveIndex();
+    }
+    /**
+     * List sessions for a user
+     */
+    async listByUser(userId) {
+        const entries = Array.from(this.indices.values()).filter((e) => e.userId === userId);
+        return entries.map((e) => ({
+            id: e.id,
+            userId: e.userId,
+            status: e.status,
+            createdAt: e.createdAt,
+            updatedAt: e.updatedAt,
+            lastActivityAt: e.updatedAt,
+            expiresAt: e.expiresAt,
+        }));
+    }
+    /**
+     * List all sessions
+     */
+    async listAll() {
+        const entries = Array.from(this.indices.values());
+        return entries.map((e) => ({
+            id: e.id,
+            userId: e.userId,
+            status: e.status,
+            createdAt: e.createdAt,
+            updatedAt: e.updatedAt,
+            lastActivityAt: e.updatedAt,
+            expiresAt: e.expiresAt,
+        }));
+    }
+    /**
+     * Get session summary without loading full context
+     */
+    async getSummary(sessionId) {
+        const entry = this.indices.get(sessionId);
+        if (!entry)
+            return null;
+        return {
+            id: entry.id,
+            userId: entry.userId,
+            status: entry.status,
+            createdAt: entry.createdAt,
+            updatedAt: entry.updatedAt,
+            lastActivityAt: entry.updatedAt,
+            expiresAt: entry.expiresAt,
+        };
+    }
+    /**
+     * Check if session exists
+     */
+    async exists(sessionId) {
+        return this.indices.has(sessionId);
+    }
+    /**
+     * Get session count
+     */
+    async count() {
+        return this.indices.size;
+    }
+    /**
+     * Get sessions by status
+     */
+    async listByStatus(status) {
+        const entries = Array.from(this.indices.values()).filter((e) => e.status === status);
+        return entries.map((e) => ({
+            id: e.id,
+            userId: e.userId,
+            status: e.status,
+            createdAt: e.createdAt,
+            updatedAt: e.updatedAt,
+            lastActivityAt: e.updatedAt,
+            expiresAt: e.expiresAt,
+        }));
+    }
+    /**
+     * Close store and cleanup
+     */
+    async close() {
+        // Clear all auto-save timers
+        for (const timer of this.autoSaveTimers.values()) {
+            clearTimeout(timer);
+        }
+        this.autoSaveTimers.clear();
+        // Save final index
+        await this.saveIndex();
+    }
+}
+// ─── Factory Function ────────────────────────────────────────────────────
+/**
+ * Create a session store
+ */
+export function createSessionStore(config) {
+    return new SessionStore(config);
+}
+//# sourceMappingURL=store.js.map