@izumisy/kyrage 1.4.0 → 1.5.0
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/dist/index.d.ts +3 -0
- package/dist/index.js +1 -1
- package/dist/main.js +8 -8
- package/package.json +2 -1
package/dist/index.d.ts
CHANGED
|
@@ -78,6 +78,7 @@ type DefinedForeignKeyConstraint = {
|
|
|
78
78
|
referencedColumns: ReadonlyArray<string>;
|
|
79
79
|
onDelete?: ReferentialActions;
|
|
80
80
|
onUpdate?: ReferentialActions;
|
|
81
|
+
inline?: boolean;
|
|
81
82
|
};
|
|
82
83
|
type ExpressionBuilders<T extends Record<string, DefinedColumn>> = {
|
|
83
84
|
/**
|
|
@@ -117,6 +118,7 @@ type ExpressionBuilders<T extends Record<string, DefinedColumn>> = {
|
|
|
117
118
|
name?: string;
|
|
118
119
|
onDelete?: ReferentialActions;
|
|
119
120
|
onUpdate?: ReferentialActions;
|
|
121
|
+
inline?: boolean;
|
|
120
122
|
}) => DefinedForeignKeyConstraint;
|
|
121
123
|
};
|
|
122
124
|
/**
|
|
@@ -188,6 +190,7 @@ declare const defineConfig: (config: DefineConfigProp) => {
|
|
|
188
190
|
name: string;
|
|
189
191
|
onDelete?: "cascade" | "set null" | "set default" | "restrict" | "no action" | undefined;
|
|
190
192
|
onUpdate?: "cascade" | "set null" | "set default" | "restrict" | "no action" | undefined;
|
|
193
|
+
inline?: boolean | undefined;
|
|
191
194
|
}[];
|
|
192
195
|
readonly database: DatabaseValue;
|
|
193
196
|
readonly tables: DefinedTables;
|
package/dist/index.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
import"./chunk-EY62Q3Z2.js";var f=(
|
|
1
|
+
import"./chunk-EY62Q3Z2.js";var f=(i,r)=>r.length===1?`${i}_${r[0]}_primary_key`:`pk_${i}_${r.join("_")}`,c=(i,r)=>r.length===1?`${i}_${r[0]}_unique`:`uq_${i}_${r.join("_")}`,p=(i,r)=>`idx_${i}_${r.join("_")}`,g=(i,r)=>`fk_${i}_${r.join("_")}`,m={primaryKey:f,unique:c,index:p,foreignKey:g};var D=(i,r)=>({type:i,notNull:r?.notNull,primaryKey:r?.primaryKey,unique:r?.unique,defaultSql:r?.defaultSql}),K=(i,r,y)=>{let t=e=>{if(e.length===0)throw new Error("At least one column must be specified.")},l=(e,n)=>`${e}_${i}_${n.join("_")}`,a=y?y({index:(e,n)=>(t(e),{kind:"index",name:n?.name??l("idx",e),columns:e,unique:n?.unique??!1}),primaryKey:(e,n)=>(t(e),{kind:"primaryKey",name:n?.name??l("pk",e),columns:e}),unique:(e,n)=>(t(e),{kind:"unique",name:n?.name??l("uq",e),columns:e}),reference:(e,n,u,o)=>{let s=Array.isArray(e)?e:[e],d=Array.isArray(u)?u:[u];return t(s),t(d),{kind:"foreignKey",name:o?.name??m.foreignKey(i,s),columns:s,referencedTable:n.tableName,referencedColumns:d,onDelete:o?.onDelete,onUpdate:o?.onUpdate,inline:o?.inline??!0}}}):[];return{tableName:i,columns:r,indexes:a.filter(e=>e.kind==="index"),primaryKeys:a.filter(e=>e.kind==="primaryKey"),uniques:a.filter(e=>e.kind==="unique"),foreignKeys:a.filter(e=>e.kind==="foreignKey")}},R=i=>{let r=new Map,y=new Map,t=new Map,l=new Map;for(let a of i.tables){let e=a.tableName;for(let n of a.indexes??[])r.set(`${e}:${n.name}`,{table:e,name:n.name,columns:n.columns,unique:n.unique});for(let n of a.primaryKeys??[])y.set(`${e}:${n.name}`,{table:e,name:n.name,columns:n.columns});for(let n of a.uniques??[])t.set(`${e}:${n.name}`,{table:e,name:n.name,columns:n.columns});for(let n of a.foreignKeys??[])l.set(`${e}:${n.name}`,{table:e,name:n.name,columns:n.columns,referencedTable:n.referencedTable,referencedColumns:n.referencedColumns,onDelete:n.onDelete,onUpdate:n.onUpdate,inline:n.inline});for(let[n,u]of Object.entries(a.columns)){if(u.primaryKey){let o=m.primaryKey(e,[n]),s=`${e}:${o}`;y.has(s)||y.set(s,{table:e,name:o,columns:[n]})}if(u.unique){let o=m.unique(e,[n]),s=`${e}:${o}`;t.has(s)||t.set(s,{table:e,name:o,columns:[n]})}}}return{...i,dev:i.dev,indexes:Array.from(r.values()),primaryKeyConstraints:Array.from(y.values()),uniqueConstraints:Array.from(t.values()),foreignKeyConstraints:Array.from(l.values())}};export{D as column,R as defineConfig,K as defineTable};
|
package/dist/main.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import{a as N,b as w}from"./chunk-EY62Q3Z2.js";import{defineCommand as
|
|
1
|
+
import{a as N,b as w}from"./chunk-EY62Q3Z2.js";import{defineCommand as Yn,runMain as Zn}from"citty";var qe="1.5.0";import{defineCommand as On}from"citty";import f from"zod";import{loadConfig as Et}from"c12";import b from"zod";var O=b.object({type:b.string()}).catchall(b.any()),h=b.object({table:b.string(),name:b.string()}),I=b.object({table:b.string(),column:b.string()}),G=b.object({...h.shape,columns:b.array(b.string()).readonly()}),J=b.object({...h.shape,columns:b.array(b.string()).readonly()}),L=b.enum(["cascade","set null","set default","restrict","no action"]),Y=b.object({...h.shape,columns:b.array(b.string()).readonly(),referencedTable:b.string(),referencedColumns:b.array(b.string()).readonly(),onDelete:L.optional(),onUpdate:L.optional(),inline:b.boolean().optional()});import{getContainerRuntimeClient as me}from"testcontainers";import Z from"zod";var k=e=>Z.object({container:Z.object({image:Z.string().default(e.defaultImage),name:Z.string().optional()})}),q=class{constructor(t,n){this.dialect=t;this.containerFactory=n}async setup(t,n){return new le({dialect:this.dialect,manageType:n,containerName:t.container.name,containerFactory:()=>this.containerFactory(t.container.image)})}async hasExisting(t){return t==="dev-start"?F(this.dialect):!1}async cleanup(){await $e()}},ce="kyrage.dialect",pe="kyrage.managed",ke="kyrage.managed-by",le=class{constructor(t){this.options=t;this.startedContainer=null;this.container=this.setupContainer()}setupContainer(){let t=this.options.containerFactory();return t.withLabels({[ce]:this.options.dialect,[pe]:"true",[ke]:this.options.manageType}),this.options.manageType==="dev-start"&&t.withReuse(),this.options.containerName&&t.withName(this.options.containerName),t}async start(){this.startedContainer=await this.container.start()}async stop(){this.startedContainer&&(await this.startedContainer.stop(),this.startedContainer=null)}async remove(){await $e(),this.startedContainer=null}getConnectionString(){if(!this.startedContainer)throw new Error("Container is not started");return this.startedContainer.getConnectionUri()}async getStatus(){let n=await(await me()).container.fetchByLabel(ce,this.options.dialect,{status:["running"]});if(n){let r=await n.inspect();return{type:"container",imageName:r.Config.Image,containerID:r.Id}}return{type:"unavailable"}}async isAvailable(){return(await this.getStatus()).type==="container"}},F=async e=>(await(await me()).container.list()).some(r=>r.Labels[pe]==="true"&&r.Labels[ke]==="dev-start"&&r.Labels[ce]===e&&r.State==="running"),$e=async()=>{let e=await me(),n=(await e.container.list()).filter(r=>r.Labels[pe]==="true").map(r=>r.Id);await Promise.allSettled(n.map(async r=>e.container.getById(r).remove({force:!0})))};var Kt=f.object({type:f.string(),primaryKey:f.boolean().optional().default(!1),notNull:f.boolean().optional().default(!1),unique:f.boolean().optional().default(!1),defaultSql:f.string().optional()}),Rt=f.object({tableName:f.string(),columns:f.record(f.string(),Kt)}),Pt=f.object({...h.shape,columns:f.array(f.string()).readonly(),unique:f.boolean().default(!1)}),It=f.enum(["postgres","cockroachdb"]),qt=f.object({dialect:It,connectionString:f.string()}),kt=f.union([k({defaultImage:""})]),$t=f.object({database:qt,dev:kt.optional(),tables:f.array(Rt).readonly(),indexes:f.array(Pt).readonly(),primaryKeyConstraints:f.array(G).readonly(),uniqueConstraints:f.array(J).readonly(),foreignKeyConstraints:f.array(Y).readonly()}),Me=async()=>{let e=await Et({name:"kyrage"});return $t.parse(e.config)};import{Kysely as Jt}from"kysely";import{PostgresDialect as Mt,sql as ue}from"kysely";import{Pool as Ut}from"pg";import{PostgreSqlContainer as jt}from"@testcontainers/postgresql";import*as v from"ramda";var Ue=(e,t)=>{let n=m=>i=>v.any(v.whereEq({name:i.name,table:i.table}),m),[r,o]=v.partition(n(e.uniqueConstraints),t.constraints.unique),[l,a]=v.partition(n(e.indexes),t.indexes),c=v.reject(n(l),o),s=v.reject(n(r),a);return{indexes:[...l,...s],uniqueConstraints:[...r,...c]}};var X=class{getName(){return"postgres"}createKyselyDialect(t){return new Mt({pool:new Ut({connectionString:t})})}createIntrospectionDriver(t){return{convertTypeName:ye,introspect:de(t)}}createDevDatabaseProvider(){return new q(this.getName(),t=>new jt(t))}parseDevDatabaseConfig(t){return k({defaultImage:"postgres:16"}).parse(t)}async hasReusableDevDatabase(){return F(this.getName())}},de=e=>async t=>{var s=[];try{let n=N(s,e.getDB(),!0);let[r,o,l]=await Promise.all([Lt(n),Ft(n),Bt(n)]);let{indexes:a,uniqueConstraints:c}=Ue(t.config,{indexes:o,constraints:l});return{tables:r,indexes:a,constraints:{...l,unique:c}}}catch(m){var i=m,p=!0}finally{var u=w(s,i,p);u&&await u}},ye=e=>({bool:"boolean",int2:"smallint",int4:"integer",int8:"bigint"})[e]??e,Lt=async e=>{let{rows:t}=await ue`
|
|
2
2
|
SELECT
|
|
3
3
|
n.nspname AS table_schema,
|
|
4
4
|
c.relname AS table_name,
|
|
@@ -18,7 +18,7 @@ import{a as N,b as w}from"./chunk-EY62Q3Z2.js";import{defineCommand as Un,runMai
|
|
|
18
18
|
AND a.attnum > 0
|
|
19
19
|
AND NOT a.attisdropped
|
|
20
20
|
ORDER BY c.relname, a.attnum;
|
|
21
|
-
`.$castTo().execute(e);return t.map(n=>({schema:n.table_schema,table:n.table_name,name:n.column_name,default:n.column_default,characterMaximumLength:n.character_maximum_length?Number(n.character_maximum_length):null}))},
|
|
21
|
+
`.$castTo().execute(e);return t.map(n=>({schema:n.table_schema,table:n.table_name,name:n.column_name,default:n.column_default,characterMaximumLength:n.character_maximum_length?Number(n.character_maximum_length):null}))},Ft=async e=>{let{rows:t}=await ue`
|
|
22
22
|
SELECT
|
|
23
23
|
t.relname AS table_name,
|
|
24
24
|
c.relname AS index_name,
|
|
@@ -33,7 +33,7 @@ import{a as N,b as w}from"./chunk-EY62Q3Z2.js";import{defineCommand as Un,runMai
|
|
|
33
33
|
AND n.nspname = 'public'
|
|
34
34
|
AND NOT i.indisprimary
|
|
35
35
|
GROUP BY t.relname, c.relname, i.indisunique, i.indisprimary, c.oid;
|
|
36
|
-
`.$castTo().execute(e);return t.map(n=>({table:n.table_name,name:n.index_name,columns:n.column_names,unique:n.is_unique}))},
|
|
36
|
+
`.$castTo().execute(e);return t.map(n=>({table:n.table_name,name:n.index_name,columns:n.column_names,unique:n.is_unique}))},Bt=async e=>{let{rows:t}=await ue`
|
|
37
37
|
SELECT
|
|
38
38
|
n.nspname AS schema,
|
|
39
39
|
t.relname AS table,
|
|
@@ -89,8 +89,8 @@ import{a as N,b as w}from"./chunk-EY62Q3Z2.js";import{defineCommand as Un,runMai
|
|
|
89
89
|
GROUP BY n.nspname, t.relname, c.conname, c.contype, c.oid, c.conkey, t.oid,
|
|
90
90
|
rt.relname, c.confkey, c.confdeltype, c.confupdtype
|
|
91
91
|
ORDER BY t.relname, c.conname;
|
|
92
|
-
`.$castTo().execute(e);return{primaryKey:t.filter(n=>n.type==="PRIMARY KEY"),unique:t.filter(n=>n.type==="UNIQUE"),foreignKey:t.filter(n=>n.type==="FOREIGN KEY").map(n=>{let{referenced_table:a,referenced_columns:o,on_delete:l,on_update:i,...c}=n;return{...c,referencedTable:a,referencedColumns:o,onDelete:l,onUpdate:i}})}};import{CockroachDbContainer as Bt}from"@testcontainers/cockroachdb";import{Pool as zt}from"pg";import{PostgresAdapter as Ft,PostgresDialect as Ht}from"kysely";var Z=class{getName(){return"cockroachdb"}createKyselyDialect(t){return new ye({pool:new zt({connectionString:t})})}createIntrospectionDriver(t){let n=["rowid"];return{convertTypeName:ue,introspect:async a=>{let o=await pe(t)(a);return{tables:o.tables.filter(i=>!n.includes(i.name)),indexes:o.indexes,constraints:o.constraints}}}}createDevDatabaseProvider(){return new I(this.getName(),t=>new Bt(t))}parseDevDatabaseConfig(t){return k({defaultImage:"cockroachdb/cockroach:latest-v24.3"}).parse(t)}async hasReusableDevDatabase(){return L(this.getName())}},de=class extends Ft{async acquireMigrationLock(t,n){await t.selectFrom(n.lockTable).selectAll().forUpdate().execute()}},ye=class extends Ht{createAdapter(){return new de}};var Vt={postgres:new Y,cockroachdb:new Z},A=e=>{let t=Vt[e];if(!t)throw new Error(`Unsupported dialect: ${e}`);return t};import{DummyDriver as Qt}from"kysely";var X=class extends Qt{constructor(n){super();this.queries=n}async acquireConnection(){return new fe(this.queries)}},fe=class{constructor(t){this.queries=t}async executeQuery(t){let n=t.query.kind;if(n==="CreateTableNode"||n==="DropTableNode"||n==="AlterTableNode"){if(t.query.table.table.identifier.name.startsWith("kysely_"))return{rows:[]};this.queries.push(t)}else(n==="CreateIndexNode"||n==="DropIndexNode")&&this.queries.push(t);return{rows:[]}}async*streamQuery(){}};var ee=e=>new ge({databaseProps:e.database,options:e.options}),ge=class{constructor(t){this.constructorProps=t}getDB(t){let n=A(this.constructorProps.databaseProps.dialect).createKyselyDialect(this.constructorProps.databaseProps.connectionString);return new be({dialect:{createAdapter:()=>n.createAdapter(),createDriver:()=>n.createDriver(),createIntrospector:a=>n.createIntrospector(a),createQueryCompiler:()=>n.createQueryCompiler()}},{isPlan:t?.plan===!0})}getDialect(){return this.constructorProps.databaseProps.dialect}},be=class extends Wt{constructor(n,a){let o=[];super({dialect:{...n.dialect,createDriver:()=>a.isPlan?new X(o):n.dialect.createDriver()}});this.plannedQueries=[];this.plannedQueries=o}getPlannedQueries(){return this.plannedQueries}};import{createConsola as Jt}from"consola";var Gt=Jt({stdout:process.stderr}),S={stdout:e=>console.log(e),reporter:{...Gt}},he={stdout:()=>{},reporter:{info:()=>{},warn:()=>{},success:()=>{},log:()=>{},error:()=>{}}};async function E(){let e=await ke();return{client:ee({database:e.database}),logger:S,config:e}}import{Migrator as dn}from"kysely";import{DEFAULT_MIGRATION_TABLE as Ct}from"kysely";import{readdir as sn,readFile as cn}from"fs/promises";import{join as ln}from"path";import F from"zod";import en from"zod";import h from"zod";import B from"zod";import{sql as Zt}from"kysely";import{isColumnDataType as Yt}from"kysely";var $=e=>{if(!Yt(e))throw new Error(`Unsupported data type: ${e}`)};var y=e=>e;var Me=y({typeName:"create_table",schema:B.object({type:B.literal("create_table"),table:B.string(),columns:B.record(B.string(),v)}),execute:async(e,t)=>{let n=e.schema.createTable(t.table);n=Ce(n,t.columns),await n.execute()}}),je=(e,t)=>({type:"create_table",table:e,columns:t});function Ce(e,t){let n=e;for(let[a,o]of Object.entries(t)){let l=o.type;$(l),n=n.addColumn(a,l,i=>{let c=i;return o.notNull&&(c=c.notNull()),typeof o.defaultSql=="string"&&(c=c.defaultTo(Zt.raw(o.defaultSql))),c})}return n}var Ue=y({typeName:"create_table_with_constraints",schema:h.object({type:h.literal("create_table_with_constraints"),table:h.string(),columns:h.record(h.string(),v),constraints:h.object({primaryKey:h.object({name:h.string(),columns:h.array(h.string()).readonly()}).optional(),unique:h.array(h.object({name:h.string(),columns:h.array(h.string()).readonly()})).readonly().optional()}).optional()}),execute:async(e,t)=>{let n=e.schema.createTable(t.table);if(n=Ce(n,t.columns),t.constraints){let{primaryKey:a,unique:o}=t.constraints;if(a&&(n=n.addPrimaryKeyConstraint(a.name,a.columns)),o)for(let l of o)n=n.addUniqueConstraint(l.name,l.columns)}await n.execute()}});import xe from"zod";var Le=y({typeName:"drop_table",schema:xe.object({type:xe.literal("drop_table"),table:xe.string()}),execute:async(e,t)=>{await e.schema.dropTable(t.table).execute()}}),Be=e=>({type:"drop_table",table:e});import ze from"zod";import{sql as Xt}from"kysely";var Fe=y({typeName:"add_column",schema:ze.object({...q.shape,type:ze.literal("add_column"),attributes:v}),execute:async(e,t)=>{let n=t.attributes.type;$(n),await e.schema.alterTable(t.table).addColumn(t.column,n,a=>{let o=a;return t.attributes.notNull&&(o=o.notNull()),typeof t.attributes.defaultSql=="string"&&(o=o.defaultTo(Xt.raw(t.attributes.defaultSql))),o}).execute()}}),He=(e,t)=>({...e,type:"add_column",attributes:t});import Ve from"zod";var Qe=y({typeName:"drop_column",schema:Ve.object({...q.shape,type:Ve.literal("drop_column"),attributes:v}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropColumn(t.column).execute()}}),We=(e,t)=>({...e,type:"drop_column",attributes:t});import Je from"zod";var Ge=y({typeName:"alter_column",schema:Je.object({...q.shape,type:Je.literal("alter_column"),before:v,after:v}),execute:async(e,t)=>{let{table:n,column:a,before:o,after:l}=t;if(o.type!==l.type){let i=l.type;$(i),await e.schema.alterTable(n).alterColumn(a,c=>c.setDataType(i)).execute()}l.notNull!==o.notNull&&(l.notNull?await e.schema.alterTable(n).alterColumn(a,i=>i.setNotNull()).execute():await e.schema.alterTable(n).alterColumn(a,i=>i.dropNotNull()).execute())}}),Ye=(e,t,n)=>({...e,type:"alter_column",before:t,after:n});import z from"zod";var Ze=y({typeName:"create_index",schema:z.object({...b.shape,type:z.literal("create_index"),columns:z.array(z.string()).readonly(),unique:z.boolean()}),execute:async(e,t)=>{let n=e.schema.createIndex(t.name).on(t.table);for(let a of t.columns)n=n.column(a);t.unique&&(n=n.unique()),await n.execute()}}),De=e=>({...e,type:"create_index"});import Xe from"zod";var et=y({typeName:"drop_index",schema:Xe.object({...b.shape,type:Xe.literal("drop_index")}),execute:async(e,t)=>{await e.schema.dropIndex(t.name).execute()}}),_e=e=>({...e,type:"drop_index"});import tt from"zod";var nt=y({typeName:"create_primary_key_constraint",schema:tt.object({...Q.shape,type:tt.literal("create_primary_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).addPrimaryKeyConstraint(t.name,t.columns).execute()}}),ve=e=>({...e,type:"create_primary_key_constraint"});import at from"zod";var rt=y({typeName:"drop_primary_key_constraint",schema:at.object({...b.shape,type:at.literal("drop_primary_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),Se=e=>({...e,type:"drop_primary_key_constraint"});import ot from"zod";var it=y({typeName:"create_unique_constraint",schema:ot.object({...W.shape,type:ot.literal("create_unique_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).addUniqueConstraint(t.name,t.columns).execute()}}),Oe=e=>({...e,type:"create_unique_constraint"});import st from"zod";var ct=y({typeName:"drop_unique_constraint",schema:st.object({...b.shape,type:st.literal("drop_unique_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),Te=e=>({...e,type:"drop_unique_constraint"});import lt from"zod";var mt=y({typeName:"create_foreign_key_constraint",schema:lt.object({...J.shape,type:lt.literal("create_foreign_key_constraint")}),execute:async(e,t)=>{let n=e.schema.alterTable(t.table).addForeignKeyConstraint(t.name,t.columns,t.referencedTable,t.referencedColumns);t.onDelete&&(n=n.onDelete(t.onDelete)),t.onUpdate&&(n=n.onUpdate(t.onUpdate)),await n.execute()}}),Ne=(e,t)=>({...e,...t,type:"create_foreign_key_constraint"});import pt from"zod";var ut=y({typeName:"drop_foreign_key_constraint",schema:pt.object({...b.shape,type:pt.literal("drop_foreign_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),we=e=>({...e,type:"drop_foreign_key_constraint"});var dt=[Ue,Me,Le,Fe,Qe,Ge,Ze,et,nt,rt,it,ct,mt,ut],yt=en.union(dt.map(e=>e.schema));async function ft(e,t){return await tn(t.type)(e,t)}function tn(e){let t=dt.find(n=>n.typeName===e);if(!t)throw new Error(`Unknown operation type: ${e}`);return t.execute}import*as bt from"ramda";var nn=e=>{let t=new Set;return e.forEach(n=>{n.type==="drop_table"&&t.add(n.table)}),t.size===0?e:e.filter(n=>n.type==="drop_table"?!0:!t.has(n.table))},an=e=>{let t=new Set;if(e.forEach(i=>{i.type==="create_table"&&t.add(i.table)}),t.size===0)return e;let n=new Map,a=new Map,o=[];e.forEach(i=>{if(i.type==="create_table")n.set(i.table,i);else if((i.type==="create_primary_key_constraint"||i.type==="create_unique_constraint")&&t.has(i.table)){let c=a.get(i.table)||[];a.set(i.table,[...c,i])}else o.push(i)});let l=[];return n.forEach((i,c)=>{let s=a.get(c)||[];if(s.length===0)l.push(i);else{let m={};s.forEach(r=>{r.type==="create_primary_key_constraint"?m.primaryKey={name:r.name,columns:r.columns}:r.type==="create_unique_constraint"&&(m.unique||(m.unique=[]),m.unique.push({name:r.name,columns:r.columns}))}),l.push({type:"create_table_with_constraints",table:i.table,columns:i.columns,constraints:m})}}),[...l,...o]},rn=e=>{let t=new Set;return e.forEach(n=>{(n.type==="drop_unique_constraint"||n.type==="drop_primary_key_constraint")&&t.add(`${n.table}.${n.name}`)}),e.filter(n=>{if(n.type==="drop_index"){let a=`${n.table}.${n.name}`;if(t.has(a))return!1}return!0})},on=e=>e.slice().sort((t,n)=>{let a=gt[t.type],o=gt[n.type];return a!==o?a-o:t.table.localeCompare(n.table)}),gt={drop_foreign_key_constraint:0,drop_unique_constraint:1,drop_primary_key_constraint:2,drop_index:3,drop_column:4,drop_table:5,create_table_with_constraints:6,create_table:7,add_column:8,alter_column:9,create_index:10,create_primary_key_constraint:11,create_unique_constraint:12,create_foreign_key_constraint:13},ht=bt.pipe(nn,an,rn,on);var xt=e=>({getMigrations:async()=>{let t=await e.migrationsResolver(),n={};return t.forEach(a=>{n[a.id]={up:async o=>{for(let l of ht(a.diff.operations))await ft(o,l)}}}),n}}),mn=F.object({operations:F.array(yt).readonly()}),pn=F.object({id:F.string(),version:F.string(),diff:mn}),M="migrations",Ae=async()=>{try{let t=(await sn(M)).filter(n=>n.endsWith(".json")).map(async n=>pn.parse(JSON.parse(await cn(ln(M,n),"utf-8"))));return await Promise.all(t)}catch(e){if(e instanceof Object&&"code"in e&&e.code==="ENOENT")return[];throw e}},R=async e=>{var l=[];try{let t=N(l,e.getDB(),!0);let n=await Ae();let a=(await t.introspection.getTables({withInternalKyselyTables:!0})).map(r=>r.name);if(!a.includes(Ct))return n;let o=await t.selectFrom(Ct).select(["name","timestamp"]).$narrowType().execute();return n.filter(r=>!o.some(p=>p.name===r.id))}catch(i){var c=i,s=!0}finally{var m=w(l,c,s);m&&await m}};import{format as yn}from"sql-formatter";async function H(e,t){var p=[];try{let{client:n,logger:a}=e;let{reporter:o}=a;let l=N(p,n.getDB({plan:t.plan}),!0);let i=xt({migrationsResolver:async()=>t.plan?await R(n):await Ae(),options:{plan:t.plan}});let c=new dn({db:l,provider:i});let{results:s,error:m}=await c.migrateToLatest();let r=l.getPlannedQueries();if(r.length>0){r.forEach(_=>{a.stdout(t.pretty?yn(_.sql):_.sql)});return}s&&s.length>0?s.forEach(_=>{_.status==="Error"?o.error(`Migration failed: ${_.migrationName}`):_.status==="Success"&&o.success(`Migration applied: ${_.migrationName}`)}):o.info("No migrations to run");if(m)throw m instanceof Error?m:new Error(`Migration error: ${m}`)}catch(u){var x=u,C=!0}finally{var D=w(p,x,C);D&&await D}}var Dt=un({meta:{name:"apply",description:"Run migrations to sync database schema"},args:{plan:{type:"boolean",description:"Plan the migration without applying it",default:!1},pretty:{type:"boolean",description:"Pretty print the migration SQL (only for --plan)",default:!1}},run:async e=>{try{let t=await E();await H(t,{plan:e.args.plan,pretty:e.args.pretty})}catch(t){S.reporter.error(t),process.exit(1)}}});import{defineCommand as vn}from"citty";import{mkdir as Sn,writeFile as On,unlink as Tn}from"fs/promises";import{join as Nn}from"path";import*as d from"ramda";var j=()=>({added:(e,t,n)=>d.pipe(d.difference(t),d.map(n))(e),removed:(e,t,n)=>d.pipe(d.difference(e),d.map(n))(t),changed:(e,t,n,a)=>d.pipe(d.intersection(e),d.filter(n),d.map(a))(t)}),te=d.prop("name"),fn=d.eqBy(e=>[e.type,e.notNull??!1,e.primaryKey??!1,e.unique??!1]);function gn(e,t){let n=[],a=j(),o=e.columns,l=t.columns,i=Object.keys(o),c=Object.keys(l);return a.added(i,c,s=>({column:s,attributes:l[s]})).forEach(s=>{n.push(He({table:e.name,column:s.column},s.attributes))}),a.removed(i,c,s=>({column:s,attributes:o[s]})).forEach(s=>{n.push(We({table:e.name,column:s.column},s.attributes))}),a.changed(i,c,s=>!fn(o[s],l[s]),s=>({column:s,before:o[s],after:l[s]})).forEach(s=>{n.push(Ye({table:e.name,column:s.column},s.before,s.after))}),n}function bn(e){let{current:t,ideal:n}=e,a=[],o=j(),l=d.map(te,t),i=d.map(te,n);return o.added(l,i,c=>n.find(m=>m.name===c)).forEach(c=>{a.push(je(c.name,c.columns))}),o.removed(l,i,d.identity).forEach(c=>{a.push(Be(c))}),d.filter(c=>n.some(s=>s.name===c.name))(t).forEach(c=>{let s=n.find(m=>m.name===c.name);a.push(...gn(c,s))}),a}function hn(e){let{current:t,ideal:n}=e,a=[],o=j(),l=r=>`${r.table}:${r.name}`,i=new Map(t.map(r=>[l(r),r])),c=new Map(n.map(r=>[l(r),r])),s=Array.from(i.keys()),m=Array.from(c.keys());return o.added(s,m,r=>c.get(r)).forEach(r=>{a.push(De(r))}),o.removed(s,m,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(_e(r))}),o.changed(s,m,r=>{let p=i.get(r),u=c.get(r);return!d.equals(p.columns,u.columns)||p.unique!==u.unique},r=>{let p=i.get(r),u=c.get(r);return{current:p,ideal:u}}).forEach(({current:r,ideal:p})=>{a.push(_e(r)),a.push(De(p))}),a}function Cn(e){let{current:t,ideal:n}=e,a=[],o=j(),l=r=>`${r.table}:${r.name}`,i=new Map(t.map(r=>[l(r),r])),c=new Map(n.map(r=>[l(r),r])),s=Array.from(i.keys()),m=Array.from(c.keys());return o.added(s,m,r=>c.get(r)).forEach(r=>{a.push(ve(r))}),o.removed(s,m,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(Se(r))}),o.changed(s,m,r=>{let p=i.get(r),u=c.get(r);return!d.equals(p.columns,u.columns)},r=>{let p=i.get(r),u=c.get(r);return{current:p,ideal:u}}).forEach(({current:r,ideal:p})=>{a.push(Se(r)),a.push(ve(p))}),a}function xn(e){let{current:t,ideal:n}=e,a=[],o=j(),l=r=>`${r.table}:${r.name}`,i=new Map(t.map(r=>[l(r),r])),c=new Map(n.map(r=>[l(r),r])),s=Array.from(i.keys()),m=Array.from(c.keys());return o.added(s,m,r=>c.get(r)).forEach(r=>{a.push(Oe(r))}),o.removed(s,m,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(Te(r))}),o.changed(s,m,r=>{let p=i.get(r),u=c.get(r);return!d.equals(p.columns,u.columns)},r=>{let p=i.get(r),u=c.get(r);return{current:p,ideal:u}}).forEach(({current:r,ideal:p})=>{a.push(Te(r)),a.push(Oe(p))}),a}function Dn(e){let t=[],n=j(),a=e.current.map(te),o=e.ideal.map(te);t.push(...n.added(a,o,i=>{let c=e.ideal.find(p=>p.name===i),{table:s,name:m,...r}=c;return Ne({table:s,name:m},r)})),t.push(...n.removed(a,o,i=>{let c=e.current.find(s=>s.name===i);return we(c)}));let l=i=>{let c=e.current.find(m=>m.name===i),s=e.ideal.find(m=>m.name===i);return!c||!s?!1:!(d.equals(c.columns,s.columns)&&c.referencedTable===s.referencedTable&&d.equals(c.referencedColumns,s.referencedColumns)&&c.onDelete===s.onDelete&&c.onUpdate===s.onUpdate)};return t.push(...n.changed(a,o,l,i=>{let c=e.current.find(u=>u.name===i),s=e.ideal.find(u=>u.name===i),{table:m,name:r,...p}=s;return[we(c),Ne({table:m,name:r},p)]}).flat()),t}function _t(e){let t=bn({current:e.current.tables,ideal:e.ideal.tables}),n=hn({current:e.current.indexes,ideal:e.ideal.indexes}),a=Cn({current:e.current.primaryKeyConstraints,ideal:e.ideal.primaryKeyConstraints}),o=xn({current:e.current.uniqueConstraints,ideal:e.ideal.uniqueConstraints}),l=Dn({current:e.current.foreignKeyConstraints,ideal:e.ideal.foreignKeyConstraints});return{operations:[...t,...n,...a,...o,...l]}}var vt=e=>{let n=A(e.getDialect()).createIntrospectionDriver(e);return{introspect:async o=>{var C=[];try{let l=N(C,e.getDB(),!0);let i=await l.introspection.getTables();let{tables:c,indexes:s,constraints:m}=await n.introspect({config:o});let r=()=>i.map(O=>{let P={};for(let U of O.columns){let oe=c.find(Re=>Re.table===O.name&&Re.name===U.name);oe&&(P[U.name]={schema:O.schema,table:O.name,name:U.name,dataType:n.convertTypeName(U.dataType),default:oe.default??null,characterMaximumLength:oe.characterMaximumLength??null,notNull:!U.isNullable})}return{schema:O.schema,name:O.name,columns:P}});let p=O=>!O.table.startsWith("kysely_");let u=()=>s.filter(p);let x=()=>({primaryKey:m.primaryKey.filter(p),unique:m.unique.filter(p),foreignKey:m.foreignKey.filter(p)});return{tables:r(),indexes:u(),constraints:x()}}catch(D){var _=D,K=!0}finally{var re=w(C,_,K);re&&await re}}}};var ne=async(e,t,n)=>{let a=A(t),o=await a.createDevDatabaseProvider().setup(a.parseDevDatabaseConfig(e),n);return{manageType:n,instance:o}};async function _n(e,t){let{config:n}=e,a=A(n.database.dialect);switch(t.mode){case"dev-start":{let{instance:o,manageType:l}=await ne(n.dev,n.database.dialect,"dev-start");return{manageType:l,manager:o,result:{reused:await o.isAvailable()}}}case"generate-dev":{let o=await a.hasReusableDevDatabase(),{instance:l,manageType:i}=await ne(n.dev,n.database.dialect,o?"dev-start":"one-off");return{manageType:i,manager:l,result:{reused:o}}}}}async function ae(e,t){let{config:n}=e,a=t.logger||he,{reporter:o}=a;if(!n.dev)throw new Error("Dev database configuration is required");let{manageType:l,manager:i,result:c}=await _n(e,t);c.reused?o.info("\u{1F504} Reusing existing dev database..."):o.info("\u{1F680} Starting dev database..."),await i.start(),o.success(`Dev database started: ${n.database.dialect}`);let s=ee({database:{dialect:n.database.dialect,connectionString:i.getConnectionString()}}),m=await R(s);return m.length>0?(o.info(`\u{1F504} Applying ${m.length} pending migrations...`),await H({client:s,logger:he,config:n},{plan:!1,pretty:!1}),o.success(`Applied ${m.length} migrations`)):o.info("No pending migrations found"),{client:s,manager:i,cleanup:async()=>{switch(t.mode){case"dev-start":o.success("Persistent dev database ready: "+n.database.dialect);break;case"generate-dev":l==="dev-start"?o.info("Dev start container remains running"):(await i.stop(),o.success("Temporary dev database stopped"));break}}}}async function wn(e,t){let{logger:n,config:a}=e,{reporter:o}=n;t.squash&&await An(e);let{client:l,cleanup:i}=t.dev?await ae(e,{mode:"generate-dev",logger:n}):{client:e.client,cleanup:async()=>{}};try{if(!t.ignorePending){let r=await R(l);if(r.length>0)if(t.dev)o.info(`Dev database has ${r.length} pending migrations. These will be applied automatically as baseline.`);else{o.warn([`There are pending migrations: ${r.map(p=>p.id).join(", ")}`,"Please apply them first before generating a new migration.","Otherwise, use --ignore-pending to skip this check."].join(`
|
|
93
|
-
`));return}}let c=await
|
|
94
|
-
`));break;case"drop_column":n.push(`-- remove_column: ${
|
|
95
|
-
`));break;case"create_index":n.push(`-- create_index: ${
|
|
96
|
-
`))},
|
|
92
|
+
`.$castTo().execute(e);return{primaryKey:t.filter(n=>n.type==="PRIMARY KEY"),unique:t.filter(n=>n.type==="UNIQUE"),foreignKey:t.filter(n=>n.type==="FOREIGN KEY").map(n=>{let{referenced_table:r,referenced_columns:o,on_delete:l,on_update:a,...c}=n;return{...c,referencedTable:r,referencedColumns:o,onDelete:l,onUpdate:a}})}};import{CockroachDbContainer as zt}from"@testcontainers/cockroachdb";import{Pool as Ht}from"pg";import{PostgresAdapter as Vt,PostgresDialect as Qt}from"kysely";var ee=class{getName(){return"cockroachdb"}createKyselyDialect(t){return new ge({pool:new Ht({connectionString:t})})}createIntrospectionDriver(t){let n=["rowid"];return{convertTypeName:ye,introspect:async r=>{let o=await de(t)(r);return{tables:o.tables.filter(a=>!n.includes(a.name)),indexes:o.indexes,constraints:o.constraints}}}}createDevDatabaseProvider(){return new q(this.getName(),t=>new zt(t))}parseDevDatabaseConfig(t){return k({defaultImage:"cockroachdb/cockroach:latest-v24.3"}).parse(t)}async hasReusableDevDatabase(){return F(this.getName())}},fe=class extends Vt{async acquireMigrationLock(t,n){await t.selectFrom(n.lockTable).selectAll().forUpdate().execute()}},ge=class extends Qt{createAdapter(){return new fe}};var Wt={postgres:new X,cockroachdb:new ee},A=e=>{let t=Wt[e];if(!t)throw new Error(`Unsupported dialect: ${e}`);return t};import{DummyDriver as Gt}from"kysely";var te=class extends Gt{constructor(n){super();this.queries=n}async acquireConnection(){return new be(this.queries)}},be=class{constructor(t){this.queries=t}async executeQuery(t){let n=t.query.kind;if(n==="CreateTableNode"||n==="DropTableNode"||n==="AlterTableNode"){if(t.query.table.table.identifier.name.startsWith("kysely_"))return{rows:[]};this.queries.push(t)}else(n==="CreateIndexNode"||n==="DropIndexNode")&&this.queries.push(t);return{rows:[]}}async*streamQuery(){}};var ne=e=>new he({databaseProps:e.database,options:e.options}),he=class{constructor(t){this.constructorProps=t}getDB(t){let n=A(this.constructorProps.databaseProps.dialect).createKyselyDialect(this.constructorProps.databaseProps.connectionString);return new Ce({dialect:{createAdapter:()=>n.createAdapter(),createDriver:()=>n.createDriver(),createIntrospector:r=>n.createIntrospector(r),createQueryCompiler:()=>n.createQueryCompiler()}},{isPlan:t?.plan===!0})}getDialect(){return this.constructorProps.databaseProps.dialect}},Ce=class extends Jt{constructor(n,r){let o=[];super({dialect:{...n.dialect,createDriver:()=>r.isPlan?new te(o):n.dialect.createDriver()}});this.plannedQueries=[];this.plannedQueries=o}getPlannedQueries(){return this.plannedQueries}};import{createConsola as Yt}from"consola";var Zt=Yt({stdout:process.stderr}),T={stdout:e=>console.log(e),reporter:{...Zt}},xe={stdout:()=>{},reporter:{info:()=>{},warn:()=>{},success:()=>{},log:()=>{},error:()=>{}}};async function E(){let e=await Me();return{client:ne({database:e.database}),logger:T,config:e}}import{Migrator as Tn}from"kysely";import{DEFAULT_MIGRATION_TABLE as Dt}from"kysely";import{readdir as hn,readFile as Cn}from"fs/promises";import{join as xn}from"path";import V from"zod";import nn from"zod";import g from"zod";import B from"zod";import{sql as en}from"kysely";import{isColumnDataType as Xt}from"kysely";var $=e=>{if(!Xt(e))throw new Error(`Unsupported data type: ${e}`)};var y=e=>e;var je=y({typeName:"create_table",schema:B.object({type:B.literal("create_table"),table:B.string(),columns:B.record(B.string(),O)}),execute:async(e,t)=>{let n=e.schema.createTable(t.table);n=De(n,t.columns),await n.execute()}}),Le=(e,t)=>({type:"create_table",table:e,columns:t});function De(e,t){let n=e;for(let[r,o]of Object.entries(t)){let l=o.type;$(l),n=n.addColumn(r,l,a=>{let c=a;return o.notNull&&(c=c.notNull()),typeof o.defaultSql=="string"&&(c=c.defaultTo(en.raw(o.defaultSql))),c})}return n}var Fe=y({typeName:"create_table_with_constraints",schema:g.object({type:g.literal("create_table_with_constraints"),table:g.string(),columns:g.record(g.string(),O),constraints:g.object({primaryKey:g.object({name:g.string(),columns:g.array(g.string()).readonly()}).optional(),unique:g.array(g.object({name:g.string(),columns:g.array(g.string()).readonly()})).readonly().optional(),foreignKeys:g.array(g.object({name:g.string(),columns:g.array(g.string()).readonly(),referencedTable:g.string(),referencedColumns:g.array(g.string()).readonly(),onDelete:L.optional(),onUpdate:L.optional(),inline:g.boolean().optional()})).readonly().optional()}).optional()}),execute:async(e,t)=>{let n=e.schema.createTable(t.table);if(n=De(n,t.columns),t.constraints){let{primaryKey:r,unique:o,foreignKeys:l}=t.constraints;if(r&&(n=n.addPrimaryKeyConstraint(r.name,r.columns)),o)for(let a of o)n=n.addUniqueConstraint(a.name,a.columns);if(l)for(let a of l)n=n.addForeignKeyConstraint(a.name,a.columns,a.referencedTable,a.referencedColumns,c=>{let s=c;return a.onDelete&&(s=s.onDelete(a.onDelete)),a.onUpdate&&(s=s.onUpdate(a.onUpdate)),s})}await n.execute()}}),Be=(e,t,n)=>({type:"create_table_with_constraints",table:e,columns:t,constraints:{primaryKey:n?.primaryKey,unique:n?.unique??[],foreignKeys:n?.foreignKeys??[]}});import _e from"zod";var ze=y({typeName:"drop_table",schema:_e.object({type:_e.literal("drop_table"),table:_e.string()}),execute:async(e,t)=>{await e.schema.dropTable(t.table).execute()}}),He=e=>({type:"drop_table",table:e});import Ve from"zod";import{sql as tn}from"kysely";var Qe=y({typeName:"add_column",schema:Ve.object({...I.shape,type:Ve.literal("add_column"),attributes:O}),execute:async(e,t)=>{let n=t.attributes.type;$(n),await e.schema.alterTable(t.table).addColumn(t.column,n,r=>{let o=r;return t.attributes.notNull&&(o=o.notNull()),typeof t.attributes.defaultSql=="string"&&(o=o.defaultTo(tn.raw(t.attributes.defaultSql))),o}).execute()}}),We=(e,t)=>({...e,type:"add_column",attributes:t});import Ge from"zod";var Je=y({typeName:"drop_column",schema:Ge.object({...I.shape,type:Ge.literal("drop_column"),attributes:O}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropColumn(t.column).execute()}}),Ye=(e,t)=>({...e,type:"drop_column",attributes:t});import Ze from"zod";var Xe=y({typeName:"alter_column",schema:Ze.object({...I.shape,type:Ze.literal("alter_column"),before:O,after:O}),execute:async(e,t)=>{let{table:n,column:r,before:o,after:l}=t;if(o.type!==l.type){let a=l.type;$(a),await e.schema.alterTable(n).alterColumn(r,c=>c.setDataType(a)).execute()}l.notNull!==o.notNull&&(l.notNull?await e.schema.alterTable(n).alterColumn(r,a=>a.setNotNull()).execute():await e.schema.alterTable(n).alterColumn(r,a=>a.dropNotNull()).execute())}}),et=(e,t,n)=>({...e,type:"alter_column",before:t,after:n});import z from"zod";var tt=y({typeName:"create_index",schema:z.object({...h.shape,type:z.literal("create_index"),columns:z.array(z.string()).readonly(),unique:z.boolean()}),execute:async(e,t)=>{let n=e.schema.createIndex(t.name).on(t.table);for(let r of t.columns)n=n.column(r);t.unique&&(n=n.unique()),await n.execute()}}),Oe=e=>({...e,type:"create_index"});import nt from"zod";var rt=y({typeName:"drop_index",schema:nt.object({...h.shape,type:nt.literal("drop_index")}),execute:async(e,t)=>{await e.schema.dropIndex(t.name).execute()}}),Te=e=>({...e,type:"drop_index"});import at from"zod";var ot=y({typeName:"create_primary_key_constraint",schema:at.object({...G.shape,type:at.literal("create_primary_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).addPrimaryKeyConstraint(t.name,t.columns).execute()}}),Se=e=>({...e,type:"create_primary_key_constraint"});import it from"zod";var st=y({typeName:"drop_primary_key_constraint",schema:it.object({...h.shape,type:it.literal("drop_primary_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),ve=e=>({...e,type:"drop_primary_key_constraint"});import ct from"zod";var lt=y({typeName:"create_unique_constraint",schema:ct.object({...J.shape,type:ct.literal("create_unique_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).addUniqueConstraint(t.name,t.columns).execute()}}),Ne=e=>({...e,type:"create_unique_constraint"});import mt from"zod";var pt=y({typeName:"drop_unique_constraint",schema:mt.object({...h.shape,type:mt.literal("drop_unique_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),we=e=>({...e,type:"drop_unique_constraint"});import ut from"zod";var dt=y({typeName:"create_foreign_key_constraint",schema:ut.object({...Y.shape,type:ut.literal("create_foreign_key_constraint")}),execute:async(e,t)=>{let n=e.schema.alterTable(t.table).addForeignKeyConstraint(t.name,t.columns,t.referencedTable,t.referencedColumns);t.onDelete&&(n=n.onDelete(t.onDelete)),t.onUpdate&&(n=n.onUpdate(t.onUpdate)),await n.execute()}}),Ae=(e,t)=>({...e,...t,type:"create_foreign_key_constraint"});import yt from"zod";var ft=y({typeName:"drop_foreign_key_constraint",schema:yt.object({...h.shape,type:yt.literal("drop_foreign_key_constraint")}),execute:async(e,t)=>{await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}}),Ee=e=>({...e,type:"drop_foreign_key_constraint"});var gt=[Fe,je,ze,Qe,Je,Xe,tt,rt,ot,st,lt,pt,dt,ft],bt=nn.union(gt.map(e=>e.schema));async function ht(e,t){return await rn(t.type)(e,t)}function rn(e){let t=gt.find(n=>n.typeName===e);if(!t)throw new Error(`Unknown operation type: ${e}`);return t.execute}import*as H from"ramda";import{Graph as an,topologicalSort as on,CycleError as sn}from"graph-data-structure";var cn=e=>{let t=new Set;return e.forEach(n=>{n.type==="drop_table"&&t.add(n.table)}),t.size===0?e:e.filter(n=>n.type==="drop_table"?!0:!t.has(n.table))},ln=e=>{let t=new Set;if(e.forEach(a=>{a.type==="create_table"&&t.add(a.table)}),t.size===0)return e;let n=new Map,r=new Map,o=[];e.forEach(a=>{a.type==="create_table"&&n.set(a.table,a)}),e.forEach(a=>{switch(a.type){case"create_table":return;case"create_primary_key_constraint":case"create_unique_constraint":if(t.has(a.table)){let c=r.get(a.table)||[];r.set(a.table,[...c,a])}else o.push(a);break;case"create_foreign_key_constraint":if(t.has(a.table)&&a.inline!==!1){let c=n.get(a.table),s=a.columns.filter(i=>!(i in c.columns));if(s.length>0)throw new Error(`Foreign key constraint "${a.name}" references non-existent columns: ${s.join(", ")}`);let m=r.get(a.table)||[];r.set(a.table,[...m,a])}else o.push(a);break;default:o.push(a);break}});let l=[];return n.forEach((a,c)=>{let s=r.get(c)||[];if(s.length===0){l.push(a);return}let m={unique:[],foreignKeys:[]};s.forEach(i=>{switch(i.type){case"create_primary_key_constraint":m.primaryKey=i;break;case"create_unique_constraint":m.unique.push(i);break;case"create_foreign_key_constraint":m.foreignKeys.push(i);break}}),l.push(Be(a.table,a.columns,m))}),[...l,...o]},mn=e=>{let t=new Set;return e.forEach(n=>{(n.type==="drop_unique_constraint"||n.type==="drop_primary_key_constraint")&&t.add(`${n.table}.${n.name}`)}),e.filter(n=>{if(n.type==="drop_index"){let r=`${n.table}.${n.name}`;if(t.has(r))return!1}return!0})},Ke=e=>"table"in e?e.table:null,pn=e=>{let t=new Set;return e.forEach(n=>{let r=Ke(n);r&&t.add(r)}),t},un=e=>{switch(e.type){case"create_foreign_key_constraint":return[{table:e.table,referencedTable:e.referencedTable}];case"create_table_with_constraints":return(e.constraints?.foreignKeys||[]).map(t=>({table:e.table,referencedTable:t.referencedTable}));default:return[]}},dn=e=>{let t=new an;return e.forEach(n=>un(n).forEach(r=>{r.table!==r.referencedTable&&(n.type!=="create_foreign_key_constraint"||n.inline!==!1)&&t.addEdge(r.referencedTable,r.table)})),t},yn=(e,t)=>{try{let n=dn(t),r=on(n),o=new Set(r);return[...Array.from(e).filter(a=>!o.has(a)).sort(),...r]}catch(n){throw n instanceof sn?new Error("Circular dependency detected in foreign key constraints. Please resolve this by setting 'inline: false' for one of the foreign key constraints in each circular relationship."):n}},fn=e=>new Map(e.map((t,n)=>[t,n])),gn=e=>(t,n)=>{let r=Ct[t.type],o=Ct[n.type];if(r!==o)return r-o;let l=Ke(t),a=Ke(n);if(l&&a){let c=e.get(l)??Number.MAX_SAFE_INTEGER,s=e.get(a)??Number.MAX_SAFE_INTEGER;if(c!==s)return c-s}return(l||"").localeCompare(a||"")},Ct={drop_foreign_key_constraint:0,drop_unique_constraint:1,drop_primary_key_constraint:2,drop_index:3,drop_column:4,drop_table:5,create_table_with_constraints:6,create_table:7,add_column:8,alter_column:9,create_index:10,create_primary_key_constraint:11,create_unique_constraint:12,create_foreign_key_constraint:13},bn=e=>H.sort(gn(H.pipe(pn,t=>yn(t,e),fn)(e)),e),xt=H.pipe(cn,ln,mn,bn);var _t=e=>({getMigrations:async()=>{let t=await e.migrationsResolver(),n={};return t.forEach(r=>{n[r.id]={up:async o=>{for(let l of xt(r.diff.operations))await ht(o,l)}}}),n}}),Dn=V.object({operations:V.array(bt).readonly()}),_n=V.object({id:V.string(),version:V.string(),diff:Dn}),M="migrations",Re=async()=>{try{let t=(await hn(M)).filter(n=>n.endsWith(".json")).map(async n=>_n.parse(JSON.parse(await Cn(xn(M,n),"utf-8"))));return await Promise.all(t)}catch(e){if(e instanceof Object&&"code"in e&&e.code==="ENOENT")return[];throw e}},K=async e=>{var l=[];try{let t=N(l,e.getDB(),!0);let n=await Re();let r=(await t.introspection.getTables({withInternalKyselyTables:!0})).map(i=>i.name);if(!r.includes(Dt))return n;let o=await t.selectFrom(Dt).select(["name","timestamp"]).$narrowType().execute();return n.filter(i=>!o.some(p=>p.name===i.id))}catch(a){var c=a,s=!0}finally{var m=w(l,c,s);m&&await m}};import{format as Sn}from"sql-formatter";async function Q(e,t){var p=[];try{let{client:n,logger:r}=e;let{reporter:o}=r;let l=N(p,n.getDB({plan:t.plan}),!0);let a=_t({migrationsResolver:async()=>t.plan?await K(n):await Re(),options:{plan:t.plan}});let c=new Tn({db:l,provider:a});let{results:s,error:m}=await c.migrateToLatest();let i=l.getPlannedQueries();if(i.length>0){i.forEach(_=>{r.stdout(t.pretty?Sn(_.sql):_.sql)});return}s&&s.length>0?s.forEach(_=>{_.status==="Error"?o.error(`Migration failed: ${_.migrationName}`):_.status==="Success"&&o.success(`Migration applied: ${_.migrationName}`)}):o.info("No migrations to run");if(m)throw m instanceof Error?m:new Error(`Migration error: ${m}`)}catch(u){var x=u,C=!0}finally{var D=w(p,x,C);D&&await D}}var Ot=On({meta:{name:"apply",description:"Run migrations to sync database schema"},args:{plan:{type:"boolean",description:"Plan the migration without applying it",default:!1},pretty:{type:"boolean",description:"Pretty print the migration SQL (only for --plan)",default:!1}},run:async e=>{try{let t=await E();await Q(t,{plan:e.args.plan,pretty:e.args.pretty})}catch(t){T.reporter.error(t),process.exit(1)}}});import{defineCommand as In}from"citty";import{mkdir as qn,writeFile as kn,unlink as $n}from"fs/promises";import{join as Mn}from"path";import*as d from"ramda";var U=()=>({added:(e,t,n)=>d.pipe(d.difference(t),d.map(n))(e),removed:(e,t,n)=>d.pipe(d.difference(e),d.map(n))(t),changed:(e,t,n,r)=>d.pipe(d.intersection(e),d.filter(n),d.map(r))(t)}),re=d.prop("name"),vn=d.eqBy(e=>[e.type,e.notNull??!1,e.primaryKey??!1,e.unique??!1]);function Nn(e,t){let n=[],r=U(),o=e.columns,l=t.columns,a=Object.keys(o),c=Object.keys(l);return r.added(a,c,s=>({column:s,attributes:l[s]})).forEach(s=>{n.push(We({table:e.name,column:s.column},s.attributes))}),r.removed(a,c,s=>({column:s,attributes:o[s]})).forEach(s=>{n.push(Ye({table:e.name,column:s.column},s.attributes))}),r.changed(a,c,s=>!vn(o[s],l[s]),s=>({column:s,before:o[s],after:l[s]})).forEach(s=>{n.push(et({table:e.name,column:s.column},s.before,s.after))}),n}function wn(e){let{current:t,ideal:n}=e,r=[],o=U(),l=d.map(re,t),a=d.map(re,n);return o.added(l,a,c=>n.find(m=>m.name===c)).forEach(c=>{r.push(Le(c.name,c.columns))}),o.removed(l,a,d.identity).forEach(c=>{r.push(He(c))}),d.filter(c=>n.some(s=>s.name===c.name))(t).forEach(c=>{let s=n.find(m=>m.name===c.name);r.push(...Nn(c,s))}),r}function An(e){let{current:t,ideal:n}=e,r=[],o=U(),l=i=>`${i.table}:${i.name}`,a=new Map(t.map(i=>[l(i),i])),c=new Map(n.map(i=>[l(i),i])),s=Array.from(a.keys()),m=Array.from(c.keys());return o.added(s,m,i=>c.get(i)).forEach(i=>{r.push(Oe(i))}),o.removed(s,m,i=>{let p=a.get(i);return{table:p.table,name:p.name}}).forEach(i=>{r.push(Te(i))}),o.changed(s,m,i=>{let p=a.get(i),u=c.get(i);return!d.equals(p.columns,u.columns)||p.unique!==u.unique},i=>{let p=a.get(i),u=c.get(i);return{current:p,ideal:u}}).forEach(({current:i,ideal:p})=>{r.push(Te(i)),r.push(Oe(p))}),r}function En(e){let{current:t,ideal:n}=e,r=[],o=U(),l=i=>`${i.table}:${i.name}`,a=new Map(t.map(i=>[l(i),i])),c=new Map(n.map(i=>[l(i),i])),s=Array.from(a.keys()),m=Array.from(c.keys());return o.added(s,m,i=>c.get(i)).forEach(i=>{r.push(Se(i))}),o.removed(s,m,i=>{let p=a.get(i);return{table:p.table,name:p.name}}).forEach(i=>{r.push(ve(i))}),o.changed(s,m,i=>{let p=a.get(i),u=c.get(i);return!d.equals(p.columns,u.columns)},i=>{let p=a.get(i),u=c.get(i);return{current:p,ideal:u}}).forEach(({current:i,ideal:p})=>{r.push(ve(i)),r.push(Se(p))}),r}function Kn(e){let{current:t,ideal:n}=e,r=[],o=U(),l=i=>`${i.table}:${i.name}`,a=new Map(t.map(i=>[l(i),i])),c=new Map(n.map(i=>[l(i),i])),s=Array.from(a.keys()),m=Array.from(c.keys());return o.added(s,m,i=>c.get(i)).forEach(i=>{r.push(Ne(i))}),o.removed(s,m,i=>{let p=a.get(i);return{table:p.table,name:p.name}}).forEach(i=>{r.push(we(i))}),o.changed(s,m,i=>{let p=a.get(i),u=c.get(i);return!d.equals(p.columns,u.columns)},i=>{let p=a.get(i),u=c.get(i);return{current:p,ideal:u}}).forEach(({current:i,ideal:p})=>{r.push(we(i)),r.push(Ne(p))}),r}function Rn(e){let t=[],n=U(),r=e.current.map(re),o=e.ideal.map(re);t.push(...n.added(r,o,a=>{let c=e.ideal.find(p=>p.name===a),{table:s,name:m,...i}=c;return Ae({table:s,name:m},i)})),t.push(...n.removed(r,o,a=>{let c=e.current.find(s=>s.name===a);return Ee(c)}));let l=a=>{let c=e.current.find(m=>m.name===a),s=e.ideal.find(m=>m.name===a);return!c||!s?!1:!(d.equals(c.columns,s.columns)&&c.referencedTable===s.referencedTable&&d.equals(c.referencedColumns,s.referencedColumns)&&c.onDelete===s.onDelete&&c.onUpdate===s.onUpdate)};return t.push(...n.changed(r,o,l,a=>{let c=e.current.find(u=>u.name===a),s=e.ideal.find(u=>u.name===a),{table:m,name:i,...p}=s;return[Ee(c),Ae({table:m,name:i},p)]}).flat()),t}function Tt(e){let t=wn({current:e.current.tables,ideal:e.ideal.tables}),n=An({current:e.current.indexes,ideal:e.ideal.indexes}),r=En({current:e.current.primaryKeyConstraints,ideal:e.ideal.primaryKeyConstraints}),o=Kn({current:e.current.uniqueConstraints,ideal:e.ideal.uniqueConstraints}),l=Rn({current:e.current.foreignKeyConstraints,ideal:e.ideal.foreignKeyConstraints});return{operations:[...t,...n,...r,...o,...l]}}var St=e=>{let n=A(e.getDialect()).createIntrospectionDriver(e);return{introspect:async o=>{var C=[];try{let l=N(C,e.getDB(),!0);let a=await l.introspection.getTables();let{tables:c,indexes:s,constraints:m}=await n.introspect({config:o});let i=()=>a.map(S=>{let P={};for(let j of S.columns){let se=c.find(Ie=>Ie.table===S.name&&Ie.name===j.name);se&&(P[j.name]={schema:S.schema,table:S.name,name:j.name,dataType:n.convertTypeName(j.dataType),default:se.default??null,characterMaximumLength:se.characterMaximumLength??null,notNull:!j.isNullable})}return{schema:S.schema,name:S.name,columns:P}});let p=S=>!S.table.startsWith("kysely_");let u=()=>s.filter(p);let x=()=>({primaryKey:m.primaryKey.filter(p),unique:m.unique.filter(p),foreignKey:m.foreignKey.filter(p)});return{tables:i(),indexes:u(),constraints:x()}}catch(D){var _=D,R=!0}finally{var ie=w(C,_,R);ie&&await ie}}}};var ae=async(e,t,n)=>{let r=A(t),o=await r.createDevDatabaseProvider().setup(r.parseDevDatabaseConfig(e),n);return{manageType:n,instance:o}};async function Pn(e,t){let{config:n}=e,r=A(n.database.dialect);switch(t.mode){case"dev-start":{let{instance:o,manageType:l}=await ae(n.dev,n.database.dialect,"dev-start");return{manageType:l,manager:o,result:{reused:await o.isAvailable()}}}case"generate-dev":{let o=await r.hasReusableDevDatabase(),{instance:l,manageType:a}=await ae(n.dev,n.database.dialect,o?"dev-start":"one-off");return{manageType:a,manager:l,result:{reused:o}}}}}async function oe(e,t){let{config:n}=e,r=t.logger||xe,{reporter:o}=r;if(!n.dev)throw new Error("Dev database configuration is required");let{manageType:l,manager:a,result:c}=await Pn(e,t);c.reused?o.info("\u{1F504} Reusing existing dev database..."):o.info("\u{1F680} Starting dev database..."),await a.start(),o.success(`Dev database started: ${n.database.dialect}`);let s=ne({database:{dialect:n.database.dialect,connectionString:a.getConnectionString()}}),m=await K(s);return m.length>0?(o.info(`\u{1F504} Applying ${m.length} pending migrations...`),await Q({client:s,logger:xe,config:n},{plan:!1,pretty:!1}),o.success(`Applied ${m.length} migrations`)):o.info("No pending migrations found"),{client:s,manager:a,cleanup:async()=>{switch(t.mode){case"dev-start":o.success("Persistent dev database ready: "+n.database.dialect);break;case"generate-dev":l==="dev-start"?o.info("Dev start container remains running"):(await a.stop(),o.success("Temporary dev database stopped"));break}}}}async function Un(e,t){let{logger:n,config:r}=e,{reporter:o}=n;t.squash&&await jn(e);let{client:l,cleanup:a}=t.dev?await oe(e,{mode:"generate-dev",logger:n}):{client:e.client,cleanup:async()=>{}};try{if(!t.ignorePending){let i=await K(l);if(i.length>0)if(t.dev)o.info(`Dev database has ${i.length} pending migrations. These will be applied automatically as baseline.`);else{o.warn([`There are pending migrations: ${i.map(p=>p.id).join(", ")}`,"Please apply them first before generating a new migration.","Otherwise, use --ignore-pending to skip this check."].join(`
|
|
93
|
+
`));return}}let c=await Ln({client:l,config:r});if(!c){o.info("No changes detected, no migration needed.");return}Fn(n,c.diff);let s=`${M}/${c.id}.json`;await qn(M,{recursive:!0}),await kn(s,JSON.stringify(c,null,2));let m=t.squash?`Generated squashed migration: ${s}`:`Migration file generated: ${s}`;o.success(m),t.dev&&await Q({...e,client:l},{plan:!1,pretty:!1})}finally{await a()}}var jn=async e=>{let{client:t,logger:n}=e,{reporter:r}=n,o=await K(t);if(o.length===0){r.info("No pending migrations found, nothing to squash.");return}r.info(`Found ${o.length} pending migrations to squash:`),o.forEach(a=>{r.info(` - ${a.id}.json`)});let l=o.map(a=>Mn(M,`${a.id}.json`));try{await Promise.all(l.map(a=>$n(a))),r.success(`\u{1F5D1}\uFE0F Removed ${l.length} pending migration files`)}catch(a){throw new Error(`Failed to remove pending migration files: ${a}`)}},Ln=async e=>{let{client:t,config:n}=e,r=St(t),{tables:o,indexes:l,constraints:a}=await r.introspect(n),c=(u,x)=>C=>C.some(D=>D.table===u&&D.columns.length===1&&D.columns[0]===x),s=o.map(u=>({name:u.name,columns:Object.fromEntries(Object.entries(u.columns).map(([x,C])=>{let D=c(u.name,x),_=D(a.primaryKey),R=D(a.unique),S=a.primaryKey.some(P=>P.table===u.name&&P.columns.length>1&&P.columns.includes(x))?!0:C.notNull;return[x,{type:C.dataType,notNull:S,defaultSql:C.default??void 0,primaryKey:_,unique:R}]}))})),m=n.tables.map(u=>({name:u.tableName,columns:Object.fromEntries(Object.entries(u.columns).map(([x,C])=>{let D=(n.primaryKeyConstraints??[]).some(R=>R.table===u.tableName&&R.columns.includes(x)),_=C.primaryKey||C.notNull||D;return[x,{...C,notNull:_}]}))})),i=Tt({current:{tables:s,indexes:l,uniqueConstraints:a.unique,primaryKeyConstraints:a.primaryKey,foreignKeyConstraints:a.foreignKey},ideal:{...n,tables:m}});return i.operations.length===0?null:{version:"1",id:Date.now()+"",diff:i}},Fn=(e,t)=>{let n=[];t.operations.forEach(r=>{switch(r.type){case"create_table":n.push(`-- create_table: ${r.table}`),Object.entries(r.columns).forEach(([o,l])=>{n.push(` -> column: ${o} (${JSON.stringify(l)})`)});break;case"drop_table":n.push(`-- remove_table: ${r.table}`);break;case"add_column":n.push([`-- add_column: ${r.table}.${r.column}`,` -> to: ${JSON.stringify(r.attributes)}`].join(`
|
|
94
|
+
`));break;case"drop_column":n.push(`-- remove_column: ${r.table}.${r.column}`);break;case"alter_column":n.push([`-- change_column: ${r.table}.${r.column}`,` -> from: ${JSON.stringify(r.before)}`,` -> to: ${JSON.stringify(r.after)}`].join(`
|
|
95
|
+
`));break;case"create_index":n.push(`-- create_index: ${r.table}.${r.name} (${r.columns.join(", ")})${r.unique?" [unique]":""}`);break;case"drop_index":n.push(`-- drop_index: ${r.table}.${r.name}`);break;case"create_primary_key_constraint":n.push(`-- create_primary_key_constraint: ${r.table}.${r.name} (${r.columns.join(", ")})`);break;case"drop_primary_key_constraint":n.push(`-- drop_primary_key_constraint: ${r.table}.${r.name}`);break;case"create_unique_constraint":n.push(`-- create_unique_constraint: ${r.table}.${r.name} (${r.columns.join(", ")})`);break;case"drop_unique_constraint":n.push(`-- drop_unique_constraint: ${r.table}.${r.name}`);break;case"create_foreign_key_constraint":n.push(`-- create_foreign_key_constraint: ${r.table}.${r.name} (${r.columns.join(", ")}) -> ${r.referencedTable} (${r.referencedColumns.join(", ")})${r.onDelete?` ON DELETE ${r.onDelete.toUpperCase()}`:""}${r.onUpdate?` ON UPDATE ${r.onUpdate.toUpperCase()}`:""}`);break;case"drop_foreign_key_constraint":n.push(`-- drop_foreign_key_constraint: ${r.table}.${r.name}`);break;default:break}}),e.reporter.log(n.join(`
|
|
96
|
+
`))},vt=In({meta:{name:"generate",description:"Generate migration files based on the current schema"},args:{"ignore-pending":{type:"boolean",description:"Ignore pending migrations and generate a new one",default:!1},dev:{type:"boolean",description:"Use dev database for safe migration generation",default:!1},squash:{type:"boolean",description:"Consolidate pending migrations into a single migration file",default:!1}},run:async e=>{try{let t=await E();await Un(t,{ignorePending:e.args["ignore-pending"],dev:e.args.dev,squash:e.args.squash})}catch(t){T.reporter.error(t),process.exit(1)}}});import{defineCommand as W}from"citty";async function Bn(e){let{manager:t,logger:n}=e,r=await t.getStatus();if(!r||r.type!=="container"){n.reporter.info("No running dev containers found");return}n.reporter.info(`Running: ${r.containerID} (${r.imageName})`)}async function zn(e){let{manager:t,logger:n}=e;if(!await t.isAvailable()){n.reporter.info("No running dev containers found");return}await t.start(),console.log(t.getConnectionString())}async function Hn(e){let{manager:t,logger:n}=e;if(!await t.isAvailable()){n.reporter.info("No dev containers found");return}await t.remove(),n.reporter.success("Cleaned up dev containers")}async function Vn(e){let{logger:t}=e,{reporter:n}=t,{manager:r}=await oe(e,{mode:"dev-start",logger:t});n.success(`\u2728 Dev database ready: ${r.getConnectionString()}`)}async function Pe(){let e=await E();if(!e.config.dev)throw new Error("No dev database configuration found");let{instance:t}=await ae(e.config.dev,e.config.database.dialect,"dev-start");return{...e,manager:t}}var Qn=W({meta:{name:"start",description:"Start development database with migrations applied"},run:async()=>{try{let e=await E();await Vn(e)}catch(e){T.reporter.error(e),process.exit(1)}}}),Wn=W({meta:{name:"status",description:"Show status of kyrage dev container"},run:async()=>{try{let e=await Pe();await Bn(e)}catch(e){T.reporter.error(e),process.exit(1)}}}),Gn=W({meta:{name:"get-url",description:"Print connection URL for running dev container"},run:async()=>{try{let e=await Pe();await zn(e)}catch(e){T.reporter.error(e),process.exit(1)}}}),Jn=W({meta:{name:"clean",description:"Remove all kyrage dev containers"},run:async()=>{try{let e=await Pe();await Hn(e)}catch(e){T.reporter.error(e),process.exit(1)}}}),Nt=W({meta:{name:"dev",description:"Manage development database containers"},subCommands:{start:Qn,status:Wn,"get-url":Gn,clean:Jn}});var Xn=Yn({meta:{name:"kyrage",version:qe,description:"Kysely migration CLI with declarative schema"},subCommands:{apply:Ot,generate:vt,dev:Nt}});Zn(Xn);
|
package/package.json
CHANGED
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
"name": "@izumisy/kyrage",
|
|
3
3
|
"private": false,
|
|
4
4
|
"type": "module",
|
|
5
|
-
"version": "1.
|
|
5
|
+
"version": "1.5.0",
|
|
6
6
|
"description": "A minimal, schema-based declarative migration tool for Node.js ecosystem",
|
|
7
7
|
"author": "Seiya Izumi (https://github.com/IzumiSy)",
|
|
8
8
|
"license": "MIT",
|
|
@@ -36,6 +36,7 @@
|
|
|
36
36
|
"c12": "^3.3.0",
|
|
37
37
|
"citty": "^0.1.6",
|
|
38
38
|
"consola": "^3.4.2",
|
|
39
|
+
"graph-data-structure": "^4.5.0",
|
|
39
40
|
"kysely": "^0.28.7",
|
|
40
41
|
"mysql2": "^3.15.0",
|
|
41
42
|
"pg": "^8.16.3",
|