45 lines
4.4 KiB
TypeScript
45 lines
4.4 KiB
TypeScript
import { sql } from "drizzle-orm"
|
|
import { pgTable, uuid, bigint, text, timestamp, boolean, integer, jsonb, index, uniqueIndex, primaryKey, check } from "drizzle-orm/pg-core"
|
|
import { tenants } from "./tenants"
|
|
import { collaborationPrincipals, collaborationRooms } from "./collaboration"
|
|
|
|
const time = (name: string) => timestamp(name, { withTimezone: true })
|
|
export const collaborationFederationIdentity = pgTable("collaboration_federation_identity", {
|
|
singleton: boolean("singleton").primaryKey().default(true), instanceId: uuid("instance_id").notNull().unique(),
|
|
keyId: uuid("key_id").notNull(), publicKey: text("public_key").notNull(), privateKeyCipher: jsonb("private_key_cipher").notNull(),
|
|
createdAt: time("created_at").notNull().defaultNow(),
|
|
}, t => ({ singleton: check("collaboration_federation_identity_singleton_check", sql`${t.singleton}`) }))
|
|
export const collaborationPartners = pgTable("collaboration_partners", {
|
|
tenantId: bigint("tenant_id", { mode: "number" }).notNull().references(() => tenants.id, { onDelete: "cascade" }),
|
|
origin: text("origin").notNull(), instanceId: uuid("instance_id").notNull(), keyId: uuid("key_id").notNull(), publicKey: text("public_key").notNull(),
|
|
active: boolean("active").notNull().default(true), updatedAt: time("updated_at").notNull().defaultNow(),
|
|
}, t => ({ pk: primaryKey({ columns: [t.tenantId, t.origin] }) }))
|
|
export const collaborationFederationNonces = pgTable("collaboration_federation_nonces", {
|
|
origin: text("origin").notNull(), nonce: uuid("nonce").notNull(), expiresAt: time("expires_at").notNull(),
|
|
}, t => ({ pk: primaryKey({ columns: [t.origin, t.nonce] }) }))
|
|
export const collaborationRemoteBindings = pgTable("collaboration_remote_bindings", {
|
|
id: uuid("id").primaryKey().defaultRandom(),
|
|
tenantId: bigint("tenant_id", { mode: "number" }).notNull().references(() => tenants.id, { onDelete: "cascade" }),
|
|
principalId: uuid("principal_id").notNull().references(() => collaborationPrincipals.id, { onDelete: "cascade" }),
|
|
origin: text("origin").notNull(), roomId: uuid("room_id").notNull(), invitationId: uuid("invitation_id").notNull(),
|
|
status: text("status").notNull().default("pending"), invitation: jsonb("invitation").notNull(), snapshot: jsonb("snapshot"),
|
|
revision: bigint("revision", { mode: "number" }).notNull().default(0), lastReadSeq: bigint("last_read_seq", { mode: "number" }).notNull().default(0),
|
|
syncSeq: bigint("sync_seq", { mode: "number" }).notNull().default(0),
|
|
syncAt: time("sync_at"), lastError: text("last_error"),
|
|
}, t => ({ identity: uniqueIndex("collaboration_remote_bindings_principal_id_origin_room_id_key").on(t.principalId, t.origin, t.roomId),
|
|
status: check("collaboration_remote_bindings_status_check", sql`${t.status} IN ('pending','active','revoked','declined')`) }))
|
|
export const collaborationRemoteMessages = pgTable("collaboration_remote_messages", {
|
|
bindingId: uuid("binding_id").notNull().references(() => collaborationRemoteBindings.id, { onDelete: "cascade" }),
|
|
id: uuid("id").notNull(), seq: bigint("seq", { mode: "number" }).notNull(), payload: jsonb("payload").notNull(),
|
|
}, t => ({ pk: primaryKey({ columns: [t.bindingId, t.id] }), seq: uniqueIndex("collaboration_remote_messages_binding_id_seq_key").on(t.bindingId, t.seq) }))
|
|
export const collaborationOutbox = pgTable("collaboration_outbox", {
|
|
id: uuid("id").primaryKey().defaultRandom(), dedupKey: text("dedup_key").notNull().unique(),
|
|
tenantId: bigint("tenant_id", { mode: "number" }).notNull().references(() => tenants.id, { onDelete: "cascade" }),
|
|
principalId: uuid("principal_id").notNull().references(() => collaborationPrincipals.id), origin: text("origin").notNull(),
|
|
bindingId: uuid("binding_id").references(() => collaborationRemoteBindings.id, { onDelete: "cascade" }),
|
|
roomId: uuid("room_id").references(() => collaborationRooms.id, { onDelete: "cascade" }), action: text("action").notNull(), payload: jsonb("payload").notNull(),
|
|
status: text("status").notNull().default("pending"), attempts: integer("attempts").notNull().default(0),
|
|
nextAttemptAt: time("next_attempt_at").notNull().defaultNow(), leaseUntil: time("lease_until"), lastError: text("last_error"), createdAt: time("created_at").notNull().defaultNow(),
|
|
}, t => ({ pending: index("collaboration_outbox_pending").on(t.nextAttemptAt).where(sql`${t.status}='pending'`),
|
|
status: check("collaboration_outbox_status_check", sql`${t.status} IN ('pending','sent','failed')`) }))
|