a-r-j · Jul 8, 2022 · Oct 31, 2026
diff --git a/out/extension.js b/out/extension.js
diff --git a/out/extension.js.map b/out/extension.js.map
diff --git a/out/test/suite/extension.test.js b/out/test/suite/extension.test.js
diff --git a/out/test/suite/extension.test.js.map b/out/test/suite/extension.test.js.map
diff --git a/package.json b/package.json
@@ -48,7 +48,8 @@
     "onCommand:protein-viewer.start",
     "onCommand:protein-viewer.activateFromFiles",
     "onCommand:protein-viewer.activateFromFolder",
-    "onCommand:protein-viewer.ESMFold"
+    "onCommand:protein-viewer.ESMFold",
+    "onCommand:protein-viewer.viewCurrentPDB"
   ],
   "main": "./out/extension.js",
   "contributes": {
@@ -68,6 +69,10 @@
       {
         "command": "protein-viewer.ESMFold",
         "title": "ESMFold Sequence"
+      },
+      {
+        "command": "protein-viewer.viewCurrentPDB",
+        "title": "View structure in Protein Viewer"
       }
     ],
     "menus": {

diff --git a/src/extension.ts b/src/extension.ts
@@ -94,7 +94,15 @@ export async function activate(context: vscode.ExtensionContext) {
 
 		});
 	});
-	context.subscriptions.push(helloCommand, activateFromFiles, activateFromFolder, ESMFold);
+
+	const viewCurrentPDB = vscode.commands.registerCommand("protein-viewer.viewCurrentPDB", () => {
+		const activeEditor = vscode.window.activeTextEditor?.document.uri;
+		if (activeEditor) {
+			ProteinViewerPanel.renderFromFiles(context.extensionUri, [activeEditor]);
+		}
+	});
+
+	context.subscriptions.push(helloCommand, activateFromFiles, activateFromFolder, ESMFold, viewCurrentPDB);
 }
 
 // this method is called when your extension is deactivated