rivet-design 0.7.0 → 0.8.0

package/dist/services/HostedDemoAuthSessionStore.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RedisHostedDemoAuthSessionStore = exports.InMemoryHostedDemoAuthSessionStore = void 0;
+const redis_1 = require("redis");
+const logger_1 = require("../utils/logger");
+const log = (0, logger_1.createLogger)('HostedDemoAuthSessionStore');
+class InMemoryHostedDemoAuthSessionStore {
+    records = new Map();
+    async initialize() {
+        return Promise.resolve();
+    }
+    async loadAll() {
+        return [...this.records.entries()].map(([sessionId, value]) => ({
+            sessionId,
+            value,
+        }));
+    }
+    async set(sessionId, value, _ttlMs) {
+        this.records.set(sessionId, value);
+    }
+    async delete(sessionId) {
+        this.records.delete(sessionId);
+    }
+    async shutdown() {
+        this.records.clear();
+    }
+}
+exports.InMemoryHostedDemoAuthSessionStore = InMemoryHostedDemoAuthSessionStore;
+class RedisHostedDemoAuthSessionStore {
+    client;
+    keyPrefix;
+    constructor(options) {
+        this.client = (0, redis_1.createClient)({ url: options.redisUrl });
+        this.keyPrefix = options.keyPrefix;
+    }
+    async initialize() {
+        if (this.client.isOpen) {
+            return;
+        }
+        this.client.on('error', (error) => {
+            log.warn('Redis auth session store error:', error);
+        });
+        await this.client.connect();
+    }
+    async loadAll() {
+        const records = [];
+        for await (const keyOrChunk of this.client.scanIterator({
+            MATCH: `${this.keyPrefix}:*`,
+            COUNT: 100,
+        })) {
+            const keys = Array.isArray(keyOrChunk) ? keyOrChunk : [keyOrChunk];
+            for (const key of keys) {
+                const raw = await this.client.get(key);
+                if (!raw) {
+                    continue;
+                }
+                const sessionId = key.slice(`${this.keyPrefix}:`.length);
+                if (!sessionId) {
+                    continue;
+                }
+                try {
+                    const value = JSON.parse(raw);
+                    records.push({ sessionId, value });
+                }
+                catch (error) {
+                    log.warn(`Failed to parse auth session payload for ${sessionId}:`, error);
+                }
+            }
+        }
+        return records;
+    }
+    async set(sessionId, value, ttlMs) {
+        await this.client.set(this.getKey(sessionId), JSON.stringify(value), {
+            PX: ttlMs,
+        });
+    }
+    async delete(sessionId) {
+        await this.client.del(this.getKey(sessionId));
+    }
+    async shutdown() {
+        if (this.client.isOpen) {
+            await this.client.quit();
+        }
+    }
+    getKey(sessionId) {
+        return `${this.keyPrefix}:${sessionId}`;
+    }
+}
+exports.RedisHostedDemoAuthSessionStore = RedisHostedDemoAuthSessionStore;
+//# sourceMappingURL=HostedDemoAuthSessionStore.js.map

package/dist/services/HostedDemoAuthSessionStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HostedDemoAuthSessionStore.js","sourceRoot":"","sources":["../../src/services/HostedDemoAuthSessionStore.ts"],"names":[],"mappings":";;;AAAA,iCAA2D;AAC3D,4CAA+C;AAE/C,MAAM,GAAG,GAAG,IAAA,qBAAY,EAAC,4BAA4B,CAAC,CAAC;AA2BvD,MAAa,kCAAkC;IAG5B,OAAO,GAAG,IAAI,GAAG,EAAoC,CAAC;IAEvE,KAAK,CAAC,UAAU;QACd,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO;QACX,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9D,SAAS;YACT,KAAK;SACN,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,GAAG,CACP,SAAiB,EACjB,KAA+B,EAC/B,MAAc;QAEd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF;AA/BD,gFA+BC;AAOD,MAAa,+BAA+B;IACzB,MAAM,CAAkB;IACxB,SAAS,CAAS;IAEnC,YAAY,OAA+C;QACzD,IAAI,CAAC,MAAM,GAAG,IAAA,oBAAY,EAAC,EAAE,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAChC,GAAG,CAAC,IAAI,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,OAAO,GAAgC,EAAE,CAAC;QAChD,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YACtD,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI;YAC5B,KAAK,EAAE,GAAG;SACX,CAAC,EAAE,CAAC;YACH,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACnE,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACvC,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,SAAS;gBACX,CAAC;gBAED,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,MAAM,CAAC,CAAC;gBACzD,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,SAAS;gBACX,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA6B,CAAC;oBAC1D,OAAO,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;gBACrC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,IAAI,CAAC,4CAA4C,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC5E,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,GAAG,CACP,SAAiB,EACjB,KAA+B,EAC/B,KAAa;QAEb,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;YACnE,EAAE,EAAE,KAAK;SACV,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,SAAiB;QAC9B,OAAO,GAAG,IAAI,CAAC,SAAS,IAAI,SAAS,EAAE,CAAC;IAC1C,CAAC;CACF;AAvED,0EAuEC"}

package/dist/services/HostedDemoSessionService.d.ts

@@ -0,0 +1,91 @@
+import { type ChildProcess } from 'child_process';
+import { TelemetryService } from './TelemetryService';
+import { type HostedDemoSessionPersistedRuntimeState, type HostedDemoSessionStore } from './HostedDemoSessionStore';
+export type HostedDemoSession = {
+    sessionId: string;
+    userId: string;
+    createdAt: number;
+    expiresAt: number;
+    lastActionAt: number;
+    worktreePath: string;
+    projectPath: string;
+    rivetPort: number;
+    runtimeState: HostedDemoSessionPersistedRuntimeState;
+};
+export type HostedDemoSessionWithUrl = HostedDemoSession & {
+    sessionUrl: string;
+};
+type RuntimeStartParams = {
+    projectPath: string;
+    rivetPort: number;
+    sessionId: string;
+};
+type HostedDemoRuntimeDeps = {
+    now?: () => number;
+    runGitCommand?: (repoPath: string, args: string[]) => Promise<void>;
+    startRuntime?: (params: RuntimeStartParams) => Promise<ChildProcess>;
+    stopRuntime?: (runtimeProcess: ChildProcess) => Promise<void>;
+    waitForRuntimeHealth?: (rivetPort: number) => Promise<void>;
+};
+export type HostedDemoSessionConfig = {
+    repoPath: string;
+    templateProjectPath: string;
+    telemetry?: TelemetryService;
+    worktreesRootPath?: string;
+    sessionTtlMs?: number;
+    sessionIdleTtlMs?: number;
+    maxActiveSessions?: number;
+    baseRivetPort?: number;
+    publicBaseUrl?: string;
+    sessionMetadataStore?: HostedDemoSessionStore;
+};
+export declare class HostedDemoSessionService {
+    private readonly config;
+    private readonly deps;
+    private readonly telemetry;
+    private readonly sessionStore;
+    private readonly sessions;
+    private pendingSessionCreations;
+    private readonly cleanupTimer;
+    private readonly rehydrationLocks;
+    constructor(config: HostedDemoSessionConfig, deps?: HostedDemoRuntimeDeps);
+    /**
+     * Loads durable session rows into memory so stopped sessions can be rehydrated after deploy.
+     */
+    initialize(): Promise<void>;
+    createSession(userId: string): Promise<HostedDemoSessionWithUrl>;
+    getActiveSessionCount(): number;
+    getTemplateProjectPath(): string;
+    getSession(sessionId: string): HostedDemoSession | null;
+    getSessionOwnerUserId(sessionId: string): string | null;
+    /**
+     * Records a Rivet user action for idle TTL (does not extend idle on passive polling).
+     */
+    recordSessionAction(sessionId: string): Promise<void>;
+    getSessionProxyTarget(sessionId: string): string | null;
+    /**
+     * Starts the per-session runtime when it was stopped for idle or after process restart.
+     */
+    ensureRuntimeReady(sessionId: string): Promise<void>;
+    closeSession(sessionId: string, options?: {
+        removeWorktree?: boolean;
+    }): Promise<void>;
+    shutdown(): Promise<void>;
+    private readonly countRunningOrStartingSessions;
+    private readonly persistSession;
+    private readonly toPersisted;
+    private readonly internalFromPersisted;
+    private readonly runRehydration;
+    private readonly stopRuntimeForIdle;
+    private readonly cleanupExpiredSessions;
+    private readonly safeRemoveWorktree;
+    private readonly toErrorType;
+    getSessionUrl(sessionId: string): string;
+    private readonly toPublicSession;
+    private readonly defaultRunGitCommand;
+    private readonly defaultStartRuntime;
+    private readonly defaultStopRuntime;
+    private readonly defaultWaitForRuntimeHealth;
+}
+export {};
+//# sourceMappingURL=HostedDemoSessionService.d.ts.map

package/dist/services/HostedDemoSessionService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HostedDemoSessionService.d.ts","sourceRoot":"","sources":["../../src/services/HostedDemoSessionService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,eAAe,CAAC;AAOnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAEL,KAAK,sCAAsC,EAE3C,KAAK,sBAAsB,EAC5B,MAAM,0BAA0B,CAAC;AAWlC,MAAM,MAAM,iBAAiB,GAAG;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,sCAAsC,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG,iBAAiB,GAAG;IACzD,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAOF,KAAK,kBAAkB,GAAG;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,KAAK,qBAAqB,GAAG;IAC3B,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,YAAY,CAAC,EAAE,CAAC,MAAM,EAAE,kBAAkB,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;IACrE,WAAW,CAAC,EAAE,CAAC,cAAc,EAAE,YAAY,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,oBAAoB,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7D,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,SAAS,CAAC,EAAE,gBAAgB,CAAC;IAC7B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,sBAAsB,CAAC;CAC/C,CAAC;AAOF,qBAAa,wBAAwB;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA2C;IAClE,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAkC;IACvD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA+B;IACzD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAyB;IACtD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAgD;IACzE,OAAO,CAAC,uBAAuB,CAAK;IACpC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAiC;IAC9D,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAoC;gBAEzD,MAAM,EAAE,uBAAuB,EAAE,IAAI,CAAC,EAAE,qBAAqB;IAkCzE;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB3B,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAyGtE,qBAAqB,IAAI,MAAM;IAI/B,sBAAsB,IAAI,MAAM;IAIhC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,iBAAiB,GAAG,IAAI;IAQvD,qBAAqB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAQvD;;OAEG;IACG,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAS3D,qBAAqB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAWvD;;OAEG;IACG,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiCpD,YAAY,CAChB,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,OAAO,CAAA;KAAE,GACrC,OAAO,CAAC,IAAI,CAAC;IA+BV,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAS/B,OAAO,CAAC,QAAQ,CAAC,8BAA8B,CAQ7C;IAEF,OAAO,CAAC,QAAQ,CAAC,cAAc,CAQ7B;IAEF,OAAO,CAAC,QAAQ,CAAC,WAAW,CAe1B;IAEF,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAgBpC;IAEF,OAAO,CAAC,QAAQ,CAAC,cAAc,CAsE7B;IAEF,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAiCjC;IAEF,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CA+BrC;IAEF,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAqBjC;IAEF,OAAO,CAAC,QAAQ,CAAC,WAAW,CAK1B;IAEF,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAQxC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAc9B;IAEF,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAKnC;IAEF,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CA2BlC;IAEF,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAkBjC;IAEF,OAAO,CAAC,QAAQ,CAAC,2BAA2B,CAmC1C;CACH"}