[TS-CODEGEN] context types generation re-work (#480)

* [TS-CODEGEN] context types generation re-work

Author: vejrj

change/@graphitation-cli-66b66740-d897-4e21-9dce-9f452b45a900.json

@@ -0,0 +1,7 @@
+{
+  "type": "minor",
+  "comment": "Context type generation re-worked",
+  "packageName": "@graphitation/cli",
+  "email": "[email protected]",
+  "dependentChangeType": "patch"
+}

change/@graphitation-ts-codegen-c09b79fd-44a4-492e-8b1d-f18202a82cd3.json

@@ -0,0 +1,7 @@
+{
+  "type": "minor",
+  "comment": "Context type generation re-worked",
+  "packageName": "@graphitation/ts-codegen",
+  "email": "[email protected]",
+  "dependentChangeType": "patch"
+}

packages/cli/src/supermassive.ts

@@ -10,15 +10,19 @@ import * as glob from "fast-glob";
 
 type GenerateInterfacesOptions = {
   outputDir?: string;
-  contextImport?: string;
-  contextName?: string;
+  contextTypePath?: string;
+  contextTypeName?: string;
   enumsImport?: string;
   legacy?: boolean;
   legacyModels?: boolean;
   useStringUnionsInsteadOfEnums?: boolean;
   enumMigrationJsonFile?: string;
   enumMigrationExceptionsJsonFile?: string;
   generateOnlyEnums?: boolean;
+  contextSubTypeNameTemplate?: string;
+  contextSubTypePathTemplate?: string;
+  defaultContextSubTypePath?: string;
+  defaultContextSubTypeName?: string;
   scope?: string;
 };
 
@@ -45,10 +49,26 @@ export function supermassive(): Command {
       "output directory relative to file, default generated",
     )
     .option(
-      "-ci, --context-import [contextImport]",
+      "-ci, --context-type-path [contextTypePath]",
       "from where to import context",
     )
-    .option("-cn, --context-name [contextName]", "Context name")
+    .option("-cn, --context-type-name [contextTypeName]", "Context type name")
+    .option(
+      "-dcp, --default-context-sub-type-path [defaultContextSubTypePath]",
+      "From where the default context type will be exported",
+    )
+    .option(
+      "-dcn, --default-context-sub-type-name [defaultContextSubTypeName]",
+      "Default context type which will extend context sub type",
+    )
+    .option(
+      "-cnt, --context-sub-type-name-template [contextSubTypeNameTemplate]",
+      "context resource name template. You need to specify ${resourceName} in the parameter eg. `${resourceName}Context`",
+    )
+    .option(
+      "-cpt, --context-sub-type-path-template [contextSubTypePathTemplate]",
+      "context resource path template. You need to specify ${resourceName} in the parameter eg. `@package/preffix-${resourceName}-suffix`",
+    )
     .option("-ei, --enums-import [enumsImport]", "from where to import enums")
     .option("-l, --legacy", "generate legacy types")
     .option("--legacy-models", "do not use models for object types")
@@ -91,16 +111,19 @@ function getFiles(inputs: Array<string>) {
     .flat()
     .filter(Boolean);
 }
-function getContextPath(outputDir: string, contextImport: string | undefined) {
-  if (!contextImport) {
+function getContextPath(
+  outputDir: string,
+  contextTypePath: string | undefined,
+) {
+  if (!contextTypePath) {
     return;
   }
 
-  if (!contextImport.startsWith(".")) {
-    return contextImport;
+  if (!contextTypePath.startsWith(".")) {
+    return contextTypePath;
   }
 
-  const contextDir = path.join(process.cwd(), contextImport);
+  const contextDir = path.join(process.cwd(), contextTypePath);
 
   return path
     .relative(outputDir, contextDir)
@@ -171,8 +194,16 @@ async function generateInterfaces(
     const result = generateTS(document, {
       outputPath,
       documentPath: fullPath,
-      contextImport: getContextPath(outputPath, options.contextImport) || null,
-      contextName: options.contextName,
+      contextTypePath:
+        getContextPath(outputPath, options.contextTypePath) || null,
+      contextTypeName: options.contextTypeName,
+      contextSubTypeNameTemplate: options.contextSubTypeNameTemplate,
+      contextSubTypePathTemplate: options.contextSubTypePathTemplate,
+      defaultContextSubTypePath: getContextPath(
+        outputPath,
+        options.defaultContextSubTypePath,
+      ),
+      defaultContextSubTypeName: options.defaultContextSubTypeName,
       enumsImport: getContextPath(outputPath, options.enumsImport) || null,
       legacyCompat: !!options.legacy,
       legacyNoModelsForObjects: !!options.legacyModels,
@@ -195,6 +226,16 @@ async function generateInterfaces(
       ),
     );
 
+    if (result.contextMappingOutput) {
+      outputs.push(
+        fs.writeFile(
+          path.join(outputPath, "schema-context-mapping-metadata.json"),
+          JSON.stringify(result.contextMappingOutput, null, 2),
+          { encoding: "utf-8" },
+        ),
+      );
+    }
+
     await Promise.all(outputs);
   }
 }