Kedro viz server removed

Signed-off-by: Jitendra Gundaniya <[email protected]>

bundled/tool/lsp_server.py

@@ -83,6 +83,8 @@
 )
 from pygls.server import LanguageServer
 
+from kedro_viz.server import load_and_populate_data
+from kedro_viz.api.rest.responses import get_json_data
 
 class KedroLanguageServer(LanguageServer):
     """Store Kedro-specific information in the language server."""
@@ -553,6 +555,21 @@ def definition_from_flowchart(ls, word):
     result = definition(LSP_SERVER, params=None, word=word)
     return result
 
+@LSP_SERVER.command("kedro.getProjectData")
+def get_porject_data_from_viz(lsClient):
+    """Get project data from kedro viz
+    """
+    data = None
+    try:
+        load_and_populate_data(Path.cwd())
+        data = get_json_data()
+        return data
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        print("Execution completed.")
+        return data
+
 ### End of  kedro-lsp
 
 

package.json

@@ -174,7 +174,6 @@
     "dependencies": {
         "@vscode/python-extension": "^1.0.5",
         "fs-extra": "^11.2.0",
-        "node-fetch": "^3.3.2",
         "vscode-languageclient": "^8.1.0"
     },
     "devDependencies": {

src/common/commands.ts

@@ -83,7 +83,6 @@ export async function executeServerDefinitionCommand(lsClient: LanguageClient |
     const result: any[] | undefined = await vscode.commands.executeCommand(
         commandName /* if your command accepts arguments you can pass them here */,
         target,
-
     );
     logger.info(`${commandName} result: ${JSON.stringify(result, undefined, 2)}`);
     if (result && result.length > 0) {
@@ -97,3 +96,19 @@ export async function executeServerDefinitionCommand(lsClient: LanguageClient |
         });
     }
 }
+
+export async function executeGetProjectDataCommand(lsClient: LanguageClient | undefined) {
+    if (!lsClient || lsClient.state !== State.Running) {
+        await vscode.window.showErrorMessage('There is no language server running.');
+        return;
+    }
+    if (!lsClient.initializeResult) {
+        await vscode.window.showErrorMessage('The Language Server fail to initialise.');
+        return;
+    }
+
+    const commandName = 'kedro.getProjectData';
+    logger.info(`executing command: '${commandName}'`);
+    const result = await vscode.commands.executeCommand(commandName);
+    return result;
+}

src/common/utilities.ts

@@ -6,7 +6,6 @@ import * as path from 'path';
 import { LogLevel, Uri, WorkspaceFolder } from 'vscode';
 import { Trace } from 'vscode-jsonrpc/node';
 import { getWorkspaceFolders } from './vscodeapi';
-import fetch from 'node-fetch';
 import KedroVizPanel from '../webview/vizWebView';
 
 function logLevelToTrace(logLevel: LogLevel): Trace {
@@ -67,10 +66,3 @@ export async function getProjectRoot(): Promise<WorkspaceFolder> {
         return rootWorkspace;
     }
 }
-
-export async function fetchAndUpdateProjectData(): Promise<void> {
-    fetch('http://127.0.0.1:3131/api/main')
-        .then((response: { text: () => any }) => response.text())
-        .then((data: string) => KedroVizPanel.currentPanel?.updateData(data))
-        .catch((err: { message: string }) => console.error('Error: ' + err.message));
-}

src/extension.ts

@@ -1,6 +1,11 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
-import { selectEnvironment, executeServerCommand, executeServerDefinitionCommand } from './common/commands';
+import {
+    selectEnvironment,
+    executeServerCommand,
+    executeServerDefinitionCommand,
+    executeGetProjectDataCommand,
+} from './common/commands';
 import * as vscode from 'vscode';
 import { LanguageClient } from 'vscode-languageclient/node';
 import { registerLogger, traceError, traceLog, traceVerbose } from './common/log/logging';
@@ -23,10 +28,8 @@ import { loadServerDefaults } from './common/setup';
 import { getLSClientTraceLevel, getProjectRoot } from './common/utilities';
 import { createOutputChannel, onDidChangeConfiguration, registerCommand } from './common/vscodeapi';
 import KedroVizPanel from './webview/vizWebView';
-import { runKedroVizServer } from './webview/vizServer';
 
 let lsClient: LanguageClient | undefined;
-let kedroVizProcess: any;
 
 export async function activate(context: vscode.ExtensionContext): Promise<void> {
     // This is required to get server name and module. This should be
@@ -58,8 +61,10 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
     );
 
     context.subscriptions.push(
-        vscode.commands.registerCommand('kedro.runKedroViz', () => {
+        vscode.commands.registerCommand('kedro.runKedroViz', async () => {
             KedroVizPanel.createOrShow(context.extensionUri, lsClient);
+            const projectData = await executeGetProjectDataCommand(lsClient);
+            KedroVizPanel.currentPanel?.updateData(projectData);
         }),
     );
 
@@ -88,12 +93,6 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
         console.log(interpreterDetails);
         console.log('===============DEBUG============');
 
-        // Start kedro viz server
-        if (kedroVizProcess) {
-            process.kill(-kedroVizProcess.pid);
-        }
-        kedroVizProcess = await runKedroVizServer();
-
         if (interpreterDetails.path) {
             traceVerbose(`Using interpreter from Python extension: ${interpreterDetails.path.join(' ')}`);
             lsClient = await restartServer(serverId, serverName, outputChannel, lsClient, env);
@@ -168,8 +167,4 @@ export async function deactivate(): Promise<void> {
     if (lsClient) {
         await lsClient.stop();
     }
-    if (kedroVizProcess) {
-        process.kill(-kedroVizProcess.pid);
-        kedroVizProcess = null; // Reset the reference after killing the process
-    }
 }

src/webview/goToDefinition.ts

@@ -11,7 +11,8 @@ export async function goToDefinition(message: Message) {
     let filePattern = '**/*.yml';
 
     if (message.type === 'task') {
-        filePattern = '**/*.py';
+        // Looking only in pipelines folders
+        filePattern = '**/pipelines/**/*.py';
     }
 
     const files = await vscode.workspace.findFiles(filePattern);

src/webview/vizWebView.ts

@@ -1,7 +1,5 @@
 import * as vscode from 'vscode';
-import fetch from 'node-fetch';
 import { goToDefinition } from './goToDefinition';
-import { fetchAndUpdateProjectData } from '../common/utilities';
 import { LanguageClient } from 'vscode-languageclient/node';
 import { executeServerDefinitionCommand } from '../common/commands';
 
@@ -37,7 +35,6 @@ export default class KedroVizPanel {
         });
 
         KedroVizPanel.currentPanel = new KedroVizPanel(panel, extensionUri, lsClient);
-        fetchAndUpdateProjectData();
     }
 
     public static revive(panel: vscode.WebviewPanel, extensionUri: vscode.Uri) {
@@ -84,11 +81,6 @@ export default class KedroVizPanel {
         );
     }
 
-    public updateTheme() {
-        // Send a message to the webview.
-        this._panel.webview.postMessage({ command: 'updateTheme', theme: 'light' });
-    }
-
     public updateData(data: any) {
         // Send a message to the webview.
         this._panel.webview.postMessage({ command: 'updateData', data });

webview/src/App.jsx

@@ -14,7 +14,7 @@ function App() {
       const message = event.data;
       switch (message.command) {
         case "updateData":
-          setData(JSON.parse(message.data));
+          setData(message.data);
           setLoading(false);
           break;
         default: