Allow multiple entry points

dldc-packages · Feb 8, 2025 · 23a62b4 · 23a62b4
1 parent 1ff8a63
commit 23a62b4
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 11 deletions.
diff --git a/src/server/engine.ts b/src/server/engine.ts
@@ -24,7 +24,7 @@ export interface TEngineOptions {
   graph: TGraphBaseAny;
   resolvers: TResolver[];
   operators?: TPrepareFromOperator[];
-  entry: string;
+  entry: string | string[]; // TODO: use graph object instead ?
 }
 
 export function createEngine(
@@ -35,6 +35,10 @@ export function createEngine(
     entry,
   }: TEngineOptions,
 ): TEngine {
+  const entries = Array.isArray(entry) ? entry : [entry];
+  if (entries.length === 0) {
+    throw new Error("You must provide at least one entry point");
+  }
   const rootStructure = graph[ROOT];
   const { getResolvers } = validateResolvers(rootStructure, resolvers);
   const operators: TPrepareFromOperator[] = [
@@ -58,7 +62,7 @@ export function createEngine(
     let variableCount = 0;
     const mid = prepare(
       {
-        entry,
+        entries,
         rootGraph: graph,
         rootStructure,
         operators,

diff --git a/src/server/erreur.ts b/src/server/erreur.ts
@@ -20,7 +20,7 @@ export type TGraphClientErreurData = {
 } | {
   kind: "InvalidEntry";
   graph: TGraphBaseAny;
-  entry: string;
+  entries: string[];
   requested: string;
 } | {
   kind: "InvalidUnionTypeQuery";
@@ -60,16 +60,20 @@ export function createInvalidQuery(
 
 export function createInvalidEntry(
   graph: TGraphBaseAny,
-  entry: string,
+  entries: string[],
   requested: string,
 ): Error {
+  const entryMessage = entries.length === 1
+    ? printValue(entries[0])
+    : `one of ${printValue(entries)}`;
+
   return GraphClientErreurPrivate.setAndReturn(
     new Error(
-      `Invalid entry, all queries should start from ${
-        printValue(entry)
-      } (requested: ${printValue(requested)})`,
+      `Invalid entry, all queries should start from ${entryMessage} (requested: ${
+        printValue(requested)
+      })`,
     ),
-    { graph, kind: "InvalidEntry", entry, requested },
+    { graph, kind: "InvalidEntry", entries, requested },
   );
 }
 

diff --git a/src/server/prepare.ts b/src/server/prepare.ts
@@ -31,7 +31,7 @@ export type TPrepareFromOperator = (
 ) => TMiddleware | null;
 
 export interface TPrepareContext {
-  entry: string;
+  entries: string[];
   rootGraph: TGraphBaseAny;
   rootStructure: TRootStructure;
   operators: TPrepareFromOperator[];
@@ -97,8 +97,8 @@ const PREPARE_BY_STRUCTURE: TByStructureKind = {
     if (typeof queryItem !== "string") {
       return null;
     }
-    if (queryItem !== context.entry) {
-      throw createInvalidEntry(graph, context.entry, queryItem);
+    if (!context.entries.includes(queryItem)) {
+      throw createInvalidEntry(graph, context.entries, queryItem);
     }
     return prepareObjectLike(context, graph, query, identity);
   },