pocketbase-zod-schema 0.2.5 → 0.3.1

package/dist/migration/generator.js

@@ -110,6 +110,53 @@ function generateTimestamp(config) {
   }
   return Math.floor(Date.now() / 1e3).toString();
 }
+function splitDiffByCollection(diff, baseTimestamp) {
+  const operations = [];
+  let currentTimestamp = parseInt(baseTimestamp, 10);
+  for (const collection of diff.collectionsToCreate) {
+    operations.push({
+      type: "create",
+      collection,
+      timestamp: currentTimestamp.toString()
+    });
+    currentTimestamp += 1;
+  }
+  for (const modification of diff.collectionsToModify) {
+    operations.push({
+      type: "modify",
+      collection: modification.collection,
+      modifications: modification,
+      timestamp: currentTimestamp.toString()
+    });
+    currentTimestamp += 1;
+  }
+  for (const collection of diff.collectionsToDelete) {
+    operations.push({
+      type: "delete",
+      collection: collection.name || collection,
+      // Handle both object and string
+      timestamp: currentTimestamp.toString()
+    });
+    currentTimestamp += 1;
+  }
+  return operations;
+}
+function generateCollectionMigrationFilename(operation) {
+  const timestamp = operation.timestamp;
+  const operationType = operation.type === "modify" ? "updated" : operation.type === "create" ? "created" : "deleted";
+  let collectionName;
+  if (typeof operation.collection === "string") {
+    collectionName = operation.collection;
+  } else {
+    collectionName = operation.collection.name;
+  }
+  const sanitizedName = collectionName.replace(/[^a-zA-Z0-9_]/g, "_").toLowerCase();
+  return `${timestamp}_${operationType}_${sanitizedName}.js`;
+}
+function incrementTimestamp(timestamp) {
+  const currentTimestamp = parseInt(timestamp, 10);
+  return (currentTimestamp + 1).toString();
+}
 function generateMigrationDescription(diff) {
   const parts = [];
   if (diff.collectionsToCreate.length > 0) {
@@ -217,14 +264,13 @@ function formatValue(value) {
     return "null";
   }
   if (typeof value === "string") {
-    return `"${value.replace(/\\/g, "\\\\").replace(/"/g, '\\"').replace(/\n/g, "\\n")}"`;
+    return JSON.stringify(value);
   }
   if (typeof value === "number" || typeof value === "boolean") {
     return String(value);
   }
   if (Array.isArray(value)) {
-    const items = value.map((v) => formatValue(v)).join(", ");
-    return `[${items}]`;
+    return JSON.stringify(value).replace(/","/g, '", "');
   }
   if (typeof value === "object") {
     const entries = Object.entries(value).map(([k, v]) => `${k}: ${formatValue(v)}`).join(", ");
@@ -232,7 +278,7 @@ function formatValue(value) {
   }
   return String(value);
 }
-function generateFieldDefinitionObject(field) {
+function generateFieldDefinitionObject(field, collectionIdMap) {
   const parts = [];
   parts.push(`      name: "${field.name}"`);
   parts.push(`      type: "${field.type}"`);
@@ -240,34 +286,47 @@ function generateFieldDefinitionObject(field) {
   if (field.unique !== void 0) {
     parts.push(`      unique: ${field.unique}`);
   }
+  if (field.type === "select") {
+    const maxSelect = field.options?.maxSelect ?? 1;
+    parts.push(`      maxSelect: ${maxSelect}`);
+    const values = field.options?.values ?? [];
+    parts.push(`      values: ${formatValue(values)}`);
+  }
   if (field.options && Object.keys(field.options).length > 0) {
     for (const [key, value] of Object.entries(field.options)) {
+      if (field.type === "select" && (key === "maxSelect" || key === "values")) {
+        continue;
+      }
       parts.push(`      ${key}: ${formatValue(value)}`);
     }
   }
   if (field.relation) {
     const isUsersCollection = field.relation.collection.toLowerCase() === "users";
-    const collectionIdPlaceholder = isUsersCollection ? '"_pb_users_auth_"' : `app.findCollectionByNameOrId("${field.relation.collection}").id`;
-    parts.push(`      collectionId: ${collectionIdPlaceholder}`);
-    if (field.relation.maxSelect !== void 0) {
-      parts.push(`      maxSelect: ${field.relation.maxSelect}`);
-    }
-    if (field.relation.minSelect !== void 0) {
-      parts.push(`      minSelect: ${field.relation.minSelect}`);
-    }
-    if (field.relation.cascadeDelete !== void 0) {
-      parts.push(`      cascadeDelete: ${field.relation.cascadeDelete}`);
+    let collectionIdValue;
+    if (isUsersCollection) {
+      collectionIdValue = '"_pb_users_auth_"';
+    } else if (collectionIdMap && collectionIdMap.has(field.relation.collection)) {
+      collectionIdValue = `"${collectionIdMap.get(field.relation.collection)}"`;
+    } else {
+      collectionIdValue = `app.findCollectionByNameOrId("${field.relation.collection}").id`;
     }
+    parts.push(`      collectionId: ${collectionIdValue}`);
+    const maxSelect = field.relation.maxSelect ?? 1;
+    parts.push(`      maxSelect: ${maxSelect}`);
+    const minSelect = field.relation.minSelect ?? null;
+    parts.push(`      minSelect: ${minSelect}`);
+    const cascadeDelete = field.relation.cascadeDelete ?? false;
+    parts.push(`      cascadeDelete: ${cascadeDelete}`);
   }
   return `    {
 ${parts.join(",\n")},
     }`;
 }
-function generateFieldsArray(fields) {
+function generateFieldsArray(fields, collectionIdMap) {
   if (fields.length === 0) {
     return "[]";
   }
-  const fieldObjects = fields.map((field) => generateFieldDefinitionObject(field));
+  const fieldObjects = fields.map((field) => generateFieldDefinitionObject(field, collectionIdMap));
   return `[
 ${fieldObjects.join(",\n")},
   ]`;
@@ -326,7 +385,7 @@ function generateIndexesArray(indexes) {
   if (!indexes || indexes.length === 0) {
     return "[]";
   }
-  const indexStrings = indexes.map((idx) => `"${idx}"`);
+  const indexStrings = indexes.map((idx) => JSON.stringify(idx));
   return `[
     ${indexStrings.join(",\n    ")},
   ]`;
@@ -380,9 +439,12 @@ function getSystemFields() {
     }
   ];
 }
-function generateCollectionCreation(collection, varName = "collection", isLast = false) {
+function generateCollectionCreation(collection, varName = "collection", isLast = false, collectionIdMap) {
   const lines = [];
   lines.push(`  const ${varName} = new Collection({`);
+  if (collection.id) {
+    lines.push(`    id: ${formatValue(collection.id)},`);
+  }
   lines.push(`    name: "${collection.name}",`);
   lines.push(`    type: "${collection.type}",`);
   const permissionsCode = generateCollectionPermissions(collection.permissions);
@@ -394,7 +456,7 @@ function generateCollectionCreation(collection, varName = "collection", isLast =
   }
   const systemFields = getSystemFields();
   const allFields = [...systemFields, ...collection.fields];
-  lines.push(`    fields: ${generateFieldsArray(allFields)},`);
+  lines.push(`    fields: ${generateFieldsArray(allFields, collectionIdMap)},`);
   lines.push(`    indexes: ${generateIndexesArray(collection.indexes)},`);
   lines.push(`  });`);
   lines.push(``);
@@ -416,42 +478,59 @@ function getFieldConstructorName(fieldType) {
   };
   return constructorMap[fieldType] || "TextField";
 }
-function generateFieldConstructorOptions(field) {
+function generateFieldConstructorOptions(field, collectionIdMap) {
   const parts = [];
   parts.push(`    name: "${field.name}"`);
   parts.push(`    required: ${field.required}`);
   if (field.unique !== void 0) {
     parts.push(`    unique: ${field.unique}`);
   }
+  if (field.type === "select") {
+    const maxSelect = field.options?.maxSelect ?? 1;
+    parts.push(`    maxSelect: ${maxSelect}`);
+    const values = field.options?.values ?? [];
+    parts.push(`    values: ${formatValue(values)}`);
+  }
   if (field.options && Object.keys(field.options).length > 0) {
     for (const [key, value] of Object.entries(field.options)) {
-      parts.push(`    ${key}: ${formatValue(value)}`);
+      if (field.type === "select" && (key === "maxSelect" || key === "values")) {
+        continue;
+      }
+      if (field.type === "number" && key === "noDecimal") {
+        parts.push(`    onlyInt: ${formatValue(value)}`);
+      } else {
+        parts.push(`    ${key}: ${formatValue(value)}`);
+      }
     }
   }
   if (field.relation && field.type === "relation") {
     const isUsersCollection = field.relation.collection.toLowerCase() === "users";
-    const collectionIdPlaceholder = isUsersCollection ? '"_pb_users_auth_"' : `app.findCollectionByNameOrId("${field.relation.collection}").id`;
-    parts.push(`    collectionId: ${collectionIdPlaceholder}`);
-    if (field.relation.maxSelect !== void 0) {
-      parts.push(`    maxSelect: ${field.relation.maxSelect}`);
-    }
-    if (field.relation.minSelect !== void 0) {
-      parts.push(`    minSelect: ${field.relation.minSelect}`);
-    }
-    if (field.relation.cascadeDelete !== void 0) {
-      parts.push(`    cascadeDelete: ${field.relation.cascadeDelete}`);
+    let collectionIdValue;
+    if (isUsersCollection) {
+      collectionIdValue = '"_pb_users_auth_"';
+    } else if (collectionIdMap && collectionIdMap.has(field.relation.collection)) {
+      collectionIdValue = `"${collectionIdMap.get(field.relation.collection)}"`;
+    } else {
+      collectionIdValue = `app.findCollectionByNameOrId("${field.relation.collection}").id`;
     }
+    parts.push(`    collectionId: ${collectionIdValue}`);
+    const maxSelect = field.relation.maxSelect ?? 1;
+    parts.push(`    maxSelect: ${maxSelect}`);
+    const minSelect = field.relation.minSelect ?? null;
+    parts.push(`    minSelect: ${minSelect}`);
+    const cascadeDelete = field.relation.cascadeDelete ?? false;
+    parts.push(`    cascadeDelete: ${cascadeDelete}`);
   }
   return parts.join(",\n");
 }
-function generateFieldAddition(collectionName, field, varName, isLast = false) {
+function generateFieldAddition(collectionName, field, varName, isLast = false, collectionIdMap) {
   const lines = [];
   const constructorName = getFieldConstructorName(field.type);
   const collectionVar = varName || `collection_${collectionName}_${field.name}`;
   lines.push(`  const ${collectionVar} = app.findCollectionByNameOrId("${collectionName}");`);
   lines.push(``);
   lines.push(`  ${collectionVar}.fields.add(new ${constructorName}({`);
-  lines.push(generateFieldConstructorOptions(field));
+  lines.push(generateFieldConstructorOptions(field, collectionIdMap));
   lines.push(`  }));`);
   lines.push(``);
   lines.push(isLast ? `  return app.save(${collectionVar});` : `  app.save(${collectionVar});`);
@@ -501,7 +580,7 @@ function generateIndexAddition(collectionName, index, varName, isLast = false) {
   const lines = [];
   const collectionVar = varName || `collection_${collectionName}_idx`;
   lines.push(`  const ${collectionVar} = app.findCollectionByNameOrId("${collectionName}");`);
-  lines.push(`  ${collectionVar}.indexes.push("${index}");`);
+  lines.push(`  ${collectionVar}.indexes.push(${JSON.stringify(index)});`);
   lines.push(isLast ? `  return app.save(${collectionVar});` : `  app.save(${collectionVar});`);
   return lines.join("\n");
 }
@@ -510,7 +589,7 @@ function generateIndexRemoval(collectionName, index, varName, isLast = false) {
   const collectionVar = varName || `collection_${collectionName}_idx`;
   const indexVar = `${collectionVar}_indexToRemove`;
   lines.push(`  const ${collectionVar} = app.findCollectionByNameOrId("${collectionName}");`);
-  lines.push(`  const ${indexVar} = ${collectionVar}.indexes.findIndex(idx => idx === "${index}");`);
+  lines.push(`  const ${indexVar} = ${collectionVar}.indexes.findIndex(idx => idx === ${JSON.stringify(index)});`);
   lines.push(`  if (${indexVar} !== -1) {`);
   lines.push(`    ${collectionVar}.indexes.splice(${indexVar}, 1);`);
   lines.push(`  }`);
@@ -539,16 +618,213 @@ function generateCollectionDeletion(collectionName, varName = "collection", isLa
   lines.push(isLast ? `  return app.delete(${varName});` : `  app.delete(${varName});`);
   return lines.join("\n");
 }
+function generateOperationUpMigration(operation, collectionIdMap) {
+  const lines = [];
+  if (operation.type === "create") {
+    const collection = operation.collection;
+    const varName = `collection_${collection.name}`;
+    lines.push(generateCollectionCreation(collection, varName, true, collectionIdMap));
+  } else if (operation.type === "modify") {
+    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;
+    for (const field of modification.fieldsToAdd) {
+      operationCount++;
+      const varName = `collection_${collectionName}_add_${field.name}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldAddition(collectionName, field, varName, isLast, collectionIdMap));
+      if (!isLast) lines.push("");
+    }
+    for (const fieldMod of modification.fieldsToModify) {
+      operationCount++;
+      const varName = `collection_${collectionName}_modify_${fieldMod.fieldName}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldModification(collectionName, fieldMod, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (const field of modification.fieldsToRemove) {
+      operationCount++;
+      const varName = `collection_${collectionName}_remove_${field.name}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldDeletion(collectionName, field.name, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (let i = 0; i < modification.indexesToAdd.length; i++) {
+      operationCount++;
+      const index = modification.indexesToAdd[i];
+      const varName = `collection_${collectionName}_addidx_${i}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateIndexAddition(collectionName, index, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (let i = 0; i < modification.indexesToRemove.length; i++) {
+      operationCount++;
+      const index = modification.indexesToRemove[i];
+      const varName = `collection_${collectionName}_rmidx_${i}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateIndexRemoval(collectionName, index, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
+      for (const permission of modification.permissionsToUpdate) {
+        operationCount++;
+        const varName = `collection_${collectionName}_perm_${permission.ruleType}`;
+        const isLast = operationCount === totalOperations;
+        lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.newValue, varName, isLast));
+        if (!isLast) lines.push("");
+      }
+    } else if (modification.rulesToUpdate.length > 0) {
+      for (const rule of modification.rulesToUpdate) {
+        operationCount++;
+        const varName = `collection_${collectionName}_rule_${rule.ruleType}`;
+        const isLast = operationCount === totalOperations;
+        lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.newValue, varName, isLast));
+        if (!isLast) lines.push("");
+      }
+    }
+  } else if (operation.type === "delete") {
+    const collectionName = typeof operation.collection === "string" ? operation.collection : operation.collection.name;
+    const varName = `collection_${collectionName}`;
+    lines.push(generateCollectionDeletion(collectionName, varName, true));
+  }
+  let code = lines.join("\n");
+  const savePattern = /^(\s*)app\.save\((\w+)\);$/gm;
+  const deletePattern = /^(\s*)app\.delete\((\w+)\);$/gm;
+  const saveMatches = [...code.matchAll(savePattern)];
+  const deleteMatches = [...code.matchAll(deletePattern)];
+  const allMatches = [
+    ...saveMatches.map((m) => ({ match: m, type: "save", index: m.index })),
+    ...deleteMatches.map((m) => ({ match: m, type: "delete", index: m.index }))
+  ].sort((a, b) => b.index - a.index);
+  if (allMatches.length > 0) {
+    const lastMatch = allMatches[0];
+    if (lastMatch.type === "save") {
+      code = code.substring(0, lastMatch.match.index) + lastMatch.match[1] + "return app.save(" + lastMatch.match[2] + ");" + code.substring(lastMatch.match.index + lastMatch.match[0].length);
+    } else {
+      code = code.substring(0, lastMatch.match.index) + lastMatch.match[1] + "return app.delete(" + lastMatch.match[2] + ");" + code.substring(lastMatch.match.index + lastMatch.match[0].length);
+    }
+  }
+  return code;
+}
+function generateOperationDownMigration(operation, collectionIdMap) {
+  const lines = [];
+  if (operation.type === "create") {
+    const collection = operation.collection;
+    const varName = `collection_${collection.name}`;
+    lines.push(generateCollectionDeletion(collection.name, varName, true));
+  } else if (operation.type === "modify") {
+    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;
+    if (modification.permissionsToUpdate && modification.permissionsToUpdate.length > 0) {
+      for (const permission of modification.permissionsToUpdate) {
+        operationCount++;
+        const varName = `collection_${collectionName}_revert_perm_${permission.ruleType}`;
+        const isLast = operationCount === totalOperations;
+        lines.push(generatePermissionUpdate(collectionName, permission.ruleType, permission.oldValue, varName, isLast));
+        if (!isLast) lines.push("");
+      }
+    } else if (modification.rulesToUpdate.length > 0) {
+      for (const rule of modification.rulesToUpdate) {
+        operationCount++;
+        const varName = `collection_${collectionName}_revert_rule_${rule.ruleType}`;
+        const isLast = operationCount === totalOperations;
+        lines.push(generateRuleUpdate(collectionName, rule.ruleType, rule.oldValue, varName, isLast));
+        if (!isLast) lines.push("");
+      }
+    }
+    for (let i = 0; i < modification.indexesToRemove.length; i++) {
+      operationCount++;
+      const index = modification.indexesToRemove[i];
+      const varName = `collection_${collectionName}_restore_idx_${i}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateIndexAddition(collectionName, index, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (let i = 0; i < modification.indexesToAdd.length; i++) {
+      operationCount++;
+      const index = modification.indexesToAdd[i];
+      const varName = `collection_${collectionName}_revert_idx_${i}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateIndexRemoval(collectionName, index, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (const field of modification.fieldsToRemove) {
+      operationCount++;
+      const varName = `collection_${collectionName}_restore_${field.name}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldAddition(collectionName, field, varName, isLast, collectionIdMap));
+      if (!isLast) lines.push("");
+    }
+    for (const fieldMod of modification.fieldsToModify) {
+      operationCount++;
+      const reverseChanges = fieldMod.changes.map((change) => ({
+        property: change.property,
+        oldValue: change.newValue,
+        newValue: change.oldValue
+      }));
+      const reverseMod = {
+        fieldName: fieldMod.fieldName,
+        currentDefinition: fieldMod.newDefinition,
+        newDefinition: fieldMod.currentDefinition,
+        changes: reverseChanges
+      };
+      const varName = `collection_${collectionName}_revert_${fieldMod.fieldName}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldModification(collectionName, reverseMod, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+    for (const field of modification.fieldsToAdd) {
+      operationCount++;
+      const varName = `collection_${collectionName}_revert_add_${field.name}`;
+      const isLast = operationCount === totalOperations;
+      lines.push(generateFieldDeletion(collectionName, field.name, varName, isLast));
+      if (!isLast) lines.push("");
+    }
+  } else if (operation.type === "delete") {
+    const collection = operation.collection;
+    if (typeof collection !== "string") {
+      const varName = `collection_${collection.name}`;
+      lines.push(generateCollectionCreation(collection, varName, true, collectionIdMap));
+    }
+  }
+  let code = lines.join("\n");
+  const savePattern = /^(\s*)app\.save\((\w+)\);$/gm;
+  const deletePattern = /^(\s*)app\.delete\((\w+)\);$/gm;
+  const saveMatches = [...code.matchAll(savePattern)];
+  const deleteMatches = [...code.matchAll(deletePattern)];
+  const allMatches = [
+    ...saveMatches.map((m) => ({ match: m, type: "save", index: m.index })),
+    ...deleteMatches.map((m) => ({ match: m, type: "delete", index: m.index }))
+  ].sort((a, b) => b.index - a.index);
+  if (allMatches.length > 0) {
+    const lastMatch = allMatches[0];
+    if (lastMatch.type === "save") {
+      code = code.substring(0, lastMatch.match.index) + lastMatch.match[1] + "return app.save(" + lastMatch.match[2] + ");" + code.substring(lastMatch.match.index + lastMatch.match[0].length);
+    } else {
+      code = code.substring(0, lastMatch.match.index) + lastMatch.match[1] + "return app.delete(" + lastMatch.match[2] + ");" + code.substring(lastMatch.match.index + lastMatch.match[0].length);
+    }
+  }
+  return code;
+}
 function generateUpMigration(diff) {
   const lines = [];
   lines.push(`  // UP MIGRATION`);
   lines.push(``);
+  const collectionIdMap = /* @__PURE__ */ new Map();
+  for (const collection of diff.collectionsToCreate) {
+    if (collection.id) {
+      collectionIdMap.set(collection.name, collection.id);
+    }
+  }
   if (diff.collectionsToCreate.length > 0) {
     lines.push(`  // Create new collections`);
     for (let i = 0; i < diff.collectionsToCreate.length; i++) {
       const collection = diff.collectionsToCreate[i];
       const varName = `collection_${collection.name}_create`;
-      lines.push(generateCollectionCreation(collection, varName));
+      lines.push(generateCollectionCreation(collection, varName, false, collectionIdMap));
       lines.push(``);
     }
   }
@@ -560,7 +836,7 @@ function generateUpMigration(diff) {
         lines.push(`  // Add fields to ${collectionName}`);
         for (const field of modification.fieldsToAdd) {
           const varName = `collection_${collectionName}_add_${field.name}`;
-          lines.push(generateFieldAddition(collectionName, field, varName));
+          lines.push(generateFieldAddition(collectionName, field, varName, false, collectionIdMap));
           lines.push(``);
         }
       }
@@ -651,12 +927,23 @@ function generateDownMigration(diff) {
   const lines = [];
   lines.push(`  // DOWN MIGRATION (ROLLBACK)`);
   lines.push(``);
+  const collectionIdMap = /* @__PURE__ */ new Map();
+  for (const collection of diff.collectionsToCreate) {
+    if (collection.id) {
+      collectionIdMap.set(collection.name, collection.id);
+    }
+  }
+  for (const collection of diff.collectionsToDelete) {
+    if (collection.id) {
+      collectionIdMap.set(collection.name, collection.id);
+    }
+  }
   if (diff.collectionsToDelete.length > 0) {
     lines.push(`  // Recreate deleted collections`);
     for (let i = 0; i < diff.collectionsToDelete.length; i++) {
       const collection = diff.collectionsToDelete[i];
       const varName = `collection_${collection.name}_recreate`;
-      lines.push(generateCollectionCreation(collection, varName));
+      lines.push(generateCollectionCreation(collection, varName, false, collectionIdMap));
       lines.push(``);
     }
   }
@@ -701,7 +988,7 @@ function generateDownMigration(diff) {
         lines.push(`  // Restore fields to ${collectionName}`);
         for (const field of modification.fieldsToRemove) {
           const varName = `collection_${collectionName}_restore_${field.name}`;
-          lines.push(generateFieldAddition(collectionName, field, varName));
+          lines.push(generateFieldAddition(collectionName, field, varName, false, collectionIdMap));
           lines.push(``);
         }
       }
@@ -770,12 +1057,33 @@ function generate(diff, config) {
   const normalizedConfig = typeof config === "string" ? { migrationDir: config } : config;
   try {
     const migrationDir = resolveMigrationDir(normalizedConfig);
-    const upCode = generateUpMigration(diff);
-    const downCode = generateDownMigration(diff);
-    const content = createMigrationFileStructure(upCode, downCode, normalizedConfig);
-    const filename = generateMigrationFilename(diff, normalizedConfig);
-    const filePath = writeMigrationFile(migrationDir, filename, content);
-    return filePath;
+    const hasChanges = diff.collectionsToCreate.length > 0 || diff.collectionsToModify.length > 0 || diff.collectionsToDelete.length > 0;
+    if (!hasChanges) {
+      return [];
+    }
+    const collectionIdMap = /* @__PURE__ */ new Map();
+    for (const collection of diff.collectionsToCreate) {
+      if (collection.id) {
+        collectionIdMap.set(collection.name, collection.id);
+      }
+    }
+    for (const collection of diff.collectionsToDelete) {
+      if (collection.id) {
+        collectionIdMap.set(collection.name, collection.id);
+      }
+    }
+    const baseTimestamp = generateTimestamp(normalizedConfig);
+    const operations = splitDiffByCollection(diff, baseTimestamp);
+    const filePaths = [];
+    for (const operation of operations) {
+      const upCode = generateOperationUpMigration(operation, collectionIdMap);
+      const downCode = generateOperationDownMigration(operation, collectionIdMap);
+      const content = createMigrationFileStructure(upCode, downCode, normalizedConfig);
+      const filename = generateCollectionMigrationFilename(operation);
+      const filePath = writeMigrationFile(migrationDir, filename, content);
+      filePaths.push(filePath);
+    }
+    return filePaths;
   } catch (error) {
     if (error instanceof MigrationGenerationError || error instanceof FileSystemError) {
       throw error;
@@ -793,7 +1101,8 @@ var MigrationGenerator = class {
     this.config = mergeConfig(config);
   }
   /**
-   * Generates a migration file from a schema diff
+   * Generates migration files from a schema diff
+   * Returns array of file paths (one per collection operation)
    */
   generate(diff) {
     return generate(diff, this.config);
@@ -818,6 +1127,6 @@ var MigrationGenerator = class {
   }
 };
 
-export { MigrationGenerator, createMigrationFileStructure, generate, generateCollectionCreation, generateCollectionPermissions, generateCollectionRules, generateDownMigration, generateFieldAddition, generateFieldDefinitionObject, generateFieldDeletion, generateFieldModification, generateFieldsArray, generateIndexesArray, generateMigrationDescription, generateMigrationFilename, generatePermissionUpdate, generateTimestamp, generateUpMigration, writeMigrationFile };
+export { MigrationGenerator, createMigrationFileStructure, generate, generateCollectionCreation, generateCollectionMigrationFilename, generateCollectionPermissions, generateCollectionRules, generateDownMigration, generateFieldAddition, generateFieldDefinitionObject, generateFieldDeletion, generateFieldModification, generateFieldsArray, generateIndexesArray, generateMigrationDescription, generateMigrationFilename, generateOperationDownMigration, generateOperationUpMigration, generatePermissionUpdate, generateTimestamp, generateUpMigration, incrementTimestamp, splitDiffByCollection, writeMigrationFile };
 //# sourceMappingURL=generator.js.map
 //# sourceMappingURL=generator.js.map