@supabase/lite 0.0.1 → 0.2.1-next.1

package/dist/db/postgres/PostgresConnection.js

@@ -1,9 +1,4 @@
-import postgres from 'postgres';
-import { PostgresJSDialect } from 'kysely-postgres-js';
-import { Kysely, sql } from 'kysely';
-import { Connection, invariant, RelationNotFoundError } from '@supabase/lite';
-
-try {
+import x from'postgres';import {PostgresJSDialect}from'kysely-postgres-js';import {Kysely,sql}from'kysely';import {Connection,invariant,RelationNotFoundError}from'@supabase/lite';try {
    /**
     * Adding this to avoid warnings from node:sqlite being experimental
     */
@@ -13,20 +8,8 @@ try {
       return emitWarning(warning, ...args);
    };
 } catch {}
-var ForceRollback = class extends Error {
-  constructor(result) {
-    super("Force rollback");
-    this.result = result;
-  }
-};
-var BasePostgresConnection = class extends Connection {
-  dialect = "postgres";
-  constructor(options) {
-    super({
-      ...options,
-      // @todo: this should be handled by the auth schema
-      baseSchema: `
-            ${options.baseSchema ?? ""}
+var E=class extends Error{constructor(o){super("Force rollback");this.result=o;}},N=class extends Connection{dialect="postgres";constructor(n){super({...n,baseSchema:`
+            ${n.baseSchema??""}
          
             CREATE SCHEMA IF NOT EXISTS auth;
 
@@ -52,16 +35,7 @@ var BasePostgresConnection = class extends Connection {
                 '{}'
               )::jsonb;
             $$ LANGUAGE SQL STABLE;
-         `
-    });
-  }
-  async introspect(options) {
-    invariant(typeof options === "object" || options === void 0, "options must be an object");
-    const useCache = options?.useCache ?? false;
-    const cached = await this.readCachedIntrospection({ useCache });
-    if (cached) return cached;
-    try {
-      const tblResult = await sql`
+         `});}async introspect(n){invariant(typeof n=="object"||n===void 0,"options must be an object");let o=n?.useCache??false,t=await this.readCachedIntrospection({useCache:o});if(t)return t;try{let c=(await sql`
             SELECT
                tbls.table_name AS "name",
                tbls.table_schema AS "schema",
@@ -74,17 +48,7 @@ var BasePostgresConnection = class extends Connection {
                tbls.table_schema NOT IN ('information_schema', 'pg_catalog')
                AND tbls.table_type IN ('BASE TABLE', 'FOREIGN TABLE')
             ORDER BY tbls.table_schema, tbls.table_name
-         `.execute(this.kysely);
-      const tables = tblResult.rows.map((r) => ({
-        name: r.name,
-        schema: r.schema,
-        type: "table",
-        rows: Number(r.rows),
-        sql: "",
-        engine: "",
-        collation: ""
-      }));
-      const colResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({name:e.name,schema:e.schema,type:"table",rows:Number(e.rows),sql:"",engine:"",collation:""})),l=(await sql`
             SELECT
                cols.table_name AS "table",
                cols.table_schema AS "schema",
@@ -136,25 +100,7 @@ var BasePostgresConnection = class extends Connection {
                LEFT JOIN pg_catalog.pg_type pt ON pt.oid = attr.atttypid
             WHERE cols.table_schema NOT IN ('information_schema', 'pg_catalog')
             ORDER BY cols.table_schema, cols.table_name, cols.ordinal_position
-         `.execute(this.kysely);
-      const columns = colResult.rows.map((r) => ({
-        table: r.table,
-        schema: r.schema,
-        name: r.name,
-        type: r.type,
-        nullable: r.nullable,
-        default_value: r.default_value,
-        is_primary_key: false,
-        // filled below
-        ordinal_position: Number(r.ordinal_position),
-        character_maximum_length: r.character_maximum_length ?? null,
-        precision: r.precision ?? null,
-        is_identity: r.is_identity,
-        collation: r.collation,
-        pg_type: r.udt_name ?? void 0,
-        is_generated: r.is_generated ?? false
-      }));
-      const fkResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({table:e.table,schema:e.schema,name:e.name,type:e.type,nullable:e.nullable,default_value:e.default_value,is_primary_key:!1,ordinal_position:Number(e.ordinal_position),character_maximum_length:e.character_maximum_length??null,precision:e.precision??null,is_identity:e.is_identity,collation:e.collation,pg_type:e.udt_name??void 0,is_generated:e.is_generated??!1})),p=(await sql`
             SELECT
                ns.nspname AS "schema",
                cl.relname AS "table",
@@ -179,19 +125,7 @@ var BasePostgresConnection = class extends Connection {
                'information_schema'
                , 'pg_catalog')
             ORDER BY ns.nspname, cl.relname, con.conname, cols.ord
-         `.execute(this.kysely);
-      const fks = fkResult.rows.map((r) => ({
-        table: r.table,
-        column: r.column,
-        schema: r.schema,
-        ref_table: r.ref_table,
-        ref_column: r.ref_column,
-        foreign_key_name: r.foreign_key_name,
-        fk_def: r.fk_def,
-        on_update: "",
-        on_delete: ""
-      }));
-      const pkResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({table:e.table,column:e.column,schema:e.schema,ref_table:e.ref_table,ref_column:e.ref_column,foreign_key_name:e.foreign_key_name,fk_def:e.fk_def,on_update:"",on_delete:""})),r=(await sql`
             SELECT
                ns.nspname AS "schema",
                cl.relname AS "table",
@@ -208,20 +142,7 @@ var BasePostgresConnection = class extends Connection {
                'information_schema'
                , 'pg_catalog')
             GROUP BY ns.nspname, cl.relname, con.oid
-         `.execute(this.kysely);
-      const pks = pkResult.rows.map((r) => ({
-        table: r.table,
-        columns: r.columns,
-        schema: r.schema,
-        field_count: r.columns.length
-      }));
-      const pkSet = new Set(
-        pks.flatMap((pk) => pk.columns.map((col) => `${pk.schema}.${pk.table}.${col}`))
-      );
-      for (const col of columns) {
-        col.is_primary_key = pkSet.has(`${col.schema}.${col.table}.${col.name}`);
-      }
-      const idxResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({table:e.table,columns:e.columns,schema:e.schema,field_count:e.columns.length})),S=new Set(r.flatMap(e=>e.columns.map(b=>`${e.schema}.${e.table}.${b}`)));for(let e of l)e.is_primary_key=S.has(`${e.schema}.${e.table}.${e.name}`);let g=(await sql`
             SELECT
                tnsp.nspname AS "schema",
                cl.relname AS "table",
@@ -242,15 +163,7 @@ var BasePostgresConnection = class extends Connection {
                'information_schema'
                , 'pg_catalog')
             GROUP BY tnsp.nspname, cl.relname, ic.relname, idx.indisunique
-         `.execute(this.kysely);
-      const indexes = idxResult.rows.map((r) => ({
-        table: r.table,
-        name: r.name,
-        unique: r.unique,
-        columns: r.columns,
-        schema: r.schema
-      }));
-      const viewResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({table:e.table,name:e.name,unique:e.unique,columns:e.columns,schema:e.schema})),O=(await sql`
             SELECT
                views.schemaname AS "schema",
                views.viewname AS "name",
@@ -258,13 +171,7 @@ var BasePostgresConnection = class extends Connection {
             FROM pg_views views
             WHERE views.schemaname NOT IN ('information_schema', 'pg_catalog')
             ORDER BY views.schemaname, views.viewname
-         `.execute(this.kysely);
-      const views = viewResult.rows.map((r) => ({
-        name: r.name,
-        schema: r.schema,
-        sql: r.sql ?? ""
-      }));
-      const chkResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({name:e.name,schema:e.schema,sql:e.sql??""})),A=(await sql`
             SELECT
                n.nspname AS "schema",
                cl.relname AS "table",
@@ -276,13 +183,7 @@ var BasePostgresConnection = class extends Connection {
                c.contype = 'c'
                AND n.nspname NOT IN ('information_schema', 'pg_catalog')
             ORDER BY n.nspname, cl.relname
-         `.execute(this.kysely);
-      const checkConstraints = chkResult.rows.map((r) => ({
-        schema: r.schema,
-        table: r.table,
-        expression: r.expression ?? ""
-      }));
-      const uqResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({schema:e.schema,table:e.table,expression:e.expression??""})),h=(await sql`
             SELECT
                n.nspname AS "schema",
                cl.relname AS "table",
@@ -297,14 +198,7 @@ var BasePostgresConnection = class extends Connection {
                c.contype = 'u'
                AND n.nspname NOT IN ('information_schema', 'pg_catalog')
             GROUP BY n.nspname, cl.relname, c.conname, c.oid
-         `.execute(this.kysely);
-      const uniqueConstraints = uqResult.rows.map((r) => ({
-        schema: r.schema,
-        table: r.table,
-        name: r.name,
-        columns: r.columns
-      }));
-      const commentResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({schema:e.schema,table:e.table,name:e.name,columns:e.columns})),f=(await sql`
             SELECT
                n.nspname AS "schema",
                cl.relname AS "table",
@@ -319,14 +213,7 @@ var BasePostgresConnection = class extends Connection {
                n.nspname NOT IN ('information_schema', 'pg_catalog')
                AND cl.relkind IN ('r', 'v', 'm', 'f', 'p')
                AND d.description IS NOT NULL
-         `.execute(this.kysely);
-      const comments = commentResult.rows.map((r) => ({
-        schema: r.schema,
-        table: r.table,
-        column: r.column ?? void 0,
-        text: r.text
-      }));
-      const enumResult = await sql`
+         `.execute(this.kysely)).rows.map(e=>({schema:e.schema,table:e.table,column:e.column??void 0,text:e.text})),R=await sql`
             SELECT
                n.nspname AS "schema",
                t.typname AS "type",
@@ -337,8 +224,7 @@ var BasePostgresConnection = class extends Connection {
                JOIN pg_namespace n ON n.oid = t.typnamespace
             WHERE n.nspname NOT IN ('pg_catalog', 'information_schema')
             GROUP BY n.nspname, t.typname
-         `.execute(this.kysely);
-      const compositeResult = await sql`
+         `.execute(this.kysely),T=await sql`
             SELECT
                n.nspname AS "schema",
                t.typname AS "type",
@@ -360,76 +246,9 @@ var BasePostgresConnection = class extends Connection {
                AND NOT a.attisdropped
                AND n.nspname NOT IN ('pg_catalog', 'information_schema')
             GROUP BY n.nspname, t.typname
-         `.execute(this.kysely);
-      const customTypes = [
-        ...enumResult.rows.map((r) => ({
-          schema: r.schema,
-          type: r.type,
-          kind: "enum",
-          values: r.values
-        })),
-        ...compositeResult.rows.map((r) => ({
-          schema: r.schema,
-          type: r.type,
-          kind: "composite",
-          fields: r.fields
-        }))
-      ];
-      const dbResult = await sql`SELECT current_database() AS "name"`.execute(this.kysely);
-      const dbName = dbResult.rows[0]?.name ?? "postgres";
-      const result = {
-        tables,
-        columns,
-        indexes,
-        foreign_keys: fks,
-        primary_keys: pks,
-        views,
-        triggers: [],
-        check_constraints: checkConstraints,
-        unique_constraints: uniqueConstraints,
-        comments,
-        custom_types: customTypes,
-        database_name: dbName,
-        version: ""
-      };
-      await this.writeCachedIntrospection(result, { useDriver: useCache });
-      return result;
-    } catch (e) {
-      console.error("Introspection failed:", e);
-      return {
-        tables: [],
-        columns: [],
-        indexes: [],
-        foreign_keys: [],
-        primary_keys: [],
-        views: [],
-        triggers: [],
-        check_constraints: [],
-        unique_constraints: [],
-        comments: [],
-        custom_types: [],
-        database_name: "postgres",
-        version: ""
-      };
-    }
-  }
-  rlsState = "unknown";
-  /**
-   * Detect if any table has RLS enabled, and if so ensure anon/authenticated
-   * roles exist with default privileges. Runs once per connection.
-   */
-  async ensureRlsContext() {
-    if (this.rlsState !== "unknown") return this.rlsState === "active";
-    const result = await sql`
+         `.execute(this.kysely),L=[...R.rows.map(e=>({schema:e.schema,type:e.type,kind:"enum",values:e.values})),...T.rows.map(e=>({schema:e.schema,type:e.type,kind:"composite",fields:e.fields}))],I=(await sql`SELECT current_database() AS "name"`.execute(this.kysely)).rows[0]?.name??"postgres",y={tables:c,columns:l,indexes:g,foreign_keys:p,primary_keys:r,views:O,triggers:[],check_constraints:A,unique_constraints:h,comments:f,custom_types:L,database_name:I,version:""};return await this.writeCachedIntrospection(y,{useDriver:o}),y}catch(s){return console.error("Introspection failed:",s),{tables:[],columns:[],indexes:[],foreign_keys:[],primary_keys:[],views:[],triggers:[],check_constraints:[],unique_constraints:[],comments:[],custom_types:[],database_name:"postgres",version:""}}}rlsState="unknown";async ensureRlsContext(){if(this.rlsState!=="unknown")return this.rlsState==="active";let n=await sql`
          SELECT count(*) ::text as count FROM pg_class WHERE relrowsecurity = true
-      `.execute(this.kysely);
-    const hasRls = Number(result.rows[0]?.count ?? 0) > 0;
-    if (!hasRls) {
-      this.rlsState = "inactive";
-      return false;
-    }
-    this.rlsState = "active";
-    await sql`
+      `.execute(this.kysely);if(!(Number(n.rows[0]?.count??0)>0))return this.rlsState="inactive",false;this.rlsState="active",await sql`
          DO $$ BEGIN
             IF NOT EXISTS (SELECT FROM pg_roles WHERE rolname = 'anon') THEN
                CREATE ROLE anon NOLOGIN;
@@ -438,174 +257,8 @@ var BasePostgresConnection = class extends Connection {
                CREATE ROLE authenticated NOLOGIN;
             END IF;
          END $$
-      `.execute(this.kysely);
-    const schemas = await sql`
+      `.execute(this.kysely);let t=await sql`
          SELECT DISTINCT n.nspname FROM pg_class c
          JOIN pg_namespace n ON n.oid = c.relnamespace
          WHERE c.relrowsecurity = true
-      `.execute(this.kysely);
-    for (const { nspname } of schemas.rows) {
-      await sql`GRANT USAGE ON SCHEMA ${sql.ref(nspname)} TO anon, authenticated`.execute(
-        this.kysely
-      );
-      await sql.raw(`GRANT ALL ON ALL TABLES IN SCHEMA "${nspname}" TO anon, authenticated`).execute(this.kysely);
-      await sql.raw(`GRANT ALL ON ALL SEQUENCES IN SCHEMA "${nspname}" TO anon, authenticated`).execute(this.kysely);
-      await sql.raw(
-        `ALTER DEFAULT PRIVILEGES IN SCHEMA "${nspname}" GRANT ALL ON TABLES TO anon, authenticated`
-      ).execute(this.kysely);
-      await sql.raw(
-        `ALTER DEFAULT PRIVILEGES IN SCHEMA "${nspname}" GRANT ALL ON SEQUENCES TO anon, authenticated`
-      ).execute(this.kysely);
-    }
-    return true;
-  }
-  async withContext(vars, fn, opts) {
-    const forceRollback = opts?.forceRollback === true;
-    if (!forceRollback && !vars) return fn(this.kysely);
-    const rlsActive = vars ? await this.ensureRlsContext() : false;
-    if (!forceRollback && !rlsActive) return fn(this.kysely);
-    try {
-      return await this.kysely.transaction().execute(async (trx) => {
-        if (vars && rlsActive) {
-          const auth = vars.auth;
-          const role = auth?.role || "anon";
-          const uid = String(auth?.uid ?? "");
-          const jwt = auth?.jwt;
-          await sql`SELECT set_config('role', ${role}, true)`.execute(trx);
-          await sql`SELECT set_config('request.jwt.claim.sub', ${uid}, true)`.execute(trx);
-          await sql`SELECT set_config('request.jwt.claim.role', ${role}, true)`.execute(trx);
-          if (jwt && typeof jwt === "object") {
-            await sql`SELECT set_config('request.jwt.claims', ${JSON.stringify(jwt)}, true)`.execute(
-              trx
-            );
-          }
-        }
-        const result = await fn(trx);
-        if (forceRollback) throw new ForceRollback(result);
-        return result;
-      });
-    } catch (error) {
-      if (error instanceof ForceRollback) return error.result;
-      throw error;
-    }
-  }
-  async onPostgrestAST(ast) {
-    if (!ast.from) return ast;
-    const schema = ast.schema ?? "public";
-    const relation = ast.from;
-    const introspection = await this.introspect({ useCache: true });
-    if (!introspection.tables.some((t) => t.name === relation && t.schema === schema) && !introspection.views.some((v) => v.name === relation && v.schema === schema)) {
-      throw new RelationNotFoundError(schema, relation);
-    }
-    return ast;
-  }
-  async transaction(statements, opts) {
-    await this.exec("BEGIN");
-    try {
-      for (const statement of statements) {
-        await this.exec(statement);
-      }
-      await this.exec("COMMIT");
-    } catch (error) {
-      await this.exec("ROLLBACK");
-      throw error;
-    } finally {
-      if (opts?.intent === "migration") {
-        await this.clearSchemaCache();
-      }
-    }
-  }
-};
-
-// src/db/postgres/PostgresConnection.ts
-var PostgresConnection = class extends BasePostgresConnection {
-  driver;
-  dialect = "postgres";
-  constructor(config) {
-    super(config);
-    this.driver = postgres(config.url ?? "", {
-      types: {
-        // Parse bigint/int8 (OID 20) as JS number instead of string
-        bigint: {
-          to: 20,
-          from: [20],
-          serialize: (x) => x.toString(),
-          parse: (x) => {
-            const n = Number(x);
-            if (Number.isSafeInteger(n)) return n;
-            return BigInt(x);
-          }
-        },
-        // json (OID 114) and jsonb (OID 3802): postgres.js leaves both as text by default.
-        // PostgREST returns them as parsed JSON, including json_agg() in embed aggregates.
-        json: {
-          to: 114,
-          from: [114],
-          serialize: (x) => JSON.stringify(x),
-          parse: (x) => JSON.parse(x)
-        },
-        jsonb: {
-          to: 3802,
-          from: [3802],
-          serialize: (x) => JSON.stringify(x),
-          parse: (x) => JSON.parse(x)
-        },
-        // numeric (OID 1700): postgres.js returns as string for precision safety.
-        // PostgREST returns as JSON number — coerce when round-trip is exact, else keep string.
-        numeric: {
-          to: 1700,
-          from: [1700],
-          serialize: (x) => String(x),
-          parse: (x) => {
-            const n = Number(x);
-            if (Number.isFinite(n) && String(n) === x) return n;
-            return x;
-          }
-        },
-        // Date/time OIDs: keep as raw strings to mirror SQLite, which stores ISO text.
-        // Auth/storage code already uses Date.parse() on these, and the spec timestamp
-        // sentinel only matches strings.
-        date: {
-          to: 1082,
-          from: [1082],
-          serialize: (x) => x,
-          parse: (x) => x
-        },
-        time: {
-          to: 1083,
-          from: [1083],
-          serialize: (x) => x,
-          parse: (x) => x
-        },
-        timestamp: {
-          to: 1114,
-          from: [1114],
-          serialize: (x) => x,
-          parse: (x) => x
-        },
-        timestamptz: {
-          to: 1184,
-          from: [1184],
-          serialize: (x) => x,
-          parse: (x) => x
-        },
-        timetz: {
-          to: 1266,
-          from: [1266],
-          serialize: (x) => x,
-          parse: (x) => x
-        }
-      }
-    });
-    const dialect = new PostgresJSDialect({ postgres: this.driver });
-    this.kysely = new Kysely({ dialect });
-  }
-  async close() {
-    await this.driver.end();
-  }
-};
-function createPostgresConnection(config) {
-  return new PostgresConnection(config);
-}
-
-export { PostgresConnection, createPostgresConnection };
+      `.execute(this.kysely);for(let{nspname:s}of t.rows)await sql`GRANT USAGE ON SCHEMA ${sql.ref(s)} TO anon, authenticated`.execute(this.kysely),await sql.raw(`GRANT ALL ON ALL TABLES IN SCHEMA "${s}" TO anon, authenticated`).execute(this.kysely),await sql.raw(`GRANT ALL ON ALL SEQUENCES IN SCHEMA "${s}" TO anon, authenticated`).execute(this.kysely),await sql.raw(`ALTER DEFAULT PRIVILEGES IN SCHEMA "${s}" GRANT ALL ON TABLES TO anon, authenticated`).execute(this.kysely),await sql.raw(`ALTER DEFAULT PRIVILEGES IN SCHEMA "${s}" GRANT ALL ON SEQUENCES TO anon, authenticated`).execute(this.kysely);return  true}async withContext(n,o,t){let s=t?.forceRollback===true;if(!s&&!n)return o(this.kysely);let c=n?await this.ensureRlsContext():false;if(!s&&!c)return o(this.kysely);try{return await this.kysely.transaction().execute(async i=>{if(n&&c){let u=n.auth,p=u?.role||"anon",d=String(u?.uid??""),r=u?.jwt;await sql`SELECT set_config('role', ${p}, true)`.execute(i),await sql`SELECT set_config('request.jwt.claim.sub', ${d}, true)`.execute(i),await sql`SELECT set_config('request.jwt.claim.role', ${p}, true)`.execute(i),r&&typeof r=="object"&&await sql`SELECT set_config('request.jwt.claims', ${JSON.stringify(r)}, true)`.execute(i);}let l=await o(i);if(s)throw new E(l);return l})}catch(i){if(i instanceof E)return i.result;throw i}}async onPostgrestAST(n){if(!n.from)return n;let o=n.schema??"public",t=n.from,s=await this.introspect({useCache:true});if(!s.tables.some(c=>c.name===t&&c.schema===o)&&!s.views.some(c=>c.name===t&&c.schema===o))throw new RelationNotFoundError(o,t);return n}async transaction(n,o){await this.exec("BEGIN");try{for(let t of n)await this.exec(t);await this.exec("COMMIT");}catch(t){throw await this.exec("ROLLBACK"),t}finally{o?.intent==="migration"&&await this.clearSchemaCache();}}};var _=class extends N{driver;dialect="postgres";constructor(n){super(n),this.driver=x(n.url??"",{types:{bigint:{to:20,from:[20],serialize:t=>t.toString(),parse:t=>{let s=Number(t);return Number.isSafeInteger(s)?s:BigInt(t)}},json:{to:114,from:[114],serialize:t=>JSON.stringify(t),parse:t=>JSON.parse(t)},jsonb:{to:3802,from:[3802],serialize:t=>JSON.stringify(t),parse:t=>JSON.parse(t)},numeric:{to:1700,from:[1700],serialize:t=>String(t),parse:t=>{let s=Number(t);return Number.isFinite(s)&&String(s)===t?s:t}},date:{to:1082,from:[1082],serialize:t=>t,parse:t=>t},time:{to:1083,from:[1083],serialize:t=>t,parse:t=>t},timestamp:{to:1114,from:[1114],serialize:t=>t,parse:t=>t},timestamptz:{to:1184,from:[1184],serialize:t=>t,parse:t=>t},timetz:{to:1266,from:[1266],serialize:t=>t,parse:t=>t}}});let o=new PostgresJSDialect({postgres:this.driver});this.kysely=new Kysely({dialect:o});}async close(){await this.driver.end();}};function z(m){return new _(m)}export{_ as PostgresConnection,z as createPostgresConnection};