Add webview

kshmidt-digma · kshmidt-digma · commit 36f084ad1b2a · 2025-09-05T12:25:29.000+02:00
diff --git a/src/extension.ts b/src/extension.ts
@@ -6,6 +6,7 @@ import { registerMcpServer } from "./registerMcpServer";
 import { SettingsManager } from "./SettingsManager";
 import type { PackageJSON } from "./types";
 import { attachIncidentFileToChatContext } from "./uris/handlers/context";
+import { openWebviewWithUrl } from "./uris/handlers/webview";
 import { UriRouter } from "./uris/UriRouter";
 
 const START_PORT = 33100;
@@ -14,6 +15,12 @@ const END_PORT = 33199;
 let digmaClient: DigmaApiClient | null = null;
 // let mcpServerDisposable: vscode.Disposable | null = null;
 
+const getBaseDomain = (url: string): string => {
+  const hostname = new URL(url).hostname;
+  const parts = hostname.split(".");
+  return parts.slice(-2).join(".");
+};
+
 const findAvailablePort = async (
   start: number,
   end: number
@@ -63,6 +70,9 @@ async function initializeClient(
 export async function activate(context: vscode.ExtensionContext) {
   const uriRouter = new UriRouter();
 
+  const settingsManager = new SettingsManager(context);
+  await initializeClient(settingsManager);
+
   uriRouter.route(
     "/chat/context/add/file/incident/:incidentId",
     async (params) => {
@@ -73,8 +83,25 @@ export async function activate(context: vscode.ExtensionContext) {
     }
   );
 
-  const settingsManager = new SettingsManager(context);
-  await initializeClient(settingsManager);
+  uriRouter.route("/webview/open", async (_, query) => {
+    const url = query?.url;
+
+    const settingsUrl = await settingsManager.getSetting<string>("url");
+    if (settingsUrl && url) {
+      const settingsBaseDomain = getBaseDomain(settingsUrl);
+      const urlBaseDomain = getBaseDomain(url);
+
+      if (settingsBaseDomain === urlBaseDomain) {
+        openWebviewWithUrl(url);
+      } else {
+        vscode.window.showErrorMessage(
+          "URL must be from the same base domain as configured"
+        );
+      }
+    } else {
+      vscode.window.showErrorMessage("URL parameter is required for webview");
+    }
+  });
 
   // Listen for configuration changes
   const configChangeListener = vscode.workspace.onDidChangeConfiguration(
diff --git a/src/uris/handlers/webview.ts b/src/uris/handlers/webview.ts
@@ -0,0 +1,53 @@
+import vscode from "vscode";
+
+export const openWebviewWithUrl = (url: string) => {
+  try {
+    const panel = vscode.window.createWebviewPanel(
+      "editorWebview",
+      "Digma",
+      vscode.ViewColumn.One,
+      {
+        enableScripts: true,
+        retainContextWhenHidden: true
+      }
+    );
+
+    panel.webview.html = getWebviewContent(url);
+  } catch {
+    vscode.window.showErrorMessage(`Invalid URL provided: ${url}`);
+  }
+};
+
+function getWebviewContent(url: string): string {
+  return `<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Digma Webview</title>
+    <style>
+        body, html {
+            margin: 0;
+            padding: 0;
+            height: 100vh;
+            overflow: hidden;
+        }
+        iframe {
+            width: 100%;
+            height: 100vh;
+            border: none;
+        }
+        .error {
+            padding: 20px;
+            background: #2d2d2d;
+        }
+    </style>
+</head>
+<body>
+    <iframe src="${url}" 
+            onload="this.style.display='block'" 
+            onerror="document.body.innerHTML='<div class=\\"error\\">Failed to load: ${url}</div>'">
+    </iframe>
+</body>
+</html>`;
+}
