Fix showing preview with command palette

Author: JamesCamel

package.json

@@ -99,15 +99,15 @@
 				}
 			},
 			{
-				"command": "note.showPreview",
+				"command": "HackMD.showPreview",
 				"title": "HackMD: Show Preview",
 				"icon": {
 					"light": "src/icon/light/view-dark.svg",
 					"dark": "src/icon/dark/view-light.svg"
 				}
 			},
 			{
-				"command": "note.showPreviewAndEditor",
+				"command": "HackMD.showPreviewAndEditor",
 				"title": "HackMD: Show Preview To Side",
 				"icon": {
 					"light": "src/icon/light/column-dark.svg",
@@ -133,12 +133,12 @@
 			],
 			"view/item/context": [
 				{
-					"command": "note.showPreview",
+					"command": "HackMD.showPreview",
 					"when": "view == mdTreeItems && viewItem == file",
 					"group": "inline"
 				},
 				{
-					"command": "note.showPreviewAndEditor",
+					"command": "HackMD.showPreviewAndEditor",
 					"when": "view == mdTreeItems && viewItem == file",
 					"group": "inline"
 				}

src/commands/treeView.ts

@@ -17,35 +17,93 @@ export async function registerTreeViewCommands(context: vscode.ExtensionContext)
     context.subscriptions.push(vscode.commands.registerCommand('clickTreeItem', async (label, noteId) => {
         if (label && noteId) {
             const content = await API.exportString(noteId, ExportType.MD);
-            if (content) {
-                const uri = vscode.Uri.parse(`hackmd:${label}.md#${noteId}`);
-                const doc = await vscode.workspace.openTextDocument(uri);
-                await vscode.window.showTextDocument(doc, { preview: false });
-            }
+            if (!checkNoteExist(content)) { return; }
+
+            const uri = vscode.Uri.parse(`hackmd:${label}.md#${noteId}`);
+            const doc = await vscode.workspace.openTextDocument(uri);
+            await vscode.window.showTextDocument(doc, { preview: false });
+
         }
     }));
 
-    context.subscriptions.push(vscode.commands.registerCommand('note.showPreview', async (noteNode: NoteTreeNode) => {
-        if (noteNode.label && noteNode.noteId) {
+    context.subscriptions.push(vscode.commands.registerCommand('HackMD.showPreview', async (noteNode: NoteTreeNode) => {
+        if (noteNode) {
             const content = await API.exportString(noteNode.noteId, ExportType.MD);
-            if (content) {
-                const uri = vscode.Uri.parse(`hackmd:${noteNode.label}.md#${noteNode.noteId}`);
-                vscode.commands.executeCommand('markdown.showPreview', uri);
-            }
+            if (!checkNoteExist(content)) { return; }
+
+            const uri = vscode.Uri.parse(`hackmd:${noteNode.label}.md#${noteNode.noteId}`);
+            vscode.commands.executeCommand('markdown.showPreview', uri);
+        } else {
+            const editor = vscode.window.activeTextEditor;
+            if (!checkEditorExist(editor)) { return; }
+
+            const noteId = editor.document.uri.fragment;
+            if (!checkNoteIdExist(noteId)) { return; }
+
+            const content = await API.exportString(noteId, ExportType.MD);
+            if (!checkNoteExist(content)) { return; }
+
+            const fileName = editor.document.fileName.split('.')[0];
+            const uri = vscode.Uri.parse(`hackmd:${fileName}.md#${noteId}`);
+            vscode.commands.executeCommand('markdown.showPreview', uri);
         }
     }));
 
-    context.subscriptions.push(vscode.commands.registerCommand('note.showPreviewAndEditor', async (noteNode: NoteTreeNode) => {
-        if (noteNode.label && noteNode.noteId) {
+    context.subscriptions.push(vscode.commands.registerCommand('HackMD.showPreviewAndEditor', async (noteNode: NoteTreeNode) => {
+        if (noteNode) {
             const content = await API.exportString(noteNode.noteId, ExportType.MD);
-            if (content) {
-                const uri = vscode.Uri.parse(`hackmd:${noteNode.label}.md#${noteNode.noteId}`);
-                const doc = await vscode.workspace.openTextDocument(uri);
-                await vscode.window.showTextDocument(doc, { preview: false });
-                vscode.commands.executeCommand('markdown.showPreviewToSide', uri);
-            }
+            if (!checkNoteExist(content)) { return; }
+
+            const uri = vscode.Uri.parse(`hackmd:${noteNode.label}.md#${noteNode.noteId}`);
+            const doc = await vscode.workspace.openTextDocument(uri);
+            await vscode.window.showTextDocument(doc, { preview: false });
+            vscode.commands.executeCommand('markdown.showPreviewToSide', uri);
+
+        } else {
+            const editor = vscode.window.activeTextEditor;
+            if (!checkEditorExist(editor)) { return; }
+
+            const noteId = editor.document.uri.fragment;
+            if (!checkNoteIdExist(noteId)) { return; }
+
+            const content = await API.exportString(noteId, ExportType.MD);
+            if (!checkNoteExist(content)) { return; }
+
+            const fileName = editor.document.fileName.split('.')[0];
+            const uri = vscode.Uri.parse(`hackmd:${fileName}.md#${noteId}`);
+            const doc = await vscode.workspace.openTextDocument(uri);
+            await vscode.window.showTextDocument(doc, { preview: false });
+            vscode.commands.executeCommand('markdown.showPreviewToSide', uri);
         }
     }));
 
     context.subscriptions.push(vscode.workspace.registerTextDocumentContentProvider('hackmd', new MdTextDocumentContentProvider()));
-}
+}
+
+const checkEditorExist = (editor) => {
+    if (editor) {
+        return true;
+    } else {
+        vscode.window.showInformationMessage('Current window is not a text editor. Please open one first.');
+        return false;
+    }
+};
+
+const checkNoteIdExist = (noteId) => {
+    if (noteId) {
+        return true;
+    } else {
+        vscode.window.showInformationMessage("Please open a note first");
+        return false;
+    }
+};
+
+const checkNoteExist = (content) => {
+    if (content) {
+        return true;
+    } else {
+        vscode.window.showInformationMessage("Can't find the note from HackMD. Make sure it's still exist.");
+        return false;
+    }
+};
+