CodeCow/stack
1
0
Fork 0
mirror of https://github.com/stack-auth/stack.git synced 2026-06-13 21:01:21 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

bug fix
Some checks failed
DB migration compat / Check if migrations changed (push) Has been cancelled
Details
DB migration compat / Back-compat — Current branch migrations with ${{ needs.check-migrations-changed.outputs.base_branch }} branch code (push) Has been cancelled
Details
DB migration compat / Forward-compat — Current branch code with ${{ needs.check-migrations-changed.outputs.base_branch }} branch migrations (push) Has been cancelled
Details
DB migration compat / No migration changes (skipped) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Aadesh Kheria 2026-04-21 10:29:02 -07:00
parent 3654af5591
commit dbc7988169
1 changed files with 17 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
apps/internal-tool/src/hooks/useSpacetimeDB.ts
View File

@ -1,26 +1,23 @@
import { captureError } from "@stackframe/stack-shared/dist/utils/errors";
import { useEffect, useState, useRef } from "react";
import type { Identity } from "spacetimedb";
import { envOrDevDefault } from "../lib/env";
import { DbConnection, type ErrorContext, type EventContext, type SubscriptionEventContext } from "../module_bindings";
import type { AiQueryLogRow, McpCallLogRow, PublishedQaRow } from "../types";
export type EnsureEnrolled = (identity: Identity) => Promise<void>;
const IS_DEV = process.env.NODE_ENV === "development";
const PLACEHOLDER = "REPLACE_ME";
const rawHost = process.env.NEXT_PUBLIC_SPACETIMEDB_HOST;
const rawDbName = process.env.NEXT_PUBLIC_SPACETIMEDB_DB_NAME;
function resolveEnv(raw: string | undefined, devDefault: string, name: string): string {
if (raw && raw !== PLACEHOLDER) return raw;
if (IS_DEV) return devDefault;
throw new Error(`${name} is not configured. Set it in .env.local or hosting platform env.`);
let cachedConfig: { host: string, dbName: string, tokenKey: string } | null = null;
function getConfig() {
if (cachedConfig) return cachedConfig;
const host = envOrDevDefault(process.env.NEXT_PUBLIC_SPACETIMEDB_HOST, "ws://localhost:8139", "NEXT_PUBLIC_SPACETIMEDB_HOST");
if (process.env.NODE_ENV !== "development" && !host.startsWith("wss://")) {
throw new Error("NEXT_PUBLIC_SPACETIMEDB_HOST must use wss:// in production");
}
const dbName = envOrDevDefault(process.env.NEXT_PUBLIC_SPACETIMEDB_DB_NAME, "stack-auth-llm", "NEXT_PUBLIC_SPACETIMEDB_DB_NAME");
cachedConfig = { host, dbName, tokenKey: `spacetimedb_${host}/${dbName}/auth_token` };
return cachedConfig;
}
const HOST = resolveEnv(rawHost, "ws://localhost:8139", "NEXT_PUBLIC_SPACETIMEDB_HOST");
if (!IS_DEV && !HOST.startsWith("wss://")) {
throw new Error("NEXT_PUBLIC_SPACETIMEDB_HOST must use wss:// in production");
}
const DB_NAME = resolveEnv(rawDbName, "stack-auth-llm", "NEXT_PUBLIC_SPACETIMEDB_DB_NAME");
const TOKEN_KEY = `spacetimedb_${HOST}/${DB_NAME}/auth_token`;
const MAX_RETRIES = 5;
const RETRY_DELAY_MS = 2000;
@ -73,14 +70,15 @@ function useTableSubscription<Row extends { id: bigint }>(
}
function connect() {
const config = getConfig();
const conn = DbConnection.builder()
.withUri(HOST)
.withDatabaseName(DB_NAME)
.withToken(localStorage.getItem(TOKEN_KEY) || undefined)
.withUri(config.host)
.withDatabaseName(config.dbName)
.withToken(localStorage.getItem(config.tokenKey) || undefined)
.onConnect((connInstance: DbConnection, identity: Identity, token: string) => {
if (cancelled) return;
retryCount = 0;
localStorage.setItem(TOKEN_KEY, token);
localStorage.setItem(config.tokenKey, token);
connRef.current = connInstance;
const startSubscription = () => {
@ -140,7 +138,7 @@ function useTableSubscription<Row extends { id: bigint }>(
const message = err instanceof Error ? err.message : "";
const looksLikeAuthFailure = /unauthor|verify token|401/i.test(message);
if (looksLikeAuthFailure) {
localStorage.removeItem(TOKEN_KEY);
localStorage.removeItem(config.tokenKey);
}
retry();
})