@izumisy/kyrage 1.3.1 → 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 +5 -4
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
|
|
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 S,b as E}from"./chunk-EY62Q3Z2.js";import{defineCommand as on,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 S,b as E}from"./chunk-EY62Q3Z2.js";import{defineCommand as on,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 S,b as E}from"./chunk-EY62Q3Z2.js";import{defineCommand as on,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 Xe}from"@testcontainers/cockroachdb";import{Pool as Ze}from"pg";import{PostgresAdapter as et,PostgresDialect as tt}from"kysely";var H=class{getName(){return"cockroachdb"}createKyselyDialect(t){return new me({pool:new Ze({connectionString:t})})}createIntrospectionDriver(t){let n=["rowid"];return{convertTypeName:ce,introspect:async a=>{let o=await se(t)(a);return{tables:o.tables.filter(i=>!n.includes(i.name)),indexes:o.indexes,constraints:o.constraints}}}}createDevDatabaseProvider(){return new R(this.getName(),t=>new Xe(t))}parseDevDatabaseConfig(t){return P({defaultImage:"cockroachdb/cockroach:latest-v24.3"}).parse(t)}async hasReusableDevDatabase(){return M(this.getName())}},le=class extends et{async acquireMigrationLock(t,n){await t.selectFrom(n.lockTable).selectAll().forUpdate().execute()}},me=class extends tt{createAdapter(){return new le}};var nt={postgres:new B,cockroachdb:new H},w=e=>{let t=nt[e];if(!t)throw new Error(`Unsupported dialect: ${e}`);return t};import{DummyDriver as at}from"kysely";var Q=class extends at{constructor(n){super();this.queries=n}async acquireConnection(){return new ue(this.queries)}},ue=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 z=e=>new pe({databaseProps:e.database,options:e.options}),pe=class{constructor(t){this.constructorProps=t}getDB(t){let n=w(this.constructorProps.databaseProps.dialect).createKyselyDialect(this.constructorProps.databaseProps.connectionString);return new de({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}},de=class extends rt{constructor(n,a){let o=[];super({dialect:{...n.dialect,createDriver:()=>a.isPlan?new Q(o):n.dialect.createDriver()}});this.plannedQueries=[];this.plannedQueries=o}getPlannedQueries(){return this.plannedQueries}};import{createConsola as ot}from"consola";var it=ot({stdout:process.stderr}),x={stdout:e=>console.log(e),reporter:{...it}},ye={stdout:()=>{},reporter:{info:()=>{},warn:()=>{},success:()=>{},log:()=>{},error:()=>{}}};async function T(){let e=await Ne();return{client:z({database:e.database}),logger:x,config:e}}import{Migrator as Rt}from"kysely";import{DEFAULT_MIGRATION_TABLE as Ee,isColumnDataType as st,sql as Te}from"kysely";import{readdir as ct,readFile as lt}from"fs/promises";import{join as mt}from"path";import U from"zod";import*as Ae from"ramda";var I="migrations",ut=U.object({operations:U.array(xe).readonly()}),pt=U.object({id:U.string(),version:U.string(),diff:ut}),ge=async()=>{try{let t=(await ct(I)).filter(n=>n.endsWith(".json")).map(async n=>pt.parse(JSON.parse(await lt(mt(I,n),"utf-8"))));return await Promise.all(t)}catch(e){if(e instanceof Object&&"code"in e&&e.code==="ENOENT")return[];throw e}},A=async e=>{var l=[];try{let t=S(l,e.getDB(),!0);let n=await ge();let a=(await t.introspection.getTables({withInternalKyselyTables:!0})).map(r=>r.name);if(!a.includes(Ee))return n;let o=await t.selectFrom(Ee).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 u=E(l,c,s);u&&await u}},Oe=e=>({getMigrations:async()=>{let t=await e.migrationsResolver(),n={};return t.forEach(a=>{n[a.id]={up:async o=>{await dt(o,a.diff)}}}),n}});async function dt(e,t){let n=Ae.pipe(Tt,At,Ot);for(let a of n(t.operations))await yt(e,a)}async function yt(e,t){switch(t.type){case"create_table":return gt(e,t);case"drop_table":return ft(e,t);case"add_column":return bt(e,t);case"drop_column":return Ct(e,t);case"alter_column":return ht(e,t);case"create_index":return Dt(e,t);case"drop_index":return xt(e,t);case"create_primary_key_constraint":return _t(e,t);case"drop_primary_key_constraint":return vt(e,t);case"create_unique_constraint":return Nt(e,t);case"drop_unique_constraint":return St(e,t);case"create_foreign_key_constraint":return Et(e,t);case"drop_foreign_key_constraint":return wt(e,t);default:throw new Error(`Unknown operation type: ${t.type}`)}}async function gt(e,t){let n=e.schema.createTable(t.table);for(let[a,o]of Object.entries(t.columns)){let l=o.type;fe(l),n=n.addColumn(a,l,i=>{let c=i;return o.notNull&&(c=c.notNull()),typeof o.defaultSql=="string"&&(c=c.defaultTo(Te.raw(o.defaultSql))),c})}await n.execute()}async function ft(e,t){await e.schema.dropTable(t.table).execute()}async function bt(e,t){let n=t.attributes.type;fe(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(Te.raw(t.attributes.defaultSql))),o}).execute()}async function Ct(e,t){await e.schema.alterTable(t.table).dropColumn(t.column).execute()}async function ht(e,t){let{table:n,column:a,before:o,after:l}=t;if(o.type!==l.type){let i=l.type;fe(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())}async function Dt(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()}async function xt(e,t){await e.schema.dropIndex(t.name).execute()}async function _t(e,t){await e.schema.alterTable(t.table).addPrimaryKeyConstraint(t.name,t.columns).execute()}async function vt(e,t){await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}async function Nt(e,t){await e.schema.alterTable(t.table).addUniqueConstraint(t.name,t.columns).execute()}async function St(e,t){await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}async function Et(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()}async function wt(e,t){await e.schema.alterTable(t.table).dropConstraint(t.name).execute()}var fe=e=>{if(!st(e))throw new Error(`Unsupported data type: ${e}`)},we={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:6,add_column:7,alter_column:8,create_index:9,create_primary_key_constraint:10,create_unique_constraint:11,create_foreign_key_constraint:12},Tt=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))},At=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})},Ot=e=>e.slice().sort((t,n)=>{let a=we[t.type],o=we[n.type];return a!==o?a-o:t.table.localeCompare(n.table)});import{format as Pt}from"sql-formatter";async function L(e,t){var p=[];try{let{client:n,logger:a}=e;let{reporter:o}=a;let l=S(p,n.getDB({plan:t.plan}),!0);let i=Oe({migrationsResolver:async()=>t.plan?await A(n):await ge(),options:{plan:t.plan}});let c=new Rt({db:l,provider:i});let{results:s,error:u}=await c.migrateToLatest();let r=l.getPlannedQueries();if(r.length>0){r.forEach(D=>{a.stdout(t.pretty?Pt(D.sql):D.sql)});return}s&&s.length>0?s.forEach(D=>{D.status==="Error"?o.error(`Migration failed: ${D.migrationName}`):D.status==="Success"&&o.success(`Migration applied: ${D.migrationName}`)}):o.info("No migrations to run");if(u)throw u instanceof Error?u:new Error(`Migration error: ${u}`)}catch(d){var C=d,b=!0}finally{var h=E(p,C,b);h&&await h}}var Ke=Kt({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 T();await L(t,{plan:e.args.plan,pretty:e.args.pretty})}catch(t){x.reporter.error(t),process.exit(1)}}});import{defineCommand as Ft}from"citty";import{mkdir as Bt,writeFile as Ht,unlink as Qt}from"fs/promises";import{join as zt}from"path";import*as y from"ramda";var q=()=>({added:(e,t,n)=>y.pipe(y.difference(t),y.map(n))(e),removed:(e,t,n)=>y.pipe(y.difference(e),y.map(n))(t),changed:(e,t,n,a)=>y.pipe(y.intersection(e),y.filter(n),y.map(a))(t)}),V=y.prop("name"),It=y.eqBy(e=>[e.type,e.notNull??!1,e.primaryKey??!1,e.unique??!1]);function qt(e,t){let n=[],a=q(),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(f.addColumn({table:e.name,column:s.column},s.attributes))}),a.removed(i,c,s=>({column:s,attributes:o[s]})).forEach(s=>{n.push(f.dropColumn({table:e.name,column:s.column},s.attributes))}),a.changed(i,c,s=>!It(o[s],l[s]),s=>({column:s,before:o[s],after:l[s]})).forEach(s=>{n.push(f.alterColumn({table:e.name,column:s.column},s.before,s.after))}),n}function kt(e){let{current:t,ideal:n}=e,a=[],o=q(),l=y.map(V,t),i=y.map(V,n);return o.added(l,i,c=>n.find(u=>u.name===c)).forEach(c=>{a.push(f.createTable(c.name,c.columns))}),o.removed(l,i,y.identity).forEach(c=>{a.push(f.dropTable(c))}),y.filter(c=>n.some(s=>s.name===c.name))(t).forEach(c=>{let s=n.find(u=>u.name===c.name);a.push(...qt(c,s))}),a}function $t(e){let{current:t,ideal:n}=e,a=[],o=q(),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()),u=Array.from(c.keys());return o.added(s,u,r=>c.get(r)).forEach(r=>{a.push(f.createIndex(r))}),o.removed(s,u,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(f.dropIndex(r))}),o.changed(s,u,r=>{let p=i.get(r),d=c.get(r);return!y.equals(p.columns,d.columns)||p.unique!==d.unique},r=>{let p=i.get(r),d=c.get(r);return{current:p,ideal:d}}).forEach(({current:r,ideal:p})=>{a.push(f.dropIndex(r)),a.push(f.createIndex(p))}),a}function Mt(e){let{current:t,ideal:n}=e,a=[],o=q(),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()),u=Array.from(c.keys());return o.added(s,u,r=>c.get(r)).forEach(r=>{a.push(f.createPrimaryKeyConstraint(r))}),o.removed(s,u,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(f.dropPrimaryKeyConstraint(r))}),o.changed(s,u,r=>{let p=i.get(r),d=c.get(r);return!y.equals(p.columns,d.columns)},r=>{let p=i.get(r),d=c.get(r);return{current:p,ideal:d}}).forEach(({current:r,ideal:p})=>{a.push(f.dropPrimaryKeyConstraint(r)),a.push(f.createPrimaryKeyConstraint(p))}),a}function Ut(e){let{current:t,ideal:n}=e,a=[],o=q(),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()),u=Array.from(c.keys());return o.added(s,u,r=>c.get(r)).forEach(r=>{a.push(f.createUniqueConstraint(r))}),o.removed(s,u,r=>{let p=i.get(r);return{table:p.table,name:p.name}}).forEach(r=>{a.push(f.dropUniqueConstraint(r))}),o.changed(s,u,r=>{let p=i.get(r),d=c.get(r);return!y.equals(p.columns,d.columns)},r=>{let p=i.get(r),d=c.get(r);return{current:p,ideal:d}}).forEach(({current:r,ideal:p})=>{a.push(f.dropUniqueConstraint(r)),a.push(f.createUniqueConstraint(p))}),a}function Lt(e){let t=[],n=q(),a=e.current.map(V),o=e.ideal.map(V);t.push(...n.added(a,o,i=>{let c=e.ideal.find(p=>p.name===i),{table:s,name:u,...r}=c;return f.createForeignKeyConstraint({table:s,name:u},r)})),t.push(...n.removed(a,o,i=>{let c=e.current.find(s=>s.name===i);return f.dropForeignKeyConstraint(c)}));let l=i=>{let c=e.current.find(u=>u.name===i),s=e.ideal.find(u=>u.name===i);return!c||!s?!1:!(y.equals(c.columns,s.columns)&&c.referencedTable===s.referencedTable&&y.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(d=>d.name===i),s=e.ideal.find(d=>d.name===i),{table:u,name:r,...p}=s;return[f.dropForeignKeyConstraint(c),f.createForeignKeyConstraint({table:u,name:r},p)]}).flat()),t}function Re(e){let t=kt({current:e.current.tables,ideal:e.ideal.tables}),n=$t({current:e.current.indexes,ideal:e.ideal.indexes}),a=Mt({current:e.current.primaryKeyConstraints,ideal:e.ideal.primaryKeyConstraints}),o=Ut({current:e.current.uniqueConstraints,ideal:e.ideal.uniqueConstraints}),l=Lt({current:e.current.foreignKeyConstraints,ideal:e.ideal.foreignKeyConstraints});return{operations:[...t,...n,...a,...o,...l]}}var Pe=e=>{let n=w(e.getDialect()).createIntrospectionDriver(e);return{introspect:async o=>{var b=[];try{let l=S(b,e.getDB(),!0);let i=await l.introspection.getTables();let{tables:c,indexes:s,constraints:u}=await n.introspect({config:o});let r=()=>i.map(_=>{let K={};for(let k of _.columns){let Y=c.find(Ce=>Ce.table===_.name&&Ce.name===k.name);Y&&(K[k.name]={schema:_.schema,table:_.name,name:k.name,dataType:n.convertTypeName(k.dataType),default:Y.default??null,characterMaximumLength:Y.characterMaximumLength??null,notNull:!k.isNullable})}return{schema:_.schema,name:_.name,columns:K}});let p=_=>!_.table.startsWith("kysely_");let d=()=>s.filter(p);let C=()=>({primaryKey:u.primaryKey.filter(p),unique:u.unique.filter(p),foreignKey:u.foreignKey.filter(p)});return{tables:r(),indexes:d(),constraints:C()}}catch(h){var D=h,O=!0}finally{var G=E(b,D,O);G&&await G}}}};var W=async(e,t,n)=>{let a=w(t),o=await a.createDevDatabaseProvider().setup(a.parseDevDatabaseConfig(e),n);return{manageType:n,instance:o}};async function jt(e,t){let{config:n}=e,a=w(n.database.dialect);switch(t.mode){case"dev-start":{let{instance:o,manageType:l}=await W(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 W(n.dev,n.database.dialect,o?"dev-start":"one-off");return{manageType:i,manager:l,result:{reused:o}}}}}async function J(e,t){let{config:n}=e,a=t.logger||ye,{reporter:o}=a;if(!n.dev)throw new Error("Dev database configuration is required");let{manageType:l,manager:i,result:c}=await jt(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=z({database:{dialect:n.database.dialect,connectionString:i.getConnectionString()}}),u=await A(s);return u.length>0?(o.info(`\u{1F504} Applying ${u.length} pending migrations...`),await L({client:s,logger:ye,config:n},{plan:!1,pretty:!1}),o.success(`Applied ${u.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 Vt(e,t){let{logger:n,config:a}=e,{reporter:o}=n;t.squash&&await Wt(e);let{client:l,cleanup:i}=t.dev?await J(e,{mode:"generate-dev",logger:n}):{client:e.client,cleanup:async()=>{}};try{if(!t.ignorePending){let r=await A(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",
|
|
@@ -33,16 +33,17 @@
|
|
|
33
33
|
"@testcontainers/cockroachdb": "^11.5.1",
|
|
34
34
|
"@testcontainers/postgresql": "^11.5.1",
|
|
35
35
|
"better-sqlite3": "^12.2.0",
|
|
36
|
-
"c12": "^3.
|
|
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
|
-
"mysql2": "^3.
|
|
41
|
+
"mysql2": "^3.15.0",
|
|
41
42
|
"pg": "^8.16.3",
|
|
42
43
|
"ramda": "^0.31.3",
|
|
43
44
|
"sql-formatter": "^15.6.9",
|
|
44
45
|
"testcontainers": "^11.5.1",
|
|
45
|
-
"zod": "^4.1.
|
|
46
|
+
"zod": "^4.1.11"
|
|
46
47
|
},
|
|
47
48
|
"devDependencies": {
|
|
48
49
|
"@swc/core": "^1.13.5",
|