fix disconnect import wallet

FLock-io · Sep 15, 2023 · 5b2b17b · 5b2b17b
1 parent 79460a2
commit 5b2b17b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/renderer/CustomConnector.tsx b/src/renderer/CustomConnector.tsx
@@ -44,6 +44,10 @@ export default class CustomConnector extends Connector<any, any> {
     localStorage.setItem('privateKey', privateKey);
   }
 
+  async deletePrivateKey() {
+    localStorage.removeItem('privateKey');
+  }
+
   async connect({ chainId }: { chainId?: number } = {}): Promise<
     Required<ConnectorData>
   > {
@@ -96,7 +100,7 @@ export default class CustomConnector extends Connector<any, any> {
 
   async disconnect() {
     // Clear the private key and provider
-    this.setPrivateKey(undefined);
+    this.deletePrivateKey();
     this.#provider = undefined;
   }
 }
diff --git a/src/renderer/context/runnerContext.tsx b/src/renderer/context/runnerContext.tsx
@@ -20,7 +20,7 @@ export const RunnerContext = createContext<IRunnerContext>(
 export function RunnerContextProvider({
   children,
 }: RunnerContextProviderProps) {
-  const { connector, connectors } = useConnect();
+  const { connectors } = useConnect();
 
   const [runningTasks, setRunningTasks] = useState<string[]>([]);
   const [logs, setLogs] = useState<Map<string, string[]>>(new Map());
@@ -36,6 +36,7 @@ export function RunnerContextProvider({
 
   const runTask = async (task: TaskType, file: File) => {
     setLogs(new Map(logs.set(task.address, [])));
+
     // @ts-ignore
     let privateKey = await connectors[1].getPrivateKey();
 
@@ -45,6 +46,7 @@ export function RunnerContextProvider({
         method: 'eth_private_key',
       });
     }
+
     window.electron.ipcRenderer.sendMessage('ipc', [
       'join',
       task.address,