73 allow to process prisma schemas programtically (#74)

* feat: allow generate prisma schema from function

* update

* fix: docs

* update

Author: nitzano

.changeset/soft-cooks-camp.md

@@ -0,0 +1,5 @@
+---
+"prisma-openapi": minor
+---
+
+Allow to generate schemas with sdk

README.md

@@ -19,7 +19,6 @@ A Prisma generator that automatically creates OpenAPI specifications from your P
 - [Usage](#usage)
 - [Examples](#examples)
   - [Basic Usage](#basic-usage)
-  - [Custom Configuration](#custom-configuration)
   - [JSDoc Integration](#jsdoc-integration)
   - [Prisma Comments as Descriptions](#prisma-comments-as-descriptions)
 - [Configuration](#configuration)
@@ -34,6 +33,7 @@ A Prisma generator that automatically creates OpenAPI specifications from your P
 - 🧩 **Relationship Support**: Properly maps Prisma relationships to OpenAPI references
 - *️⃣ **Enum Support**: Full support for Prisma enums in your API documentation
 - 📝 **JSDoc Generation**: Create JSDoc comments for your TypeScript types based on the Prisma schema
+- 🧰 **Programmatic Schema Generation**: Generate schema directly from code
 
 ## Setup
 
@@ -150,6 +150,36 @@ components:
         - published
         - author
         - authorId
+
+### Programmatic Schema Generation
+
+Generate a schema programmatically from a Prisma schema string:
+
+```ts
+import { generateOpenApiSchema } from "prisma-openapi";
+
+const schema = `
+datasource db {
+  provider = "postgresql"
+}
+
+model User {
+  id    Int    @id @default(autoincrement())
+  email String @unique
+  name  String?
+}
+`;
+
+const { yaml, json } = await generateOpenApiSchema({
+  prismaSchema: schema,
+  title: "My API",
+  description: "Generated via SDK",
+  generateYaml: true,
+  generateJson: true,
+});
+
+console.log(yaml);
+```
 ```
 
 ### Custom Configuration

package.json

@@ -9,7 +9,16 @@
     "prisma-openapi": "./dist/index.js"
   },
   "type": "module",
-  "exports": "./dist/index.js",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./lib": {
+      "types": "./dist/lib/index.d.ts",
+      "default": "./dist/lib/index.js"
+    }
+  },
   "types": "./dist/index.d.ts",
   "scripts": {
     "clean": "rimraf dist prisma/openapi",

src/lib/index.ts

@@ -0,0 +1,60 @@
+import {getDMMF} from '@prisma/internals';
+import {generateOpenApiSpec} from '../on-generate/generate-open-api-spec.js';
+import {
+	defaultOptions,
+	parseCommaSeparatedList,
+	type PrismaOpenApiOptions,
+} from '../on-generate/generator-options.js';
+
+export type PrismaOpenApiSchemaOptions = Partial<PrismaOpenApiOptions>;
+
+export async function generateOpenApiSchema(
+	prismaSchema: string,
+	options: PrismaOpenApiSchemaOptions = {},
+): Promise<string> {
+	if (!prismaSchema.trim()) {
+		throw new Error('Prisma schema must be a non-empty string.');
+	}
+
+	const dmmf = await getDMMF({datamodel: prismaSchema});
+	const prismaOpenApiOptions: PrismaOpenApiOptions = {
+		...defaultOptions,
+		...options,
+	};
+
+	let filteredModels = [...dmmf.datamodel.models];
+	const includeModelsList = parseCommaSeparatedList(
+		prismaOpenApiOptions.includeModels,
+	);
+	const excludeModelsList = parseCommaSeparatedList(
+		prismaOpenApiOptions.excludeModels,
+	);
+
+	if (includeModelsList && includeModelsList.length > 0) {
+		filteredModels = filteredModels.filter((model) =>
+			includeModelsList.includes(model.name),
+		);
+	}
+
+	if (excludeModelsList && excludeModelsList.length > 0) {
+		filteredModels = filteredModels.filter(
+			(model) => !excludeModelsList.includes(model.name),
+		);
+	}
+
+	const openApiBuilder = generateOpenApiSpec(
+		filteredModels,
+		dmmf.datamodel.enums,
+		prismaOpenApiOptions,
+	);
+
+	if (prismaOpenApiOptions.generateJson && !prismaOpenApiOptions.generateYaml) {
+		return openApiBuilder.getSpecAsJson();
+	}
+
+	if (prismaOpenApiOptions.generateYaml) {
+		return openApiBuilder.getSpecAsYaml();
+	}
+
+	return openApiBuilder.getSpecAsJson();
+}