Something went wrong. Try again.
a tool for shared writing and social publishing
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120import { drizzle } from "drizzle-orm/node-postgres";import { sql, type SQL } from "drizzle-orm";import { v7 } from "uuid";import { pool } from "supabase/pool";
export type LeafletFact = { entity: string; attribute: string; data: unknown };
// Insert a new leaflet: an entity set holding `entityIds`, a permission token// with full rights rooted at `rootEntityId`, and `facts`. All inserts run in a// single statement (FK checks fire at end-of-statement and see rows from// sibling CTEs); callers can append a tail CTE (e.g. to link the new token to// something) for the same round trip.export async function insertLeaflet({ rootEntityId, entityIds, facts, title = null, description = null, tailCte,}: { rootEntityId: string; entityIds: string[]; facts: LeafletFact[]; title?: string | null; description?: string | null; tailCte?: (ids: { permTokenId: string; rootEntityId: string }) => SQL;}): Promise<{ permTokenId: string }> { const entitySetId = v7(); const permTokenId = v7();
const entityValues = sql.join( entityIds.map((id) => sql`(${id}, ${entitySetId})`), sql`, `, ); const factValues = sql.join( facts.map( (f) => sql`(${v7()}, ${f.entity}, ${f.attribute}, ${JSON.stringify(f.data)}::jsonb)`, ), sql`, `, ); const tail = tailCte?.({ permTokenId, rootEntityId }) ?? sql``;
const client = await pool.connect(); const db = drizzle(client); try { await db.execute(sql` WITH new_set AS ( INSERT INTO entity_sets (id) VALUES (${entitySetId}) ), new_entities AS ( INSERT INTO entities (id, set) VALUES ${entityValues} ), new_token AS ( INSERT INTO permission_tokens (id, root_entity, title, description) VALUES (${permTokenId}, ${rootEntityId}, ${title}, ${description}) ), new_rights AS ( INSERT INTO permission_token_rights (token, entity_set, read, write, create_token, change_entity_set) VALUES (${permTokenId}, ${entitySetId}, true, true, true, true) )${ facts.length > 0 ? sql`, new_facts AS ( INSERT INTO facts (id, entity, attribute, data) VALUES ${factValues} )` : sql`` }${tail} SELECT 1 `); } finally { client.release(); }
return { permTokenId };}
// Add entities and facts to an existing leaflet's entity set, in one// statement for the same reason as above.export async function appendToLeaflet({ entitySetId, entityIds, facts,}: { entitySetId: string; entityIds: string[]; facts: LeafletFact[];}): Promise<void> { if (entityIds.length === 0 && facts.length === 0) return; const entityValues = sql.join( entityIds.map((id) => sql`(${id}, ${entitySetId})`), sql`, `, ); const factValues = sql.join( facts.map( (f) => sql`(${v7()}, ${f.entity}, ${f.attribute}, ${JSON.stringify(f.data)}::jsonb)`, ), sql`, `, ); const client = await pool.connect(); const db = drizzle(client); try { await db.execute(sql` WITH new_entities AS ( INSERT INTO entities (id, set) VALUES ${entityValues} )${ facts.length > 0 ? sql`, new_facts AS ( INSERT INTO facts (id, entity, attribute, data) VALUES ${factValues} )` : sql`` } SELECT 1 `); } finally { client.release(); }}