pocketbase-zod-schema 0.7.0 → 0.7.2

package/dist/migration/generator.cjs

@@ -224,6 +224,13 @@ function formatValue(value) {
   }
   return String(value);
 }
+function formatSqlTemplate(query, indent = "      ") {
+  const escaped = query.replace(/\\/g, "\\\\").replace(/`/g, "\\`").replace(/\$\{/g, "\\${");
+  const body = escaped.split("\n").map((line) => line.trim() === "" ? "" : `${indent}${line}`).join("\n");
+  return `\`
+${body}
+${indent.slice(0, -2)}\``;
+}
 function getFieldConstructorName(fieldType) {
   const constructorMap = {
     text: "TextField",
@@ -262,6 +269,36 @@ function getSystemFields() {
     }
   ];
 }
+function getSystemTimestampFields() {
+  return [
+    {
+      name: "created",
+      id: "autodate2990389176",
+      type: "autodate",
+      required: false,
+      options: {
+        hidden: false,
+        onCreate: true,
+        onUpdate: false,
+        presentable: false,
+        system: true
+      }
+    },
+    {
+      name: "updated",
+      id: "autodate3332085495",
+      type: "autodate",
+      required: false,
+      options: {
+        hidden: false,
+        onCreate: true,
+        onUpdate: true,
+        presentable: false,
+        system: true
+      }
+    }
+  ];
+}
 function getAuthSystemFields() {
   return [
     {
@@ -554,6 +591,15 @@ function generateCollectionRules(rules, collectionType = "base") {
   if (!rules) {
     return "";
   }
+  if (collectionType === "view") {
+    return [
+      `"listRule": ${formatValue(rules.listRule ?? null)}`,
+      `"viewRule": ${formatValue(rules.viewRule ?? null)}`,
+      `"createRule": null`,
+      `"updateRule": null`,
+      `"deleteRule": null`
+    ].join(",\n    ");
+  }
   const parts = [];
   if (rules.listRule !== void 0) {
     parts.push(`"listRule": ${formatValue(rules.listRule)}`);
@@ -579,6 +625,15 @@ function generateCollectionPermissions(permissions, collectionType = "base") {
   if (!permissions) {
     return "";
   }
+  if (collectionType === "view") {
+    return [
+      `"listRule": ${formatValue(permissions.listRule ?? null)}`,
+      `"viewRule": ${formatValue(permissions.viewRule ?? null)}`,
+      `"createRule": null`,
+      `"updateRule": null`,
+      `"deleteRule": null`
+    ].join(",\n    ");
+  }
   const parts = [];
   if (permissions.listRule !== void 0) {
     parts.push(`"listRule": ${formatValue(permissions.listRule)}`);
@@ -616,6 +671,28 @@ function generatePermissionUpdate(collectionName, ruleType, newValue, varName, i
   lines.push(isLast ? `  return app.save(${collectionVar});` : `  app.save(${collectionVar});`);
   return lines.join("\n");
 }
+function generateViewQueryUpdate(collectionName, viewQuery, varName, isLast = false, collectionIdMap) {
+  const lines = [];
+  const collectionVar = varName || `collection_${collectionName}_viewQuery`;
+  lines.push(`  const ${collectionVar} = ${generateFindCollectionCode(collectionName, collectionIdMap)};`);
+  lines.push(`  unmarshal({`);
+  lines.push(`    "viewQuery": ${formatSqlTemplate(viewQuery, "      ")},`);
+  lines.push(`  }, ${collectionVar})`);
+  lines.push(isLast ? `  return app.save(${collectionVar});` : `  app.save(${collectionVar});`);
+  return lines.join("\n");
+}
+function generateGroupedRuleUpdates(collectionName, entries, varSuffix, isLast = false, collectionIdMap) {
+  const collectionVar = `collection_${collectionName}_${varSuffix}`;
+  const lines = [];
+  lines.push(`  const ${collectionVar} = ${generateFindCollectionCode(collectionName, collectionIdMap)};`);
+  lines.push(`  unmarshal({`);
+  for (const entry of entries) {
+    lines.push(`    "${entry.ruleType}": ${formatValue(entry.value)},`);
+  }
+  lines.push(`  }, ${collectionVar})`);
+  lines.push(isLast ? `  return app.save(${collectionVar});` : `  app.save(${collectionVar});`);
+  return lines.join("\n");
+}
 
 // src/migration/generator/collections.ts
 function generateCollectionCreation(collection, varName = "collection", isLast = false, collectionIdMap) {
@@ -634,16 +711,24 @@ function generateCollectionCreation(collection, varName = "collection", isLast =
   } else if (rulesCode) {
     lines.push(`    ${rulesCode},`);
   }
+  if (collection.type === "view") {
+    lines.push(`    "viewQuery": ${formatSqlTemplate(collection.viewQuery ?? "")},`);
+    lines.push(`  });`);
+    lines.push(``);
+    lines.push(isLast ? `  return app.save(${varName});` : `  app.save(${varName});`);
+    return lines.join("\n");
+  }
   const systemFieldNames = ["created", "updated", "id"];
   if (collection.type === "auth") {
     systemFieldNames.push(...getAuthSystemFields().map((f) => f.name));
   }
   const userFields = collection.fields.filter((f) => !systemFieldNames.includes(f.name));
-  const allFields = [...getSystemFields().filter((f) => f.name === "id")];
+  const allFields = [...getSystemFields()];
   if (collection.type === "auth") {
     allFields.push(...getAuthSystemFields());
   }
   allFields.push(...userFields);
+  allFields.push(...getSystemTimestampFields());
   lines.push(`    "fields": ${generateFieldsArray(allFields, collectionIdMap)},`);
   let allIndexes = [...collection.indexes || []];
   if (collection.type === "auth") {
@@ -673,7 +758,21 @@ function generateOperationUpMigration(operation, collectionIdMap) {
     const modification = operation.modifications;
     const collectionName = typeof operation.collection === "string" ? operation.collection : operation.collection?.name ?? modification.collection;
     let operationCount = 0;
-    const totalOperations = modification.fieldsToAdd.length + modification.fieldsToModify.length + modification.fieldsToRemove.length + modification.indexesToAdd.length + modification.indexesToRemove.length + modification.rulesToUpdate.length + modification.permissionsToUpdate.length;
+    const totalOperations = modification.fieldsToAdd.length + modification.fieldsToModify.length + modification.fieldsToRemove.length + modification.indexesToAdd.length + modification.indexesToRemove.length + (modification.viewQueryUpdate ? 1 : 0) + (modification.permissionsToUpdate.length > 0 ? 1 : modification.rulesToUpdate.length > 0 ? 1 : 0);
+    if (modification.viewQueryUpdate) {
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      lines.push(
+        generateViewQueryUpdate(
+          collectionName,
+          modification.viewQueryUpdate.newValue,
+          void 0,
+          isLast,
+          collectionIdMap
+        )
+      );
+      if (!isLast) lines.push("");
+    }
     for (let i = 0; i < modification.fieldsToAdd.length; i++) {
       const field = modification.fieldsToAdd[i];
       operationCount++;
@@ -713,23 +812,29 @@ function generateOperationUpMigration(operation, collectionIdMap) {
       if (!isLast) lines.push("");
     }
     if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
-      for (const permission of modification.permissionsToUpdate) {
-        operationCount++;
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      if (modification.permissionsToUpdate.length >= 2) {
+        const entries = modification.permissionsToUpdate.map((p) => ({ ruleType: p.ruleType, value: p.newValue }));
+        lines.push(generateGroupedRuleUpdates(collectionName, entries, "rules", isLast, collectionIdMap));
+      } else {
+        const permission = modification.permissionsToUpdate[0];
         const varName = `collection_${collectionName}_perm_${permission.ruleType}`;
-        const isLast = operationCount === totalOperations;
-        lines.push(
-          generatePermissionUpdate(collectionName, permission.ruleType, permission.newValue, varName, isLast, collectionIdMap)
-        );
-        if (!isLast) lines.push("");
+        lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.newValue, varName, isLast, collectionIdMap));
       }
+      if (!isLast) lines.push("");
     } else if (modification.rulesToUpdate.length > 0) {
-      for (const rule of modification.rulesToUpdate) {
-        operationCount++;
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      if (modification.rulesToUpdate.length >= 2) {
+        const entries = modification.rulesToUpdate.map((r) => ({ ruleType: r.ruleType, value: r.newValue }));
+        lines.push(generateGroupedRuleUpdates(collectionName, entries, "rules", isLast, collectionIdMap));
+      } else {
+        const rule = modification.rulesToUpdate[0];
         const varName = `collection_${collectionName}_rule_${rule.ruleType}`;
-        const isLast = operationCount === totalOperations;
         lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.newValue, varName, isLast, collectionIdMap));
-        if (!isLast) lines.push("");
       }
+      if (!isLast) lines.push("");
     }
   } else if (operation.type === "delete") {
     const collectionName = typeof operation.collection === "string" ? operation.collection : operation.collection.name;
@@ -768,25 +873,45 @@ function generateOperationDownMigration(operation, collectionIdMap) {
     const modification = operation.modifications;
     const collectionName = typeof operation.collection === "string" ? operation.collection : operation.collection?.name ?? modification.collection;
     let operationCount = 0;
-    const totalOperations = modification.fieldsToAdd.length + modification.fieldsToModify.length + modification.fieldsToRemove.length + modification.indexesToAdd.length + modification.indexesToRemove.length + modification.rulesToUpdate.length + modification.permissionsToUpdate.length;
+    const totalOperations = modification.fieldsToAdd.length + modification.fieldsToModify.length + modification.fieldsToRemove.length + modification.indexesToAdd.length + modification.indexesToRemove.length + (modification.viewQueryUpdate ? 1 : 0) + (modification.permissionsToUpdate.length > 0 ? 1 : modification.rulesToUpdate.length > 0 ? 1 : 0);
+    if (modification.viewQueryUpdate) {
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      lines.push(
+        generateViewQueryUpdate(
+          collectionName,
+          modification.viewQueryUpdate.oldValue ?? "",
+          `collection_${collectionName}_revert_viewQuery`,
+          isLast,
+          collectionIdMap
+        )
+      );
+      if (!isLast) lines.push("");
+    }
     if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
-      for (const permission of modification.permissionsToUpdate) {
-        operationCount++;
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      if (modification.permissionsToUpdate.length >= 2) {
+        const entries = modification.permissionsToUpdate.map((p) => ({ ruleType: p.ruleType, value: p.oldValue }));
+        lines.push(generateGroupedRuleUpdates(collectionName, entries, "revert_rules", isLast, collectionIdMap));
+      } else {
+        const permission = modification.permissionsToUpdate[0];
         const varName = `collection_${collectionName}_revert_perm_${permission.ruleType}`;
-        const isLast = operationCount === totalOperations;
-        lines.push(
-          generatePermissionUpdate(collectionName, permission.ruleType, permission.oldValue, varName, isLast, collectionIdMap)
-        );
-        if (!isLast) lines.push("");
+        lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.oldValue, varName, isLast, collectionIdMap));
       }
+      if (!isLast) lines.push("");
     } else if (modification.rulesToUpdate.length > 0) {
-      for (const rule of modification.rulesToUpdate) {
-        operationCount++;
+      operationCount++;
+      const isLast = operationCount === totalOperations;
+      if (modification.rulesToUpdate.length >= 2) {
+        const entries = modification.rulesToUpdate.map((r) => ({ ruleType: r.ruleType, value: r.oldValue }));
+        lines.push(generateGroupedRuleUpdates(collectionName, entries, "revert_rules", isLast, collectionIdMap));
+      } else {
+        const rule = modification.rulesToUpdate[0];
         const varName = `collection_${collectionName}_revert_rule_${rule.ruleType}`;
-        const isLast = operationCount === totalOperations;
         lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.oldValue, varName, isLast, collectionIdMap));
-        if (!isLast) lines.push("");
       }
+      if (!isLast) lines.push("");
     }
     for (let i = 0; i < modification.indexesToRemove.length; i++) {
       operationCount++;
@@ -893,6 +1018,19 @@ function generateUpMigration(diff) {
     lines.push(`  // Modify existing collections`);
     for (const modification of diff.collectionsToModify) {
       const collectionName = modification.collection;
+      if (modification.viewQueryUpdate) {
+        lines.push(`  // Update the view query of ${collectionName}`);
+        lines.push(
+          generateViewQueryUpdate(
+            collectionName,
+            modification.viewQueryUpdate.newValue,
+            void 0,
+            false,
+            collectionIdMap
+          )
+        );
+        lines.push(``);
+      }
       if (modification.fieldsToAdd.length > 0) {
         lines.push(`  // Add fields to ${collectionName}`);
         for (let i = 0; i < modification.fieldsToAdd.length; i++) {
@@ -938,20 +1076,26 @@ function generateUpMigration(diff) {
       }
       if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
         lines.push(`  // Update permissions for ${collectionName}`);
-        for (const permission of modification.permissionsToUpdate) {
+        if (modification.permissionsToUpdate.length >= 2) {
+          const entries = modification.permissionsToUpdate.map((p) => ({ ruleType: p.ruleType, value: p.newValue }));
+          lines.push(generateGroupedRuleUpdates(collectionName, entries, "rules", false, collectionIdMap));
+        } else {
+          const permission = modification.permissionsToUpdate[0];
           const varName = `collection_${collectionName}_perm_${permission.ruleType}`;
-          lines.push(
-            generatePermissionUpdate(collectionName, permission.ruleType, permission.newValue, varName, false, collectionIdMap)
-          );
-          lines.push(``);
+          lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.newValue, varName, false, collectionIdMap));
         }
+        lines.push(``);
       } else if (modification.rulesToUpdate.length > 0) {
         lines.push(`  // Update rules for ${collectionName}`);
-        for (const rule of modification.rulesToUpdate) {
+        if (modification.rulesToUpdate.length >= 2) {
+          const entries = modification.rulesToUpdate.map((r) => ({ ruleType: r.ruleType, value: r.newValue }));
+          lines.push(generateGroupedRuleUpdates(collectionName, entries, "rules", false, collectionIdMap));
+        } else {
+          const rule = modification.rulesToUpdate[0];
           const varName = `collection_${collectionName}_rule_${rule.ruleType}`;
           lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.newValue, varName, false, collectionIdMap));
-          lines.push(``);
         }
+        lines.push(``);
       }
     }
   }
@@ -1023,22 +1167,41 @@ function generateDownMigration(diff) {
     lines.push(`  // Revert modifications`);
     for (const modification of diff.collectionsToModify) {
       const collectionName = modification.collection;
+      if (modification.viewQueryUpdate) {
+        lines.push(`  // Restore the view query of ${collectionName}`);
+        lines.push(
+          generateViewQueryUpdate(
+            collectionName,
+            modification.viewQueryUpdate.oldValue ?? "",
+            `collection_${collectionName}_revert_viewQuery`,
+            false,
+            collectionIdMap
+          )
+        );
+        lines.push(``);
+      }
       if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
         lines.push(`  // Revert permissions for ${collectionName}`);
-        for (const permission of modification.permissionsToUpdate) {
+        if (modification.permissionsToUpdate.length >= 2) {
+          const entries = modification.permissionsToUpdate.map((p) => ({ ruleType: p.ruleType, value: p.oldValue }));
+          lines.push(generateGroupedRuleUpdates(collectionName, entries, "revert_rules", false, collectionIdMap));
+        } else {
+          const permission = modification.permissionsToUpdate[0];
           const varName = `collection_${collectionName}_revert_perm_${permission.ruleType}`;
-          lines.push(
-            generatePermissionUpdate(collectionName, permission.ruleType, permission.oldValue, varName, false, collectionIdMap)
-          );
-          lines.push(``);
+          lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.oldValue, varName, false, collectionIdMap));
         }
+        lines.push(``);
       } else if (modification.rulesToUpdate.length > 0) {
         lines.push(`  // Revert rules for ${collectionName}`);
-        for (const rule of modification.rulesToUpdate) {
+        if (modification.rulesToUpdate.length >= 2) {
+          const entries = modification.rulesToUpdate.map((r) => ({ ruleType: r.ruleType, value: r.oldValue }));
+          lines.push(generateGroupedRuleUpdates(collectionName, entries, "revert_rules", false, collectionIdMap));
+        } else {
+          const rule = modification.rulesToUpdate[0];
           const varName = `collection_${collectionName}_revert_rule_${rule.ruleType}`;
           lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.oldValue, varName, false, collectionIdMap));
-          lines.push(``);
         }
+        lines.push(``);
       }
       if (modification.indexesToRemove.length > 0) {
         lines.push(`  // Restore indexes to ${collectionName}`);
@@ -1368,6 +1531,7 @@ exports.DEFAULT_CONFIG = DEFAULT_CONFIG;
 exports.DEFAULT_TEMPLATE = DEFAULT_TEMPLATE;
 exports.MigrationGenerator = MigrationGenerator;
 exports.createMigrationFileStructure = createMigrationFileStructure;
+exports.formatSqlTemplate = formatSqlTemplate;
 exports.formatValue = formatValue;
 exports.generate = generate;
 exports.generateCollectionCreation = generateCollectionCreation;
@@ -1383,6 +1547,7 @@ exports.generateFieldDeletion = generateFieldDeletion;
 exports.generateFieldModification = generateFieldModification;
 exports.generateFieldsArray = generateFieldsArray;
 exports.generateFindCollectionCode = generateFindCollectionCode;
+exports.generateGroupedRuleUpdates = generateGroupedRuleUpdates;
 exports.generateIndexAddition = generateIndexAddition;
 exports.generateIndexRemoval = generateIndexRemoval;
 exports.generateIndexesArray = generateIndexesArray;
@@ -1394,10 +1559,12 @@ exports.generatePermissionUpdate = generatePermissionUpdate;
 exports.generateRuleUpdate = generateRuleUpdate;
 exports.generateTimestamp = generateTimestamp;
 exports.generateUpMigration = generateUpMigration;
+exports.generateViewQueryUpdate = generateViewQueryUpdate;
 exports.getAuthSystemFields = getAuthSystemFields;
 exports.getAuthSystemIndexes = getAuthSystemIndexes;
 exports.getFieldConstructorName = getFieldConstructorName;
 exports.getSystemFields = getSystemFields;
+exports.getSystemTimestampFields = getSystemTimestampFields;
 exports.incrementTimestamp = incrementTimestamp;
 exports.mergeConfig = mergeConfig;
 exports.resolveMigrationDir = resolveMigrationDir;