{"version":3,"sources":["../../src/utilities/buildCreateMigration.ts"],"sourcesContent":["import type { DrizzleSnapshotJSON } from 'drizzle-kit/api'\nimport type { CreateMigration, Payload } from 'payload'\n\nimport fs from 'fs'\nimport path from 'path'\nimport { getPredefinedMigration, writeMigrationIndex } from 'payload'\nimport prompts from 'prompts'\n\nimport type { DrizzleAdapter } from '../types.js'\n\nimport { getMigrationTemplate } from './getMigrationTemplate.js'\n\nexport const buildCreateMigration = ({\n  executeMethod,\n  filename,\n  sanitizeStatements,\n}: {\n  executeMethod: string\n  filename: string\n  sanitizeStatements: (args: { sqlExecute: string; statements: string[] }) => string\n}): CreateMigration => {\n  const dirname = path.dirname(filename)\n  return async function createMigration(\n    this: DrizzleAdapter,\n    { file, forceAcceptWarning, migrationName, payload, skipEmpty },\n  ) {\n    const dir = payload.db.migrationDir\n    if (!fs.existsSync(dir)) {\n      fs.mkdirSync(dir)\n    }\n\n    const { generateDrizzleJson, generateMigration, upSnapshot } = this.requireDrizzleKit()\n\n    const drizzleJsonAfter = await generateDrizzleJson(this.schema)\n\n    const [yyymmdd, hhmmss] = new Date().toISOString().split('T')\n    const formattedDate = yyymmdd.replace(/\\D/g, '')\n    const formattedTime = hhmmss.split('.')[0].replace(/\\D/g, '')\n    let imports: string = ''\n    let downSQL: string\n    let upSQL: string\n\n    const predefinedMigration = await getPredefinedMigration({\n      dirname,\n      file,\n      migrationName,\n      payload,\n    })\n\n    imports = predefinedMigration.imports\n    downSQL = predefinedMigration.downSQL\n    upSQL = predefinedMigration.upSQL\n\n    const timestamp = `${formattedDate}_${formattedTime}`\n\n    const name = migrationName || file?.split('/').slice(2).join('/')\n    const fileName = `${timestamp}${name ? `_${name.replace(/\\W/g, '_')}` : ''}`\n\n    const filePath = `${dir}/${fileName}`\n\n    if (typeof predefinedMigration.dynamic === 'function') {\n      const dynamicResult = await predefinedMigration.dynamic({ filePath, payload })\n\n      if (dynamicResult.upSQL) {\n        upSQL = dynamicResult.upSQL\n      }\n\n      if (dynamicResult.downSQL) {\n        downSQL = dynamicResult.downSQL\n      }\n\n      if (dynamicResult.imports) {\n        imports = dynamicResult.imports\n      }\n    }\n\n    let drizzleJsonBefore = this.defaultDrizzleSnapshot as DrizzleSnapshotJSON\n\n    if (this.schemaName) {\n      drizzleJsonBefore.schemas = {\n        [this.schemaName]: this.schemaName,\n      }\n    }\n\n    if (!upSQL) {\n      // Get latest migration snapshot\n      const latestSnapshot = fs\n        .readdirSync(dir)\n        .filter((file) => file.endsWith('.json'))\n        .sort()\n        .reverse()?.[0]\n\n      if (latestSnapshot) {\n        drizzleJsonBefore = JSON.parse(fs.readFileSync(`${dir}/${latestSnapshot}`, 'utf8'))\n\n        if (upSnapshot && drizzleJsonBefore.version < drizzleJsonAfter.version) {\n          drizzleJsonBefore = upSnapshot(drizzleJsonBefore)\n        }\n      }\n\n      payload.logger.info('Starting migration: generating UP statements...')\n      const sqlStatementsUp = await generateMigration(drizzleJsonBefore, drizzleJsonAfter)\n\n      payload.logger.info('Migration UP complete. Generating DOWN statements...')\n      const sqlStatementsDown = await generateMigration(drizzleJsonAfter, drizzleJsonBefore)\n\n      payload.logger.info('Migration DOWN statements generation complete.')\n\n      const sqlExecute = `await db.${executeMethod}(` + 'sql`'\n\n      if (sqlStatementsUp?.length) {\n        upSQL = sanitizeStatements({ sqlExecute, statements: sqlStatementsUp })\n      }\n      if (sqlStatementsDown?.length) {\n        downSQL = sanitizeStatements({ sqlExecute, statements: sqlStatementsDown })\n      }\n\n      if (!upSQL?.length && !downSQL?.length && !forceAcceptWarning) {\n        if (skipEmpty) {\n          process.exit(0)\n        }\n\n        const { confirm: shouldCreateBlankMigration } = await prompts(\n          {\n            name: 'confirm',\n            type: 'confirm',\n            initial: false,\n            message: 'No schema changes detected. Would you like to create a blank migration file?',\n          },\n          {\n            onCancel: () => {\n              process.exit(0)\n            },\n          },\n        )\n\n        if (!shouldCreateBlankMigration) {\n          process.exit(0)\n        }\n      }\n\n      // write schema\n      fs.writeFileSync(`${filePath}.json`, JSON.stringify(drizzleJsonAfter, null, 2))\n    }\n\n    const data = getMigrationTemplate({\n      downSQL: downSQL || `  // Migration code`,\n      imports,\n      packageName: payload.db.packageName,\n      upSQL: upSQL || `  // Migration code`,\n    })\n\n    const fullPath = `${filePath}.ts`\n\n    // write migration\n    fs.writeFileSync(fullPath, data)\n\n    writeMigrationIndex({ migrationsDir: payload.db.migrationDir })\n\n    payload.logger.info({ msg: `Migration created at ${fullPath}` })\n  }\n}\n"],"names":["fs","path","getPredefinedMigration","writeMigrationIndex","prompts","getMigrationTemplate","buildCreateMigration","executeMethod","filename","sanitizeStatements","dirname","createMigration","file","forceAcceptWarning","migrationName","payload","skipEmpty","dir","db","migrationDir","existsSync","mkdirSync","generateDrizzleJson","generateMigration","upSnapshot","requireDrizzleKit","drizzleJsonAfter","schema","yyymmdd","hhmmss","Date","toISOString","split","formattedDate","replace","formattedTime","imports","downSQL","upSQL","predefinedMigration","timestamp","name","slice","join","fileName","filePath","dynamic","dynamicResult","drizzleJsonBefore","defaultDrizzleSnapshot","schemaName","schemas","latestSnapshot","readdirSync","filter","endsWith","sort","reverse","JSON","parse","readFileSync","version","logger","info","sqlStatementsUp","sqlStatementsDown","sqlExecute","length","statements","process","exit","confirm","shouldCreateBlankMigration","type","initial","message","onCancel","writeFileSync","stringify","data","packageName","fullPath","migrationsDir","msg"],"mappings":"AAGA,OAAOA,QAAQ,KAAI;AACnB,OAAOC,UAAU,OAAM;AACvB,SAASC,sBAAsB,EAAEC,mBAAmB,QAAQ,UAAS;AACrE,OAAOC,aAAa,UAAS;AAI7B,SAASC,oBAAoB,QAAQ,4BAA2B;AAEhE,OAAO,MAAMC,uBAAuB,CAAC,EACnCC,aAAa,EACbC,QAAQ,EACRC,kBAAkB,EAKnB;IACC,MAAMC,UAAUT,KAAKS,OAAO,CAACF;IAC7B,OAAO,eAAeG,gBAEpB,EAAEC,IAAI,EAAEC,kBAAkB,EAAEC,aAAa,EAAEC,OAAO,EAAEC,SAAS,EAAE;QAE/D,MAAMC,MAAMF,QAAQG,EAAE,CAACC,YAAY;QACnC,IAAI,CAACnB,GAAGoB,UAAU,CAACH,MAAM;YACvBjB,GAAGqB,SAAS,CAACJ;QACf;QAEA,MAAM,EAAEK,mBAAmB,EAAEC,iBAAiB,EAAEC,UAAU,EAAE,GAAG,IAAI,CAACC,iBAAiB;QAErF,MAAMC,mBAAmB,MAAMJ,oBAAoB,IAAI,CAACK,MAAM;QAE9D,MAAM,CAACC,SAASC,OAAO,GAAG,IAAIC,OAAOC,WAAW,GAAGC,KAAK,CAAC;QACzD,MAAMC,gBAAgBL,QAAQM,OAAO,CAAC,OAAO;QAC7C,MAAMC,gBAAgBN,OAAOG,KAAK,CAAC,IAAI,CAAC,EAAE,CAACE,OAAO,CAAC,OAAO;QAC1D,IAAIE,UAAkB;QACtB,IAAIC;QACJ,IAAIC;QAEJ,MAAMC,sBAAsB,MAAMrC,uBAAuB;YACvDQ;YACAE;YACAE;YACAC;QACF;QAEAqB,UAAUG,oBAAoBH,OAAO;QACrCC,UAAUE,oBAAoBF,OAAO;QACrCC,QAAQC,oBAAoBD,KAAK;QAEjC,MAAME,YAAY,GAAGP,cAAc,CAAC,EAAEE,eAAe;QAErD,MAAMM,OAAO3B,iBAAiBF,MAAMoB,MAAM,KAAKU,MAAM,GAAGC,KAAK;QAC7D,MAAMC,WAAW,GAAGJ,YAAYC,OAAO,CAAC,CAAC,EAAEA,KAAKP,OAAO,CAAC,OAAO,MAAM,GAAG,IAAI;QAE5E,MAAMW,WAAW,GAAG5B,IAAI,CAAC,EAAE2B,UAAU;QAErC,IAAI,OAAOL,oBAAoBO,OAAO,KAAK,YAAY;YACrD,MAAMC,gBAAgB,MAAMR,oBAAoBO,OAAO,CAAC;gBAAED;gBAAU9B;YAAQ;YAE5E,IAAIgC,cAAcT,KAAK,EAAE;gBACvBA,QAAQS,cAAcT,KAAK;YAC7B;YAEA,IAAIS,cAAcV,OAAO,EAAE;gBACzBA,UAAUU,cAAcV,OAAO;YACjC;YAEA,IAAIU,cAAcX,OAAO,EAAE;gBACzBA,UAAUW,cAAcX,OAAO;YACjC;QACF;QAEA,IAAIY,oBAAoB,IAAI,CAACC,sBAAsB;QAEnD,IAAI,IAAI,CAACC,UAAU,EAAE;YACnBF,kBAAkBG,OAAO,GAAG;gBAC1B,CAAC,IAAI,CAACD,UAAU,CAAC,EAAE,IAAI,CAACA,UAAU;YACpC;QACF;QAEA,IAAI,CAACZ,OAAO;YACV,gCAAgC;YAChC,MAAMc,iBAAiBpD,GACpBqD,WAAW,CAACpC,KACZqC,MAAM,CAAC,CAAC1C,OAASA,KAAK2C,QAAQ,CAAC,UAC/BC,IAAI,GACJC,OAAO,IAAI,CAAC,EAAE;YAEjB,IAAIL,gBAAgB;gBAClBJ,oBAAoBU,KAAKC,KAAK,CAAC3D,GAAG4D,YAAY,CAAC,GAAG3C,IAAI,CAAC,EAAEmC,gBAAgB,EAAE;gBAE3E,IAAI5B,cAAcwB,kBAAkBa,OAAO,GAAGnC,iBAAiBmC,OAAO,EAAE;oBACtEb,oBAAoBxB,WAAWwB;gBACjC;YACF;YAEAjC,QAAQ+C,MAAM,CAACC,IAAI,CAAC;YACpB,MAAMC,kBAAkB,MAAMzC,kBAAkByB,mBAAmBtB;YAEnEX,QAAQ+C,MAAM,CAACC,IAAI,CAAC;YACpB,MAAME,oBAAoB,MAAM1C,kBAAkBG,kBAAkBsB;YAEpEjC,QAAQ+C,MAAM,CAACC,IAAI,CAAC;YAEpB,MAAMG,aAAa,CAAC,SAAS,EAAE3D,cAAc,CAAC,CAAC,GAAG;YAElD,IAAIyD,iBAAiBG,QAAQ;gBAC3B7B,QAAQ7B,mBAAmB;oBAAEyD;oBAAYE,YAAYJ;gBAAgB;YACvE;YACA,IAAIC,mBAAmBE,QAAQ;gBAC7B9B,UAAU5B,mBAAmB;oBAAEyD;oBAAYE,YAAYH;gBAAkB;YAC3E;YAEA,IAAI,CAAC3B,OAAO6B,UAAU,CAAC9B,SAAS8B,UAAU,CAACtD,oBAAoB;gBAC7D,IAAIG,WAAW;oBACbqD,QAAQC,IAAI,CAAC;gBACf;gBAEA,MAAM,EAAEC,SAASC,0BAA0B,EAAE,GAAG,MAAMpE,QACpD;oBACEqC,MAAM;oBACNgC,MAAM;oBACNC,SAAS;oBACTC,SAAS;gBACX,GACA;oBACEC,UAAU;wBACRP,QAAQC,IAAI,CAAC;oBACf;gBACF;gBAGF,IAAI,CAACE,4BAA4B;oBAC/BH,QAAQC,IAAI,CAAC;gBACf;YACF;YAEA,eAAe;YACftE,GAAG6E,aAAa,CAAC,GAAGhC,SAAS,KAAK,CAAC,EAAEa,KAAKoB,SAAS,CAACpD,kBAAkB,MAAM;QAC9E;QAEA,MAAMqD,OAAO1E,qBAAqB;YAChCgC,SAASA,WAAW,CAAC,mBAAmB,CAAC;YACzCD;YACA4C,aAAajE,QAAQG,EAAE,CAAC8D,WAAW;YACnC1C,OAAOA,SAAS,CAAC,mBAAmB,CAAC;QACvC;QAEA,MAAM2C,WAAW,GAAGpC,SAAS,GAAG,CAAC;QAEjC,kBAAkB;QAClB7C,GAAG6E,aAAa,CAACI,UAAUF;QAE3B5E,oBAAoB;YAAE+E,eAAenE,QAAQG,EAAE,CAACC,YAAY;QAAC;QAE7DJ,QAAQ+C,MAAM,CAACC,IAAI,CAAC;YAAEoB,KAAK,CAAC,qBAAqB,EAAEF,UAAU;QAAC;IAChE;AACF,EAAC"}