Clear login status when opening the extension if the cookie is deleted

Author: JamesCamel

package.json

@@ -46,12 +46,12 @@
 			"properties": {
 				"Hackmd.serverURL": {
 					"type": "string",
-					"default": "https://hackmd.io/",
+					"default": "https://hackmd.io",
 					"markdownDescription": "Specify the self-hosted server URL for HackMD EE or CodiMD."
 				},
 				"Hackmd.enterprise": {
 					"type": "boolean",
-					"default": false,
+					"default": true,
 					"description": "Check this if you are connecting to HackMD Enterprise server which contains more features."
 				}
 			}

src/commands/index.ts

@@ -1,11 +1,10 @@
 import * as vscode from 'vscode';
-import { registerUserCommands } from './user'
-import { registerTreeViewCommands } from './treeView'
+import { registerUserCommands } from './user';
+import { registerTreeViewCommands } from './treeView';
 import { registerNoteCommands } from './note';
-import { Store } from '../store';
 
-export function registerCommand(context: vscode.ExtensionContext, store: Store) {
-    registerUserCommands(context, store);
-    registerTreeViewCommands(context, store);
+export function registerCommands(context: vscode.ExtensionContext) {
+    registerUserCommands(context);
+    registerTreeViewCommands(context);
     registerNoteCommands(context);
 }

src/commands/treeView.ts

@@ -1,15 +1,18 @@
 import * as vscode from 'vscode';
-import { Store } from '../store';
 import { HackMDTreeViewProvider } from './../tree/index';
 import { NoteTreeNode } from './../tree/nodes';
 import { MdTextDocumentContentProvider } from './../mdTextDocument';
-import { refreshHistoryList } from './../utils';
+import { refreshHistoryList, refreshLoginStatus, refreshLoginCredential, getLoginCredential } from './../utils';
 import { API, ExportType } from './../api';
 
-export async function registerTreeViewCommands(context: vscode.ExtensionContext, store: Store) {
-    const hackMDTreeViewProvider = new HackMDTreeViewProvider(store);
+export async function registerTreeViewCommands(context: vscode.ExtensionContext) {
+    const hackMDTreeViewProvider = new HackMDTreeViewProvider();
     context.subscriptions.push(vscode.window.registerTreeDataProvider('mdTreeItems', hackMDTreeViewProvider));
-    context.subscriptions.push(vscode.commands.registerCommand('treeView.refreshList', async () => await refreshHistoryList()));
+    context.subscriptions.push(vscode.commands.registerCommand('treeView.refreshList', async () => {
+        await refreshLoginStatus();
+        await refreshHistoryList();
+        await refreshLoginCredential(context);
+    }));
 
     context.subscriptions.push(vscode.commands.registerCommand('clickTreeItem', async (label, noteId) => {
         if (label && noteId) {
@@ -36,7 +39,7 @@ export async function registerTreeViewCommands(context: vscode.ExtensionContext,
     context.subscriptions.push(vscode.commands.registerCommand('note.showPreviewAndEditor', async (node: NoteTreeNode) => {
         const noteNode = node;
         if (noteNode.label && noteNode.noteId) {
-            const content = await API.exportString(noteNode.noteId,ExportType.MD);
+            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);

src/commands/user.ts

@@ -1,9 +1,8 @@
 import * as vscode from 'vscode';
-import { checkLogin, login, refreshHistoryList } from './../utils'
-import { Store } from '../store';
+import { checkLogin, login, refreshHistoryList, refreshLoginStatus } from './../utils';
 import { API } from './../api';
 
-export async function registerUserCommands(context: vscode.ExtensionContext, store: Store) {
+export async function registerUserCommands(context: vscode.ExtensionContext) {
     context.subscriptions.push(vscode.commands.registerCommand('HackMD.login', async () => {
         if (await checkLogin()) {
             vscode.window.showInformationMessage('Already logged in, please log out first.');
@@ -50,8 +49,8 @@ export async function registerUserCommands(context: vscode.ExtensionContext, sto
           return;
         }
         await API.logout();
-        store.isLogin = false;
-        vscode.window.showInformationMessage('Successfully logged out.');
+        await refreshLoginStatus();
         await refreshHistoryList();
+        vscode.window.showInformationMessage('Successfully logged out.');
       }));
 }

src/extension.ts

@@ -4,10 +4,9 @@ import axios from 'axios';
 import * as vscode from 'vscode';
 import * as markdownitContainer from 'markdown-it-container';
 import * as S from 'string';
-import { store } from './store';
 import { initializeStorage } from './store/storage';
 import * as Prism from 'prismjs';
-import { registerCommand } from './commands';
+import { registerCommands } from './commands';
 
 require('prismjs/components/prism-wiki');
 require('prismjs/components/prism-haskell');
@@ -228,8 +227,8 @@ let highlight;
 axios.defaults.withCredentials = true;
 
 export async function activate(context: vscode.ExtensionContext) {
-  initializeStorage(context);
-  registerCommand(context, store);
+  initializeStorage();
+  registerCommands(context);
 
   return {
     extendMarkdownIt(md: any) {

src/store/storage.ts

@@ -1,18 +1,8 @@
-import { reaction } from "mobx";
 import { store } from '.';
-import * as vscode from 'vscode';
+import { checkLogin } from "../utils";
+import { API } from './../api';
 
-export async function initializeStorage(context: vscode.ExtensionContext) {
-    store.history = context.globalState.get('history');
-    store.isLogin = context.globalState.get('isLogin');
-    reaction(
-        () => [store.history],
-        () => {
-            context.globalState.update('history', store.history);
-            vscode.commands.executeCommand("setContext", 'history', store.history);
-
-            context.globalState.update('isLogin', store.isLogin);
-            vscode.commands.executeCommand("setContext", 'isLogin', store.isLogin);
-        }
-    );
+export async function initializeStorage() {
+    store.history = store.history = (await API.getHistory()).history.reverse();
+    store.isLogin = await checkLogin();
 }

src/tree/index.ts

@@ -1,17 +1,18 @@
 import * as vscode from 'vscode';
-import { Store, store } from '../store';
+import { store } from '../store';
 import { reaction } from 'mobx';
 import { TreeNode, NoteTreeNode } from './nodes';
 
 export class HackMDTreeViewProvider implements vscode.TreeDataProvider<TreeNode> {
     private _onDidChangeTreeData = new vscode.EventEmitter<TreeNode>();
     public readonly onDidChangeTreeData: vscode.Event<TreeNode> = this._onDidChangeTreeData.event;
-    constructor(private store: Store) {
+    constructor() {
         reaction(
             () => [
-                store.history
+                store.history,
+                store.isLogin
             ],
-            () => {
+            async () => {
                 this.refresh();
             }
         );
@@ -24,9 +25,9 @@ export class HackMDTreeViewProvider implements vscode.TreeDataProvider<TreeNode>
     getChildren(element?: TreeNode): vscode.ProviderResult<TreeNode[]> {
         if (store.isLogin) {
             if (element === undefined) {
-                return [new TreeNode("history", vscode.TreeItemCollapsibleState.Collapsed)]
+                return [new TreeNode("history", vscode.TreeItemCollapsibleState.Collapsed)];
             } else {
-                return this.store.history.map(item =>
+                return store.history.map(item =>
                     new NoteTreeNode(item.id, item.text, vscode.TreeItemCollapsibleState.None)
                 );
             }

src/utils.ts

@@ -8,6 +8,18 @@ export const refreshHistoryList = async () => {
     } else {
         store.history = [{}];
     }
+
+};
+
+export const refreshLoginStatus = async () => {
+    store.isLogin = await checkLogin();
+};
+
+export const refreshLoginCredential = async (context: vscode.ExtensionContext) => {
+    if (!(await checkLogin())) {
+        context.globalState.update('email', undefined);
+        context.globalState.update('password', undefined);
+    }
 };
 
 export const checkLogin = async () => {
@@ -17,7 +29,7 @@ export const checkLogin = async () => {
 export const login = async (context: vscode.ExtensionContext) => {
     const { email, password } = getLoginCredential(context);
     if (!email || !password) {
-        vscode.window.showInformationMessage('Please enter your email and password to use HackMD extension!')
+        vscode.window.showInformationMessage('Please enter your email and password to use HackMD extension!');
         return;
     }
     await API.login(email, password);