WIP

Author: EasyRhinoMSFT

src/extension/index.ts

@@ -56,19 +56,34 @@ export async function activate(context: ExtensionContext) {
 
     // URI handler
     window.registerUriHandler({
-        async handleUri(uri: Uri) {
-            const params = new URLSearchParams(uri.query);
-            const alertRequestUri = params.get('requestUri');
-
-            if (alertRequestUri) {
-                const url = new URL(alertRequestUri);
-                if (url) {
-                    await loadAlertSarif(url);
-                }
-            }
+        async handleUri() {
+            startWebServer();
         }
     });
 
+    async function startWebServer() {
+        const port = 4169;
+        createServer((request, response) => {
+            outputChannel.appendLine(`${request.method} request received for ${request.url}}`);
+            if (request.method === 'POST') {
+                const filePath = `${os.tmpdir}\\${(new Date()).getTime()}.sarif`;
+                const stream = request
+                    //.pipe(meter(1024 * 1024)) // Cap at 1 MB
+                    .pipe(fs.createWriteStream(filePath));
+                stream.on('finish', async () => {
+                    store.logs.push(...await loadLogs([Uri.file(filePath)]));
+                    if (store.results.length) panel.show();
+                });
+    
+                response.statusCode = 200;
+                response.end();
+    
+            }
+        }).listen(port, () => {
+            outputChannel.appendLine(`Server running at http://localhost:${port}`);
+        });
+    }
+
     async function loadAlertSarif(url: URL) {
         try {
             //const azureAccount: AzureAccountExtensionApi = (<AzureAccountExtensionApi>extensions.getExtension('ms-vscode.azure-account')?.exports); //.getApi('1.0.0'); // = (<AzureExtensionApiProvider>extensions.getExtension('ms-vscode.azure-account')!.exports).getApi('1.0.0');
@@ -127,27 +142,6 @@ export async function activate(context: ExtensionContext) {
 
     //await loadAlertSarif(new URL('https://advsec.codedev.ms/cmeyer/_apis/advancedsecurity/alerts/18/?project=ProjektEins&repository=1213f32a-071a-4282-b9a2-2f6141590138'));
 
-    const port = 4169;
-    createServer((request, response) => {
-        outputChannel.appendLine(`${request.method} request received for ${request.url}}`);
-        if (request.method === 'POST') {
-            const filePath = `${os.tmpdir}\\${(new Date()).getTime()}.sarif`;
-            const stream = request
-                //.pipe(meter(1024 * 1024)) // Cap at 1 MB
-                .pipe(fs.createWriteStream(filePath));
-            stream.on('finish', async () => {
-                store.logs.push(...await loadLogs([Uri.file(filePath)]));
-                if (store.results.length) panel.show();
-            });
-
-            response.statusCode = 200;
-            response.end();
-
-        }
-    }).listen(port, () => {
-        outputChannel.appendLine(`Server running at http://localhost:${port}`);
-    });
-
     // General Activation
     activateSarifStatusBarItem(disposables);
     activateDiagnostics(disposables, store, baser, outputChannel);