joist-codegen 2.3.0-next.7 → 2.3.0-next.71

package/build/index.cjs

@@ -0,0 +1,135 @@
+#!/usr/bin/env node
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const require_runtime = require("./_virtual/_rolldown/runtime.cjs");
+const require_codemods_index = require("./codemods/index.cjs");
+const require_utils = require("./utils.cjs");
+const require_inheritance = require("./inheritance.cjs");
+const require_logger = require("./logger.cjs");
+const require_config = require("./config.cjs");
+const require_assignTags = require("./assignTags.cjs");
+const require_foreignKeyOrdering = require("./foreignKeyOrdering.cjs");
+const require_generate = require("./generate.cjs");
+const require_generateFlushFunction = require("./generateFlushFunction.cjs");
+const require_installSkills = require("./installSkills.cjs");
+const require_scanEntityFiles = require("./scanEntityFiles.cjs");
+const require_loadMetadata = require("./loadMetadata.cjs");
+const require_EntityDbMetadata = require("./EntityDbMetadata.cjs");
+let joist_utils = require("joist-utils");
+let pg_structure = require("pg-structure");
+pg_structure = require_runtime.__toESM(pg_structure, 1);
+let ts_poet = require("ts-poet");
+let pg = require("pg");
+let node_process = require("node:process");
+node_process = require_runtime.__toESM(node_process, 1);
+//#region src/index.ts
+const { default: pgStructure } = pg_structure.default;
+async function joistCodegen() {
+	const config = await require_config.loadConfig();
+	maybeSetDatabaseUrl(config);
+	if (!node_process.default.env.DATABASE_URL && !node_process.default.env.DB_USER) {
+		console.log(`Database connection information not found, please set either:`);
+		console.log(`  - the DATABASE_URL env variable (i.e. using .env and dotenv), or`);
+		console.log(`  - the databaseUrl key in joist-config.json`);
+		return;
+	}
+	const pgConfig = (0, joist_utils.newPgConnectionConfig)();
+	const client = new pg.Client(pgConfig);
+	await client.connect();
+	const dbMetadata = await loadSchemaMetadata(config, client);
+	const { entities, enums, totalTables } = dbMetadata;
+	console.log(`Found ${totalTables} total tables, ${entities.length} entity tables, ${Object.entries(enums).length} enum tables`);
+	console.log("");
+	require_inheritance.applyInheritanceUpdates(config, dbMetadata);
+	require_EntityDbMetadata.resolveNameConflicts(config, dbMetadata);
+	require_assignTags.assignTags(config, dbMetadata);
+	validateTagDelimiter(config, entities);
+	await require_scanEntityFiles.scanEntityFiles(config, dbMetadata);
+	await require_foreignKeyOrdering.maybeSetForeignKeyOrdering(config, dbMetadata.entities);
+	await maybeGenerateFlushFunctions(config, client, pgConfig, dbMetadata);
+	await client.end();
+	await require_codemods_index.maybeRunTransforms(config);
+	for (const entity of entities) require_EntityDbMetadata.failIfOverlappingFieldNames(entity);
+	require_config.warnInvalidConfigEntries(config, dbMetadata);
+	await generateAndSaveFiles(config, dbMetadata);
+	if (config.skills !== false) await require_installSkills.installSkills();
+	require_config.stripStiPlaceholders(config, entities);
+	await require_config.writeConfig(config);
+}
+/** Uses entities and enums from the `db` schema and saves them into our entities directory. */
+async function generateAndSaveFiles(config, dbMeta) {
+	const files = await require_generate.generateFiles(config, dbMeta);
+	const esmExt = config.esm ? config.allowImportingTsExtensions ? "ts" : "js" : "";
+	await (0, ts_poet.saveFiles)({
+		toolName: "joist-codegen",
+		directory: config.entitiesDirectory,
+		files,
+		toStringOpts: { importExtensions: esmExt || false }
+	});
+}
+async function maybeGenerateFlushFunctions(config, client, pgConfig, db) {
+	if (config.createFlushFunction !== false) {
+		if (Array.isArray(config.createFlushFunction)) {
+			console.log("Creating flush_database functions");
+			await Promise.all(config.createFlushFunction.map(async (dbName) => {
+				const client = new pg.Client({
+					...pgConfig,
+					database: dbName
+				});
+				await client.connect();
+				await require_generateFlushFunction.createFlushFunction(client, db);
+				await client.end();
+			}));
+		} else {
+			console.log("Creating flush_database function");
+			await require_generateFlushFunction.createFlushFunction(client, db);
+		}
+	}
+}
+async function loadSchemaMetadata(config, client) {
+	const db = await pgStructure(client);
+	const enums = await require_loadMetadata.loadEnumMetadata(db, client, config);
+	const pgEnums = await require_loadMetadata.loadPgEnumMetadata(db, client, config);
+	const entities = db.tables.filter((t) => require_utils.isEntityTable(config, t)).sortBy("name").map((table) => new require_EntityDbMetadata.EntityDbMetadata(config, table, enums));
+	const totalTables = db.tables.length;
+	const joinTables = db.tables.filter((t) => require_utils.isJoinTable(config, t)).map((t) => t.name);
+	const otherTables = db.tables.filter((t) => !require_utils.isEntityTable(config, t) && !require_utils.isEnumTable(config, t) && !require_utils.isJoinTable(config, t)).map((t) => t.name);
+	return {
+		entities,
+		entitiesByName: Object.fromEntries(entities.map((e) => [e.name, e])),
+		enums,
+		pgEnums,
+		totalTables,
+		joinTables,
+		otherTables
+	};
+}
+function maybeSetDatabaseUrl(config) {
+	if (!node_process.default.env.DATABASE_URL && config.databaseUrl) node_process.default.env.DATABASE_URL = config.databaseUrl;
+}
+function maybeSetExitCode() {
+	if (!node_process.default.argv.includes("--always-exit-code-zero") && (node_process.default.argv.includes("--strict") && require_logger.loggerMaxWarningLevelHit >= require_logger.LOG_LEVELS.warn || require_logger.loggerMaxWarningLevelHit >= require_logger.LOG_LEVELS.error)) node_process.default.exitCode = 1;
+}
+/** Validates that tags can be parsed unambiguously with the configured delimiter. */
+function validateTagDelimiter(config, entities) {
+	const tagDelimiter = config.tagDelimiter ?? ":";
+	for (const entity of entities) if (tagDelimiter !== "" && `${entity.tagName}${tagDelimiter}`.indexOf(tagDelimiter) !== entity.tagName.length) throw new Error(`Tagged id delimiter '${tagDelimiter}' cannot occur in or overlap tag '${entity.tagName}' for ${entity.name}`);
+	else if (tagDelimiter === "" && !/^[a-z]+$/i.test(entity.tagName)) throw new Error(`Delimiterless ids require an alphabetic tag, got '${entity.tagName}' for ${entity.name}`);
+	else if (tagDelimiter === "" && entity.primaryKey.columnType !== "int" && entity.primaryKey.columnType !== "bigint") throw new Error(`Delimiterless ids require an int or bigint primary key, got '${entity.primaryKey.columnType}' for ${entity.name}`);
+}
+if (typeof module !== "undefined" && require.main === module) joistCodegen().then(() => maybeSetExitCode()).catch((err) => {
+	console.error(err);
+	node_process.default.exit(1);
+});
+//#endregion
+exports.EntityDbMetadata = require_EntityDbMetadata.EntityDbMetadata;
+exports.dateCode = require_utils.dateCode;
+exports.generateAndSaveFiles = generateAndSaveFiles;
+exports.joistCodegen = joistCodegen;
+exports.makeEntity = require_EntityDbMetadata.makeEntity;
+exports.mapSimpleDbTypeToTypescriptType = require_utils.mapSimpleDbTypeToTypescriptType;
+exports.maybeSetExitCode = maybeSetExitCode;
+exports.plainDateCode = require_utils.plainDateCode;
+exports.plainDateTimeCode = require_utils.plainDateTimeCode;
+exports.zonedDateTimeCode = require_utils.zonedDateTimeCode;
+
+//# sourceMappingURL=index.cjs.map

package/build/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["pgStructureModule","loadConfig","process","newPgConnectionConfig","Client","scanEntityFiles","maybeSetForeignKeyOrdering","maybeRunTransforms","installSkills","writeConfig","generateFiles","saveFiles","createFlushFunction","loadEnumMetadata","loadPgEnumMetadata","isEntityTable","EntityDbMetadata","isJoinTable","isEnumTable","loggerMaxWarningLevelHit","LOG_LEVELS"],"sources":["../src/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport process from \"node:process\";\n\nimport { type ConnectionConfig, newPgConnectionConfig } from \"joist-utils\";\nimport { Client } from \"pg\";\nimport pgStructureModule from \"pg-structure\";\nimport { saveFiles } from \"ts-poet\";\n\nimport { assignTags } from \"./assignTags.ts\";\nimport { maybeRunTransforms } from \"./codemods/index.ts\";\nimport { type Config, loadConfig, stripStiPlaceholders, warnInvalidConfigEntries, writeConfig } from \"./config.ts\";\nimport {\n  type DbMetadata,\n  EntityDbMetadata,\n  failIfOverlappingFieldNames,\n  resolveNameConflicts,\n} from \"./EntityDbMetadata.ts\";\nimport { maybeSetForeignKeyOrdering } from \"./foreignKeyOrdering.ts\";\nimport { generateFiles } from \"./generate.ts\";\nimport { createFlushFunction } from \"./generateFlushFunction.ts\";\nimport { applyInheritanceUpdates } from \"./inheritance.ts\";\nimport { installSkills } from \"./installSkills.ts\";\nimport { loadEnumMetadata, loadPgEnumMetadata } from \"./loadMetadata.ts\";\nimport { LOG_LEVELS, loggerMaxWarningLevelHit } from \"./logger.ts\";\nimport { scanEntityFiles } from \"./scanEntityFiles.ts\";\nimport { isEntityTable, isEnumTable, isJoinTable, mapSimpleDbTypeToTypescriptType } from \"./utils.ts\";\n\nconst { default: pgStructure } = pgStructureModule;\n\nexport {\n  type DbMetadata,\n  type EnumField,\n  makeEntity,\n  type ManyToManyField,\n  type ManyToOneField,\n  type OneToManyField,\n  type OneToOneField,\n  type PolymorphicField,\n  type PrimitiveField,\n  type PrimitiveTypescriptType,\n} from \"./EntityDbMetadata.ts\";\nexport type { EnumMetadata, EnumRow, EnumTableData, PgEnumData, PgEnumMetadata } from \"./loadMetadata.ts\";\nexport { dateCode, plainDateCode, plainDateTimeCode, zonedDateTimeCode } from \"./utils.ts\";\nexport { type Config, EntityDbMetadata, mapSimpleDbTypeToTypescriptType };\n\nexport async function joistCodegen() {\n  const config = await loadConfig();\n\n  maybeSetDatabaseUrl(config);\n  if (!process.env.DATABASE_URL && !process.env.DB_USER) {\n    console.log(`Database connection information not found, please set either:`);\n    console.log(`  - the DATABASE_URL env variable (i.e. using .env and dotenv), or`);\n    console.log(`  - the databaseUrl key in joist-config.json`);\n    return;\n  }\n  const pgConfig = newPgConnectionConfig();\n\n  const client = new Client(pgConfig);\n  await client.connect();\n\n  const dbMetadata = await loadSchemaMetadata(config, client);\n  const { entities, enums, totalTables } = dbMetadata;\n  console.log(\n    `Found ${totalTables} total tables, ${entities.length} entity tables, ${Object.entries(enums).length} enum tables`,\n  );\n  console.log(\"\");\n\n  // Look for STI tables to synthesize separate metas\n  applyInheritanceUpdates(config, dbMetadata);\n\n  // Now that all entities (incl. STI subtypes) are known, fix up any colliding codegen'd type names\n  resolveNameConflicts(config, dbMetadata);\n\n  // Assign any new tags and write them back to the config file\n  assignTags(config, dbMetadata);\n  validateTagDelimiter(config, entities);\n\n  // Scan `*.ts` files after we've expanded `Task` -> `TaskOld.ts`\n  await scanEntityFiles(config, dbMetadata);\n\n  // If we're not using deferred FKs, determine our DAG insert order\n  await maybeSetForeignKeyOrdering(config, dbMetadata.entities);\n\n  // Generate the flush function for tests\n  await maybeGenerateFlushFunctions(config, client, pgConfig, dbMetadata);\n\n  await client.end();\n\n  // Apply any codemods to the user's codebase, if we have them\n  await maybeRunTransforms(config);\n\n  // Do some warnings\n  for (const entity of entities) failIfOverlappingFieldNames(entity);\n  warnInvalidConfigEntries(config, dbMetadata);\n\n  // Finally actually generate the files (even if we found a fatal error)\n  await generateAndSaveFiles(config, dbMetadata);\n\n  // Install our bundled Agent Skills so coding agents can find them, unless explicitly disabled\n  if (config.skills !== false) await installSkills();\n\n  stripStiPlaceholders(config, entities);\n  await writeConfig(config);\n}\n\n/** Uses entities and enums from the `db` schema and saves them into our entities directory. */\nexport async function generateAndSaveFiles(config: Config, dbMeta: DbMetadata): Promise<void> {\n  const files = await generateFiles(config, dbMeta);\n  const esmExt = config.esm ? (config.allowImportingTsExtensions ? \"ts\" : \"js\") : \"\";\n  await saveFiles({\n    toolName: \"joist-codegen\",\n    directory: config.entitiesDirectory,\n    files,\n    toStringOpts: { importExtensions: esmExt || false },\n  });\n}\n\nasync function maybeGenerateFlushFunctions(config: Config, client: Client, pgConfig: ConnectionConfig, db: DbMetadata) {\n  // In graphql-service we have our own custom flush function, so allow skipping this\n  if (config.createFlushFunction !== false) {\n    // Look for multiple test databases\n    if (Array.isArray(config.createFlushFunction)) {\n      console.log(\"Creating flush_database functions\");\n      await Promise.all(\n        config.createFlushFunction.map(async (dbName) => {\n          const client = new Client({ ...pgConfig, database: dbName });\n          await client.connect();\n          await createFlushFunction(client, db);\n          await client.end();\n        }),\n      );\n    } else {\n      console.log(\"Creating flush_database function\");\n      await createFlushFunction(client, db);\n    }\n  }\n}\n\nasync function loadSchemaMetadata(config: Config, client: Client): Promise<DbMetadata> {\n  // Here we load all schemas, to avoid pg-structure failing on cross-schema foreign keys\n  // like our cyanaudit triggers (https://github.com/ozum/pg-structure/issues/85), and then\n  // later filter them non-public schema tables out.\n  const db = await pgStructure(client);\n  const enums = await loadEnumMetadata(db, client, config);\n  const pgEnums = await loadPgEnumMetadata(db, client, config);\n  const entities = db.tables\n    .filter((t) => isEntityTable(config, t))\n    .sortBy(\"name\")\n    .map((table) => new EntityDbMetadata(config, table, enums));\n  const totalTables = db.tables.length;\n  const joinTables = db.tables.filter((t) => isJoinTable(config, t)).map((t) => t.name);\n  const otherTables = db.tables\n    .filter((t) => !isEntityTable(config, t) && !isEnumTable(config, t) && !isJoinTable(config, t))\n    .map((t) => t.name);\n  const entitiesByName = Object.fromEntries(entities.map((e) => [e.name, e]));\n  return { entities, entitiesByName, enums, pgEnums, totalTables, joinTables, otherTables };\n}\n\nfunction maybeSetDatabaseUrl(config: Config): void {\n  if (!process.env.DATABASE_URL && config.databaseUrl) {\n    process.env.DATABASE_URL = config.databaseUrl;\n  }\n}\n\nexport function maybeSetExitCode(): void {\n  if (\n    !process.argv.includes(\"--always-exit-code-zero\") &&\n    // strict mode + warnings or greater\n    ((process.argv.includes(\"--strict\") && loggerMaxWarningLevelHit >= LOG_LEVELS.warn) ||\n      // otherwise errors or greater\n      loggerMaxWarningLevelHit >= LOG_LEVELS.error)\n  ) {\n    process.exitCode = 1;\n  }\n}\n\n/** Validates that tags can be parsed unambiguously with the configured delimiter. */\nfunction validateTagDelimiter(config: Config, entities: EntityDbMetadata[]): void {\n  const tagDelimiter = config.tagDelimiter ?? \":\";\n  for (const entity of entities) {\n    if (tagDelimiter !== \"\" && `${entity.tagName}${tagDelimiter}`.indexOf(tagDelimiter) !== entity.tagName.length) {\n      throw new Error(\n        `Tagged id delimiter '${tagDelimiter}' cannot occur in or overlap tag '${entity.tagName}' for ${entity.name}`,\n      );\n    } else if (tagDelimiter === \"\" && !/^[a-z]+$/i.test(entity.tagName)) {\n      throw new Error(`Delimiterless ids require an alphabetic tag, got '${entity.tagName}' for ${entity.name}`);\n    } else if (\n      tagDelimiter === \"\" &&\n      entity.primaryKey.columnType !== \"int\" &&\n      entity.primaryKey.columnType !== \"bigint\"\n    ) {\n      throw new Error(\n        `Delimiterless ids require an int or bigint primary key, got '${entity.primaryKey.columnType}' for ${entity.name}`,\n      );\n    }\n  }\n}\n\nif (typeof module !== \"undefined\" && require.main === module) {\n  joistCodegen()\n    .then(() => maybeSetExitCode())\n    .catch((err) => {\n      console.error(err);\n      process.exit(1);\n    });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AA4BA,MAAM,EAAE,SAAS,gBAAgBA,aAAAA;AAkBjC,eAAsB,eAAe;CACnC,MAAM,SAAS,MAAMC,eAAAA,WAAW;CAEhC,oBAAoB,MAAM;CAC1B,IAAI,CAACC,aAAAA,QAAQ,IAAI,gBAAgB,CAACA,aAAAA,QAAQ,IAAI,SAAS;EACrD,QAAQ,IAAI,+DAA+D;EAC3E,QAAQ,IAAI,oEAAoE;EAChF,QAAQ,IAAI,8CAA8C;EAC1D;CACF;CACA,MAAM,YAAA,GAAWC,YAAAA,sBAAAA,CAAsB;CAEvC,MAAM,SAAS,IAAIC,GAAAA,OAAO,QAAQ;CAClC,MAAM,OAAO,QAAQ;CAErB,MAAM,aAAa,MAAM,mBAAmB,QAAQ,MAAM;CAC1D,MAAM,EAAE,UAAU,OAAO,gBAAgB;CACzC,QAAQ,IACN,SAAS,YAAY,iBAAiB,SAAS,OAAO,kBAAkB,OAAO,QAAQ,KAAK,CAAC,CAAC,OAAO,aACvG;CACA,QAAQ,IAAI,EAAE;CAGd,oBAAA,wBAAwB,QAAQ,UAAU;CAG1C,yBAAA,qBAAqB,QAAQ,UAAU;CAGvC,mBAAA,WAAW,QAAQ,UAAU;CAC7B,qBAAqB,QAAQ,QAAQ;CAGrC,MAAMC,wBAAAA,gBAAgB,QAAQ,UAAU;CAGxC,MAAMC,2BAAAA,2BAA2B,QAAQ,WAAW,QAAQ;CAG5D,MAAM,4BAA4B,QAAQ,QAAQ,UAAU,UAAU;CAEtE,MAAM,OAAO,IAAI;CAGjB,MAAMC,uBAAAA,mBAAmB,MAAM;CAG/B,KAAK,MAAM,UAAU,UAAU,yBAAA,4BAA4B,MAAM;CACjE,eAAA,yBAAyB,QAAQ,UAAU;CAG3C,MAAM,qBAAqB,QAAQ,UAAU;CAG7C,IAAI,OAAO,WAAW,OAAO,MAAMC,sBAAAA,cAAc;CAEjD,eAAA,qBAAqB,QAAQ,QAAQ;CACrC,MAAMC,eAAAA,YAAY,MAAM;AAC1B;;AAGA,eAAsB,qBAAqB,QAAgB,QAAmC;CAC5F,MAAM,QAAQ,MAAMC,iBAAAA,cAAc,QAAQ,MAAM;CAChD,MAAM,SAAS,OAAO,MAAO,OAAO,6BAA6B,OAAO,OAAQ;CAChF,OAAA,GAAMC,QAAAA,UAAAA,CAAU;EACd,UAAU;EACV,WAAW,OAAO;EAClB;EACA,cAAc,EAAE,kBAAkB,UAAU,MAAM;CACpD,CAAC;AACH;AAEA,eAAe,4BAA4B,QAAgB,QAAgB,UAA4B,IAAgB;CAErH,IAAI,OAAO,wBAAwB,OAAO;EAExC,IAAI,MAAM,QAAQ,OAAO,mBAAmB,GAAG;GAC7C,QAAQ,IAAI,mCAAmC;GAC/C,MAAM,QAAQ,IACZ,OAAO,oBAAoB,IAAI,OAAO,WAAW;IAC/C,MAAM,SAAS,IAAIP,GAAAA,OAAO;KAAE,GAAG;KAAU,UAAU;IAAO,CAAC;IAC3D,MAAM,OAAO,QAAQ;IACrB,MAAMQ,8BAAAA,oBAAoB,QAAQ,EAAE;IACpC,MAAM,OAAO,IAAI;GACnB,CAAC,CACH;EACF,OAAO;GACL,QAAQ,IAAI,kCAAkC;GAC9C,MAAMA,8BAAAA,oBAAoB,QAAQ,EAAE;EACtC;CACF;AACF;AAEA,eAAe,mBAAmB,QAAgB,QAAqC;CAIrF,MAAM,KAAK,MAAM,YAAY,MAAM;CACnC,MAAM,QAAQ,MAAMC,qBAAAA,iBAAiB,IAAI,QAAQ,MAAM;CACvD,MAAM,UAAU,MAAMC,qBAAAA,mBAAmB,IAAI,QAAQ,MAAM;CAC3D,MAAM,WAAW,GAAG,OACjB,QAAQ,MAAMC,cAAAA,cAAc,QAAQ,CAAC,CAAC,CAAC,CACvC,OAAO,MAAM,CAAC,CACd,KAAK,UAAU,IAAIC,yBAAAA,iBAAiB,QAAQ,OAAO,KAAK,CAAC;CAC5D,MAAM,cAAc,GAAG,OAAO;CAC9B,MAAM,aAAa,GAAG,OAAO,QAAQ,MAAMC,cAAAA,YAAY,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,IAAI;CACpF,MAAM,cAAc,GAAG,OACpB,QAAQ,MAAM,CAACF,cAAAA,cAAc,QAAQ,CAAC,KAAK,CAACG,cAAAA,YAAY,QAAQ,CAAC,KAAK,CAACD,cAAAA,YAAY,QAAQ,CAAC,CAAC,CAAC,CAC9F,KAAK,MAAM,EAAE,IAAI;CAEpB,OAAO;EAAE;EAAU,gBADI,OAAO,YAAY,SAAS,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC,CACzC;EAAG;EAAO;EAAS;EAAa;EAAY;CAAY;AAC1F;AAEA,SAAS,oBAAoB,QAAsB;CACjD,IAAI,CAACf,aAAAA,QAAQ,IAAI,gBAAgB,OAAO,aACtC,aAAA,QAAQ,IAAI,eAAe,OAAO;AAEtC;AAEA,SAAgB,mBAAyB;CACvC,IACE,CAACA,aAAAA,QAAQ,KAAK,SAAS,yBAAyB,MAE9CA,aAAAA,QAAQ,KAAK,SAAS,UAAU,KAAKiB,eAAAA,4BAA4BC,eAAAA,WAAW,QAE5ED,eAAAA,4BAA4BC,eAAAA,WAAW,QAEzC,aAAA,QAAA,WAAmB;AAEvB;;AAGA,SAAS,qBAAqB,QAAgB,UAAoC;CAChF,MAAM,eAAe,OAAO,gBAAgB;CAC5C,KAAK,MAAM,UAAU,UACnB,IAAI,iBAAiB,MAAM,GAAG,OAAO,UAAU,eAAe,QAAQ,YAAY,MAAM,OAAO,QAAQ,QACrG,MAAM,IAAI,MACR,wBAAwB,aAAa,oCAAoC,OAAO,QAAQ,QAAQ,OAAO,MACzG;MACK,IAAI,iBAAiB,MAAM,CAAC,YAAY,KAAK,OAAO,OAAO,GAChE,MAAM,IAAI,MAAM,qDAAqD,OAAO,QAAQ,QAAQ,OAAO,MAAM;MACpG,IACL,iBAAiB,MACjB,OAAO,WAAW,eAAe,SACjC,OAAO,WAAW,eAAe,UAEjC,MAAM,IAAI,MACR,gEAAgE,OAAO,WAAW,WAAW,QAAQ,OAAO,MAC9G;AAGN;AAEA,IAAI,OAAO,WAAW,eAAe,QAAQ,SAAS,QACpD,aAAa,CAAC,CACX,WAAW,iBAAiB,CAAC,CAAC,CAC9B,OAAO,QAAQ;CACd,QAAQ,MAAM,GAAG;CACjB,aAAA,QAAQ,KAAK,CAAC;AAChB,CAAC"}

package/build/index.d.cts

@@ -0,0 +1,12 @@
+import { Config } from "./config.cjs";
+import { dateCode, mapSimpleDbTypeToTypescriptType, plainDateCode, plainDateTimeCode, zonedDateTimeCode } from "./utils.cjs";
+import { EnumMetadata, EnumRow, EnumTableData, PgEnumData, PgEnumMetadata } from "./loadMetadata.cjs";
+import { DbMetadata, EntityDbMetadata, EnumField, ManyToManyField, ManyToOneField, OneToManyField, OneToOneField, PolymorphicField, PrimitiveField, PrimitiveTypescriptType, makeEntity } from "./EntityDbMetadata.cjs";
+//#region src/index.d.ts
+declare function joistCodegen(): Promise<void>;
+/** Uses entities and enums from the `db` schema and saves them into our entities directory. */
+declare function generateAndSaveFiles(config: Config, dbMeta: DbMetadata): Promise<void>;
+declare function maybeSetExitCode(): void;
+//#endregion
+export { type Config, type DbMetadata, EntityDbMetadata, type EnumField, type EnumMetadata, type EnumRow, type EnumTableData, type ManyToManyField, type ManyToOneField, type OneToManyField, type OneToOneField, type PgEnumData, type PgEnumMetadata, type PolymorphicField, type PrimitiveField, type PrimitiveTypescriptType, dateCode, generateAndSaveFiles, joistCodegen, makeEntity, mapSimpleDbTypeToTypescriptType, maybeSetExitCode, plainDateCode, plainDateTimeCode, zonedDateTimeCode };
+//# sourceMappingURL=index.d.cts.map

package/build/index.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.cts","names":[],"sources":["../src/index.ts"],"mappings":";;;;;iBA8CsB,gBAAY;;iBA6DZ,qBAAqB,QAAQ,QAAQ,QAAQ,aAAa;iBA0DhE"}

package/build/index.d.mts

@@ -0,0 +1,12 @@
+import { Config } from "./config.mjs";
+import { dateCode, mapSimpleDbTypeToTypescriptType, plainDateCode, plainDateTimeCode, zonedDateTimeCode } from "./utils.mjs";
+import { EnumMetadata, EnumRow, EnumTableData, PgEnumData, PgEnumMetadata } from "./loadMetadata.mjs";
+import { DbMetadata, EntityDbMetadata, EnumField, ManyToManyField, ManyToOneField, OneToManyField, OneToOneField, PolymorphicField, PrimitiveField, PrimitiveTypescriptType, makeEntity } from "./EntityDbMetadata.mjs";
+//#region src/index.d.ts
+declare function joistCodegen(): Promise<void>;
+/** Uses entities and enums from the `db` schema and saves them into our entities directory. */
+declare function generateAndSaveFiles(config: Config, dbMeta: DbMetadata): Promise<void>;
+declare function maybeSetExitCode(): void;
+//#endregion
+export { type Config, type DbMetadata, EntityDbMetadata, type EnumField, type EnumMetadata, type EnumRow, type EnumTableData, type ManyToManyField, type ManyToOneField, type OneToManyField, type OneToOneField, type PgEnumData, type PgEnumMetadata, type PolymorphicField, type PrimitiveField, type PrimitiveTypescriptType, dateCode, generateAndSaveFiles, joistCodegen, makeEntity, mapSimpleDbTypeToTypescriptType, maybeSetExitCode, plainDateCode, plainDateTimeCode, zonedDateTimeCode };
+//# sourceMappingURL=index.d.mts.map

package/build/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/index.ts"],"mappings":";;;;;iBA8CsB,gBAAY;;iBA6DZ,qBAAqB,QAAQ,QAAQ,QAAQ,aAAa;iBA0DhE"}

package/build/index.js

@@ -1,142 +1,123 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapSimpleDbTypeToTypescriptType = exports.EntityDbMetadata = exports.makeEntity = void 0;
-exports.joistCodegen = joistCodegen;
-exports.generateAndSaveFiles = generateAndSaveFiles;
-exports.maybeSetExitCode = maybeSetExitCode;
-const joist_utils_1 = require("joist-utils");
-const node_process_1 = __importDefault(require("node:process"));
-const pg_1 = require("pg");
-const pg_structure_1 = __importDefault(require("pg-structure"));
-const ts_poet_1 = require("ts-poet");
-const EntityDbMetadata_1 = require("./EntityDbMetadata");
-Object.defineProperty(exports, "EntityDbMetadata", { enumerable: true, get: function () { return EntityDbMetadata_1.EntityDbMetadata; } });
-const assignTags_1 = require("./assignTags");
-const codemods_1 = require("./codemods");
-const config_1 = require("./config");
-const foreignKeyOrdering_1 = require("./foreignKeyOrdering");
-const generate_1 = require("./generate");
-const generateFlushFunction_1 = require("./generateFlushFunction");
-const inheritance_1 = require("./inheritance");
-const loadMetadata_1 = require("./loadMetadata");
-const logger_1 = require("./logger");
-const scanEntityFiles_1 = require("./scanEntityFiles");
-const utils_1 = require("./utils");
-Object.defineProperty(exports, "mapSimpleDbTypeToTypescriptType", { enumerable: true, get: function () { return utils_1.mapSimpleDbTypeToTypescriptType; } });
-var EntityDbMetadata_2 = require("./EntityDbMetadata");
-Object.defineProperty(exports, "makeEntity", { enumerable: true, get: function () { return EntityDbMetadata_2.makeEntity; } });
+#!/usr/bin/env node
+import { __require } from "./_virtual/_rolldown/runtime.js";
+import { maybeRunTransforms } from "./codemods/index.js";
+import { dateCode, isEntityTable, isEnumTable, isJoinTable, mapSimpleDbTypeToTypescriptType, plainDateCode, plainDateTimeCode, zonedDateTimeCode } from "./utils.js";
+import { applyInheritanceUpdates } from "./inheritance.js";
+import { LOG_LEVELS, loggerMaxWarningLevelHit } from "./logger.js";
+import { loadConfig, stripStiPlaceholders, warnInvalidConfigEntries, writeConfig } from "./config.js";
+import { assignTags } from "./assignTags.js";
+import { maybeSetForeignKeyOrdering } from "./foreignKeyOrdering.js";
+import { generateFiles } from "./generate.js";
+import { createFlushFunction } from "./generateFlushFunction.js";
+import { installSkills } from "./installSkills.js";
+import { scanEntityFiles } from "./scanEntityFiles.js";
+import { loadEnumMetadata, loadPgEnumMetadata } from "./loadMetadata.js";
+import { EntityDbMetadata, failIfOverlappingFieldNames, makeEntity, resolveNameConflicts } from "./EntityDbMetadata.js";
+import { newPgConnectionConfig } from "joist-utils";
+import pgStructureModule from "pg-structure";
+import { saveFiles } from "ts-poet";
+import { Client } from "pg";
+import process from "node:process";
+//#region src/index.ts
+const { default: pgStructure } = pgStructureModule;
 async function joistCodegen() {
-    const config = await (0, config_1.loadConfig)();
-    maybeSetDatabaseUrl(config);
-    if (!node_process_1.default.env.DATABASE_URL && !node_process_1.default.env.DB_USER) {
-        console.log(`Database connection information not found, please set either:`);
-        console.log(`  - the DATABASE_URL env variable (i.e. using .env and dotenv), or`);
-        console.log(`  - the databaseUrl key in joist-config.json`);
-        return;
-    }
-    const pgConfig = (0, joist_utils_1.newPgConnectionConfig)();
-    const client = new pg_1.Client(pgConfig);
-    await client.connect();
-    const dbMetadata = await loadSchemaMetadata(config, client);
-    const { entities, enums, totalTables } = dbMetadata;
-    console.log(`Found ${totalTables} total tables, ${entities.length} entity tables, ${Object.entries(enums).length} enum tables`);
-    console.log("");
-    // Look for STI tables to synthesize separate metas
-    (0, inheritance_1.applyInheritanceUpdates)(config, dbMetadata);
-    // Now that all entities (incl. STI subtypes) are known, fix up any colliding codegen'd type names
-    (0, EntityDbMetadata_1.resolveNameConflicts)(config, dbMetadata);
-    // Assign any new tags and write them back to the config file
-    (0, assignTags_1.assignTags)(config, dbMetadata);
-    // Scan `*.ts` files after we've expanded `Task` -> `TaskOld.ts`
-    await (0, scanEntityFiles_1.scanEntityFiles)(config, dbMetadata);
-    // If we're not using deferred FKs, determine our DAG insert order
-    await (0, foreignKeyOrdering_1.maybeSetForeignKeyOrdering)(config, dbMetadata.entities);
-    // Generate the flush function for tests
-    await maybeGenerateFlushFunctions(config, client, pgConfig, dbMetadata);
-    await client.end();
-    // Apply any codemods to the user's codebase, if we have them
-    await (0, codemods_1.maybeRunTransforms)(config);
-    // Do some warnings
-    for (const entity of entities)
-        (0, EntityDbMetadata_1.failIfOverlappingFieldNames)(entity);
-    (0, config_1.warnInvalidConfigEntries)(config, dbMetadata);
-    // Finally actually generate the files (even if we found a fatal error)
-    await generateAndSaveFiles(config, dbMetadata);
-    (0, config_1.stripStiPlaceholders)(config, entities);
-    await (0, config_1.writeConfig)(config);
+	const config = await loadConfig();
+	maybeSetDatabaseUrl(config);
+	if (!process.env.DATABASE_URL && !process.env.DB_USER) {
+		console.log(`Database connection information not found, please set either:`);
+		console.log(`  - the DATABASE_URL env variable (i.e. using .env and dotenv), or`);
+		console.log(`  - the databaseUrl key in joist-config.json`);
+		return;
+	}
+	const pgConfig = newPgConnectionConfig();
+	const client = new Client(pgConfig);
+	await client.connect();
+	const dbMetadata = await loadSchemaMetadata(config, client);
+	const { entities, enums, totalTables } = dbMetadata;
+	console.log(`Found ${totalTables} total tables, ${entities.length} entity tables, ${Object.entries(enums).length} enum tables`);
+	console.log("");
+	applyInheritanceUpdates(config, dbMetadata);
+	resolveNameConflicts(config, dbMetadata);
+	assignTags(config, dbMetadata);
+	validateTagDelimiter(config, entities);
+	await scanEntityFiles(config, dbMetadata);
+	await maybeSetForeignKeyOrdering(config, dbMetadata.entities);
+	await maybeGenerateFlushFunctions(config, client, pgConfig, dbMetadata);
+	await client.end();
+	await maybeRunTransforms(config);
+	for (const entity of entities) failIfOverlappingFieldNames(entity);
+	warnInvalidConfigEntries(config, dbMetadata);
+	await generateAndSaveFiles(config, dbMetadata);
+	if (config.skills !== false) await installSkills();
+	stripStiPlaceholders(config, entities);
+	await writeConfig(config);
 }
 /** Uses entities and enums from the `db` schema and saves them into our entities directory. */
 async function generateAndSaveFiles(config, dbMeta) {
-    const files = await (0, generate_1.generateFiles)(config, dbMeta);
-    const esmExt = config.esm ? (config.allowImportingTsExtensions ? "ts" : "js") : "";
-    await (0, ts_poet_1.saveFiles)({
-        toolName: "joist-codegen",
-        directory: config.entitiesDirectory,
-        files,
-        toStringOpts: { importExtensions: esmExt || false },
-    });
+	const files = await generateFiles(config, dbMeta);
+	const esmExt = config.esm ? config.allowImportingTsExtensions ? "ts" : "js" : "";
+	await saveFiles({
+		toolName: "joist-codegen",
+		directory: config.entitiesDirectory,
+		files,
+		toStringOpts: { importExtensions: esmExt || false }
+	});
 }
 async function maybeGenerateFlushFunctions(config, client, pgConfig, db) {
-    // In graphql-service we have our own custom flush function, so allow skipping this
-    if (config.createFlushFunction !== false) {
-        // Look for multiple test databases
-        if (Array.isArray(config.createFlushFunction)) {
-            console.log("Creating flush_database functions");
-            await Promise.all(config.createFlushFunction.map(async (dbName) => {
-                const client = new pg_1.Client({ ...pgConfig, database: dbName });
-                await client.connect();
-                await (0, generateFlushFunction_1.createFlushFunction)(client, db);
-                await client.end();
-            }));
-        }
-        else {
-            console.log("Creating flush_database function");
-            await (0, generateFlushFunction_1.createFlushFunction)(client, db);
-        }
-    }
+	if (config.createFlushFunction !== false) {
+		if (Array.isArray(config.createFlushFunction)) {
+			console.log("Creating flush_database functions");
+			await Promise.all(config.createFlushFunction.map(async (dbName) => {
+				const client = new Client({
+					...pgConfig,
+					database: dbName
+				});
+				await client.connect();
+				await createFlushFunction(client, db);
+				await client.end();
+			}));
+		} else {
+			console.log("Creating flush_database function");
+			await createFlushFunction(client, db);
+		}
+	}
 }
 async function loadSchemaMetadata(config, client) {
-    // Here we load all schemas, to avoid pg-structure failing on cross-schema foreign keys
-    // like our cyanaudit triggers (https://github.com/ozum/pg-structure/issues/85), and then
-    // later filter them non-public schema tables out.
-    const db = await (0, pg_structure_1.default)(client);
-    const enums = await (0, loadMetadata_1.loadEnumMetadata)(db, client, config);
-    const pgEnums = await (0, loadMetadata_1.loadPgEnumMetadata)(db, client, config);
-    const entities = db.tables
-        .filter((t) => (0, utils_1.isEntityTable)(config, t))
-        .sortBy("name")
-        .map((table) => new EntityDbMetadata_1.EntityDbMetadata(config, table, enums));
-    const totalTables = db.tables.length;
-    const joinTables = db.tables.filter((t) => (0, utils_1.isJoinTable)(config, t)).map((t) => t.name);
-    const otherTables = db.tables
-        .filter((t) => !(0, utils_1.isEntityTable)(config, t) && !(0, utils_1.isEnumTable)(config, t) && !(0, utils_1.isJoinTable)(config, t))
-        .map((t) => t.name);
-    const entitiesByName = Object.fromEntries(entities.map((e) => [e.name, e]));
-    return { entities, entitiesByName, enums, pgEnums, totalTables, joinTables, otherTables };
+	const db = await pgStructure(client);
+	const enums = await loadEnumMetadata(db, client, config);
+	const pgEnums = await loadPgEnumMetadata(db, client, config);
+	const entities = db.tables.filter((t) => isEntityTable(config, t)).sortBy("name").map((table) => new EntityDbMetadata(config, table, enums));
+	const totalTables = db.tables.length;
+	const joinTables = db.tables.filter((t) => isJoinTable(config, t)).map((t) => t.name);
+	const otherTables = db.tables.filter((t) => !isEntityTable(config, t) && !isEnumTable(config, t) && !isJoinTable(config, t)).map((t) => t.name);
+	return {
+		entities,
+		entitiesByName: Object.fromEntries(entities.map((e) => [e.name, e])),
+		enums,
+		pgEnums,
+		totalTables,
+		joinTables,
+		otherTables
+	};
 }
 function maybeSetDatabaseUrl(config) {
-    if (!node_process_1.default.env.DATABASE_URL && config.databaseUrl) {
-        node_process_1.default.env.DATABASE_URL = config.databaseUrl;
-    }
+	if (!process.env.DATABASE_URL && config.databaseUrl) process.env.DATABASE_URL = config.databaseUrl;
 }
 function maybeSetExitCode() {
-    if (!node_process_1.default.argv.includes("--always-exit-code-zero") &&
-        // strict mode + warnings or greater
-        ((node_process_1.default.argv.includes("--strict") && logger_1.loggerMaxWarningLevelHit >= logger_1.LOG_LEVELS.warn) ||
-            // otherwise errors or greater
-            logger_1.loggerMaxWarningLevelHit >= logger_1.LOG_LEVELS.error)) {
-        node_process_1.default.exitCode = 1;
-    }
+	if (!process.argv.includes("--always-exit-code-zero") && (process.argv.includes("--strict") && loggerMaxWarningLevelHit >= LOG_LEVELS.warn || loggerMaxWarningLevelHit >= LOG_LEVELS.error)) process.exitCode = 1;
 }
-if (require.main === module) {
-    joistCodegen()
-        .then(() => maybeSetExitCode())
-        .catch((err) => {
-        console.error(err);
-        node_process_1.default.exit(1);
-    });
+/** Validates that tags can be parsed unambiguously with the configured delimiter. */
+function validateTagDelimiter(config, entities) {
+	const tagDelimiter = config.tagDelimiter ?? ":";
+	for (const entity of entities) if (tagDelimiter !== "" && `${entity.tagName}${tagDelimiter}`.indexOf(tagDelimiter) !== entity.tagName.length) throw new Error(`Tagged id delimiter '${tagDelimiter}' cannot occur in or overlap tag '${entity.tagName}' for ${entity.name}`);
+	else if (tagDelimiter === "" && !/^[a-z]+$/i.test(entity.tagName)) throw new Error(`Delimiterless ids require an alphabetic tag, got '${entity.tagName}' for ${entity.name}`);
+	else if (tagDelimiter === "" && entity.primaryKey.columnType !== "int" && entity.primaryKey.columnType !== "bigint") throw new Error(`Delimiterless ids require an int or bigint primary key, got '${entity.primaryKey.columnType}' for ${entity.name}`);
 }
+if (typeof module !== "undefined" && __require.main === module) joistCodegen().then(() => maybeSetExitCode()).catch((err) => {
+	console.error(err);
+	process.exit(1);
+});
+//#endregion
+export { EntityDbMetadata, dateCode, generateAndSaveFiles, joistCodegen, makeEntity, mapSimpleDbTypeToTypescriptType, maybeSetExitCode, plainDateCode, plainDateTimeCode, zonedDateTimeCode };
+
 //# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;AAiCA,oCAsDC;AAGD,oDASC;AAiDD,4CAUC;AA9JD,6CAAsE;AACtE,gEAAmC;AACnC,2BAA4B;AAC5B,gEAAuC;AACvC,qCAAoC;AACpC,yDAAqH;AA0BpG,iGA1BI,mCAAgB,OA0BJ;AAzBjC,6CAA0C;AAC1C,yCAAgD;AAChD,qCAA2G;AAC3G,6DAAkE;AAClE,yCAA2C;AAC3C,mEAA8D;AAC9D,+CAAwD;AACxD,iDAAsE;AACtE,qCAAgE;AAChE,uDAAoD;AACpD,mCAAmG;AAehE,gHAfe,uCAA+B,OAef;AAblE,uDAW4B;AAR1B,8GAAA,UAAU,OAAA;AAYL,KAAK,UAAU,YAAY;IAChC,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAU,GAAE,CAAC;IAElC,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC5B,IAAI,CAAC,sBAAO,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,sBAAO,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACtD,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;QAC7E,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;QAClF,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;QAC5D,OAAO;IACT,CAAC;IACD,MAAM,QAAQ,GAAG,IAAA,mCAAqB,GAAE,CAAC;IAEzC,MAAM,MAAM,GAAG,IAAI,WAAM,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;IAEvB,MAAM,UAAU,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5D,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,UAAU,CAAC;IACpD,OAAO,CAAC,GAAG,CACT,SAAS,WAAW,kBAAkB,QAAQ,CAAC,MAAM,mBAAmB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,cAAc,CACnH,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,mDAAmD;IACnD,IAAA,qCAAuB,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE5C,kGAAkG;IAClG,IAAA,uCAAoB,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAEzC,6DAA6D;IAC7D,IAAA,uBAAU,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE/B,gEAAgE;IAChE,MAAM,IAAA,iCAAe,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE1C,kEAAkE;IAClE,MAAM,IAAA,+CAA0B,EAAC,MAAM,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC;IAE9D,wCAAwC;IACxC,MAAM,2BAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;IAExE,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;IAEnB,6DAA6D;IAC7D,MAAM,IAAA,6BAAkB,EAAC,MAAM,CAAC,CAAC;IAEjC,mBAAmB;IACnB,KAAK,MAAM,MAAM,IAAI,QAAQ;QAAE,IAAA,8CAA2B,EAAC,MAAM,CAAC,CAAC;IACnE,IAAA,iCAAwB,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE7C,uEAAuE;IACvE,MAAM,oBAAoB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAE/C,IAAA,6BAAoB,EAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACvC,MAAM,IAAA,oBAAW,EAAC,MAAM,CAAC,CAAC;AAC5B,CAAC;AAED,+FAA+F;AACxF,KAAK,UAAU,oBAAoB,CAAC,MAAc,EAAE,MAAkB;IAC3E,MAAM,KAAK,GAAG,MAAM,IAAA,wBAAa,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClD,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACnF,MAAM,IAAA,mBAAS,EAAC;QACd,QAAQ,EAAE,eAAe;QACzB,SAAS,EAAE,MAAM,CAAC,iBAAiB;QACnC,KAAK;QACL,YAAY,EAAE,EAAE,gBAAgB,EAAE,MAAM,IAAI,KAAK,EAAE;KACpD,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,2BAA2B,CAAC,MAAc,EAAE,MAAc,EAAE,QAA0B,EAAE,EAAc;IACnH,mFAAmF;IACnF,IAAI,MAAM,CAAC,mBAAmB,KAAK,KAAK,EAAE,CAAC;QACzC,mCAAmC;QACnC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;YACjD,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;gBAC9C,MAAM,MAAM,GAAG,IAAI,WAAM,CAAC,EAAE,GAAG,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC7D,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvB,MAAM,IAAA,2CAAmB,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACtC,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;YAChD,MAAM,IAAA,2CAAmB,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,MAAc,EAAE,MAAc;IAC9D,uFAAuF;IACvF,yFAAyF;IACzF,kDAAkD;IAClD,MAAM,EAAE,GAAG,MAAM,IAAA,sBAAW,EAAC,MAAM,CAAC,CAAC;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,+BAAgB,EAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,MAAM,IAAA,iCAAkB,EAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM;SACvB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,qBAAa,EAAC,MAAM,EAAE,CAAC,CAAC,CAAC;SACvC,MAAM,CAAC,MAAM,CAAC;SACd,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,mCAAgB,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9D,MAAM,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC;IACrC,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,mBAAW,EAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtF,MAAM,WAAW,GAAG,EAAE,CAAC,MAAM;SAC1B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAA,qBAAa,EAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,IAAA,mBAAW,EAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,IAAA,mBAAW,EAAC,MAAM,EAAE,CAAC,CAAC,CAAC;SAC9F,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtB,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5E,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC;AAC5F,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAc;IACzC,IAAI,CAAC,sBAAO,CAAC,GAAG,CAAC,YAAY,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACpD,sBAAO,CAAC,GAAG,CAAC,YAAY,GAAG,MAAM,CAAC,WAAW,CAAC;IAChD,CAAC;AACH,CAAC;AAED,SAAgB,gBAAgB;IAC9B,IACE,CAAC,sBAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC;QACjD,oCAAoC;QACpC,CAAC,CAAC,sBAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,iCAAwB,IAAI,mBAAU,CAAC,IAAI,CAAC;YACjF,8BAA8B;YAC9B,iCAAwB,IAAI,mBAAU,CAAC,KAAK,CAAC,EAC/C,CAAC;QACD,sBAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;IAC5B,YAAY,EAAE;SACX,IAAI,CAAC,GAAG,EAAE,CAAC,gBAAgB,EAAE,CAAC;SAC9B,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QACb,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACnB,sBAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACP,CAAC"}
+{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport process from \"node:process\";\n\nimport { type ConnectionConfig, newPgConnectionConfig } from \"joist-utils\";\nimport { Client } from \"pg\";\nimport pgStructureModule from \"pg-structure\";\nimport { saveFiles } from \"ts-poet\";\n\nimport { assignTags } from \"./assignTags.ts\";\nimport { maybeRunTransforms } from \"./codemods/index.ts\";\nimport { type Config, loadConfig, stripStiPlaceholders, warnInvalidConfigEntries, writeConfig } from \"./config.ts\";\nimport {\n  type DbMetadata,\n  EntityDbMetadata,\n  failIfOverlappingFieldNames,\n  resolveNameConflicts,\n} from \"./EntityDbMetadata.ts\";\nimport { maybeSetForeignKeyOrdering } from \"./foreignKeyOrdering.ts\";\nimport { generateFiles } from \"./generate.ts\";\nimport { createFlushFunction } from \"./generateFlushFunction.ts\";\nimport { applyInheritanceUpdates } from \"./inheritance.ts\";\nimport { installSkills } from \"./installSkills.ts\";\nimport { loadEnumMetadata, loadPgEnumMetadata } from \"./loadMetadata.ts\";\nimport { LOG_LEVELS, loggerMaxWarningLevelHit } from \"./logger.ts\";\nimport { scanEntityFiles } from \"./scanEntityFiles.ts\";\nimport { isEntityTable, isEnumTable, isJoinTable, mapSimpleDbTypeToTypescriptType } from \"./utils.ts\";\n\nconst { default: pgStructure } = pgStructureModule;\n\nexport {\n  type DbMetadata,\n  type EnumField,\n  makeEntity,\n  type ManyToManyField,\n  type ManyToOneField,\n  type OneToManyField,\n  type OneToOneField,\n  type PolymorphicField,\n  type PrimitiveField,\n  type PrimitiveTypescriptType,\n} from \"./EntityDbMetadata.ts\";\nexport type { EnumMetadata, EnumRow, EnumTableData, PgEnumData, PgEnumMetadata } from \"./loadMetadata.ts\";\nexport { dateCode, plainDateCode, plainDateTimeCode, zonedDateTimeCode } from \"./utils.ts\";\nexport { type Config, EntityDbMetadata, mapSimpleDbTypeToTypescriptType };\n\nexport async function joistCodegen() {\n  const config = await loadConfig();\n\n  maybeSetDatabaseUrl(config);\n  if (!process.env.DATABASE_URL && !process.env.DB_USER) {\n    console.log(`Database connection information not found, please set either:`);\n    console.log(`  - the DATABASE_URL env variable (i.e. using .env and dotenv), or`);\n    console.log(`  - the databaseUrl key in joist-config.json`);\n    return;\n  }\n  const pgConfig = newPgConnectionConfig();\n\n  const client = new Client(pgConfig);\n  await client.connect();\n\n  const dbMetadata = await loadSchemaMetadata(config, client);\n  const { entities, enums, totalTables } = dbMetadata;\n  console.log(\n    `Found ${totalTables} total tables, ${entities.length} entity tables, ${Object.entries(enums).length} enum tables`,\n  );\n  console.log(\"\");\n\n  // Look for STI tables to synthesize separate metas\n  applyInheritanceUpdates(config, dbMetadata);\n\n  // Now that all entities (incl. STI subtypes) are known, fix up any colliding codegen'd type names\n  resolveNameConflicts(config, dbMetadata);\n\n  // Assign any new tags and write them back to the config file\n  assignTags(config, dbMetadata);\n  validateTagDelimiter(config, entities);\n\n  // Scan `*.ts` files after we've expanded `Task` -> `TaskOld.ts`\n  await scanEntityFiles(config, dbMetadata);\n\n  // If we're not using deferred FKs, determine our DAG insert order\n  await maybeSetForeignKeyOrdering(config, dbMetadata.entities);\n\n  // Generate the flush function for tests\n  await maybeGenerateFlushFunctions(config, client, pgConfig, dbMetadata);\n\n  await client.end();\n\n  // Apply any codemods to the user's codebase, if we have them\n  await maybeRunTransforms(config);\n\n  // Do some warnings\n  for (const entity of entities) failIfOverlappingFieldNames(entity);\n  warnInvalidConfigEntries(config, dbMetadata);\n\n  // Finally actually generate the files (even if we found a fatal error)\n  await generateAndSaveFiles(config, dbMetadata);\n\n  // Install our bundled Agent Skills so coding agents can find them, unless explicitly disabled\n  if (config.skills !== false) await installSkills();\n\n  stripStiPlaceholders(config, entities);\n  await writeConfig(config);\n}\n\n/** Uses entities and enums from the `db` schema and saves them into our entities directory. */\nexport async function generateAndSaveFiles(config: Config, dbMeta: DbMetadata): Promise<void> {\n  const files = await generateFiles(config, dbMeta);\n  const esmExt = config.esm ? (config.allowImportingTsExtensions ? \"ts\" : \"js\") : \"\";\n  await saveFiles({\n    toolName: \"joist-codegen\",\n    directory: config.entitiesDirectory,\n    files,\n    toStringOpts: { importExtensions: esmExt || false },\n  });\n}\n\nasync function maybeGenerateFlushFunctions(config: Config, client: Client, pgConfig: ConnectionConfig, db: DbMetadata) {\n  // In graphql-service we have our own custom flush function, so allow skipping this\n  if (config.createFlushFunction !== false) {\n    // Look for multiple test databases\n    if (Array.isArray(config.createFlushFunction)) {\n      console.log(\"Creating flush_database functions\");\n      await Promise.all(\n        config.createFlushFunction.map(async (dbName) => {\n          const client = new Client({ ...pgConfig, database: dbName });\n          await client.connect();\n          await createFlushFunction(client, db);\n          await client.end();\n        }),\n      );\n    } else {\n      console.log(\"Creating flush_database function\");\n      await createFlushFunction(client, db);\n    }\n  }\n}\n\nasync function loadSchemaMetadata(config: Config, client: Client): Promise<DbMetadata> {\n  // Here we load all schemas, to avoid pg-structure failing on cross-schema foreign keys\n  // like our cyanaudit triggers (https://github.com/ozum/pg-structure/issues/85), and then\n  // later filter them non-public schema tables out.\n  const db = await pgStructure(client);\n  const enums = await loadEnumMetadata(db, client, config);\n  const pgEnums = await loadPgEnumMetadata(db, client, config);\n  const entities = db.tables\n    .filter((t) => isEntityTable(config, t))\n    .sortBy(\"name\")\n    .map((table) => new EntityDbMetadata(config, table, enums));\n  const totalTables = db.tables.length;\n  const joinTables = db.tables.filter((t) => isJoinTable(config, t)).map((t) => t.name);\n  const otherTables = db.tables\n    .filter((t) => !isEntityTable(config, t) && !isEnumTable(config, t) && !isJoinTable(config, t))\n    .map((t) => t.name);\n  const entitiesByName = Object.fromEntries(entities.map((e) => [e.name, e]));\n  return { entities, entitiesByName, enums, pgEnums, totalTables, joinTables, otherTables };\n}\n\nfunction maybeSetDatabaseUrl(config: Config): void {\n  if (!process.env.DATABASE_URL && config.databaseUrl) {\n    process.env.DATABASE_URL = config.databaseUrl;\n  }\n}\n\nexport function maybeSetExitCode(): void {\n  if (\n    !process.argv.includes(\"--always-exit-code-zero\") &&\n    // strict mode + warnings or greater\n    ((process.argv.includes(\"--strict\") && loggerMaxWarningLevelHit >= LOG_LEVELS.warn) ||\n      // otherwise errors or greater\n      loggerMaxWarningLevelHit >= LOG_LEVELS.error)\n  ) {\n    process.exitCode = 1;\n  }\n}\n\n/** Validates that tags can be parsed unambiguously with the configured delimiter. */\nfunction validateTagDelimiter(config: Config, entities: EntityDbMetadata[]): void {\n  const tagDelimiter = config.tagDelimiter ?? \":\";\n  for (const entity of entities) {\n    if (tagDelimiter !== \"\" && `${entity.tagName}${tagDelimiter}`.indexOf(tagDelimiter) !== entity.tagName.length) {\n      throw new Error(\n        `Tagged id delimiter '${tagDelimiter}' cannot occur in or overlap tag '${entity.tagName}' for ${entity.name}`,\n      );\n    } else if (tagDelimiter === \"\" && !/^[a-z]+$/i.test(entity.tagName)) {\n      throw new Error(`Delimiterless ids require an alphabetic tag, got '${entity.tagName}' for ${entity.name}`);\n    } else if (\n      tagDelimiter === \"\" &&\n      entity.primaryKey.columnType !== \"int\" &&\n      entity.primaryKey.columnType !== \"bigint\"\n    ) {\n      throw new Error(\n        `Delimiterless ids require an int or bigint primary key, got '${entity.primaryKey.columnType}' for ${entity.name}`,\n      );\n    }\n  }\n}\n\nif (typeof module !== \"undefined\" && require.main === module) {\n  joistCodegen()\n    .then(() => maybeSetExitCode())\n    .catch((err) => {\n      console.error(err);\n      process.exit(1);\n    });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA4BA,MAAM,EAAE,SAAS,gBAAgB;AAkBjC,eAAsB,eAAe;CACnC,MAAM,SAAS,MAAM,WAAW;CAEhC,oBAAoB,MAAM;CAC1B,IAAI,CAAC,QAAQ,IAAI,gBAAgB,CAAC,QAAQ,IAAI,SAAS;EACrD,QAAQ,IAAI,+DAA+D;EAC3E,QAAQ,IAAI,oEAAoE;EAChF,QAAQ,IAAI,8CAA8C;EAC1D;CACF;CACA,MAAM,WAAW,sBAAsB;CAEvC,MAAM,SAAS,IAAI,OAAO,QAAQ;CAClC,MAAM,OAAO,QAAQ;CAErB,MAAM,aAAa,MAAM,mBAAmB,QAAQ,MAAM;CAC1D,MAAM,EAAE,UAAU,OAAO,gBAAgB;CACzC,QAAQ,IACN,SAAS,YAAY,iBAAiB,SAAS,OAAO,kBAAkB,OAAO,QAAQ,KAAK,CAAC,CAAC,OAAO,aACvG;CACA,QAAQ,IAAI,EAAE;CAGd,wBAAwB,QAAQ,UAAU;CAG1C,qBAAqB,QAAQ,UAAU;CAGvC,WAAW,QAAQ,UAAU;CAC7B,qBAAqB,QAAQ,QAAQ;CAGrC,MAAM,gBAAgB,QAAQ,UAAU;CAGxC,MAAM,2BAA2B,QAAQ,WAAW,QAAQ;CAG5D,MAAM,4BAA4B,QAAQ,QAAQ,UAAU,UAAU;CAEtE,MAAM,OAAO,IAAI;CAGjB,MAAM,mBAAmB,MAAM;CAG/B,KAAK,MAAM,UAAU,UAAU,4BAA4B,MAAM;CACjE,yBAAyB,QAAQ,UAAU;CAG3C,MAAM,qBAAqB,QAAQ,UAAU;CAG7C,IAAI,OAAO,WAAW,OAAO,MAAM,cAAc;CAEjD,qBAAqB,QAAQ,QAAQ;CACrC,MAAM,YAAY,MAAM;AAC1B;;AAGA,eAAsB,qBAAqB,QAAgB,QAAmC;CAC5F,MAAM,QAAQ,MAAM,cAAc,QAAQ,MAAM;CAChD,MAAM,SAAS,OAAO,MAAO,OAAO,6BAA6B,OAAO,OAAQ;CAChF,MAAM,UAAU;EACd,UAAU;EACV,WAAW,OAAO;EAClB;EACA,cAAc,EAAE,kBAAkB,UAAU,MAAM;CACpD,CAAC;AACH;AAEA,eAAe,4BAA4B,QAAgB,QAAgB,UAA4B,IAAgB;CAErH,IAAI,OAAO,wBAAwB,OAAO;EAExC,IAAI,MAAM,QAAQ,OAAO,mBAAmB,GAAG;GAC7C,QAAQ,IAAI,mCAAmC;GAC/C,MAAM,QAAQ,IACZ,OAAO,oBAAoB,IAAI,OAAO,WAAW;IAC/C,MAAM,SAAS,IAAI,OAAO;KAAE,GAAG;KAAU,UAAU;IAAO,CAAC;IAC3D,MAAM,OAAO,QAAQ;IACrB,MAAM,oBAAoB,QAAQ,EAAE;IACpC,MAAM,OAAO,IAAI;GACnB,CAAC,CACH;EACF,OAAO;GACL,QAAQ,IAAI,kCAAkC;GAC9C,MAAM,oBAAoB,QAAQ,EAAE;EACtC;CACF;AACF;AAEA,eAAe,mBAAmB,QAAgB,QAAqC;CAIrF,MAAM,KAAK,MAAM,YAAY,MAAM;CACnC,MAAM,QAAQ,MAAM,iBAAiB,IAAI,QAAQ,MAAM;CACvD,MAAM,UAAU,MAAM,mBAAmB,IAAI,QAAQ,MAAM;CAC3D,MAAM,WAAW,GAAG,OACjB,QAAQ,MAAM,cAAc,QAAQ,CAAC,CAAC,CAAC,CACvC,OAAO,MAAM,CAAC,CACd,KAAK,UAAU,IAAI,iBAAiB,QAAQ,OAAO,KAAK,CAAC;CAC5D,MAAM,cAAc,GAAG,OAAO;CAC9B,MAAM,aAAa,GAAG,OAAO,QAAQ,MAAM,YAAY,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,IAAI;CACpF,MAAM,cAAc,GAAG,OACpB,QAAQ,MAAM,CAAC,cAAc,QAAQ,CAAC,KAAK,CAAC,YAAY,QAAQ,CAAC,KAAK,CAAC,YAAY,QAAQ,CAAC,CAAC,CAAC,CAC9F,KAAK,MAAM,EAAE,IAAI;CAEpB,OAAO;EAAE;EAAU,gBADI,OAAO,YAAY,SAAS,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC,CACzC;EAAG;EAAO;EAAS;EAAa;EAAY;CAAY;AAC1F;AAEA,SAAS,oBAAoB,QAAsB;CACjD,IAAI,CAAC,QAAQ,IAAI,gBAAgB,OAAO,aACtC,QAAQ,IAAI,eAAe,OAAO;AAEtC;AAEA,SAAgB,mBAAyB;CACvC,IACE,CAAC,QAAQ,KAAK,SAAS,yBAAyB,MAE9C,QAAQ,KAAK,SAAS,UAAU,KAAK,4BAA4B,WAAW,QAE5E,4BAA4B,WAAW,QAEzC,QAAQ,WAAW;AAEvB;;AAGA,SAAS,qBAAqB,QAAgB,UAAoC;CAChF,MAAM,eAAe,OAAO,gBAAgB;CAC5C,KAAK,MAAM,UAAU,UACnB,IAAI,iBAAiB,MAAM,GAAG,OAAO,UAAU,eAAe,QAAQ,YAAY,MAAM,OAAO,QAAQ,QACrG,MAAM,IAAI,MACR,wBAAwB,aAAa,oCAAoC,OAAO,QAAQ,QAAQ,OAAO,MACzG;MACK,IAAI,iBAAiB,MAAM,CAAC,YAAY,KAAK,OAAO,OAAO,GAChE,MAAM,IAAI,MAAM,qDAAqD,OAAO,QAAQ,QAAQ,OAAO,MAAM;MACpG,IACL,iBAAiB,MACjB,OAAO,WAAW,eAAe,SACjC,OAAO,WAAW,eAAe,UAEjC,MAAM,IAAI,MACR,gEAAgE,OAAO,WAAW,WAAW,QAAQ,OAAO,MAC9G;AAGN;AAEA,IAAI,OAAO,WAAW,eAAA,UAAuB,SAAS,QACpD,aAAa,CAAC,CACX,WAAW,iBAAiB,CAAC,CAAC,CAC9B,OAAO,QAAQ;CACd,QAAQ,MAAM,GAAG;CACjB,QAAQ,KAAK,CAAC;AAChB,CAAC"}