Update lsp to have set root path

compiler/NubLang.LSP/Program.cs

@@ -18,17 +18,7 @@ var server = await LanguageServer.From(options => options
     .WithHandler<HoverHandler>()
     .WithHandler<CompletionHandler>()
     .WithHandler<DefinitionHandler>()
-    .OnInitialize((server, request, ct) =>
-    {
-        var workspaceManager = server.GetRequiredService<WorkspaceManager>();
-
-        if (request.RootPath != null)
-        {
-            workspaceManager.Init(request.RootPath);
-        }
-
-        return Task.CompletedTask;
-    })
+    .WithHandler<SetRootPathCommandHandler>()
 );
 
 await server.WaitForExit;

compiler/NubLang.LSP/SetRootPathCommandHandler.cs

@@ -0,0 +1,31 @@
+using MediatR;
+using OmniSharp.Extensions.LanguageServer.Protocol.Client.Capabilities;
+using OmniSharp.Extensions.LanguageServer.Protocol.Models;
+using OmniSharp.Extensions.LanguageServer.Protocol.Workspace;
+
+namespace NubLang.LSP;
+
+public class SetRootPathCommandHandler(WorkspaceManager workspaceManager) : ExecuteCommandHandlerBase
+{
+    protected override ExecuteCommandRegistrationOptions CreateRegistrationOptions(ExecuteCommandCapability capability, ClientCapabilities clientCapabilities)
+    {
+        return new ExecuteCommandRegistrationOptions
+        {
+            Commands = new Container<string>("nub.setRootPath")
+        };
+    }
+
+    public override Task<Unit> Handle(ExecuteCommandParams request, CancellationToken cancellationToken)
+    {
+        if (request is { Command: "nub.setRootPath", Arguments.Count: > 0 })
+        {
+            var newRoot = request.Arguments[0].ToString();
+            if (!string.IsNullOrEmpty(newRoot))
+            {
+                workspaceManager.SetRootPath(newRoot);
+            }
+        }
+
+        return Unit.Task;
+    }
+}

compiler/NubLang.LSP/WorkspaceManager.cs

@@ -17,7 +17,7 @@ public class WorkspaceManager(DiagnosticsPublisher diagnosticsPublisher)
     private readonly Dictionary<string, List<TopLevelNode>> _possiblyOutdatedTopLevelNodes = [];
     private ModuleRepository _repository = new([]);
 
-    public void Init(string rootPath)
+    public void SetRootPath(string rootPath)
     {
         _rootPath = rootPath;
         Update();