rake-db 2.30.3 → 2.30.6

package/dist/index.mjs

@@ -1787,33 +1787,46 @@ const changeAstToQuery = ({ name, from, to }) => {
   const w = [];
   for (const key in to) {
     let value = to[key];
-    if (key !== "config") {
+    if (key !== "config" && key !== "name") {
       if (value instanceof Date) value = value.toISOString();
-      let other = from[key];
-      if (other instanceof Date) other = other.toISOString();
-      if (value !== other && key in serializers) {
-        w.push(serializers[key](value));
-      }
+      w.push(serializers[key](value));
     }
   }
   if (w.length) {
     queries.push(`ALTER ROLE "${name}" WITH ${w.join(" ")}`);
   }
-  const config = to.config;
-  if (config) {
-    const fromConfig = from.config ?? emptyObject;
-    for (const key in config) {
-      const value = config[key];
-      const other = fromConfig[key];
-      if (value !== other) {
-        queries.push(
-          `ALTER ROLE "${name}" ${value ? `SET ${key} = '${value}'` : `RESET ${key}`}`
-        );
-      }
+  if (to.config) {
+    for (const key in to.config) {
+      const value = to.config[key];
+      queries.push(
+        `ALTER ROLE "${name}" ${value ? `SET ${key} = '${value}'` : `RESET ${key}`}`
+      );
+    }
+  } else if (from.config) {
+    for (const key in from.config) {
+      queries.push(`ALTER ROLE "${name}" RESET ${key}`);
     }
   }
   return queries.join(";\n");
 };
+const renameRole = async (migration, up, from, to) => {
+  if (!up) {
+    const f = from;
+    from = to;
+    to = f;
+  }
+  const ast = makeRenameAst(from, to);
+  const sql = renameAstToQuery(ast);
+  if (sql) await migration.adapter.arrays(sql);
+};
+const makeRenameAst = (from, to) => ({
+  type: "renameRole",
+  from,
+  to
+});
+const renameAstToQuery = (ast) => {
+  return `ALTER ROLE "${ast.from}" RENAME TO "${ast.to}"`;
+};
 
 const createMigrationInterface = (tx, up, config) => {
   const adapter = Object.create(tx);
@@ -2704,6 +2717,9 @@ class Migration {
   dropRole(name, params) {
     return createOrDropRole(this, !this.up, name, params);
   }
+  renameRole(from, to) {
+    return renameRole(this, this.up, from, to);
+  }
   changeRole(name, params) {
     return changeRole(
       this,
@@ -4084,7 +4100,8 @@ const astToGenerateItem = (config, ast, currentSchema) => {
       break;
     }
     case "role":
-    case "changeRole": {
+    case "changeRole":
+    case "renameRole": {
       break;
     }
     default:
@@ -4669,19 +4686,24 @@ const astEncoders = {
     return code;
   },
   role(ast) {
-    const params = ast.action === "create" ? roleParams(ast) : void 0;
+    const params = roleParams(ast.name, ast);
     const arr = [
-      `await db.${ast.action}Role(${singleQuote(ast.name)}${params?.length ? ", {" : ");"}`
+      `await db.${ast.action}Role(${singleQuote(ast.name)}${params.length ? ", {" : ");"}`
     ];
-    if (params?.length) {
+    if (params.length) {
       arr.push(params);
       arr.push("});");
     }
     return arr;
   },
+  renameRole(ast) {
+    return `await db.renameRole(${singleQuote(ast.from)}, ${singleQuote(
+      ast.to
+    )});`;
+  },
   changeRole(ast) {
-    const from = roleParams(ast.from, ast.to);
-    const to = roleParams(ast.to, ast.from, true);
+    const from = roleParams(ast.name, ast.from, ast.to);
+    const to = roleParams(ast.name, ast.to, ast.from, true);
     return [
       `await db.changeRole(${singleQuote(ast.name)}, {`,
       [...from.length ? ["from: {", from, "},"] : [], "to: {", to, "},"],
@@ -4689,7 +4711,7 @@ const astEncoders = {
     ];
   }
 };
-const roleParams = (ast, compare, to) => {
+const roleParams = (name, ast, compare, to) => {
   const params = [];
   for (const key of [
     "name",
@@ -4704,7 +4726,7 @@ const roleParams = (ast, compare, to) => {
     "bypassRls",
     "config"
   ]) {
-    if (key === "name" && !to) continue;
+    if (key === "name" && (!to || ast.name === name)) continue;
     let value = ast[key];
     if (!compare && (!value || key === "connLimit" && value === -1)) {
       continue;
@@ -5128,18 +5150,19 @@ const collationsSql = (version) => `SELECT
 FROM pg_collation
 JOIN pg_namespace n on pg_collation.collnamespace = n.oid
 WHERE ${filterSchema("n.nspname")}`;
-const roleSql = (params) => `SELECT json_agg(json_build_object(
+const roleSql = (params) => `SELECT COALESCE(json_agg(json_build_object(
   'name', rolname,
   'super', rolsuper,
   'inherit', rolinherit,
   'createRole', rolcreaterole,
+  'createDb', rolcreatedb,
   'canLogin', rolcanlogin,
   'replication', rolreplication,
   'connLimit', rolconnlimit,
   'validUntil', rolvaliduntil,
   'bypassRls', rolbypassrls,
   'config', rolconfig
-)) FROM pg_roles WHERE ${params.whereSql ?? `name != 'postgres' AND name !~ '^pg_'`}`;
+)), '[]') FROM pg_roles WHERE ${params.whereSql ?? `rolname != 'postgres' AND rolname !~ '^pg_'`}`;
 const sql = (version, params) => `SELECT (${schemasSql}) AS "schemas", ${jsonAgg(
   tablesSql,
   "tables"
@@ -5243,7 +5266,9 @@ async function introspectDbSchema(db, params) {
   if (result.roles) {
     for (const role of result.roles) {
       nullsToUndefined(role);
-      if (role.validUntil) role.validUntil = new Date(role.validUntil);
+      if (role.validUntil) {
+        role.validUntil = role.validUntil === "infinity" ? void 0 : new Date(role.validUntil);
+      }
       if (role.config) {
         const arr = role.config;
         role.config = Object.fromEntries(