@supabase/lite 0.6.1-next.5 → 0.6.2-next.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LIMITATIONS.md +1 -1
- package/README.md +15 -6
- package/STATUS.md +5 -3
- package/dist/cli/index.js +105 -105
- package/dist/cli/lib.js +35 -35
- package/dist/db/postgres/pglite/PgliteConnection.js +23 -23
- package/dist/index.d.ts +1 -0
- package/dist/index.js +36 -36
- package/package.json +1 -1
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {PGlite}from'@electric-sql/pglite';export{PGlite}from'@electric-sql/pglite';import {Kysely,sql,PostgresQueryCompiler,PostgresIntrospector,PostgresAdapter,CompiledQuery}from'kysely';import {Connection,invariant,RelationNotFoundError,DataLossError}from'@supabase/lite';import {isn}from'@electric-sql/pglite/contrib/isn';import {ltree}from'@electric-sql/pglite/contrib/ltree';import {file_fdw}from'@electric-sql/pglite/contrib/file_fdw';import {pgcrypto}from'@electric-sql/pglite/contrib/pgcrypto';try {
|
|
1
|
+
import {PGlite}from'@electric-sql/pglite';export{PGlite}from'@electric-sql/pglite';import oe from'node:fs';import {Kysely,sql,PostgresQueryCompiler,PostgresIntrospector,PostgresAdapter,CompiledQuery}from'kysely';import {Connection,invariant,RelationNotFoundError,DataLossError}from'@supabase/lite';import {isn}from'@electric-sql/pglite/contrib/isn';import {ltree}from'@electric-sql/pglite/contrib/ltree';import {file_fdw}from'@electric-sql/pglite/contrib/file_fdw';import {pgcrypto}from'@electric-sql/pglite/contrib/pgcrypto';try {
|
|
2
2
|
/**
|
|
3
3
|
* Adding this to avoid warnings from node:sqlite being experimental
|
|
4
4
|
*/
|
|
@@ -8,13 +8,13 @@ import {PGlite}from'@electric-sql/pglite';export{PGlite}from'@electric-sql/pglit
|
|
|
8
8
|
return emitWarning(warning, ...args);
|
|
9
9
|
};
|
|
10
10
|
} catch {}
|
|
11
|
-
|
|
11
|
+
var v=class{#e;#t;constructor(e,t){this.#e=e,this.#t=t;}async acquireConnection(){return new D(this.#e)}async beginTransaction(e,t){await e.executeQuery(CompiledQuery.raw("BEGIN"));}async commitTransaction(e){await e.executeQuery(CompiledQuery.raw("COMMIT"));}async rollbackTransaction(e){await e.executeQuery(CompiledQuery.raw("ROLLBACK"));}async destroy(){this.#t&&await this.#e.close();}async init(){}async releaseConnection(e){}},D=class{#e;constructor(e){this.#e=e;}async executeQuery(e){let t=await this.#e.query(e.sql,[...e.parameters]),s=t.affectedRows;return {rows:t.rows,numAffectedRows:typeof s=="number"?BigInt(s):void 0}}async*streamQuery(){throw new Error("PGlite does not support streaming.")}},T=class{client;#e=true;constructor(e,t){let s={...t};if(typeof e=="object"&&e instanceof PGlite){this.client=e,this.#e=false;return}typeof e=="string"?s={dataDir:e,...s}:s=e??{},this.client=new PGlite(s);}dialect={createAdapter:()=>new PostgresAdapter,createDriver:()=>new v(this.client,this.#e),createIntrospector:e=>new PostgresIntrospector(e),createQueryCompiler:()=>new PostgresQueryCompiler}};var O=["anon","authenticated","service_role"];function Ne(n){return n==="service_role"?"NOLOGIN BYPASSRLS":"NOLOGIN"}function C(n){return n.size===0?"":`DO $$ BEGIN${[...n].map(t=>`
|
|
12
12
|
IF NOT EXISTS (SELECT FROM pg_roles WHERE rolname = '${t}') THEN
|
|
13
|
-
CREATE ROLE ${t} ${
|
|
13
|
+
CREATE ROLE ${t} ${Ne(t)};
|
|
14
14
|
END IF;`).join("")}
|
|
15
15
|
END $$;
|
|
16
|
-
`}var K=C(new Set(O));function A(n,e){return n?n.quoted?n.value===e:n.value.toLowerCase()===e:false}function
|
|
17
|
-
`;)t++;continue}if(s==="/"&&
|
|
16
|
+
`}var K=C(new Set(O));function A(n,e){return n?n.quoted?n.value===e:n.value.toLowerCase()===e:false}function y(n){return n?.quoted?n.value:n?.value.toLowerCase()}function we(n,e){let t="",s=e+1;for(;s<n.length;){if(n[s]==='"'&&n[s+1]==='"'){t+='"',s+=2;continue}if(n[s]==='"')return {token:{value:t,quoted:true},end:s+1};t+=n[s],s++;}return {token:{value:t,quoted:true},end:s}}function Ae(n,e){let t=e+1;for(;t<n.length;){if(n[t]==="'"&&n[t+1]==="'"){t+=2;continue}if(n[t]==="'")return t+1;t++;}return t}function Re(n,e){let t=/^\$[A-Za-z_][A-Za-z0-9_]*\$|^\$\$/.exec(n.slice(e))?.[0];if(!t)return;let s=e+t.length,o=n.indexOf(t,s);return o===-1?n.length:o+t.length}function $(n){let e=[],t=0;for(;t<n.length;){let s=n[t],o=n[t+1];if(/\s/.test(s)){t++;continue}if(s==="-"&&o==="-"){for(t+=2;t<n.length&&n[t]!==`
|
|
17
|
+
`;)t++;continue}if(s==="/"&&o==="*"){for(t+=2;t<n.length&&!(n[t]==="*"&&n[t+1]==="/");)t++;t=t<n.length?t+2:t;continue}if(s==="'"){t=Ae(n,t);continue}if(s==="$"){let a=Re(n,t);if(a!==void 0){t=a;continue}}if(s==='"'){let{token:a,end:c}=we(n,t);e.push(a),t=c;continue}if(s===","||s===";"){e.push({value:s,quoted:false}),t++;continue}if(/[A-Za-z_]/.test(s)){let a=t;for(t++;t<n.length&&/[A-Za-z0-9_$]/.test(n[t]);)t++;e.push({value:n.slice(a,t).toLowerCase(),quoted:false});continue}t++;}return e}function Q(n,e){let t=y(n[e]),s=y(n[e+1]);if((t==="create"||t==="alter")&&(s==="role"||s==="user"||s==="group"))return n[e+2]}function L(n){let e=new Set,t=$(n);for(let s=0;s<t.length;s++){if(y(t[s])!=="create")continue;let o=Q(t,s);for(let a of O)A(o,a)&&e.add(a);}return e}function P(n,e,t){let s=new Set([";","as","by","check","for","granted","in","on","option","using","where","with"]);for(let o=e;o<n.length;o++){let a=y(n[o]);if(!a||s.has(a))return;for(let c of O)A(n[o],c)&&t.add(c);}}function be(n){let e=[],t=[];for(let s of n){if(y(s)===";"){t.length>0&&e.push(t),t=[];continue}t.push(s);}return t.length>0&&e.push(t),e}function j(n,e){return n.some(t=>y(t)===e)}function Te(n){let e=new Set,t=be($(n));for(let s of t){let o=y(s[0]);for(let r=0;r<s.length;r++){let l=y(s[r]);l==="to"&&(j(s,"grant")||j(s,"policy")||y(s[r-1])==="owner")&&P(s,r+1,e),l==="from"&&o==="revoke"&&P(s,r+1,e);}let a=s.findIndex(r=>y(r)==="to"),c=s.findIndex(r=>y(r)==="on");o==="grant"&&a>1&&(c===-1||a<c)&&P(s,1,e);}return e}function F(n){let e=L(n);return new Set([...Te(n)].filter(t=>!e.has(t)))}function Y(n){let e=L(n);return new Set(O.filter(t=>!e.has(t)))}function V(n,e){let t=$(n);if(t.length===0)return false;let s=Q(t,0);if(s){for(let c of e)if(A(s,c))return true;return false}let o=y(t[0]),a=y(t[1]);if(o==="drop"&&(a==="role"||a==="user"||a==="group")){for(let c of t.slice(2))for(let r of e)if(A(c,r))return true}if(o==="comment"&&y(t[1])==="on"&&y(t[2])==="role"){for(let c of e)if(A(t[3],c))return true}return false}function M(n){let e=n.trim();return e.startsWith('"')&&e.endsWith('"')||e.startsWith("`")&&e.endsWith("`")||e.startsWith("[")&&e.endsWith("]")?e.slice(1,-1):e}function Oe(n){let e=[],t="",s=0,o=null;for(let a=0;a<n.length;a++){let c=n[a];if(o){t+=c,c===o&&(o=null);continue}if(c==='"'||c==="`"||c==="'"){o=c,t+=c;continue}if(c==="("?s++:c===")"&&s>0&&s--,c===","&&s===0){e.push(t.trim()),t="";continue}t+=c;}return t.trim()&&e.push(t.trim()),e}function Ce(n){let e=n.match(/^(.*?)\s+as\s+(.+)$/i),t=e?e[1].trim():n.trim(),s=e?e[2].trim():void 0;if(!s){let c=n.match(/^(.+?)\s+((?:"[^"]+"|`[^`]+`|\[[^\]]+\]|\w+))$/);c&&!/[()]/.test(c[1])&&(t=c[1].trim(),s=c[2].trim());}let o=t.split(".").map(M);if(o.length>2)return null;let a=o.at(-1);return !a||!/^[A-Za-z_][\w$]*$/.test(a)?null:{baseColumn:a,viewColumn:s?M(s):a}}function z(n){return !!(n&&/[A-Za-z0-9_$]/.test(n))}function Le(n){let e=/\bselect\b/i.exec(n);if(!e)return null;let t=0,s=null;for(let o=e.index+e[0].length;o<n.length;o++){let a=n[o];if(s){a===s&&(s=null);continue}if(a==='"'||a==="`"||a==="'"){s=a;continue}if(a==="("){t++;continue}if(a===")"&&t>0){t--;continue}if(t===0&&n.slice(o,o+4).toLowerCase()==="from"&&!z(n[o-1])&&!z(n[o+4]))return {selectList:n.slice(e.index+e[0].length,o).trim(),fromRest:n.slice(o+4).trim()}}return null}function Ie(n){let e=0,t=[];for(;e<n.length;){for(;/\s/.test(n[e]??"");)e++;let s=n[e],o="";if(s==='"'||s==="`"){let r=n.indexOf(s,e+1);if(r===-1)return null;o=n.slice(e,r+1),e=r+1;}else if(s==="["){let r=n.indexOf("]",e+1);if(r===-1)return null;o=n.slice(e,r+1),e=r+1;}else {let r=/^[A-Za-z_][\w$]*/.exec(n.slice(e));if(!r)return null;o=r[0],e+=o.length;}t.push(o);let a=n.slice(e),c=/^\s*\./.exec(a);if(!c)break;e+=c[0].length;}return t.length===0?null:{raw:t.join("."),rest:n.slice(e).trim()}}function X(n,e,t){let s=n.get(e)??[];s.includes(t)||s.push(t),n.set(e,s);}function xe(n,e){if(!n)return null;let t=Le(n);if(!t)return null;let s=Ie(t.fromRest);if(!s)return null;let{raw:o,rest:a}=s;if(/\b(join|union|intersect|except)\b/i.test(a))return null;let c=o.split(".").map(M).at(-1);if(!c)return null;let r=new Map,l=e.get(c)??new Set;for(let f of Oe(t.selectList)){let u=f.trim();if(u==="*"||u.endsWith(".*")){for(let d of l)X(r,d,d);continue}let p=Ce(f);p&&X(r,p.baseColumn,p.viewColumn);}return r.size===0?null:{baseTable:c,columns:r}}function ke(n,e){let t=new Map;for(let[s,o]of e){let a=[];for(let c of o){let r=n.get(c);if(r)for(let l of r)a.includes(l)||a.push(l);}a.length>0&&t.set(s,a);}return t}function Be(n){let e=new Map,t=new Map;for(let r of n.columns??[]){let l=t.get(r.table)??new Set;l.add(r.name),t.set(r.table,l);}let s=(r,l)=>{let f=e.get(r)??[];f.some(u=>u.name===l.name&&u.schema===l.schema)||f.push(l),e.set(r,f);};for(let r of n.tables??[]){if(r.type!=="table")continue;let l=t.get(r.name);l&&s(r.name,{name:r.name,schema:r.schema,columnByBaseColumn:new Map([...l].map(f=>[f,[f]]))});}let o=new Map;for(let r of n.views??[]){let l=xe(r.sql,t);if(!l)continue;let f=t.get(r.name);if(!f)continue;let u=new Map;for(let[d,h]of l.columns){let g=h.filter(_=>f.has(_));g.length>0&&u.set(d,g);}if(u.size===0)continue;let p={name:r.name,schema:r.schema,columnByBaseColumn:u,fromRelation:l.baseTable};o.set(r.name,p);}let a=16;function c(r,l,f){if(l>a)return null;let u=r.fromRelation;if(e.has(u))return {physicalBase:u,columnByBaseColumn:r.columnByBaseColumn};let p=o.get(u);if(!p||f.has(u))return null;f.add(u);let d=c(p,l+1,f);if(f.delete(u),!d)return null;let h=ke(r.columnByBaseColumn,d.columnByBaseColumn);return h.size===0?null:{physicalBase:d.physicalBase,columnByBaseColumn:h}}for(let r of o.values()){let l=c(r,0,new Set([r.name]));l&&s(l.physicalBase,{name:r.name,schema:r.schema,columnByBaseColumn:l.columnByBaseColumn});}return e}function Z(n){return `${n.foreign_key_group??""}:${n.table}.${n.column}->${n.ref_table}.${n.ref_column}`}function ve(n){let e=new Map;for(let t of n){let s=`${t.schema}.${t.table}.${t.foreign_key_name}.${t.ref_schema??""}.${t.ref_table}`,o=e.get(s);o?o.push(t):e.set(s,[t]);}return [...e.values()]}function De(n){let e=[[]];for(let t of n){let s=[];for(let o of e)for(let a of t)s.push([...o,a]);e=s;}return e}function Pe(n,e,t,s){let o=s.map(c=>c.childColumn).join(","),a=s.map(c=>c.parentColumn).join(",");return `${t}:${n.name}(${o})->${e.name}(${a})`}function ee(n){if(!n.foreign_keys?.length||!n.views?.length)return n;let e=Be(n),t=new Set(n.foreign_keys.map(Z)),s=[];for(let o of ve(n.foreign_keys)){let a=o[0],c=e.get(a.table)??[],r=e.get(a.ref_table)??[];for(let l of c)for(let f of r){if(l.schema!==f.schema)continue;let u=[];for(let p of o){let d=l.columnByBaseColumn.get(p.column)??[],h=f.columnByBaseColumn.get(p.ref_column)??[],g=d.flatMap(_=>h.map(S=>({fk:p,childColumn:_,parentColumn:S})));if(g.length===0){u.length=0;break}u.push(g);}for(let p of De(u)){let h=l.name===a.table&&f.name===a.ref_table&&p.every(g=>g.childColumn===g.fk.column&&g.parentColumn===g.fk.ref_column)?void 0:Pe(l,f,a.foreign_key_name,p);for(let g of p){let _={...g.fk,table:l.name,column:g.childColumn,schema:l.schema,ref_table:f.name,ref_column:g.parentColumn,ref_schema:f.schema,foreign_key_group:h},S=Z(_);t.has(S)||(t.add(S),s.push(_));}}}}return s.length?{...n,foreign_keys:[...n.foreign_keys,...s]}:n}var R="anon, authenticated, service_role",Me=new Set(["anon","authenticated","service_role"]);function te(n){return `"${n.replace(/"/g,'""')}"`}var N=class extends Error{constructor(t){super("Force rollback");this.result=t;}},I=class extends Connection{dialect="postgres";harnessHoldingOuterTx=false;constructor(e){super({...e,baseSchema:`
|
|
18
18
|
${e.baseSchema??""}
|
|
19
19
|
|
|
20
20
|
CREATE SCHEMA IF NOT EXISTS auth;
|
|
@@ -41,7 +41,7 @@ END $$;
|
|
|
41
41
|
'{}'
|
|
42
42
|
)::jsonb;
|
|
43
43
|
$$ LANGUAGE SQL STABLE;
|
|
44
|
-
`});}async introspect(e){invariant(typeof e=="object"||e===void 0,"options must be an object");let t=e?.useCache??false,s=await this.readCachedIntrospection({useCache:t});if(s)return s;try{let
|
|
44
|
+
`});}async introspect(e){invariant(typeof e=="object"||e===void 0,"options must be an object");let t=e?.useCache??false,s=await this.readCachedIntrospection({useCache:t});if(s)return s;try{let o=await sql`
|
|
45
45
|
SELECT
|
|
46
46
|
tbls.table_name AS "name",
|
|
47
47
|
tbls.table_schema AS "schema",
|
|
@@ -65,7 +65,7 @@ END $$;
|
|
|
65
65
|
AND pc.relispartition
|
|
66
66
|
)
|
|
67
67
|
ORDER BY tbls.table_schema, tbls.table_name
|
|
68
|
-
`.execute(this.kysely),
|
|
68
|
+
`.execute(this.kysely),a=await sql`
|
|
69
69
|
SELECT
|
|
70
70
|
mv.schemaname AS "schema",
|
|
71
71
|
mv.matviewname AS "name",
|
|
@@ -93,7 +93,7 @@ END $$;
|
|
|
93
93
|
AND a.attnum > 0
|
|
94
94
|
AND NOT a.attisdropped
|
|
95
95
|
ORDER BY n.nspname, c.relname, a.attnum
|
|
96
|
-
`.execute(this.kysely),r=
|
|
96
|
+
`.execute(this.kysely),r=o.rows.map(i=>({name:i.name,schema:i.schema,type:"table",rows:Number(i.rows),sql:"",engine:"",collation:""}));for(let i of a.rows)r.push({name:i.name,schema:i.schema,type:"table",rows:0,sql:"",engine:"",collation:""});let f=(await sql`
|
|
97
97
|
SELECT
|
|
98
98
|
cols.table_name AS "table",
|
|
99
99
|
cols.table_schema AS "schema",
|
|
@@ -177,7 +177,7 @@ END $$;
|
|
|
177
177
|
AND NOT cl.relispartition
|
|
178
178
|
AND NOT ref_cl.relispartition
|
|
179
179
|
ORDER BY ns.nspname, cl.relname, con.conname, cols.ord
|
|
180
|
-
`.execute(this.kysely)).rows.map(i=>({table:i.table,column:i.column,schema:i.schema,ref_table:i.ref_table,ref_column:i.ref_column,foreign_key_name:i.foreign_key_name,fk_def:i.fk_def,on_update:"",on_delete:""})),
|
|
180
|
+
`.execute(this.kysely)).rows.map(i=>({table:i.table,column:i.column,schema:i.schema,ref_table:i.ref_table,ref_column:i.ref_column,foreign_key_name:i.foreign_key_name,fk_def:i.fk_def,on_update:"",on_delete:""})),h=(await sql`
|
|
181
181
|
SELECT
|
|
182
182
|
ns.nspname AS "schema",
|
|
183
183
|
cl.relname AS "table",
|
|
@@ -194,7 +194,7 @@ END $$;
|
|
|
194
194
|
'information_schema'
|
|
195
195
|
, 'pg_catalog')
|
|
196
196
|
GROUP BY ns.nspname, cl.relname, con.oid
|
|
197
|
-
`.execute(this.kysely)).rows.map(i=>({table:i.table,columns:i.columns,schema:i.schema,field_count:i.columns.length})),g=new Set(
|
|
197
|
+
`.execute(this.kysely)).rows.map(i=>({table:i.table,columns:i.columns,schema:i.schema,field_count:i.columns.length})),g=new Set(h.flatMap(i=>i.columns.map(w=>`${i.schema}.${i.table}.${w}`)));for(let i of f)i.is_primary_key=g.has(`${i.schema}.${i.table}.${i.name}`);let S=(await sql`
|
|
198
198
|
SELECT
|
|
199
199
|
tnsp.nspname AS "schema",
|
|
200
200
|
cl.relname AS "table",
|
|
@@ -223,7 +223,7 @@ END $$;
|
|
|
223
223
|
FROM pg_views views
|
|
224
224
|
WHERE views.schemaname NOT IN ('information_schema', 'pg_catalog')
|
|
225
225
|
ORDER BY views.schemaname, views.viewname
|
|
226
|
-
`.execute(this.kysely)).rows.map(i=>({name:i.name,schema:i.schema,sql:i.sql??""}));for(let i of
|
|
226
|
+
`.execute(this.kysely)).rows.map(i=>({name:i.name,schema:i.schema,sql:i.sql??""}));for(let i of a.rows)J.push({name:i.name,schema:i.schema,sql:i.sql??""});let ie=(await sql`
|
|
227
227
|
SELECT
|
|
228
228
|
n.nspname AS "schema",
|
|
229
229
|
cl.relname AS "table",
|
|
@@ -235,7 +235,7 @@ END $$;
|
|
|
235
235
|
c.contype = 'c'
|
|
236
236
|
AND n.nspname NOT IN ('information_schema', 'pg_catalog')
|
|
237
237
|
ORDER BY n.nspname, cl.relname
|
|
238
|
-
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,expression:i.expression??""})),
|
|
238
|
+
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,expression:i.expression??""})),re=(await sql`
|
|
239
239
|
SELECT
|
|
240
240
|
n.nspname AS "schema",
|
|
241
241
|
cl.relname AS "table",
|
|
@@ -250,7 +250,7 @@ END $$;
|
|
|
250
250
|
c.contype = 'u'
|
|
251
251
|
AND n.nspname NOT IN ('information_schema', 'pg_catalog')
|
|
252
252
|
GROUP BY n.nspname, cl.relname, c.conname, c.oid
|
|
253
|
-
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,name:i.name,columns:i.columns})),
|
|
253
|
+
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,name:i.name,columns:i.columns})),ce=(await sql`
|
|
254
254
|
SELECT
|
|
255
255
|
n.nspname AS "schema",
|
|
256
256
|
cl.relname AS "table",
|
|
@@ -265,7 +265,7 @@ END $$;
|
|
|
265
265
|
n.nspname NOT IN ('information_schema', 'pg_catalog')
|
|
266
266
|
AND cl.relkind IN ('r', 'v', 'm', 'f', 'p')
|
|
267
267
|
AND d.description IS NOT NULL
|
|
268
|
-
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,column:i.column??void 0,text:i.text})),
|
|
268
|
+
`.execute(this.kysely)).rows.map(i=>({schema:i.schema,table:i.table,column:i.column??void 0,text:i.text})),le=await sql`
|
|
269
269
|
SELECT
|
|
270
270
|
n.nspname AS "schema",
|
|
271
271
|
t.typname AS "type",
|
|
@@ -276,7 +276,7 @@ END $$;
|
|
|
276
276
|
JOIN pg_namespace n ON n.oid = t.typnamespace
|
|
277
277
|
WHERE n.nspname NOT IN ('pg_catalog', 'information_schema')
|
|
278
278
|
GROUP BY n.nspname, t.typname
|
|
279
|
-
`.execute(this.kysely),
|
|
279
|
+
`.execute(this.kysely),ue=await sql`
|
|
280
280
|
SELECT
|
|
281
281
|
n.nspname AS "schema",
|
|
282
282
|
t.typname AS "type",
|
|
@@ -298,7 +298,7 @@ END $$;
|
|
|
298
298
|
AND NOT a.attisdropped
|
|
299
299
|
AND n.nspname NOT IN ('pg_catalog', 'information_schema')
|
|
300
300
|
GROUP BY n.nspname, t.typname
|
|
301
|
-
`.execute(this.kysely),
|
|
301
|
+
`.execute(this.kysely),me=[...le.rows.map(i=>({schema:i.schema,type:i.type,kind:"enum",values:i.values})),...ue.rows.map(i=>({schema:i.schema,type:i.type,kind:"composite",fields:i.fields}))],pe=(await sql`
|
|
302
302
|
SELECT
|
|
303
303
|
n.nspname AS "schema",
|
|
304
304
|
p.proname AS "name",
|
|
@@ -321,7 +321,7 @@ END $$;
|
|
|
321
321
|
WHERE n.nspname NOT IN ('pg_catalog', 'information_schema')
|
|
322
322
|
AND p.prokind = 'f'
|
|
323
323
|
ORDER BY n.nspname, p.proname
|
|
324
|
-
`.execute(this.kysely)).rows.map(i=>{let
|
|
324
|
+
`.execute(this.kysely)).rows.map(i=>{let w=i.arg_names_raw??[],E=i.arg_modes_raw??null,ye=E&&E.length===w.length?w.filter((b,k)=>E[k]==="i"||E[k]==="b"||E[k]==="v"):w,he=!!E?.some(b=>b==="o"||b==="b"||b==="t");return {schema:i.schema,name:i.name,arg_names:ye,arg_types:i.arg_types_raw??[],arg_defaults:Number(i.arg_defaults??0),has_variadic:i.has_variadic===!0,volatility:i.volatility??"v",return_type:i.return_type??"",return_is_setof:i.return_is_setof===!0,return_rows:Number(i.return_rows??0),return_typtype:i.return_typtype??"",return_base_type:i.return_base_type??void 0,has_out_args:he}}),fe=(await sql`
|
|
325
325
|
SELECT
|
|
326
326
|
c.relname AS "name",
|
|
327
327
|
n.nspname AS "schema",
|
|
@@ -331,23 +331,23 @@ END $$;
|
|
|
331
331
|
JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
|
|
332
332
|
JOIN pg_catalog.pg_class p ON p.oid = i.inhparent
|
|
333
333
|
WHERE n.nspname NOT IN ('information_schema', 'pg_catalog')
|
|
334
|
-
`.execute(this.kysely)).rows.map(i=>({name:i.name,schema:i.schema,parent:i.parent})),
|
|
334
|
+
`.execute(this.kysely)).rows.map(i=>({name:i.name,schema:i.schema,parent:i.parent})),ge=(await sql`SELECT name FROM pg_catalog.pg_timezone_names`.execute(this.kysely)).rows.map(i=>i.name),de=(await sql`SELECT current_database() AS "name"`.execute(this.kysely)).rows[0]?.name??"postgres",W=ee({tables:r,columns:f,indexes:S,foreign_keys:p,primary_keys:h,views:J,triggers:[],functions:pe,check_constraints:ie,unique_constraints:re,comments:ce,custom_types:me,partitions:fe,database_name:de,version:"",default_schema:this.config?.translation?.schemas?.defaultSchema??"public",timezones:ge});return await this.writeCachedIntrospection(W,{useDriver:t}),W}catch(o){return console.error("Introspection failed:",o),{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 e=await sql`
|
|
335
335
|
SELECT count(*) ::text as count FROM pg_class WHERE relrowsecurity = true
|
|
336
336
|
`.execute(this.kysely);if(!(Number(e.rows[0]?.count??0)>0))return this.rlsState="inactive",false;this.rlsState="active",await sql.raw(K).execute(this.kysely);let s=await sql`
|
|
337
337
|
SELECT DISTINCT n.nspname FROM pg_class c
|
|
338
338
|
JOIN pg_namespace n ON n.oid = c.relnamespace
|
|
339
339
|
WHERE c.relrowsecurity = true
|
|
340
|
-
`.execute(this.kysely);for(let{nspname:
|
|
340
|
+
`.execute(this.kysely);for(let{nspname:o}of s.rows){let a=te(o);await sql.raw(`GRANT USAGE ON SCHEMA ${a} TO ${R}`).execute(this.kysely),await sql.raw(`GRANT ALL ON ALL TABLES IN SCHEMA ${a} TO ${R}`).execute(this.kysely),await sql.raw(`GRANT ALL ON ALL SEQUENCES IN SCHEMA ${a} TO ${R}`).execute(this.kysely),await sql.raw(`ALTER DEFAULT PRIVILEGES IN SCHEMA ${a} GRANT ALL ON TABLES TO ${R}`).execute(this.kysely),await sql.raw(`ALTER DEFAULT PRIVILEGES IN SCHEMA ${a} GRANT ALL ON SEQUENCES TO ${R}`).execute(this.kysely);}return true}async postgresTransactionActive(e){try{return await sql`SAVEPOINT __supalite_outer_tx_check`.execute(e),await sql`RELEASE SAVEPOINT __supalite_outer_tx_check`.execute(e),!0}catch{return false}}async applyJwtSessionContext(e,t,s){let o=t.auth,a=o?.role||"anon",c=String(o?.uid??""),r=o?.jwt;await sql`SELECT set_config('role', ${a}, true)`.execute(e),await sql`SELECT set_config('request.jwt.claim.sub', ${c}, true)`.execute(e),await sql`SELECT set_config('request.jwt.claim.role', ${a}, true)`.execute(e),s&&await sql.raw(`SET LOCAL ROLE ${te(a)}`).execute(e),r&&typeof r=="object"&&await sql`SELECT set_config('request.jwt.claims', ${JSON.stringify(r)}, true)`.execute(e);}async applyRpcRequestGucs(e,t){await sql`SELECT set_config('response.status', '', true)`.execute(e),await sql`SELECT set_config('response.headers', '', true)`.execute(e),await sql`SELECT set_config('request.method', ${t.method}, true)`.execute(e),await sql`SELECT set_config('request.path', ${t.path}, true)`.execute(e),await sql`SELECT set_config('request.headers', ${JSON.stringify(t.requestHeaders??{})}, true)`.execute(e);}async withContext(e,t,s){let o=s?.forceRollback===true;if(!o&&!e)return t(this.kysely);let a=e?await this.ensureRlsContext():false,r=e?.auth?.role||"anon",l=typeof r=="string"&&/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(r)&&!Me.has(r),f=s?.rpc;if(f){let u=async p=>{e&&(a||l)&&await this.applyJwtSessionContext(p,e,l),await this.applyRpcRequestGucs(p,f),f.readOnly&&await sql`SET LOCAL transaction_read_only = on`.execute(p);};if(this.harnessHoldingOuterTx)return invariant(await this.postgresTransactionActive(this.kysely),"harnessHoldingOuterTx is set but the connection has no open PostgreSQL transaction; wrap REST requests in BEGIN/ROLLBACK in the spec harness"),await u(this.kysely),t(this.kysely);try{return await this.kysely.transaction().execute(async p=>{await u(p);let d=await t(p);if(o)throw new N(d);return d})}catch(p){if(p instanceof N)return p.result;throw p}}if(!o&&!a&&!l)return t(this.kysely);if(!o&&l)return this.harnessHoldingOuterTx?(invariant(await this.postgresTransactionActive(this.kysely),"harnessHoldingOuterTx is set but the connection has no open PostgreSQL transaction; wrap REST requests in BEGIN/ROLLBACK in the spec harness"),await this.applyJwtSessionContext(this.kysely,e,true),t(this.kysely)):this.kysely.transaction().execute(async u=>(await this.applyJwtSessionContext(u,e,true),t(u)));if(!o&&a&&!l)return this.kysely.transaction().execute(async u=>(await this.applyJwtSessionContext(u,e,true),t(u)));try{return await this.kysely.transaction().execute(async u=>{e&&(a||l)&&await this.applyJwtSessionContext(u,e,l);let p=await t(u);if(o)throw new N(p);return p})}catch(u){if(u instanceof N)return u.result;throw u}}async onPostgrestAST(e){if(!e.from)return e;let t=e.schema??"public",s=e.from,o=await this.introspect({useCache:true});if(!o.tables.some(a=>a.name===s&&a.schema===t)&&!o.views.some(a=>a.name===s&&a.schema===t))throw new RelationNotFoundError(t,s);return e}async transaction(e,t){await this.exec("BEGIN");try{for(let s of e)await this.exec(s);await this.exec("COMMIT");}catch(s){throw await this.exec("ROLLBACK"),s}finally{t?.intent==="migration"&&await this.clearSchemaCache();}}get supportsRpc(){return true}async viewOptionsMetadata(e,t){let{rows:s}=await sql`
|
|
341
341
|
select is_insertable_into, is_updatable,
|
|
342
342
|
is_trigger_insertable_into, is_trigger_updatable, is_trigger_deletable
|
|
343
343
|
from information_schema.views
|
|
344
344
|
where table_schema = ${t} and table_name = ${e}
|
|
345
|
-
`.execute(this.kysely),
|
|
345
|
+
`.execute(this.kysely),o=s[0],a=r=>r==="YES",{rows:c}=await sql`
|
|
346
346
|
select table_schema, table_name from information_schema.view_table_usage
|
|
347
347
|
where view_schema = ${t} and view_name = ${e}
|
|
348
|
-
`.execute(this.kysely);return {canInsert:o
|
|
348
|
+
`.execute(this.kysely);return {canInsert:a(o?.is_insertable_into)||a(o?.is_trigger_insertable_into),canUpdate:a(o?.is_updatable)||a(o?.is_trigger_updatable),canDelete:a(o?.is_updatable)||a(o?.is_trigger_deletable),baseTables:c.map(r=>({schema:r.table_schema,name:r.table_name}))}}};var ne=(n,e=t=>t)=>{if(!n||n==="{}")return [];let t=n.slice(1,-1);if(t==="")return [];let s=[],o="",a=false,c=0;for(let r=0;r<t.length;r++){let l=t[r];l==='"'&&t[r-1]!=="\\"?(a=!a,o+=l):l==="{"&&!a?(c++,o+=l):l==="}"&&!a?(c--,o+=l):l===","&&!a&&c===0?(s.push(e(o)),o=""):o+=l;}return o!==""&&s.push(e(o)),s},He=n=>n==="NULL"?null:n.startsWith('"')&&n.endsWith('"')?n.slice(1,-1).replace(/\\(.)/g,"$1"):n,Ue=n=>n==="NULL"?null:Number.parseInt(n,10);function se(n){return typeof n=="object"&&n!==null&&"text"in n&&"values"in n}function Ge(n,e){if(n==null)return n;switch(e){case 20:return typeof n=="bigint"?n:BigInt(n);case 22:return Array.isArray(n)?n:typeof n=="string"?n.split(" ").map(Number).filter(t=>!Number.isNaN(t)):n;case 1002:case 1009:case 1015:return typeof n=="string"?ne(n,He):n;case 1005:case 1007:case 1016:return typeof n=="string"?ne(n,Ue):n;default:return n}}function U(n){return {async query(e,t){let s=se(e)?e.text:e,o=se(e)?e.values:t??[],a=await n.driver.query(s,[...o]);return {rows:a.rows.map(r=>Object.fromEntries(a.fields.map(l=>[l.name,Ge(r[l.name],l.dataTypeID)])))}}}}var Je=new Set(["public","supabase_migrations"]),We=/\bCREATE\s+SCHEMA\b(?:\s+IF\s+NOT\s+EXISTS)?\s+("?[\w$]+"?)/gi;function qe(n,e){let t=e.replace(/[.*+?^${}()|[\]\\]/g,"\\$&"),s=new RegExp(`(?<![\\w$"])${t}\\.`);return n.some(o=>s.test(o))}function ae(n,e){let t=new Set;for(let s of e)for(let o of s.matchAll(We))t.add(o[1]);return Object.values(n.schemas).map(s=>s.name).filter(s=>!Je.has(s)&&!t.has(s)&&qe(e,s)).map(s=>`CREATE SCHEMA IF NOT EXISTS ${s};`)}async function Ke(){try{return await import('@supabase/pg-delta')}catch(n){throw new Error("Postgres/PGlite schema diffing now requires a Bun/Node-compatible runtime with pg-delta available.",{cause:n})}}var Qe=["supabase_migrations"];function Ye(n){let e=/^drop table (.+)$/.exec(n);if(e)return {table:e[1],reason:"drop table"};let t=/^drop column (.+) on (.+)$/.exec(n);if(t)return {table:t[2],reason:`drop column ${t[1]}`};let s=/^drop sequence (.+)$/.exec(n);return s?{table:s[1],reason:"drop sequence"}:{table:"unknown",reason:n}}var x=class{constructor(e,t){this.from=e;this.desiredSchema=t;}#e;async#t(e){if(this.#e)return this.#e;let t=new this.from.constructor;try{let s=C(F(this.desiredSchema));return s&&await t.exec(s),await t.exec(this.desiredSchema),this.#e=await e(U(t)),this.#e}finally{await t.close();}}async diff(){let{createPlan:e,extractCatalog:t}=await Ke(),s=U(this.from),o=await this.#t(t),a=await t(s),c=await e(a,o,{filter:{not:{schema:Qe}}});if(!c)return {diff:"",plan:{steps:[],warnings:[],unsafe:false}};let r=c.plan.risk?.level==="data_loss"?c.plan.risk.statements.map(Ye):[],l=Y(this.desiredSchema),f=c.plan.statements.filter(p=>!V(p,l)),u=[...ae(o,f),...f];return {diff:u.join(`
|
|
349
349
|
|
|
350
350
|
`),plan:{steps:u.map(p=>({sql:p})),warnings:r,unsafe:c.plan.risk?.level==="data_loss"}}}async migratePlan(e,t){if(e.steps.length===0)return;if(e.unsafe&&!t?.force)throw new DataLossError(e.warnings??[]);let s=e.steps.map(r=>r.sql).join(`
|
|
351
|
-
`),
|
|
351
|
+
`),o=L(s),a=new Set(e.steps.flatMap(r=>[...F(r.sql)]).filter(r=>!o.has(r))),c=C(a);c&&await this.from.exec(c),await this.from.transaction(e.steps.map(r=>r.sql),{intent:"migration"});}async migrate(e){let t=await this.diff();return await this.migratePlan(t.plan,e),t}safeSortPlanSteps(e){return e}};var G=class extends I{driver;dialect="postgres";#e=false;constructor(e={}){let t=e.url;t&&!t.includes("://")&&!oe.existsSync(t)&&oe.mkdirSync(t,{recursive:true});let s=new PGlite({...e.pgliteOptions,dataDir:e.url??void 0,extensions:{...e.pgliteOptions?.extensions,isn:isn,ltree:ltree,file_fdw:file_fdw,pgcrypto:pgcrypto},parsers:{20:a=>{let c=Number(a);return Number.isSafeInteger(c)?c:BigInt(a)},114:a=>JSON.parse(a),3802:a=>JSON.parse(a),1700:a=>{let c=Number(a);return Number.isFinite(c)&&String(c)===a?c:a},1082:a=>a,1083:a=>a,1114:a=>a,1184:a=>a,1266:a=>a,...e.pgliteOptions?.parsers}}),{dialect:o}=new T(s);super(e),this.driver=s,this.kysely=new Kysely({dialect:o});}async exec(e,...t){try{if(t.length>0){let{rows:o}=await sql(e,...t??[]).execute(this.kysely);return {rows:o}}return {rows:(await this.driver.exec(e))[0].rows}}catch(s){throw console.error(s),new Error(`Failed to execute query: ${e}`)}}async close(){this.#e||(this.#e=true,await this.clearSchemaCache(),await this.kysely.destroy(),await this.driver.close());}createMigrator(e){let t=[this.config.baseSchema,e].filter(Boolean).join(`
|
|
352
352
|
|
|
353
|
-
`);return new x(this,t)}};async function
|
|
353
|
+
`);return new x(this,t)}};async function Mt(n={}){return new G(n)}export{G as PgliteConnection,Mt as createPgliteConnection};
|
package/dist/index.d.ts
CHANGED
|
@@ -157,6 +157,7 @@ interface DeparseOptions extends DeparserOptions {
|
|
|
157
157
|
declare function translatePostgresDdl(pgSql: string, options?: DeparseOptions): Promise<string>;
|
|
158
158
|
declare function deparsePostgresDdl(pgSql: string, options?: DeparseOptions & {
|
|
159
159
|
strict?: boolean;
|
|
160
|
+
defaultSchema?: string;
|
|
160
161
|
}): Promise<{
|
|
161
162
|
ddl: string;
|
|
162
163
|
enums: CollectedEnums;
|