Be smarter about reloading

Author: firelizzard18

package.json

@@ -3,7 +3,7 @@
   "publisher": "ethan-reesor",
   "displayName": "Go Companion",
   "description": "An unofficial companion to the official Go extension, providing experimental features",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "icon": "docs/assets/icon.png",
   "license": "MIT",
   "homepage": "https://github.com/firelizzard18/exp-vscode-go",

src/go-generate/manager.ts

@@ -35,8 +35,13 @@ export class GoGenerateManager {
 			storageUri: ctx.storageUri,
 			output: window.createOutputChannel('Go Generate', { log: true }),
 			commands: {
-				modules: (args) => commands.executeCommand('gopls.modules', args),
-				packages: (args) => commands.executeCommand('gopls.packages', args),
+				// These should never be called.
+				modules: () => {
+					throw new Error('Internal error');
+				},
+				packages: () => {
+					throw new Error('Internal error');
+				},
 			},
 		};
 		const { event } = helpers(ctx, testCtx, commands);

src/test/manager.ts

@@ -56,18 +56,6 @@ export class TestManager {
 				createTestController(id: string, label: string): TestController;
 			},
 	) {
-		// Verify that gopls is new enough to support the packages command
-		try {
-			await this.context.commands.packages({ Files: [] });
-		} catch (error) {
-			if (!`${error}`.match(/^Error: command '.*' not found$/)) {
-				throw error;
-			}
-
-			await args.showWarningMessage('gopls is not installed or does not support test discovery');
-			return;
-		}
-
 		// Register the legacy code lens provider
 		this.#disposable.push(
 			args.registerCodeLensProvider({ language: 'go', scheme: 'file', pattern: '**/*_test.go' }, this.#codeLens),

src/test/register.ts

@@ -107,8 +107,16 @@ async function registerTestController(ctx: ExtensionContext, testCtx: Context) {
 		await maybeChangedEnabled();
 	});
 
-	// [Event] File open
-	event(workspace.onDidOpenTextDocument, 'opened document', (e) => manager.enabled && manager.reloadUri(e.uri));
+	// [Event] The user opened a file in an editor
+	let seenDocuments = new Set<string>();
+	event(window.onDidChangeVisibleTextEditors, 'opened document', (editors) => {
+		for (const editor of editors) {
+			if (seenDocuments.has(`${editor.document.uri}`)) continue;
+			manager.reloadUri(editor.document.uri);
+		}
+
+		seenDocuments = new Set(editors.map((x) => `${x.document.uri}`));
+	});
 
 	// [Event] File change
 	event(workspace.onDidChangeTextDocument, 'updated document', async (e) => {