Fix issues

Author: Danimester23

plugins/csharp/service/src/csharpservice.cpp

@@ -73,6 +73,7 @@ void CsharpServiceHandler::getAstNodeInfo(
   if (!file)
   {
     LOG(warning) << "[csharpservice] getAstNodeInfo: file not found for path: " << return_.range.file;
+    return_.range.file = ""; // File ID not found safely clear the file path instead of propagating string path
     return;
   }
   std::stringstream ss;

webgui-new/src/components/codemirror-editor/codemirror-editor.tsx

@@ -141,9 +141,9 @@ export const CodeMirrorEditor = (): JSX.Element => {
 
   const highlightExtension = () => {
     return EditorView.decorations.of((view) => {
-      const decorations = highlightRanges.map((pos) =>
-        createHighlightDecoration(view, pos, highlightColor),
-      ) as never;
+      const decorations = highlightRanges
+        .map((pos) => createHighlightDecoration(view, pos, highlightColor))
+        .filter((d) => d !== undefined) as never;
       return Decoration.set(decorations, true);
     });
   };
@@ -164,9 +164,15 @@ export const CodeMirrorEditor = (): JSX.Element => {
   const updateHighlights = async (astNode: AstNodeInfo) => {
     const refTypes = await getReferenceTypes(astNode.id as string);
     if (visitedLastAstNode?.id !== astNode.id) {
+      const usageReferenceType = refTypes.get("Usage");
+      if (usageReferenceType === undefined) {
+        setHighlightRanges([]);
+        setVisitedLastAstNode(null);
+        return;
+      }
       const allReferences = await getReferences(
         astNode.id as string,
-        refTypes.get("Usage") as number,
+        usageReferenceType,
         [],
       );
       const referencesInFile = allReferences.filter(

webgui-new/src/components/diagrams/diagrams.tsx

@@ -8,6 +8,7 @@ import React, {
   useState,
   MouseEvent,
 } from "react";
+import { toast } from "react-toastify";
 import {
   createClient,
   getAstNodeInfo,
@@ -65,6 +66,11 @@ export const Diagrams = (): JSX.Element => {
             ? await getFileInfo(appCtx.projectFileId)
             : undefined;
 
+      if (!fileInfoForClient?.type) {
+        toast.error("File type is required to generate this diagram.");
+        return;
+      }
+
       createClient(appCtx.workspaceId, fileInfoForClient?.type);
 
       const initDiagramInfo =
@@ -121,6 +127,11 @@ export const Diagrams = (): JSX.Element => {
       const parsedDiagram = parser.parseFromString(diagram, "text/xml");
       const diagramSvg = parsedDiagram.getElementsByTagName("svg")[0];
 
+      if (!diagramSvg) {
+        toast.error("Failed to parse diagram or diagram is empty.");
+        return;
+      }
+
       diagramSvg.style.height = "100%";
       diagramSvg.style.cursor = "pointer";
 
@@ -154,6 +165,11 @@ export const Diagrams = (): JSX.Element => {
           ? await getFileInfo(appCtx.projectFileId)
           : undefined;
 
+    if (!fileInfoForClient?.type) {
+      toast.error("File type is required to navigate to this diagram.");
+      return;
+    }
+
     createClient(appCtx.workspaceId, fileInfoForClient?.type);
 
     const initDiagramInfo =
@@ -215,6 +231,11 @@ export const Diagrams = (): JSX.Element => {
     );
     const diagramLegendSvg = parsedDiagramLegend.getElementsByTagName("svg")[0];
 
+    if (!diagramLegendSvg) {
+      toast.error("Failed to parse legend diagram or it is empty.");
+      return;
+    }
+
     diagramLegendSvg.style.height = "100%";
     diagramLegendContainerRef.current.replaceChildren(diagramLegendSvg);
 

webgui-new/src/components/file-context-menu/file-context-menu.tsx

@@ -46,7 +46,8 @@ export const FileContextMenu = ({
   );
 
   useEffect(() => {
-    if (!fileInfo) return;
+    if (!fileInfo || fileInfo.isDirectory || fileInfo.type === "Unknown")
+      return;
     const init = async () => {
       createClient(appCtx.workspaceId, fileInfo?.type);
       const initDiagramTypes = await getFileDiagramTypes(fileInfo.id as string);

webgui-new/src/service/language-service.ts

@@ -8,6 +8,7 @@ export const createClient = (
   workspace: string,
   fileType: string | undefined,
 ) => {
+  client = undefined;
   if (!config || !fileType) return;
 
   const service = () => {