rake-db 2.3.25 → 2.3.26

package/dist/index.mjs

@@ -1050,6 +1050,12 @@ class MigrationBase {
   dropExtension(name, options = {}) {
     return createExtension$1(this, !this.up, name, options);
   }
+  createEnum(name, values, options) {
+    return createEnum(this, this.up, name, values, options);
+  }
+  dropEnum(name, values, options) {
+    return createEnum(this, !this.up, name, values, options);
+  }
   async tableExists(tableName) {
     return queryExists(this, {
       text: `SELECT 1 FROM "information_schema"."tables" WHERE "table_name" = $1`,
@@ -1118,13 +1124,35 @@ const createExtension$1 = async (migration, up, name, options) => {
   }, options);
   let query;
   if (ast.action === "drop") {
-    query = `DROP EXTENSION${ast.ifNotExists ? " IF EXISTS" : ""} "${ast.name}"${ast.cascade ? " CASCADE" : ""}`;
+    query = `DROP EXTENSION${ast.dropIfExists ? " IF EXISTS" : ""} "${ast.name}"${ast.cascade ? " CASCADE" : ""}`;
   } else {
-    query = `CREATE EXTENSION${ast.ifExists ? " IF NOT EXISTS" : ""} "${ast.name}"${ast.schema ? ` SCHEMA "${ast.schema}"` : ""}${ast.version ? ` VERSION '${ast.version}'` : ""}${ast.cascade ? " CASCADE" : ""}`;
+    query = `CREATE EXTENSION${ast.createIfNotExists ? " IF NOT EXISTS" : ""} "${ast.name}"${ast.schema ? ` SCHEMA "${ast.schema}"` : ""}${ast.version ? ` VERSION '${ast.version}'` : ""}${ast.cascade ? " CASCADE" : ""}`;
   }
   await migration.adapter.query(query);
   await runCodeUpdater(migration, ast);
 };
+const createEnum = async (migration, up, name, values, options = {}) => {
+  const [schema, enumName] = getSchemaAndTableFromName(name);
+  const ast = __spreadValues$2({
+    type: "enum",
+    action: up ? "create" : "drop",
+    schema,
+    name: enumName,
+    values
+  }, options);
+  let text;
+  const quotedName = quoteWithSchema(ast);
+  if (ast.action === "create") {
+    text = `CREATE TYPE ${quotedName} AS ENUM (${values.map((_, i) => `$${i + 1}`).join(", ")})`;
+  } else {
+    text = `DROP TYPE${ast.dropIfExists ? " IF EXISTS" : ""} ${quotedName}${ast.cascade ? " CASCADE" : ""}`;
+  }
+  await migration.adapter.query({
+    text,
+    values
+  });
+  await runCodeUpdater(migration, ast);
+};
 const queryExists = (db, sql) => {
   return db.adapter.query(sql).then(({ rowCount }) => rowCount > 0);
 };