Minor Fixes

Author: rcosta358

client/src/webview/diagram.ts

@@ -21,7 +21,7 @@ let startY = 0;
  * @param sm 
  * @returns Mermaid diagram string
  */
-export function createMermaidDiagram(sm: LJStateMachine, orientation: "LR" | "TB"): string {
+export function createMermaidDiagram(sm: LJStateMachine | undefined, orientation: "LR" | "TB"): string {
     if (!sm) return '';
 
     const lines: string[] = [];

client/src/webview/script.ts

@@ -7,7 +7,7 @@ import type { Range } from "../types/context";
 import type { LJStateMachine } from "../types/fsm";
 import type { NavTab } from "./views/sections";
 import { renderDiagnosticsView } from "./views/diagnostics/diagnostics";
-import { LJContext } from "../types/context";
+import type { LJContext } from "../types/context";
 import { ContextSectionState, renderContextView } from "./views/context/context";
 
 /**

client/src/webview/views/context/context.ts

@@ -1,5 +1,5 @@
-import { LJContext } from "../../../types/context";
-import { LJDiagnostic, RefinementMismatchError } from "../../../types/diagnostics";
+import type { LJContext } from "../../../types/context";
+import type { RefinementMismatchError } from "../../../types/diagnostics";
 import { renderMainHeader } from "../sections";
 import { renderContextAliases } from "./aliases";
 import { renderContextGhosts } from "./ghosts";
@@ -11,12 +11,10 @@ export type ContextSectionState = {
     vars: boolean;
 }
 
-export function renderContextView(context: LJContext, currentFile: string, sectionState: ContextSectionState, errorAtCursor?: RefinementMismatchError): string {
-    if (!context || !currentFile) return "";
-
-    const allVars = context.allVars || [];
-    const ghosts = context.ghosts?.filter(ghost => ghost.file === currentFile) || [];
-    const aliases = context.aliases || [];
+export function renderContextView(context: LJContext | undefined, currentFile: string | undefined, sectionState: ContextSectionState, errorAtCursor?: RefinementMismatchError): string {
+    const allVars = context?.allVars || [];
+    const ghosts = context?.ghosts?.filter(ghost => ghost.file === currentFile) || [];
+    const aliases = context?.aliases || [];
     const total = allVars.length + ghosts.length + aliases.length;
     return /*html*/`
         <div>

client/src/webview/views/diagnostics/diagnostics.ts

@@ -6,7 +6,7 @@ import { renderWarnings } from "./warnings";
 export function renderDiagnosticsView(
     diagnostics: LJDiagnostic[],
     showAll: boolean,
-    currentFile: string,
+    currentFile: string | undefined,
     expandedErrors: Set<number>,
 ): string {
     const fileDiagnostics = diagnostics.filter(diagnostic => diagnostic.file?.toLowerCase() === currentFile?.toLowerCase() || !diagnostic.file);

client/src/webview/views/fsm/fsm.ts

@@ -1,7 +1,7 @@
 import type { LJStateMachine } from "../../../types/fsm";
 import { renderMainHeader } from "../sections";
 
-export function renderStateMachineView(sm: LJStateMachine, diagram: string, orientation: "LR" | "TB"): string {
+export function renderStateMachineView(sm: LJStateMachine | undefined, diagram: string, orientation: "LR" | "TB"): string {
     return /*html*/`
         <div>
             ${renderMainHeader("", 'fsm')}