Create model filenames for framework mode

extensions/ql-vscode/src/data-extensions-editor/data-extensions-editor-view.ts

@@ -36,7 +36,8 @@ import { decodeBqrsToExternalApiUsages } from "./bqrs";
 import { redactableError } from "../common/errors";
 import { readQueryResults, runQuery } from "./external-api-usage-query";
 import {
-  createDataExtensionYamlsPerLibrary,
+  createDataExtensionYamlsForApplicationMode,
+  createDataExtensionYamlsForFrameworkMode,
   createFilenameForLibrary,
   loadDataExtensionYaml,
 } from "./yaml";
@@ -197,11 +198,26 @@ export class DataExtensionsEditorView extends AbstractWebview<
     externalApiUsages: ExternalApiUsage[],
     modeledMethods: Record<string, ModeledMethod>,
   ): Promise<void> {
-    const yamls = createDataExtensionYamlsPerLibrary(
-      this.databaseItem.language,
-      externalApiUsages,
-      modeledMethods,
-    );
+    let yamls: Record<string, string>;
+    switch (this.mode) {
+      case Mode.Application:
+        yamls = createDataExtensionYamlsForApplicationMode(
+          this.databaseItem.language,
+          externalApiUsages,
+          modeledMethods,
+        );
+        break;
+      case Mode.Framework:
+        yamls = createDataExtensionYamlsForFrameworkMode(
+          this.databaseItem.name,
+          this.databaseItem.language,
+          externalApiUsages,
+          modeledMethods,
+        );
+        break;
+      default:
+        assertNever(this.mode);
+    }
 
     for (const [filename, yaml] of Object.entries(yamls)) {
       await outputFile(join(this.extensionPack.path, filename), yaml);

extensions/ql-vscode/src/data-extensions-editor/extension-pack-name.ts

@@ -37,7 +37,7 @@ export function autoNameExtensionPack(
   };
 }
 
-function sanitizeExtensionPackName(name: string) {
+export function sanitizeExtensionPackName(name: string) {
   // Lowercase everything
   name = name.toLowerCase();
 

extensions/ql-vscode/src/data-extensions-editor/yaml.ts

@@ -10,6 +10,7 @@ import {
 } from "./predicates";
 
 import * as dataSchemaJson from "./data-schema.json";
+import { sanitizeExtensionPackName } from "./extension-pack-name";
 
 const ajv = new Ajv({ allErrors: true });
 const dataSchemaValidate = ajv.compile(dataSchemaJson);
@@ -79,7 +80,7 @@ export function createDataExtensionYaml(
 ${extensions.join("\n")}`;
 }
 
-export function createDataExtensionYamlsPerLibrary(
+export function createDataExtensionYamlsForApplicationMode(
   language: string,
   externalApiUsages: ExternalApiUsage[],
   modeledMethods: Record<string, ModeledMethod>,
@@ -116,6 +117,33 @@ export function createDataExtensionYamlsPerLibrary(
   return result;
 }
 
+export function createDataExtensionYamlsForFrameworkMode(
+  databaseName: string,
+  language: string,
+  externalApiUsages: ExternalApiUsage[],
+  modeledMethods: Record<string, ModeledMethod>,
+  prefix = "models/",
+  suffix = ".model",
+): Record<string, string> {
+  const parts = databaseName.split("/");
+  const libraryName = parts
+    .slice(1)
+    .map((part) => sanitizeExtensionPackName(part))
+    .join("-");
+
+  const methods = externalApiUsages.map((externalApiUsage) => ({
+    externalApiUsage,
+    modeledMethod: modeledMethods[externalApiUsage.signature],
+  }));
+
+  return {
+    [`${prefix}${libraryName}${suffix}.yml`]: createDataExtensionYaml(
+      language,
+      methods,
+    ),
+  };
+}
+
 // From the semver package using
 // const { re, t } = require("semver/internal/re");
 // console.log(re[t.LOOSE]);