@geekmidas/testkit 1.0.2 → 1.0.4

package/dist/initScript.mjs

@@ -0,0 +1,93 @@
+import pg from "pg";
+import { readFileSync } from "node:fs";
+
+//#region src/initScript.ts
+const { Client } = pg;
+/**
+* Parse a shell init script (like docker/postgres/init.sh) and extract
+* SQL blocks from heredoc sections (<<-EOSQL ... EOSQL).
+*
+* @param content - The shell script content
+* @param env - Environment variables to substitute ($VAR_NAME references)
+* @returns Array of SQL strings ready to execute
+* @internal Exported for testing
+*/
+function parseInitScript(content, env) {
+	const blocks = [];
+	const lines = content.split("\n");
+	let inHeredoc = false;
+	let currentBlock = [];
+	for (const line of lines) if (inHeredoc) if (/^\s*EOSQL\s*$/.test(line)) {
+		const sql = substituteEnvVars(currentBlock.join("\n"), env);
+		blocks.push(sql);
+		currentBlock = [];
+		inHeredoc = false;
+	} else currentBlock.push(line);
+	else if (line.includes("<<-EOSQL") || line.includes("<< EOSQL") || line.includes("<<EOSQL")) {
+		inHeredoc = true;
+		currentBlock = [];
+	}
+	return blocks;
+}
+/**
+* Replace shell variable references ($VAR_NAME and ${VAR_NAME})
+* with values from the provided env object.
+*/
+function substituteEnvVars(sql, env) {
+	let result = sql.replace(/\$\{(\w+)\}/g, (_, name) => env[name] ?? "");
+	result = result.replace(/\$(\w+)/g, (_, name) => env[name] ?? "");
+	return result;
+}
+/**
+* Read a postgres init script, parse out the SQL blocks,
+* substitute environment variables, and execute against a database.
+*
+* This is intended to run `docker/postgres/init.sh` against a test database
+* so that per-app users and schemas are created (matching what Docker does
+* on first volume initialization).
+*
+* Uses `CREATE ... IF NOT EXISTS` and `DO $$ ... END $$` wrappers where
+* needed so the script is idempotent.
+*
+* @param scriptPath - Path to the init.sh file
+* @param databaseUrl - PostgreSQL connection URL (should point to the test database)
+*
+* @example
+* ```typescript
+* // In your globalSetup.ts
+* import { runInitScript } from '@geekmidas/testkit/postgres';
+* import { Credentials } from '@geekmidas/envkit/credentials';
+*
+* const cleanup = await migrator.start();
+*
+* // Create per-app users in the test database
+* await runInitScript('docker/postgres/init.sh', Credentials.DATABASE_URL, {
+*   ...process.env,
+*   ...Credentials,
+* });
+* ```
+*/
+async function runInitScript(scriptPath, databaseUrl, env) {
+	const content = readFileSync(scriptPath, "utf-8");
+	const resolvedEnv = env ?? { ...process.env };
+	const blocks = parseInitScript(content, resolvedEnv);
+	if (blocks.length === 0) return;
+	const url = new URL(databaseUrl);
+	const client = new Client({
+		user: url.username,
+		password: decodeURIComponent(url.password),
+		host: url.hostname,
+		port: parseInt(url.port, 10),
+		database: url.pathname.slice(1)
+	});
+	try {
+		await client.connect();
+		for (const sql of blocks) await client.query(sql);
+	} finally {
+		await client.end();
+	}
+}
+
+//#endregion
+export { parseInitScript, runInitScript };
+//# sourceMappingURL=initScript.mjs.map

package/dist/initScript.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"initScript.mjs","names":["content: string","env: Record<string, string>","blocks: string[]","currentBlock: string[]","sql: string","scriptPath: string","databaseUrl: string","env?: Record<string, string>"],"sources":["../src/initScript.ts"],"sourcesContent":["import { readFileSync } from 'node:fs';\nimport pg from 'pg';\n\nconst { Client } = pg;\n\n/**\n * Parse a shell init script (like docker/postgres/init.sh) and extract\n * SQL blocks from heredoc sections (<<-EOSQL ... EOSQL).\n *\n * @param content - The shell script content\n * @param env - Environment variables to substitute ($VAR_NAME references)\n * @returns Array of SQL strings ready to execute\n * @internal Exported for testing\n */\nexport function parseInitScript(\n\tcontent: string,\n\tenv: Record<string, string>,\n): string[] {\n\tconst blocks: string[] = [];\n\tconst lines = content.split('\\n');\n\tlet inHeredoc = false;\n\tlet currentBlock: string[] = [];\n\n\tfor (const line of lines) {\n\t\tif (inHeredoc) {\n\t\t\t// Check for heredoc terminator (EOSQL at start of line, with optional leading whitespace)\n\t\t\tif (/^\\s*EOSQL\\s*$/.test(line)) {\n\t\t\t\tconst sql = substituteEnvVars(currentBlock.join('\\n'), env);\n\t\t\t\tblocks.push(sql);\n\t\t\t\tcurrentBlock = [];\n\t\t\t\tinHeredoc = false;\n\t\t\t} else {\n\t\t\t\tcurrentBlock.push(line);\n\t\t\t}\n\t\t} else if (\n\t\t\tline.includes('<<-EOSQL') ||\n\t\t\tline.includes('<< EOSQL') ||\n\t\t\tline.includes('<<EOSQL')\n\t\t) {\n\t\t\tinHeredoc = true;\n\t\t\tcurrentBlock = [];\n\t\t}\n\t}\n\n\treturn blocks;\n}\n\n/**\n * Replace shell variable references ($VAR_NAME and ${VAR_NAME})\n * with values from the provided env object.\n */\nfunction substituteEnvVars(sql: string, env: Record<string, string>): string {\n\t// Replace ${VAR_NAME} syntax\n\tlet result = sql.replace(/\\$\\{(\\w+)\\}/g, (_, name) => env[name] ?? '');\n\t// Replace $VAR_NAME syntax (word boundary after)\n\tresult = result.replace(/\\$(\\w+)/g, (_, name) => env[name] ?? '');\n\treturn result;\n}\n\n/**\n * Read a postgres init script, parse out the SQL blocks,\n * substitute environment variables, and execute against a database.\n *\n * This is intended to run `docker/postgres/init.sh` against a test database\n * so that per-app users and schemas are created (matching what Docker does\n * on first volume initialization).\n *\n * Uses `CREATE ... IF NOT EXISTS` and `DO $$ ... END $$` wrappers where\n * needed so the script is idempotent.\n *\n * @param scriptPath - Path to the init.sh file\n * @param databaseUrl - PostgreSQL connection URL (should point to the test database)\n *\n * @example\n * ```typescript\n * // In your globalSetup.ts\n * import { runInitScript } from '@geekmidas/testkit/postgres';\n * import { Credentials } from '@geekmidas/envkit/credentials';\n *\n * const cleanup = await migrator.start();\n *\n * // Create per-app users in the test database\n * await runInitScript('docker/postgres/init.sh', Credentials.DATABASE_URL, {\n *   ...process.env,\n *   ...Credentials,\n * });\n * ```\n */\nexport async function runInitScript(\n\tscriptPath: string,\n\tdatabaseUrl: string,\n\tenv?: Record<string, string>,\n): Promise<void> {\n\tconst content = readFileSync(scriptPath, 'utf-8');\n\tconst resolvedEnv = env ?? ({ ...process.env } as Record<string, string>);\n\tconst blocks = parseInitScript(content, resolvedEnv);\n\n\tif (blocks.length === 0) {\n\t\treturn;\n\t}\n\n\tconst url = new URL(databaseUrl);\n\tconst client = new Client({\n\t\tuser: url.username,\n\t\tpassword: decodeURIComponent(url.password),\n\t\thost: url.hostname,\n\t\tport: parseInt(url.port, 10),\n\t\tdatabase: url.pathname.slice(1),\n\t});\n\n\ttry {\n\t\tawait client.connect();\n\t\tfor (const sql of blocks) {\n\t\t\tawait client.query(sql);\n\t\t}\n\t} finally {\n\t\tawait client.end();\n\t}\n}\n"],"mappings":";;;;AAGA,MAAM,EAAE,QAAQ,GAAG;;;;;;;;;;AAWnB,SAAgB,gBACfA,SACAC,KACW;CACX,MAAMC,SAAmB,CAAE;CAC3B,MAAM,QAAQ,QAAQ,MAAM,KAAK;CACjC,IAAI,YAAY;CAChB,IAAIC,eAAyB,CAAE;AAE/B,MAAK,MAAM,QAAQ,MAClB,KAAI,UAEH,KAAI,gBAAgB,KAAK,KAAK,EAAE;EAC/B,MAAM,MAAM,kBAAkB,aAAa,KAAK,KAAK,EAAE,IAAI;AAC3D,SAAO,KAAK,IAAI;AAChB,iBAAe,CAAE;AACjB,cAAY;CACZ,MACA,cAAa,KAAK,KAAK;UAGxB,KAAK,SAAS,WAAW,IACzB,KAAK,SAAS,WAAW,IACzB,KAAK,SAAS,UAAU,EACvB;AACD,cAAY;AACZ,iBAAe,CAAE;CACjB;AAGF,QAAO;AACP;;;;;AAMD,SAAS,kBAAkBC,KAAaH,KAAqC;CAE5E,IAAI,SAAS,IAAI,QAAQ,gBAAgB,CAAC,GAAG,SAAS,IAAI,SAAS,GAAG;AAEtE,UAAS,OAAO,QAAQ,YAAY,CAAC,GAAG,SAAS,IAAI,SAAS,GAAG;AACjE,QAAO;AACP;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+BD,eAAsB,cACrBI,YACAC,aACAC,KACgB;CAChB,MAAM,UAAU,aAAa,YAAY,QAAQ;CACjD,MAAM,cAAc,OAAQ,EAAE,GAAG,QAAQ,IAAK;CAC9C,MAAM,SAAS,gBAAgB,SAAS,YAAY;AAEpD,KAAI,OAAO,WAAW,EACrB;CAGD,MAAM,MAAM,IAAI,IAAI;CACpB,MAAM,SAAS,IAAI,OAAO;EACzB,MAAM,IAAI;EACV,UAAU,mBAAmB,IAAI,SAAS;EAC1C,MAAM,IAAI;EACV,MAAM,SAAS,IAAI,MAAM,GAAG;EAC5B,UAAU,IAAI,SAAS,MAAM,EAAE;CAC/B;AAED,KAAI;AACH,QAAM,OAAO,SAAS;AACtB,OAAK,MAAM,OAAO,OACjB,OAAM,OAAO,MAAM,IAAI;CAExB,UAAS;AACT,QAAM,OAAO,KAAK;CAClB;AACD"}

package/dist/kysely.cjs

@@ -1,8 +1,8 @@
 require('./Factory-BhjUOBWN.cjs');
 const require_faker = require('./faker-B14IEMIN.cjs');
 const require_KyselyFactory = require('./KyselyFactory-BFqVIn_0.cjs');
-require('./PostgresMigrator-CHiBYEg_.cjs');
-const require_PostgresKyselyMigrator = require('./PostgresKyselyMigrator-C7ljZYvq.cjs');
+require('./PostgresMigrator-M9jpzOvN.cjs');
+const require_PostgresKyselyMigrator = require('./PostgresKyselyMigrator-CqNs3qX8.cjs');
 const require_VitestTransactionIsolator = require('./VitestTransactionIsolator-CMfJXZP8.cjs');
 const require_VitestKyselyTransactionIsolator = require('./VitestKyselyTransactionIsolator-D7RRXOBa.cjs');
 

package/dist/kysely.d.cts

@@ -1,8 +1,8 @@
-import { FakerFactory, faker } from "./faker-CbYiF-8_.cjs";
-import { ExtractSeedAttrs, FactorySeed } from "./Factory-D28yjUj5.cjs";
-import { KyselyFactory } from "./KyselyFactory-OUH03l2q.cjs";
-import "./PostgresMigrator-Bres0U6E.cjs";
-import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-CBltSOq5.cjs";
+import { FakerFactory, faker } from "./faker-Dg3trU4a.cjs";
+import { ExtractSeedAttrs, FactorySeed } from "./Factory-Cwzho3c8.cjs";
+import { KyselyFactory } from "./KyselyFactory-hMcVtvJq.cjs";
+import "./PostgresMigrator-CeYy-eHF.cjs";
+import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-CTNftoZK.cjs";
 import { DatabaseConnection, DatabaseFixtures, ExtendedDatabaseFixtures, FixtureCreators, IsolationLevel, TestWithExtendedFixtures, TransactionWrapperOptions } from "./VitestTransactionIsolator-CSroc7Df.cjs";
 import { VitestKyselyTransactionIsolator } from "./VitestKyselyTransactionIsolator-CduJlHoT.cjs";
 import { Kysely, Transaction } from "kysely";

package/dist/kysely.d.mts

@@ -1,8 +1,8 @@
-import { FakerFactory, faker } from "./faker-D9gz7KjY.mjs";
-import { ExtractSeedAttrs, FactorySeed } from "./Factory-C6W78ulZ.mjs";
-import { KyselyFactory } from "./KyselyFactory-Cc2UmOJk.mjs";
-import "./PostgresMigrator-S-YYosAC.mjs";
-import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-DrVWncqd.mjs";
+import { FakerFactory, faker } from "./faker-DsYCplsG.mjs";
+import { ExtractSeedAttrs, FactorySeed } from "./Factory-CVR3GdkW.mjs";
+import { KyselyFactory } from "./KyselyFactory-DRlMv-WT.mjs";
+import "./PostgresMigrator-C_QQ6q35.mjs";
+import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-x0uvWs3U.mjs";
 import { DatabaseConnection, DatabaseFixtures, ExtendedDatabaseFixtures, FixtureCreators, IsolationLevel, TestWithExtendedFixtures, TransactionWrapperOptions } from "./VitestTransactionIsolator-BNWJqh9f.mjs";
 import { VitestKyselyTransactionIsolator } from "./VitestKyselyTransactionIsolator-Cswnnj0k.mjs";
 import { Kysely, Transaction } from "kysely";

package/dist/kysely.mjs

@@ -1,8 +1,8 @@
 import "./Factory-BFVnMMCC.mjs";
 import { faker } from "./faker-BGKYFoCT.mjs";
 import { KyselyFactory } from "./KyselyFactory-DMswpwji.mjs";
-import "./PostgresMigrator-DcP1o-T6.mjs";
-import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-B4pScubb.mjs";
+import "./PostgresMigrator-DVAY04qN.mjs";
+import { PostgresKyselyMigrator } from "./PostgresKyselyMigrator-BIKd7G5C.mjs";
 import { IsolationLevel, extendWithFixtures } from "./VitestTransactionIsolator-DQ7tLqgV.mjs";
 import { VitestKyselyTransactionIsolator } from "./VitestKyselyTransactionIsolator-DceyIqr4.mjs";
 

package/dist/objection.cjs

@@ -1,8 +1,8 @@
 require('./Factory-BhjUOBWN.cjs');
 const require_faker = require('./faker-B14IEMIN.cjs');
 const require_ObjectionFactory = require('./ObjectionFactory-BeFBYcan.cjs');
-require('./PostgresMigrator-CHiBYEg_.cjs');
-const require_PostgresObjectionMigrator = require('./PostgresObjectionMigrator-BXLAVVwm.cjs');
+require('./PostgresMigrator-M9jpzOvN.cjs');
+const require_PostgresObjectionMigrator = require('./PostgresObjectionMigrator-DSaPhwjY.cjs');
 const require_VitestTransactionIsolator = require('./VitestTransactionIsolator-CMfJXZP8.cjs');
 const require_VitestObjectionTransactionIsolator = require('./VitestObjectionTransactionIsolator-CdLRrzNf.cjs');
 

package/dist/objection.d.cts

@@ -1,8 +1,8 @@
-import { FakerFactory, faker } from "./faker-CbYiF-8_.cjs";
-import { ExtractSeedAttrs, FactorySeed } from "./Factory-D28yjUj5.cjs";
-import { ObjectionFactory } from "./ObjectionFactory-B1WkcfZY.cjs";
-import "./PostgresMigrator-Bres0U6E.cjs";
-import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-CPfBAP7r.cjs";
+import { FakerFactory, faker } from "./faker-Dg3trU4a.cjs";
+import { ExtractSeedAttrs, FactorySeed } from "./Factory-Cwzho3c8.cjs";
+import { ObjectionFactory } from "./ObjectionFactory-DW-qwnqO.cjs";
+import "./PostgresMigrator-CeYy-eHF.cjs";
+import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-DHVC9h_P.cjs";
 import { DatabaseConnection, DatabaseFixtures, ExtendedDatabaseFixtures, FixtureCreators, IsolationLevel, TestWithExtendedFixtures, TransactionWrapperOptions } from "./VitestTransactionIsolator-CSroc7Df.cjs";
 import { VitestObjectionTransactionIsolator } from "./VitestObjectionTransactionIsolator-BXoR6xdG.cjs";
 import { Knex } from "knex";

package/dist/objection.d.mts

@@ -1,8 +1,8 @@
-import { FakerFactory, faker } from "./faker-D9gz7KjY.mjs";
-import { ExtractSeedAttrs, FactorySeed } from "./Factory-C6W78ulZ.mjs";
-import { ObjectionFactory } from "./ObjectionFactory-BYnPr9ZP.mjs";
-import "./PostgresMigrator-S-YYosAC.mjs";
-import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-DVEqB5tp.mjs";
+import { FakerFactory, faker } from "./faker-DsYCplsG.mjs";
+import { ExtractSeedAttrs, FactorySeed } from "./Factory-CVR3GdkW.mjs";
+import { ObjectionFactory } from "./ObjectionFactory-DkJUf-uM.mjs";
+import "./PostgresMigrator-C_QQ6q35.mjs";
+import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-1j6YIB1c.mjs";
 import { DatabaseConnection, DatabaseFixtures, ExtendedDatabaseFixtures, FixtureCreators, IsolationLevel, TestWithExtendedFixtures, TransactionWrapperOptions } from "./VitestTransactionIsolator-BNWJqh9f.mjs";
 import { VitestObjectionTransactionIsolator } from "./VitestObjectionTransactionIsolator-x6hY5j4u.mjs";
 import { TestAPI } from "vitest";

package/dist/objection.mjs

@@ -1,8 +1,8 @@
 import "./Factory-BFVnMMCC.mjs";
 import { faker } from "./faker-BGKYFoCT.mjs";
 import { ObjectionFactory } from "./ObjectionFactory-QCJ7u0Ql.mjs";
-import "./PostgresMigrator-DcP1o-T6.mjs";
-import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-BJ5X48U8.mjs";
+import "./PostgresMigrator-DVAY04qN.mjs";
+import { PostgresObjectionMigrator } from "./PostgresObjectionMigrator-DjPKdUbm.mjs";
 import { IsolationLevel, extendWithFixtures } from "./VitestTransactionIsolator-DQ7tLqgV.mjs";
 import { VitestObjectionTransactionIsolator } from "./VitestObjectionTransactionIsolator-OF2osYY5.mjs";
 

package/dist/os/directory.d.cts

@@ -1,2 +1,2 @@
-import { DirectoryFixtures, itWithDir } from "../directory-DGOcVlKD.cjs";
+import { DirectoryFixtures, itWithDir } from "../directory-DAnMWi50.cjs";
 export { DirectoryFixtures, itWithDir };

package/dist/os/directory.d.mts

@@ -1,2 +1,2 @@
-import { DirectoryFixtures, itWithDir } from "../directory-CVrfTq1I.mjs";
+import { DirectoryFixtures, itWithDir } from "../directory-CYXmolu1.mjs";
 export { DirectoryFixtures, itWithDir };

package/dist/os/index.d.cts

@@ -1,2 +1,2 @@
-import { itWithDir } from "../directory-DGOcVlKD.cjs";
+import { itWithDir } from "../directory-DAnMWi50.cjs";
 export { itWithDir };

package/dist/os/index.d.mts

@@ -1,2 +1,2 @@
-import { itWithDir } from "../directory-CVrfTq1I.mjs";
+import { itWithDir } from "../directory-CYXmolu1.mjs";
 export { itWithDir };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@geekmidas/testkit",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "private": false,
   "type": "module",
   "exports": {
@@ -48,6 +48,11 @@
       "types": "./dist/benchmark.d.ts",
       "import": "./dist/benchmark.mjs",
       "require": "./dist/benchmark.cjs"
+    },
+    "./postgres": {
+      "types": "./dist/initScript.d.ts",
+      "import": "./dist/initScript.mjs",
+      "require": "./dist/initScript.cjs"
     }
   },
   "dependencies": {

package/src/PostgresKyselyMigrator.ts

@@ -40,6 +40,18 @@ const logger = console;
  * const cleanup = await migrator.start();
  * // Run tests...
  * await cleanup();
+ *
+ * // With afterCreate hook to create per-app users
+ * import { runInitScript } from '@geekmidas/testkit/postgres';
+ *
+ * const migrator = new PostgresKyselyMigrator({
+ *   uri: 'postgresql://localhost:5432/test_db',
+ *   db,
+ *   provider,
+ *   afterCreate: async (uri) => {
+ *     await runInitScript('docker/postgres/init.sh', uri, env);
+ *   },
+ * });
  * ```
  */
 export class PostgresKyselyMigrator extends PostgresMigrator {
@@ -50,15 +62,17 @@ export class PostgresKyselyMigrator extends PostgresMigrator {
 	 * @param options.uri - PostgreSQL connection URI
 	 * @param options.db - Kysely database instance
 	 * @param options.provider - Migration provider for locating migration files
+	 * @param options.afterCreate - Optional hook called after database creation but before migrations
 	 */
 	constructor(
 		private options: {
 			uri: string;
 			db: Kysely<any>;
 			provider: MigrationProvider;
+			afterCreate?: (uri: string) => Promise<void>;
 		},
 	) {
-		super(options.uri);
+		super(options.uri, options.afterCreate);
 	}
 
 	/**

package/src/PostgresMigrator.ts

@@ -57,6 +57,17 @@ const logger = console;
  * const migrator = new MyMigrator('postgresql://localhost:5432/test_db');
  * const cleanup = await migrator.start();
  *
+ * // With afterCreate hook (e.g. run init script to create per-app users)
+ * import { runInitScript } from '@geekmidas/testkit/postgres';
+ *
+ * const migrator = new MyMigrator(
+ *   'postgresql://localhost:5432/test_db',
+ *   async (uri) => {
+ *     await runInitScript('docker/postgres/init.sh', uri, env);
+ *   },
+ * );
+ * const cleanup = await migrator.start();
+ *
  * // Run tests...
  *
  * // Clean up
@@ -68,8 +79,12 @@ export abstract class PostgresMigrator {
 	 * Creates a new PostgresMigrator instance.
 	 *
 	 * @param uri - PostgreSQL connection URI
+	 * @param afterCreate - Optional hook called after database creation but before migrations
 	 */
-	constructor(private uri: string) {}
+	constructor(
+		private uri: string,
+		private afterCreate?: (uri: string) => Promise<void>,
+	) {}
 
 	/**
 	 * Abstract method to be implemented by subclasses.
@@ -161,7 +176,9 @@ export abstract class PostgresMigrator {
 		const { database, db } = await setupClient(this.uri);
 		try {
 			await PostgresMigrator.create(this.uri);
-			// Implement migration logic here
+			if (this.afterCreate) {
+				await this.afterCreate(this.uri);
+			}
 			await this.migrate();
 			logger.log(`Migrating database: ${database}`);
 			// Example: await db.query('CREATE TABLE example (id SERIAL PRIMARY KEY)');

package/src/__tests__/PostgresKyselyMigrator.spec.ts

@@ -577,6 +577,110 @@ describe('PostgresKyselyMigrator', () => {
 		});
 	});
 
+	describe('afterCreate hook', () => {
+		it('should call afterCreate before migrations run', async () => {
+			const hookDbName = `test_kysely_hook_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${hookDbName}`;
+			const callOrder: string[] = [];
+
+			const db = new Kysely<TestSchema>({
+				dialect: new PostgresDialect({
+					pool: new Pool({
+						host: 'localhost',
+						port: 5432,
+						user: 'geekmidas',
+						password: 'geekmidas',
+						database: hookDbName,
+					}),
+				}),
+			});
+
+			const provider = new TestMigrationProvider();
+			provider.addMigration('001_create_table', {
+				up: async (db) => {
+					callOrder.push('migrate');
+					await db.schema
+						.createTable('items')
+						.addColumn('id', 'serial', (col) => col.primaryKey())
+						.execute();
+				},
+			});
+
+			const migrator = new PostgresKyselyMigrator({
+				uri,
+				db,
+				provider,
+				afterCreate: async (receivedUri) => {
+					callOrder.push('afterCreate');
+					expect(receivedUri).toBe(uri);
+				},
+			});
+
+			const cleanup = await migrator.start();
+
+			expect(callOrder).toEqual(['afterCreate', 'migrate']);
+
+			await cleanup();
+		});
+
+		it('should allow creating users and schemas before migrations', async () => {
+			const schemaDbName = `test_kysely_schema_hook_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${schemaDbName}`;
+
+			const db = new Kysely<TestSchema>({
+				dialect: new PostgresDialect({
+					pool: new Pool({
+						host: 'localhost',
+						port: 5432,
+						user: 'geekmidas',
+						password: 'geekmidas',
+						database: schemaDbName,
+					}),
+				}),
+			});
+
+			const provider = new TestMigrationProvider();
+			provider.addMigration('001_use_schema', {
+				up: async (db) => {
+					// Verify schema exists (created by afterCreate)
+					const result =
+						await sql`SELECT schema_name FROM information_schema.schemata WHERE schema_name = 'app'`.execute(
+							db,
+						);
+					if (result.rows.length === 0) {
+						throw new Error('Expected app schema to exist');
+					}
+					await db.schema
+						.createTable('items')
+						.addColumn('id', 'serial', (col) => col.primaryKey())
+						.execute();
+				},
+			});
+
+			const migrator = new PostgresKyselyMigrator({
+				uri,
+				db,
+				provider,
+				afterCreate: async () => {
+					const client = new Client({
+						host: 'localhost',
+						port: 5432,
+						user: 'geekmidas',
+						password: 'geekmidas',
+						database: schemaDbName,
+					});
+					await client.connect();
+					await client.query('CREATE SCHEMA IF NOT EXISTS app');
+					await client.end();
+				},
+			});
+
+			const cleanup = await migrator.start();
+
+			await cleanup();
+		});
+	});
+
 	describe('error scenarios', () => {
 		it('should handle provider errors', async () => {
 			const providerErrorDbName = `test_provider_error_${Date.now()}`;

package/src/__tests__/PostgresMigrator.spec.ts

@@ -18,6 +18,10 @@ class TestPostgresMigrator extends PostgresMigrator {
 	public migrateError?: Error;
 	public customMigrations: Array<() => Promise<void>> = [];
 
+	constructor(uri: string, afterCreate?: (uri: string) => Promise<void>) {
+		super(uri, afterCreate);
+	}
+
 	async migrate(): Promise<void> {
 		this.migrateCalled = true;
 		if (this.migrateError) {
@@ -285,6 +289,111 @@ describe('PostgresMigrator', () => {
 		});
 	});
 
+	describe('afterCreate hook', () => {
+		it('should call afterCreate after database creation and before migrate', async () => {
+			const hookDbName = `test_after_create_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${hookDbName}`;
+			const callOrder: string[] = [];
+
+			const migrator = new TestPostgresMigrator(uri, async (receivedUri) => {
+				callOrder.push('afterCreate');
+				expect(receivedUri).toBe(uri);
+			});
+
+			migrator.addMigration(async () => {
+				callOrder.push('migrate');
+			});
+
+			const cleanup = await migrator.start();
+
+			expect(callOrder).toEqual(['afterCreate', 'migrate']);
+
+			await cleanup();
+		});
+
+		it('should skip afterCreate when not provided', async () => {
+			const noHookDbName = `test_no_hook_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${noHookDbName}`;
+			const migrator = new TestPostgresMigrator(uri);
+
+			const cleanup = await migrator.start();
+
+			expect(migrator.migrateCalled).toBe(true);
+
+			await cleanup();
+		});
+
+		it('should propagate afterCreate errors and not run migrate', async () => {
+			const errorHookDbName = `test_hook_error_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${errorHookDbName}`;
+
+			const migrator = new TestPostgresMigrator(uri, async () => {
+				throw new Error('afterCreate failed');
+			});
+
+			await expect(migrator.start()).rejects.toThrow('afterCreate failed');
+			expect(migrator.migrateCalled).toBe(false);
+
+			// Cleanup the created database
+			const cleanupClient = new Client({
+				host: 'localhost',
+				port: 5432,
+				user: 'geekmidas',
+				password: 'geekmidas',
+				database: 'postgres',
+			});
+			try {
+				await cleanupClient.connect();
+				await cleanupClient.query(
+					`DROP DATABASE IF EXISTS "${errorHookDbName}"`,
+				);
+			} finally {
+				await cleanupClient.end();
+			}
+		});
+
+		it('should allow afterCreate to set up database objects before migrations', async () => {
+			const setupDbName = `test_hook_setup_${Date.now()}`;
+			const uri = `postgresql://geekmidas:geekmidas@localhost:5432/${setupDbName}`;
+
+			const migrator = new TestPostgresMigrator(uri, async () => {
+				const client = new Client({
+					host: 'localhost',
+					port: 5432,
+					user: 'geekmidas',
+					password: 'geekmidas',
+					database: setupDbName,
+				});
+				await client.connect();
+				await client.query(
+					"CREATE SCHEMA IF NOT EXISTS app; COMMENT ON SCHEMA app IS 'Created by afterCreate hook';",
+				);
+				await client.end();
+			});
+
+			migrator.addMigration(async () => {
+				// Verify schema exists during migration
+				const client = new Client({
+					host: 'localhost',
+					port: 5432,
+					user: 'geekmidas',
+					password: 'geekmidas',
+					database: setupDbName,
+				});
+				await client.connect();
+				const result = await client.query(
+					"SELECT schema_name FROM information_schema.schemata WHERE schema_name = 'app'",
+				);
+				expect(result.rowCount).toBe(1);
+				await client.end();
+			});
+
+			const cleanup = await migrator.start();
+
+			await cleanup();
+		});
+	});
+
 	describe('abstract method', () => {
 		it('should require concrete implementation of migrate method', () => {
 			// TypeScript ensures abstract methods are implemented