fix: wire lineage Export button to SaaS via extension host (#1935)

Author: ralphstodomingo

src/altimate.ts

@@ -458,6 +458,13 @@ export class AltimateRequest {
     });
   }
 
+  async exportLineage(req: { name: string; lineage_data: unknown }) {
+    return this.fetch<{ url: string }>("dbt/v4/export-lineage", {
+      method: "POST",
+      body: JSON.stringify(req),
+    });
+  }
+
   async runModeller(req: SQLToModelRequest) {
     return this.fetch<SQLToModelResponse>("dbt/v1/sqltomodel", {
       method: "POST",

src/webview_provider/newLineagePanel.ts

@@ -246,6 +246,39 @@ export class NewLineagePanel
       return;
     }
 
+    if (command === "exportLineage") {
+      try {
+        const body = await this.altimate.exportLineage({
+          name: params.name,
+          lineage_data: params.lineage_data,
+        });
+        this._panel?.webview.postMessage({
+          command: "response",
+          args: { id, syncRequestId, body, status: true },
+        });
+      } catch (error) {
+        this._panel?.webview.postMessage({
+          command: "response",
+          args: {
+            id,
+            syncRequestId,
+            error: (error as Error).message,
+            status: false,
+          },
+        });
+        window.showErrorMessage(
+          extendErrorWithSupportLinks(
+            "Could not export lineage: " + (error as Error).message,
+          ),
+        );
+        this.telemetry.sendTelemetryError(
+          "altimateLineageExportLineageError",
+          error,
+        );
+      }
+      return;
+    }
+
     if (command === "columnLineage") {
       this.handleColumnLineage(args, () => {
         this._panel?.webview.postMessage({

webview_panels/src/modules/lineage/LineageView.tsx

@@ -67,6 +67,17 @@ const LineageView = (): JSX.Element | null => {
           break;
       }
     };
+    // @ts-expect-error TODO: add type generic for executeRequestInSync
+    ApiHelper.post = async (url: string, data?: Record<string, unknown>) => {
+      switch (url) {
+        case "dbt/v4/export-lineage":
+          return executeRequestInSync("exportLineage", {
+            args: { params: data ?? {} },
+          });
+        default:
+          break;
+      }
+    };
     setIsApiHelperInitialized(true);
   }, [isComponentsApiInitialized]);