Add parser option to getSchema

src/getConfig.ts

@@ -4,8 +4,9 @@ import {
   type LilconfigResult as ConfigResultRaw,
 } from 'lilconfig';
 
+export type PrismaAstParserConfig = Pick<IParserConfig, 'nodeLocationTracking'>;
 export interface PrismaAstConfig {
-  parser: Pick<IParserConfig, 'nodeLocationTracking'>;
+  parser: PrismaAstParserConfig;
 }
 
 type ConfigResult<T> = Omit<ConfigResultRaw, 'config'> & {

src/getSchema.ts

@@ -1,7 +1,7 @@
 import { PrismaLexer } from './lexer';
-import { PrismaVisitor } from './visitor';
-import { parser } from './parser';
+import { PrismaVisitor, defaultVisitor } from './visitor';
 import type { CstNodeLocation } from 'chevrotain';
+import { PrismaParser, defaultParser } from './parser';
 
 /**
  * Parses a string containing a prisma schema's source code and returns an
@@ -14,12 +14,21 @@ import type { CstNodeLocation } from 'chevrotain';
  * // ... make changes to schema object ...
  * const changedSource = printSchema(schema)
  * */
-export function getSchema(source: string): Schema {
+export function getSchema(
+  source: string,
+  options?: {
+    parser: PrismaParser;
+    visitor: PrismaVisitor;
+  }
+): Schema {
   const lexingResult = PrismaLexer.tokenize(source);
+
+  const parser = options?.parser ?? defaultParser;
   parser.input = lexingResult.tokens;
   const cstNode = parser.schema();
   if (parser.errors.length > 0) throw parser.errors[0];
-  const visitor = new PrismaVisitor();
+
+  const visitor = options?.visitor ?? defaultVisitor;
   return visitor.visit(cstNode);
 }
 

src/index.ts

@@ -3,3 +3,5 @@ export * from './printSchema';
 export * from './PrismaSchemaBuilder';
 export type { PrismaAstConfig } from './getConfig';
 export type { CstNodeLocation } from 'chevrotain';
+export { VisitorClassFactory } from './visitor';
+export { PrismaParser } from './parser';

src/parser.ts

@@ -1,12 +1,15 @@
 import { CstParser } from 'chevrotain';
-import getConfig from './getConfig';
+import getConfig, { PrismaAstParserConfig } from './getConfig';
 import * as lexer from './lexer';
 
 type ComponentType = 'datasource' | 'generator' | 'model' | 'view' | 'enum';
 export class PrismaParser extends CstParser {
-  constructor() {
-    super(lexer.multiModeTokens, getConfig().parser);
+  readonly config: PrismaAstParserConfig;
+
+  constructor(config: PrismaAstParserConfig) {
+    super(lexer.multiModeTokens, config);
     this.performSelfAnalysis();
+    this.config = config;
   }
 
   private break = this.RULE('break', () => {
@@ -229,4 +232,4 @@ export class PrismaParser extends CstParser {
   });
 }
 
-export const parser = new PrismaParser();
+export const defaultParser = new PrismaParser(getConfig().parser);

src/schemaUtils.ts

@@ -1,5 +1,4 @@
 import type { CstNode, IToken } from 'chevrotain';
-import getConfig from './getConfig';
 import * as schema from './getSchema';
 
 const schemaObjects = ['model', 'view'];
@@ -28,9 +27,6 @@ export function appendLocationData<T extends Record<string, unknown>>(
   data: T,
   ...tokens: IToken[]
 ): T {
-  const { parser } = getConfig();
-  if (parser.nodeLocationTracking === 'none') return data;
-
   const location = tokens.reduce((memo, token) => {
     if (!token) return memo;