Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/openstatus. 🫖 Status page with uptime monitoring & API monitoring as code 🫖 openstatus.dev
bun drizzle-orm monitoring monitoring-as-code nextjs observability on-call open-source shadcn-ui status-page statuspage synthetic-monitoring tinybird turso uptime uptime-checker uptime-monitor
Something went wrong. Try again.
7.7 kB · 243 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244import { and, db as defaultDb, eq, isNull, sql } from "@openstatus/db";import { monitor, notification, page, pageComponent, selectWorkspaceSchema, statusReport, user, usersToWorkspaces, workspace,} from "@openstatus/db/src/schema";
import { type DB, type ServiceContext, batchReads } from "../context";import { NotFoundError } from "../errors";import type { Workspace } from "../types";import { GetWorkspaceByStripeIdInput, GetWorkspaceForMemberInput, type GetWorkspaceUsageInput, ListWorkspaceOwnersInput, ListWorkspacesInput, OwnedWorkspacesInput,} from "./schemas";
/** * Usage snapshot shown on the workspace settings page — counts the workspace's * active (non-soft-deleted) monitors, notifications and pages, plus the total * page-component count across all pages. */export type WorkspaceUsage = { monitors: number; notifications: number; pages: number; pageComponents: number; statusReports: number; checks: number;};
/** Load the workspace the caller is scoped to. */export async function getWorkspace(args: { ctx: ServiceContext;}): Promise<Workspace> { const { ctx } = args; const db = ctx.db ?? defaultDb;
const result = await db.query.workspace.findFirst({ where: eq(workspace.id, ctx.workspace.id), });
// Shouldn't be reachable in practice — `ctx.workspace` was already // resolved upstream — but guard explicitly so callers see the same // `NotFoundError` shape every other service throws when a row is // missing, rather than a `ZodError` from `parse(undefined)`. if (!result) throw new NotFoundError("workspace", ctx.workspace.id);
return selectWorkspaceSchema.parse(result);}
// Counts, not rows: the previous relational read materialized every page,// component, monitor (with its config blobs) and notification just to call// `.length` on them. All five are single-table and index-covered.function usageCountQueries(db: DB, workspaceId: number) { const total = sql<number>`count(*)`; return [ db .select({ count: total }) .from(monitor) .where( and(eq(monitor.workspaceId, workspaceId), isNull(monitor.deletedAt)), ), db .select({ count: total }) .from(notification) .where(eq(notification.workspaceId, workspaceId)), db .select({ count: total }) .from(page) .where(eq(page.workspaceId, workspaceId)), db .select({ count: total }) .from(pageComponent) .where(eq(pageComponent.workspaceId, workspaceId)), db .select({ count: total }) .from(statusReport) .where(eq(statusReport.workspaceId, workspaceId)), ] as const;}
function toUsage(rows: { count: number }[][]): WorkspaceUsage { const [monitors, notifications, pages, pageComponents, statusReports] = rows; return { monitors: monitors?.[0]?.count ?? 0, notifications: notifications?.[0]?.count ?? 0, pages: pages?.[0]?.count ?? 0, pageComponents: pageComponents?.[0]?.count ?? 0, statusReports: statusReports?.[0]?.count ?? 0, // Parity with the legacy router — checks usage was previously commented // out pending a real source and left as 0. Preserved here. checks: 0, };}
/** * The usage counts the dashboard surfaces alongside plan limits. Active * monitors only (`deletedAt IS NULL`); notifications / pages / page-components * are unconditional counts scoped to the workspace. */export async function getWorkspaceUsage(args: { ctx: ServiceContext; input?: GetWorkspaceUsageInput;}): Promise<WorkspaceUsage> { const { ctx } = args; const db = ctx.db ?? defaultDb;
return toUsage(await batchReads(db, usageCountQueries(db, ctx.workspace.id)));}
/** * Resolve a workspace by its Stripe customer id. Runs before a * `ctx.workspace` exists (the Stripe webhook only holds the customer id), * so it takes an optional `db`/tx rather than a `ServiceContext`. Returns * `null` when no workspace maps to the customer — an expected case (an * event for a customer we don't own), which the caller maps to its own * error rather than a thrown `NotFoundError`. */export async function getWorkspaceByStripeId(args: { input: GetWorkspaceByStripeIdInput; db?: DB;}): Promise<Workspace | null> { const { stripeId } = GetWorkspaceByStripeIdInput.parse(args.input); const db = args.db ?? defaultDb;
const row = await db.query.workspace.findFirst({ where: eq(workspace.stripeId, stripeId), });
return row ? selectWorkspaceSchema.parse(row) : null;}
/** * Workspaces the given user belongs to. Called before `ctx.workspace` is * meaningful (list runs across every workspace the user has access to), so * the user id is passed explicitly in the input rather than read from ctx. */export async function listWorkspaces(args: { ctx: ServiceContext; input: ListWorkspacesInput;}): Promise<Workspace[]> { const input = ListWorkspacesInput.parse(args.input); const db = args.ctx.db ?? defaultDb;
const rows = await db.query.usersToWorkspaces.findMany({ where: eq(usersToWorkspaces.userId, input.userId), with: { workspace: true }, });
return selectWorkspaceSchema .array() .parse(rows.map(({ workspace }) => workspace));}
/** * A workspace by slug, only if the user is a member. Billing procedures take * the slug as input, so this runs before a `ctx.workspace` exists for it. * Returns the member's email alongside for the Stripe customer record. */export async function getWorkspaceForMember(args: { input: GetWorkspaceForMemberInput; db?: DB;}): Promise<{ workspace: Workspace; email: string | null } | null> { const input = GetWorkspaceForMemberInput.parse(args.input); const db = args.db ?? defaultDb;
const row = await db .select({ workspace, email: user.email }) .from(usersToWorkspaces) .innerJoin(workspace, eq(workspace.id, usersToWorkspaces.workspaceId)) .innerJoin(user, eq(user.id, usersToWorkspaces.userId)) .where( and( eq(workspace.slug, input.slug), eq(usersToWorkspaces.userId, input.userId), ), ) .get();
if (!row) return null; return { workspace: selectWorkspaceSchema.parse(row.workspace), email: row.email, };}
/** Workspaces the user owns. Account deletion checks these for a paid plan. */export async function listOwnedWorkspaces(args: { input: OwnedWorkspacesInput; db?: DB;}): Promise<Workspace[]> { const input = OwnedWorkspacesInput.parse(args.input); const db = args.db ?? defaultDb;
const rows = await db .select({ workspace }) .from(usersToWorkspaces) .innerJoin(workspace, eq(workspace.id, usersToWorkspaces.workspaceId)) .where( and( eq(usersToWorkspaces.userId, input.userId), eq(usersToWorkspaces.role, "owner"), ), ) .all();
return selectWorkspaceSchema.array().parse(rows.map((r) => r.workspace));}
/** * Owners of a workspace — billing mail recipients and trial attribution. * Account deletion keeps the owner membership and only soft-deletes the * user, so those rows are filtered out here. */export async function listWorkspaceOwners(args: { input: ListWorkspaceOwnersInput; db?: DB;}): Promise<{ id: number; email: string | null }[]> { const input = ListWorkspaceOwnersInput.parse(args.input); const db = args.db ?? defaultDb;
return db .select({ id: user.id, email: user.email }) .from(usersToWorkspaces) .innerJoin(user, eq(user.id, usersToWorkspaces.userId)) .where( and( eq(usersToWorkspaces.workspaceId, input.workspaceId), eq(usersToWorkspaces.role, "owner"), isNull(user.deletedAt), ), ) .all();}