Improve the experience of switch endpoint (LeetCode-OpenSource#92)

jdneo · web-flow · commit 8bd1feb2cb74 · 2019-01-21T18:52:21.000+08:00
diff --git a/package.json b/package.json
@@ -43,37 +43,40 @@
         "commands": [
             {
                 "command": "leetcode.deleteCache",
-                "title": "Delete cache",
+                "title": "Delete Cache",
                 "category": "LeetCode"
             },
             {
                 "command": "leetcode.toogleLeetCodeCn",
-                "title": "Switch endpoint",
+                "title": "Switch Endpoint",
                 "category": "LeetCode",
-                "icon": "resources/cn.png"
+                "icon": {
+                    "light": "resources/light/endpoint.svg",
+                    "dark": "resources/dark/endpoint.svg"
+                }
             },
             {
                 "command": "leetcode.signin",
-                "title": "Sign in",
+                "title": "Sign In",
                 "category": "LeetCode",
                 "icon": {
-                    "light": "resources/light/signin.png",
-                    "dark": "resources/dark/signin.png"
+                    "light": "resources/light/signin.svg",
+                    "dark": "resources/dark/signin.svg"
                 }
             },
             {
                 "command": "leetcode.signout",
-                "title": "Sign out",
+                "title": "Sign Out",
                 "category": "LeetCode"
             },
             {
                 "command": "leetcode.selectSessions",
-                "title": "Select session",
+                "title": "Select Session",
                 "category": "LeetCode"
             },
             {
                 "command": "leetcode.createSession",
-                "title": "Create new session",
+                "title": "Create New Session",
                 "category": "LeetCode"
             },
             {
@@ -87,14 +90,17 @@
             },
             {
                 "command": "leetcode.showProblem",
-                "title": "Show problem",
+                "title": "Show Problem",
                 "category": "LeetCode"
             },
             {
                 "command": "leetcode.searchProblem",
                 "title": "Search Problem",
                 "category": "LeetCode",
-                "icon": "resources/search.png"
+                "icon": {
+                    "light": "resources/light/search.svg",
+                    "dark": "resources/dark/search.svg"
+                }
             },
             {
                 "command": "leetcode.testSolution",
diff --git a/resources/cn.png b/resources/cn.png
diff --git a/resources/dark/endpoint.svg b/resources/dark/endpoint.svg
@@ -0,0 +1,6 @@
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" viewBox="0 0 426.667 426.667" style="enable-background:new 0 0 426.667 426.667;" xml:space="preserve" width="512px" height="512px" class=""><g><g>
+	<g>
+		<path d="M213.333,0C95.467,0,0,95.467,0,213.333s95.467,213.333,213.333,213.333S426.667,331.2,426.667,213.333S331.2,0,213.333,0    z M192,382.507c-84.16-10.453-149.333-82.133-149.333-169.173c0-13.12,1.6-25.92,4.48-38.187l102.187,102.187v21.333    c0,23.573,19.093,42.667,42.667,42.667V382.507z M339.093,328.427c-5.44-17.28-21.44-29.76-40.427-29.76h-21.333v-64    c0-11.733-9.6-21.333-21.333-21.333H128v-42.667h42.667c11.733,0,21.333-9.6,21.333-21.333v-42.667h42.667    c23.573,0,42.667-19.093,42.667-42.667v-8.853C339.84,80.427,384,141.653,384,213.333    C384,257.707,366.933,298.027,339.093,328.427z" data-original="#000000" class="active-path" data-old_color="#c8c8c8" fill="#c8c8c8"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/dark/refresh.svg b/resources/dark/refresh.svg
@@ -1 +1,6 @@
-<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16"><path d="M13.451 5.609l-.579-.939-1.068.812-.076.094c-.335.415-.927 1.341-1.124 2.876l-.021.165.033.163.071.345c0 1.654-1.346 3-3 3-.795 0-1.545-.311-2.107-.868-.563-.567-.873-1.317-.873-2.111 0-1.431 1.007-2.632 2.351-2.929v2.926s2.528-2.087 2.984-2.461h.012l3.061-2.582-4.919-4.1h-1.137v2.404c-3.429.318-6.121 3.211-6.121 6.721 0 1.809.707 3.508 1.986 4.782 1.277 1.282 2.976 1.988 4.784 1.988 3.722 0 6.75-3.028 6.75-6.75 0-1.245-.349-2.468-1.007-3.536z" fill="#2D2D30"/><path d="M12.6 6.134l-.094.071c-.269.333-.746 1.096-.91 2.375.057.277.092.495.092.545 0 2.206-1.794 4-4 4-1.098 0-2.093-.445-2.817-1.164-.718-.724-1.163-1.718-1.163-2.815 0-2.206 1.794-4 4-4l.351.025v1.85s1.626-1.342 1.631-1.339l1.869-1.577-3.5-2.917v2.218l-.371-.03c-3.176 0-5.75 2.574-5.75 5.75 0 1.593.648 3.034 1.695 4.076 1.042 1.046 2.482 1.694 4.076 1.694 3.176 0 5.75-2.574 5.75-5.75-.001-1.106-.318-2.135-.859-3.012z" fill="#C5C5C5"/></svg>
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" viewBox="0 0 341.333 341.333" style="enable-background:new 0 0 341.333 341.333;" xml:space="preserve" width="512px" height="512px"><g><g>
+	<g>
+		<path d="M341.227,149.333V0l-50.133,50.133C260.267,19.2,217.707,0,170.56,0C76.267,0,0.107,76.373,0.107,170.667    s76.16,170.667,170.453,170.667c79.467,0,146.027-54.4,164.907-128h-44.373c-17.6,49.707-64.747,85.333-120.533,85.333    c-70.72,0-128-57.28-128-128s57.28-128,128-128c35.307,0,66.987,14.72,90.133,37.867l-68.8,68.8H341.227z" data-original="#000000" class="active-path" data-old_color="#c8c8c8" fill="#c8c8c8"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/dark/search.svg b/resources/dark/search.svg
@@ -0,0 +1,6 @@
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" width="512px" height="512px" viewBox="0 0 446.25 446.25" style="enable-background:new 0 0 446.25 446.25;" xml:space="preserve" class=""><g><g>
+	<g id="search">
+		<path d="M318.75,280.5h-20.4l-7.649-7.65c25.5-28.05,40.8-66.3,40.8-107.1C331.5,73.95,257.55,0,165.75,0S0,73.95,0,165.75    S73.95,331.5,165.75,331.5c40.8,0,79.05-15.3,107.1-40.8l7.65,7.649v20.4L408,446.25L446.25,408L318.75,280.5z M165.75,280.5    C102,280.5,51,229.5,51,165.75S102,51,165.75,51S280.5,102,280.5,165.75S229.5,280.5,165.75,280.5z" data-original="#000000" class="active-path" data-old_color="#c8c8c8" fill="#c8c8c8"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/dark/signin.png b/resources/dark/signin.png
diff --git a/resources/dark/signin.svg b/resources/dark/signin.svg
@@ -0,0 +1,7 @@
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" width="512px" height="512px" viewBox="0 0 438.542 438.542" style="enable-background:new 0 0 438.542 438.542;" xml:space="preserve" class=""><g><g>
+	<g>
+		<path d="M414.41,60.676c-16.084-16.083-35.446-24.125-58.102-24.125h-91.357c-2.478,0-4.429,0.614-5.853,1.855    c-1.431,1.237-2.382,3.093-2.854,5.568c-0.479,2.474-0.76,4.709-0.853,6.707c-0.096,1.997-0.051,4.521,0.144,7.565    c0.186,3.046,0.281,4.949,0.281,5.713c0.571,1.524,0.767,2.81,0.571,3.855c-0.191,1.043,0.476,1.903,1.998,2.568    c1.52,0.666,2.279,1.191,2.279,1.569c0,0.378,1.096,0.662,3.285,0.855c2.19,0.192,3.289,0.284,3.289,0.284h3.713h3.142h82.228    c12.56,0,23.312,4.471,32.258,13.422c8.952,8.945,13.422,19.696,13.422,32.261V319.77c0,12.563-4.47,23.312-13.422,32.258    c-8.945,8.948-19.698,13.421-32.258,13.421h-91.357c-2.478,0-4.436,0.62-5.853,1.855c-1.43,1.242-2.382,3.094-2.857,5.564    c-0.476,2.478-0.763,4.716-0.855,6.714c-0.092,1.995-0.051,4.518,0.144,7.563c0.195,3.046,0.288,4.948,0.288,5.708    c0,2.478,0.896,4.613,2.707,6.427c1.81,1.807,3.949,2.71,6.427,2.71h91.357c22.648,0,42.018-8.042,58.095-24.133    c16.084-16.077,24.126-35.446,24.126-58.095V118.769C438.533,96.118,430.491,76.754,414.41,60.676z" data-original="#000000" class="active-path" data-old_color="#c8c8c8" fill="#c8c8c8"/>
+		<path d="M338.047,219.27c0-4.948-1.813-9.233-5.427-12.85L177.302,51.101c-3.616-3.612-7.895-5.424-12.847-5.424    c-4.952,0-9.233,1.812-12.85,5.424c-3.615,3.617-5.424,7.904-5.424,12.85v82.226H18.274c-4.952,0-9.235,1.812-12.851,5.424    C1.807,155.219,0,159.503,0,164.452v109.635c0,4.949,1.807,9.233,5.424,12.848c3.619,3.61,7.902,5.427,12.851,5.427h127.906    v82.225c0,4.945,1.809,9.233,5.424,12.847c3.621,3.617,7.902,5.428,12.85,5.428c4.949,0,9.231-1.811,12.847-5.428l155.318-155.312    C336.234,228.501,338.047,224.216,338.047,219.27z" data-original="#000000" class="active-path" data-old_color="#c8c8c8" fill="#c8c8c8"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/light/endpoint.svg b/resources/light/endpoint.svg
@@ -0,0 +1,6 @@
+?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" viewBox="0 0 426.667 426.667" style="enable-background:new 0 0 426.667 426.667;" xml:space="preserve" width="512px" height="512px" class=""><g><g>
+	<g>
+		<path d="M213.333,0C95.467,0,0,95.467,0,213.333s95.467,213.333,213.333,213.333S426.667,331.2,426.667,213.333S331.2,0,213.333,0    z M192,382.507c-84.16-10.453-149.333-82.133-149.333-169.173c0-13.12,1.6-25.92,4.48-38.187l102.187,102.187v21.333    c0,23.573,19.093,42.667,42.667,42.667V382.507z M339.093,328.427c-5.44-17.28-21.44-29.76-40.427-29.76h-21.333v-64    c0-11.733-9.6-21.333-21.333-21.333H128v-42.667h42.667c11.733,0,21.333-9.6,21.333-21.333v-42.667h42.667    c23.573,0,42.667-19.093,42.667-42.667v-8.853C339.84,80.427,384,141.653,384,213.333    C384,257.707,366.933,298.027,339.093,328.427z" data-original="#000000" class="active-path" fill="#000000"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/light/refresh.svg b/resources/light/refresh.svg
@@ -1 +1,6 @@
-<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16"><path d="M13.451 5.609l-.579-.939-1.068.812-.076.094c-.335.415-.927 1.341-1.124 2.876l-.021.165.033.163.071.345c0 1.654-1.346 3-3 3-.795 0-1.545-.311-2.107-.868-.563-.567-.873-1.317-.873-2.111 0-1.431 1.007-2.632 2.351-2.929v2.926s2.528-2.087 2.984-2.461h.012l3.061-2.582-4.919-4.1h-1.137v2.404c-3.429.318-6.121 3.211-6.121 6.721 0 1.809.707 3.508 1.986 4.782 1.277 1.282 2.976 1.988 4.784 1.988 3.722 0 6.75-3.028 6.75-6.75 0-1.245-.349-2.468-1.007-3.536z" fill="#F6F6F6"/><path d="M12.6 6.134l-.094.071c-.269.333-.746 1.096-.91 2.375.057.277.092.495.092.545 0 2.206-1.794 4-4 4-1.098 0-2.093-.445-2.817-1.164-.718-.724-1.163-1.718-1.163-2.815 0-2.206 1.794-4 4-4l.351.025v1.85s1.626-1.342 1.631-1.339l1.869-1.577-3.5-2.917v2.218l-.371-.03c-3.176 0-5.75 2.574-5.75 5.75 0 1.593.648 3.034 1.695 4.076 1.042 1.046 2.482 1.694 4.076 1.694 3.176 0 5.75-2.574 5.75-5.75-.001-1.106-.318-2.135-.859-3.012z" fill="#424242"/></svg>
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" viewBox="0 0 341.333 341.333" style="enable-background:new 0 0 341.333 341.333;" xml:space="preserve" width="512px" height="512px" class=""><g><g>
+	<g>
+		<path d="M341.227,149.333V0l-50.133,50.133C260.267,19.2,217.707,0,170.56,0C76.267,0,0.107,76.373,0.107,170.667    s76.16,170.667,170.453,170.667c79.467,0,146.027-54.4,164.907-128h-44.373c-17.6,49.707-64.747,85.333-120.533,85.333    c-70.72,0-128-57.28-128-128s57.28-128,128-128c35.307,0,66.987,14.72,90.133,37.867l-68.8,68.8H341.227z" data-original="#000000" class="active-path" data-old_color="#C8C8C8" fill="#000000"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/light/search.svg b/resources/light/search.svg
@@ -0,0 +1,6 @@
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" width="512px" height="512px" viewBox="0 0 446.25 446.25" style="enable-background:new 0 0 446.25 446.25;" xml:space="preserve" class=""><g><g>
+	<g id="search">
+		<path d="M318.75,280.5h-20.4l-7.649-7.65c25.5-28.05,40.8-66.3,40.8-107.1C331.5,73.95,257.55,0,165.75,0S0,73.95,0,165.75    S73.95,331.5,165.75,331.5c40.8,0,79.05-15.3,107.1-40.8l7.65,7.649v20.4L408,446.25L446.25,408L318.75,280.5z M165.75,280.5    C102,280.5,51,229.5,51,165.75S102,51,165.75,51S280.5,102,280.5,165.75S229.5,280.5,165.75,280.5z" data-original="#000000" class="active-path" fill="#000000"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/light/signin.png b/resources/light/signin.png
diff --git a/resources/light/signin.svg b/resources/light/signin.svg
@@ -0,0 +1,7 @@
+<?xml version="1.0"?>
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" version="1.1" id="Capa_1" x="0px" y="0px" width="512px" height="512px" viewBox="0 0 438.542 438.542" style="enable-background:new 0 0 438.542 438.542;" xml:space="preserve"><g><g>
+	<g>
+		<path d="M414.41,60.676c-16.084-16.083-35.446-24.125-58.102-24.125h-91.357c-2.478,0-4.429,0.614-5.853,1.855    c-1.431,1.237-2.382,3.093-2.854,5.568c-0.479,2.474-0.76,4.709-0.853,6.707c-0.096,1.997-0.051,4.521,0.144,7.565    c0.186,3.046,0.281,4.949,0.281,5.713c0.571,1.524,0.767,2.81,0.571,3.855c-0.191,1.043,0.476,1.903,1.998,2.568    c1.52,0.666,2.279,1.191,2.279,1.569c0,0.378,1.096,0.662,3.285,0.855c2.19,0.192,3.289,0.284,3.289,0.284h3.713h3.142h82.228    c12.56,0,23.312,4.471,32.258,13.422c8.952,8.945,13.422,19.696,13.422,32.261V319.77c0,12.563-4.47,23.312-13.422,32.258    c-8.945,8.948-19.698,13.421-32.258,13.421h-91.357c-2.478,0-4.436,0.62-5.853,1.855c-1.43,1.242-2.382,3.094-2.857,5.564    c-0.476,2.478-0.763,4.716-0.855,6.714c-0.092,1.995-0.051,4.518,0.144,7.563c0.195,3.046,0.288,4.948,0.288,5.708    c0,2.478,0.896,4.613,2.707,6.427c1.81,1.807,3.949,2.71,6.427,2.71h91.357c22.648,0,42.018-8.042,58.095-24.133    c16.084-16.077,24.126-35.446,24.126-58.095V118.769C438.533,96.118,430.491,76.754,414.41,60.676z" data-original="#000000" class="active-path" fill="#000000"/>
+		<path d="M338.047,219.27c0-4.948-1.813-9.233-5.427-12.85L177.302,51.101c-3.616-3.612-7.895-5.424-12.847-5.424    c-4.952,0-9.233,1.812-12.85,5.424c-3.615,3.617-5.424,7.904-5.424,12.85v82.226H18.274c-4.952,0-9.235,1.812-12.851,5.424    C1.807,155.219,0,159.503,0,164.452v109.635c0,4.949,1.807,9.233,5.424,12.848c3.619,3.61,7.902,5.427,12.851,5.427h127.906    v82.225c0,4.945,1.809,9.233,5.424,12.847c3.621,3.617,7.902,5.428,12.85,5.428c4.949,0,9.231-1.811,12.847-5.428l155.318-155.312    C336.234,228.501,338.047,224.216,338.047,219.27z" data-original="#000000" class="active-path" fill="#000000"/>
+	</g>
+</g></g> </svg>
diff --git a/resources/search.png b/resources/search.png
diff --git a/src/commands/plugin.ts b/src/commands/plugin.ts
@@ -1,31 +1,28 @@
 // Copyright (c) jdneo. All rights reserved.
 // Licensed under the MIT license.
 
-import * as fse from "fs-extra";
-import * as os from "os";
-import * as path from "path";
 import * as vscode from "vscode";
 import { leetCodeExecutor } from "../leetCodeExecutor";
 import { IQuickItemEx } from "../shared";
 import { Endpoint } from "../shared";
 import { DialogType, promptForOpenOutputChannel, promptForSignIn } from "../utils/uiUtils";
 import { deleteCache } from "./cache";
 
-export async function toogleLeetCodeCn(): Promise<void> {
-    const isCnEnbaled: boolean = isLeetCodeCnEnabled();
+export async function switchEndpoint(): Promise<void> {
+    const isCnEnbaled: boolean = getLeetCodeEndpoint() === Endpoint.LeetCodeCN;
     const picks: Array<IQuickItemEx<string>> = [];
     picks.push(
         {
-            label: `${isCnEnbaled ? "$(check) " : ""}On`,
-            description: "",
-            detail: `Enable ${Endpoint.LeetCodeCN}.`,
-            value: "on",
+            label: `${isCnEnbaled ? "" : "$(check) "}LeetCode`,
+            description: "leetcode.com",
+            detail: `Enable LeetCode US`,
+            value: Endpoint.LeetCode,
         },
         {
-            label: `${isCnEnbaled ? "" : "$(check) "}Off`,
-            description: "",
-            detail: `Disable ${Endpoint.LeetCodeCN}.`,
-            value: "off",
+            label: `${isCnEnbaled ? "$(check) " : ""}力扣`,
+            description: "leetcode-cn.com",
+            detail: `启用中国版 LeetCode`,
+            value: Endpoint.LeetCodeCN,
         },
     );
     const choice: IQuickItemEx<string> | undefined = await vscode.window.showQuickPick(picks);
@@ -34,9 +31,8 @@ export async function toogleLeetCodeCn(): Promise<void> {
     }
     const leetCodeConfig: vscode.WorkspaceConfiguration = vscode.workspace.getConfiguration("leetcode");
     try {
-        const enabled: boolean = choice.value === "on";
-        const endpoint: string = enabled ? Endpoint.LeetCodeCN : Endpoint.LeetCode;
-        await leetCodeExecutor.toggleLeetCodeCn(enabled);
+        const endpoint: string = choice.value;
+        await leetCodeExecutor.switchEndpoint(endpoint);
         await leetCodeConfig.update("endpoint", endpoint, true /* UserSetting */);
         vscode.window.showInformationMessage(`Switched the endpoint to ${endpoint}`);
     } catch (error) {
@@ -52,29 +48,7 @@ export async function toogleLeetCodeCn(): Promise<void> {
     }
 }
 
-export async function initializeEndpoint(): Promise<void> {
-    const isCnEnabledInExtension: boolean = isLeetCodeCnEnabled();
-    const isCnEnabledInCli: boolean = await isLeetCodeCnEnabledInCli();
-    await leetCodeExecutor.toggleLeetCodeCn(isCnEnabledInExtension);
-    if (isCnEnabledInCli !== isCnEnabledInExtension) {
-        await deleteCache();
-    }
-}
-
-export function isLeetCodeCnEnabled(): boolean {
+export function getLeetCodeEndpoint(): string {
     const leetCodeConfig: vscode.WorkspaceConfiguration = vscode.workspace.getConfiguration("leetcode");
-    const endpoint: string | undefined = leetCodeConfig.get<string>("endpoint");
-    if (endpoint && endpoint === Endpoint.LeetCodeCN) {
-        return true;
-    }
-    return false;
-}
-
-async function isLeetCodeCnEnabledInCli(): Promise<boolean> {
-    const pluginsStatusFile: string = path.join(os.homedir(), ".lc", "plugins.json");
-    if (!await fse.pathExists(pluginsStatusFile)) {
-        return false;
-    }
-    const pluginsObj: {} = await fse.readJson(pluginsStatusFile);
-    return pluginsObj["leetcode.cn"];
+    return leetCodeConfig.get<string>("endpoint", Endpoint.LeetCode);
 }
diff --git a/src/extension.ts b/src/extension.ts
@@ -36,7 +36,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
         vscode.window.registerTreeDataProvider("leetCodeExplorer", leetCodeTreeDataProvider),
         vscode.languages.registerCodeLensProvider({ scheme: "file" }, codeLensProvider),
         vscode.commands.registerCommand("leetcode.deleteCache", () => cache.deleteCache()),
-        vscode.commands.registerCommand("leetcode.toogleLeetCodeCn", () => plugin.toogleLeetCodeCn()),
+        vscode.commands.registerCommand("leetcode.toogleLeetCodeCn", () => plugin.switchEndpoint()),
         vscode.commands.registerCommand("leetcode.signin", () => leetCodeManager.signIn()),
         vscode.commands.registerCommand("leetcode.signout", () => leetCodeManager.signOut()),
         vscode.commands.registerCommand("leetcode.selectSessions", () => session.selectSession()),
@@ -48,7 +48,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>
         vscode.commands.registerCommand("leetcode.submitSolution", (uri?: vscode.Uri) => submit.submitSolution(uri)),
     );
 
-    await plugin.initializeEndpoint();
+    await leetCodeExecutor.switchEndpoint(plugin.getLeetCodeEndpoint());
     leetCodeManager.getLoginStatus();
 }
 
diff --git a/src/leetCodeExecutor.ts b/src/leetCodeExecutor.ts
@@ -4,6 +4,7 @@
 import * as cp from "child_process";
 import * as path from "path";
 import * as vscode from "vscode";
+import { Endpoint } from "./shared";
 import { executeCommand, executeCommandWithProgress } from "./utils/cpUtils";
 import { DialogOptions, openUrl } from "./utils/uiUtils";
 import * as wsl from "./utils/wslUtils";
@@ -89,11 +90,14 @@ class LeetCodeExecutor {
         return await this.executeCommandWithProgressEx("Submitting to LeetCode...", "node", [await this.getLeetCodeBinaryPath(), "test", `"${filePath}"`]);
     }
 
-    public async toggleLeetCodeCn(isEnable: boolean): Promise<string> {
-        if (isEnable) {
-            return await this.executeCommandEx("node", [await this.getLeetCodeBinaryPath(), "plugin", "-e", "leetcode.cn"]);
+    public async switchEndpoint(endpoint: string): Promise<string> {
+        switch (endpoint) {
+            case Endpoint.LeetCodeCN:
+                return await this.executeCommandEx("node", [await this.getLeetCodeBinaryPath(), "plugin", "-e", "leetcode.cn"]);
+            case Endpoint.LeetCode:
+            default:
+                return await this.executeCommandEx("node", [await this.getLeetCodeBinaryPath(), "plugin", "-d", "leetcode.cn"]);
         }
-        return await this.executeCommandEx("node", [await this.getLeetCodeBinaryPath(), "plugin", "-d", "leetcode.cn"]);
     }
 
     private async executeCommandEx(command: string, args: string[], options: cp.SpawnOptions = { shell: true }): Promise<string> {
diff --git a/src/utils/uiUtils.ts b/src/utils/uiUtils.ts
