@trycompai/db 1.3.22 → 2.0.1

package/dist/client.d.ts

@@ -1,3 +1,3 @@
 import { PrismaClient } from '@prisma/client';
-export declare const db: PrismaClient<import("@prisma/client").Prisma.PrismaClientOptions, never, import("@prisma/client/runtime/library").DefaultArgs>;
+export declare const db: PrismaClient<import("@prisma/client").Prisma.PrismaClientOptions, never, import("@prisma/client/runtime/client").DefaultArgs>;
 //# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAI9C,eAAO,MAAM,EAAE,gIAIX,CAAC"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AA6B9C,eAAO,MAAM,EAAE,+HAAiD,CAAC"}

package/dist/client.js

@@ -2,10 +2,30 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.db = void 0;
 const client_1 = require("@prisma/client");
+const adapter_pg_1 = require("@prisma/adapter-pg");
 const globalForPrisma = global;
-exports.db = globalForPrisma.prisma ||
-    new client_1.PrismaClient({
-        datasourceUrl: process.env.DATABASE_URL,
+function stripSslMode(connectionString) {
+    const url = new URL(connectionString);
+    url.searchParams.delete('sslmode');
+    return url.toString();
+}
+function createPrismaClient() {
+    const rawUrl = process.env.DATABASE_URL;
+    const isLocalhost = /localhost|127\.0\.0\.1|::1/.test(rawUrl);
+    // Use verified SSL when NODE_EXTRA_CA_CERTS is set (Docker with RDS CA bundle),
+    // otherwise fall back to unverified SSL (Trigger.dev, Vercel, other environments).
+    const hasCABundle = !!process.env.NODE_EXTRA_CA_CERTS;
+    const ssl = isLocalhost ? undefined : hasCABundle ? true : { rejectUnauthorized: false };
+    // Strip sslmode from the connection string to avoid conflicts with the explicit ssl option
+    const url = ssl !== undefined ? stripSslMode(rawUrl) : rawUrl;
+    const adapter = new adapter_pg_1.PrismaPg({ connectionString: url, ssl });
+    return new client_1.PrismaClient({
+        adapter,
+        transactionOptions: {
+            timeout: 60000,
+        },
     });
+}
+exports.db = globalForPrisma.prisma || createPrismaClient();
 if (process.env.NODE_ENV !== 'production')
     globalForPrisma.prisma = exports.db;

package/dist/index.d.ts

@@ -1,3 +1,2 @@
 export * from '@prisma/client';
-export { db } from './client';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC;AAC/B,OAAO,EAAE,EAAE,EAAE,MAAM,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC"}

package/dist/index.js

@@ -14,7 +14,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.db = void 0;
 __exportStar(require("@prisma/client"), exports);
-var client_1 = require("./client");
-Object.defineProperty(exports, "db", { enumerable: true, get: function () { return client_1.db; } });

package/dist/postinstall.js

@@ -22,7 +22,7 @@ function generatePrismaClient(options = {}) {
     (0, node_fs_1.mkdirSync)(schemaDir, { recursive: true });
     (0, node_fs_1.copyFileSync)(resolution.path, schemaDestination);
     log(`Copied schema from ${resolution.path} to ${schemaDestination}`);
-    const clientEntryPoint = (0, node_path_1.resolve)(projectRoot, 'node_modules/.prisma/client/default.js');
+    const clientEntryPoint = (0, node_path_1.resolve)(projectRoot, 'src', 'generated', 'prisma', 'client.ts');
     if (!options.force && (0, node_fs_1.existsSync)(clientEntryPoint)) {
         log('Prisma client already exists. Skipping generation.');
         return { schema: schemaDestination };

package/dist/scripts/backfill-framework-versions.d.ts

@@ -0,0 +1,6 @@
+export interface BackfillResult {
+    versionsCreated: number;
+    instancesBackfilled: number;
+}
+export declare function backfillFrameworkVersions(): Promise<BackfillResult>;
+//# sourceMappingURL=backfill-framework-versions.d.ts.map

package/dist/scripts/backfill-framework-versions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"backfill-framework-versions.d.ts","sourceRoot":"","sources":["../../src/scripts/backfill-framework-versions.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,cAAc;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAkBD,wBAAsB,yBAAyB,IAAI,OAAO,CAAC,cAAc,CAAC,CAiEzE"}

package/dist/scripts/backfill-framework-versions.js

@@ -0,0 +1,142 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.backfillFrameworkVersions = backfillFrameworkVersions;
+const client_1 = require("@prisma/client");
+const client_2 = require("../client");
+async function backfillFrameworkVersions() {
+    const frameworks = await client_2.db.frameworkEditorFramework.findMany({
+        include: {
+            requirements: {
+                include: {
+                    controlTemplates: {
+                        include: {
+                            requirements: true,
+                            policyTemplates: true,
+                            taskTemplates: true,
+                        },
+                    },
+                },
+            },
+        },
+    });
+    let versionsCreated = 0;
+    for (const framework of frameworks) {
+        const manifest = buildManifestFromFramework(framework);
+        try {
+            await client_2.db.frameworkVersion.create({
+                data: {
+                    frameworkId: framework.id,
+                    version: '1.0.0',
+                    manifest: manifest,
+                    releaseNotes: 'Initial version (backfilled).',
+                },
+            });
+            versionsCreated += 1;
+        }
+        catch (err) {
+            if (err instanceof client_1.Prisma.PrismaClientKnownRequestError && err.code === 'P2002') {
+                // Raced with another backfill run — already created. Not counted.
+            }
+            else {
+                throw err;
+            }
+        }
+    }
+    // Backfill instances
+    const versions = await client_2.db.frameworkVersion.findMany({
+        where: { version: '1.0.0' },
+        select: { id: true, frameworkId: true },
+    });
+    const byFrameworkId = new Map(versions.map((v) => [v.frameworkId, v.id]));
+    const toBackfill = await client_2.db.frameworkInstance.findMany({
+        where: { currentVersionId: null, frameworkId: { not: null } },
+        select: { id: true, frameworkId: true },
+    });
+    let instancesBackfilled = 0;
+    for (const inst of toBackfill) {
+        const versionId = byFrameworkId.get(inst.frameworkId);
+        if (!versionId)
+            continue;
+        await client_2.db.frameworkInstance.update({
+            where: { id: inst.id },
+            data: { currentVersionId: versionId },
+        });
+        instancesBackfilled += 1;
+    }
+    return { versionsCreated, instancesBackfilled };
+}
+function buildManifestFromFramework(framework) {
+    // Collect all unique control templates across all requirements
+    const controlTemplateMap = new Map();
+    for (const req of framework.requirements) {
+        for (const ct of req.controlTemplates) {
+            if (!controlTemplateMap.has(ct.id)) {
+                controlTemplateMap.set(ct.id, ct);
+            }
+        }
+    }
+    const controlTemplates = [...controlTemplateMap.values()];
+    // Filter each control's requirementIds down to requirements belonging to
+    // this framework — control templates are shared across frameworks via M:N,
+    // so `ct.requirements` can contain IDs for requirements on other frameworks.
+    const ownRequirementIds = new Set(framework.requirements.map((r) => r.id));
+    return {
+        framework: {
+            id: framework.id,
+            name: framework.name,
+            catalogVersion: framework.version,
+            description: framework.description,
+        },
+        requirements: framework.requirements.map((r) => ({
+            id: r.id,
+            identifier: r.identifier,
+            name: r.name,
+            description: r.description,
+        })),
+        controls: controlTemplates.map((c) => ({
+            id: c.id,
+            name: c.name,
+            description: c.description,
+            requirementIds: c.requirements
+                .map((r) => r.id)
+                .filter((id) => ownRequirementIds.has(id)),
+            policyIds: c.policyTemplates.map((p) => p.id),
+            taskIds: c.taskTemplates.map((t) => t.id),
+            documentTypes: [...c.documentTypes],
+        })),
+        policies: dedupeById(controlTemplates.flatMap((c) => c.policyTemplates.map((p) => ({
+            id: p.id,
+            name: p.name,
+            description: p.description,
+            content: p.content,
+            frequency: p.frequency,
+            department: p.department,
+        })))),
+        tasks: dedupeById(controlTemplates.flatMap((c) => c.taskTemplates.map((t) => ({
+            id: t.id,
+            name: t.name,
+            description: t.description,
+            frequency: t.frequency,
+            department: t.department,
+        })))),
+    };
+}
+function dedupeById(items) {
+    const seen = new Map();
+    for (const item of items) {
+        if (!seen.has(item.id))
+            seen.set(item.id, item);
+    }
+    return [...seen.values()];
+}
+if (require.main === module) {
+    backfillFrameworkVersions()
+        .then((result) => {
+        console.log('Backfill complete:', result);
+        process.exit(0);
+    })
+        .catch((err) => {
+        console.error('Backfill failed:', err);
+        process.exit(1);
+    });
+}

package/package.json

@@ -1,59 +1,62 @@
 {
-  "name": "@trycompai/db",
-  "description": "Database package with Prisma client and schema for Comp AI",
-  "version": "1.3.22",
-  "dependencies": {
-    "@prisma/client": "6.18.0",
-    "dotenv": "^16.4.5",
-    "zod": "^4.1.12"
-  },
-  "devDependencies": {
-    "@types/node": "^24.2.0",
-    "prisma": "6.18.0",
-    "ts-node": "^10.9.2",
-    "typescript": "^5.9.2"
-  },
-  "exports": {
-    ".": {
-      "import": "./dist/index.js",
-      "types": "./src/index.ts",
-      "default": "./dist/index.js"
+    "name": "@trycompai/db",
+    "description": "Database package with Prisma client and schema for Comp AI",
+    "version": "2.0.1",
+    "dependencies": {
+        "@prisma/adapter-pg": "7.6.0",
+        "@prisma/client": "7.6.0",
+        "dotenv": "^16.4.5",
+        "zod": "^4.3.6"
     },
-    "./postinstall": {
-      "import": "./dist/postinstall.js",
-      "types": "./src/postinstall.ts",
-      "default": "./dist/postinstall.js"
+    "devDependencies": {
+        "@types/node": "^24.2.0",
+        "prisma": "7.6.0",
+        "ts-node": "^10.9.2",
+        "typescript": "^5.9.2"
+    },
+    "exports": {
+        ".": {
+            "types": "./src/index.ts",
+            "import": "./dist/index.js",
+            "default": "./dist/index.js"
+        },
+        "./postinstall": {
+            "import": "./dist/postinstall.js",
+            "types": "./src/postinstall.ts",
+            "default": "./dist/postinstall.js"
+        }
+    },
+    "bin": {
+        "comp-prisma-postinstall": "./dist/postinstall.js"
+    },
+    "files": [
+        "dist",
+        "README.md",
+        "INTEGRATION_GUIDE.md"
+    ],
+    "publishConfig": {
+        "access": "public"
+    },
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/trycompai/comp.git",
+        "directory": "packages/db"
+    },
+    "scripts": {
+        "backfill:framework-versions": "bun src/scripts/backfill-framework-versions.ts",
+        "build": "rm -rf dist && node scripts/generate-prisma-client-js.js && tsc",
+        "check-types": "tsc --noEmit",
+        "db:generate": "node scripts/generate-prisma-client-js.js",
+        "db:migrate": "prisma migrate dev",
+        "db:migrate:reset": "prisma migrate reset",
+        "db:push": "prisma db push",
+        "db:seed": "bun prisma/seed/seed.ts",
+        "db:studio": "prisma studio",
+        "docker:clean": "docker compose down -v",
+        "docker:down": "docker compose down",
+        "docker:up": "docker compose up -d",
+        "lint": "prettier --check 'src/**/*.{ts,tsx,js,jsx,json}' 'prisma/**/*.prisma' && tsc --noEmit",
+        "postinstall": "node scripts/generate-prisma-client-js.js || true",
+        "prepublishOnly": "bun run build"
     }
-  },
-  "bin": {
-    "comp-prisma-postinstall": "./dist/postinstall.js"
-  },
-  "files": [
-    "dist",
-    "README.md",
-    "INTEGRATION_GUIDE.md"
-  ],
-  "publishConfig": {
-    "access": "public"
-  },
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/trycompai/comp.git",
-    "directory": "packages/db"
-  },
-  "scripts": {
-    "build": "rm -rf dist && rm -rf ./prisma/generated && node scripts/combine-schemas.js && prisma generate --schema=dist/schema.prisma && tsc",
-    "check-types": "tsc --noEmit",
-    "db:generate": "node scripts/combine-schemas.js && prisma generate --schema=dist/schema.prisma",
-    "db:migrate": "prisma migrate dev",
-    "db:migrate:reset": "prisma migrate reset",
-    "db:push": "prisma db push",
-    "db:seed": "bun prisma/seed/seed.ts",
-    "db:studio": "prisma studio",
-    "docker:clean": "docker compose down -v",
-    "docker:down": "docker compose down",
-    "docker:up": "docker compose up -d",
-    "lint": "prettier --check 'src/**/*.{ts,tsx,js,jsx,json}' 'prisma/**/*.prisma' && tsc --noEmit",
-    "prepublishOnly": "bun run build"
-  }
 }

package/dist/client.ts

@@ -1,4 +0,0 @@
-import { PrismaClient } from '@prisma/client';
-const globalForPrisma = global as unknown as { prisma: PrismaClient };
-export const db = globalForPrisma.prisma || new PrismaClient();
-if (process.env.NODE_ENV !== 'production') globalForPrisma.prisma = db;

package/dist/index.ts

@@ -1,2 +0,0 @@
-export { db } from './client'
-export * from '@prisma/client';