fix: datasource and generator blocks generation

src/schemaGenerator.ts

@@ -51,7 +51,7 @@ export async function generateIfNoSchema(
       recursive: true,
     });
     // fs.writeFileSync(prismaFilePath, result);
-    await formatValidateAndWrite(schemaString, prismaFilePath);
+    await formatValidateAndWrite(result, prismaFilePath);
     console.log("🚀 Prisma schema generated successfully!");
     migrateModels.push(...jsonData.schema.map((model) => model.schemaName));
   } catch (err) {

src/utils/utils.ts

@@ -194,8 +194,8 @@ export async function formatValidateAndWrite(
     if (fs.existsSync(filePath + ".bak")) fs.unlinkSync(filePath + ".bak");
   } catch (err) {
     console.log("Error while running prisma format and validate: ", err);
-    fs.unlinkSync(filePath);
     if (fileExists) {
+      fs.unlinkSync(filePath);
       fs.renameSync(filePath + ".bak", filePath);
     }
     throw new Error(

test/schema-generation.spec.ts

@@ -32,6 +32,24 @@ describe("tests for schema generation", () => {
       expect(models).toContain("Comment");
     });
   });
+
+  it("should generate a schema.prisma with dummy_id for a schema with no id and only unique field being optional", async () => {
+    const fileContent = readJsonFile("./test/schemas/optional_unique.json");
+    await generatePrismaSchemaFile(fileContent).then((migrationModels) => {
+      expect(migrationModels).toBeDefined();
+      expect(fs.existsSync("./prisma/schema.prisma")).toBe(true);
+      const schemaPrismaContent = fs.readFileSync(
+        "./prisma/schema.prisma",
+        "utf8"
+      );
+      expect(schemaPrismaContent).toBeDefined();
+      const models = parsePrismaSchemaModels(schemaPrismaContent);
+      expect(models).toContain("User");
+      expect(models).toContain("Post");
+      expect(models).toContain("Comment");
+    });
+  });
+
   it("should generate a schema.prisma for a basic file with different casing for types ", async () => {
     const fileContent = readJsonFile("./test/schemas/casing.json");
     await generatePrismaSchemaFile(fileContent).then((migrationModels) => {

test/schemas/optional_unique.json

@@ -0,0 +1,45 @@
+{
+  "schema": [
+    {
+      "schemaName": "questions01",
+      "fields": [
+        {
+          "fieldName": "question",
+          "type": "String",
+          "description": "",
+          "maxLength": 10,
+          "isNullable": true,
+          "isUnique": false,
+          "isVectorEmbed": true,
+          "isUuid": true,
+          "embeddingAlgo": "text-embedding-ada-002"
+        },
+        {
+          "fieldName": "answer",
+          "type": "String",
+          "description": "",
+          "maxLength": "",
+          "isNullable": true,
+          "isId": false,
+          "isUuid": false,
+          "isUnique": true,
+          "isVectorEmbed": false
+        }
+      ],
+      "description": "schema for questions and answers"
+    }
+  ],
+  "dataSource": {
+    "name": "db",
+    "provider": "postgresql",
+    "url": {
+      "fromEnv": "DATABASE_URL"
+    },
+    "extensions": ["vector", "pg_trgm"]
+  },
+  "generator": {
+    "provider": "prisma-client-js",
+    "generatorName": "client",
+    "previewFeatures": ["postgresqlExtensions", "views"]
+  }
+}