Fix bug with IDL Notebooks and MCP

Author: znorman-harris

CHANGELOG.md

@@ -18,6 +18,10 @@ Document some advanced types so users may try them out and provide feedback. The
 
 Auto-complete for blocks re-work: Partial implementation of auto-complete for blocks that works better than the default snippets that exists. This functions for if-then-else only right now to verify the user experience is what it needs to be.
 
+## 5.0.1 = July 2024
+
+Fix a bug with IDL Notebooks when you would open an empty notebook.
+
 ## 5.0.0 – July 2024
 
 This release introduces a significant evolution in extension functionality, warranting the major version bump to 5.0.0.

libs/mcp/server-tools/src/index.ts

@@ -1,3 +1,6 @@
 export * from './lib/helpers/track-envi-task-for-mcp-server';
 export * from './lib/mcp-tool-registry.class';
 export * from './lib/register-all-mcp-tools';
+export * from './lib/tools/register-tool-query-envi-task-parameters';
+export * from './lib/tools/register-tool-query-envi-tasks';
+export * from './lib/tools/register-tool-run-envi-task';

libs/mcp/server-tools/src/lib/register-all-mcp-tools.ts

@@ -6,9 +6,6 @@ import { RegisterToolCreateIDLNotebook } from './tools/register-tool-create-idl-
 import { RegisterToolExecuteIDLCode } from './tools/register-tool-execute-idl-code';
 import { RegisterToolExecuteIDLFile } from './tools/register-tool-execute-idl-file';
 import { RegisterToolOpenInENVI } from './tools/register-tool-open-in-envi';
-import { RegisterToolQueryENVITaskParameters } from './tools/register-tool-query-envi-task-parameters';
-import { RegisterToolQueryENVITasks } from './tools/register-tool-query-envi-tasks';
-import { RegisterToolRunENVITask } from './tools/register-tool-run-envi-task';
 import { RegisterToolStartENVI } from './tools/register-tool-start-envi';
 import { RegisterToolStartIDL } from './tools/register-tool-start-idl';
 
@@ -41,9 +38,6 @@ export function RegisterAllMCPTools(messenger: VSCodeLanguageServerMessenger) {
   RegisterToolExecuteIDLCode(messenger);
   RegisterToolExecuteIDLFile(messenger);
   RegisterToolOpenInENVI(messenger);
-  RegisterToolQueryENVITasks(messenger);
-  RegisterToolQueryENVITaskParameters(messenger);
-  RegisterToolRunENVITask(messenger);
   RegisterToolStartENVI(messenger);
   RegisterToolStartIDL(messenger);
 

libs/shared/extension/src/lib/version.interface.ts

@@ -1,4 +1,4 @@
 /**
  * Version of the extension
  */
-export const VERSION = '5.1.0';
+export const VERSION = '5.0.1';

libs/vscode/server/src/lib/helpers/resolve-fspath-and-code-for-uri.ts

@@ -59,6 +59,11 @@ export async function ResolveFSPathAndCodeForURI(
   // get the cell index
   const idx = nb.cells.findIndex((cell) => cell.document === url);
 
+  // return if we don't have a cell
+  if (!nb.cells[idx]) {
+    return undefined;
+  }
+
   // get cell document
   const cellDoc = NOTEBOOK_MANAGER.getCellTextDocument(nb.cells[idx]);
 

libs/vscode/server/src/lib/mcp/register-user-mcp-tools.ts

@@ -1,10 +1,16 @@
 import { IDL_LSP_LOG } from '@idl/logger';
-import { TrackENVITaskForMCPServer } from '@idl/mcp/server-tools';
+import {
+  RegisterToolQueryENVITaskParameters,
+  RegisterToolQueryENVITasks,
+  RegisterToolRunENVITask,
+  TrackENVITaskForMCPServer,
+} from '@idl/mcp/server-tools';
 import {
   GLOBAL_TOKEN_TYPES,
   GlobalStructureToken,
   IGlobalIndexedToken,
 } from '@idl/types/core';
+import { VSCodeLanguageServerMessenger } from '@idl/vscode/events/server';
 
 import { IDL_INDEX } from '../events/initialize-document-manager';
 import {
@@ -18,13 +24,20 @@ import { FilterMCPENVITasks } from './filter-mcp-envi-tasks';
  *
  * WIP: Not complete and GitHub Copilot isn't the best here
  */
-export async function RegisterUserMCPTools() {
+export async function RegisterUserMCPTools(
+  messenger: VSCodeLanguageServerMessenger
+) {
   IDL_LANGUAGE_SERVER_LOGGER.log({
     log: IDL_LSP_LOG,
     type: 'info',
     content: 'Registering MCP user tools',
   });
 
+  // register additional tools
+  RegisterToolQueryENVITasks(messenger);
+  RegisterToolQueryENVITaskParameters(messenger);
+  RegisterToolRunENVITask(messenger);
+
   /** Get all structures that we know about */
   const structures =
     IDL_INDEX.globalIndex.globalTokensByTypeByName[

package.json

@@ -2,7 +2,7 @@
   "name": "idl-for-vscode",
   "displayName": "%packageJSON.displayName%",
   "description": "%packageJSON.description%",
-  "version": "5.1.0",
+  "version": "5.0.1",
   "publisher": "idl",
   "license": "MIT",
   "encryption": "yes",