From 473eab1c8c96c4c2d561f7219de5dec97b0feaf7 Mon Sep 17 00:00:00 2001
From: Levi Yan <leviyanx@gmail.com>
Date: Wed, 23 Apr 2025 09:13:42 +0800
Subject: [PATCH] feat: clean - clean ssh key and logout

---
 src/main.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/main.ts b/src/main.ts
index 346a6d2..075ce29 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -1,4 +1,5 @@
 import * as vscode from 'vscode';
+import * as fs from 'fs';
 import QuickAccessTreeProvider from './views/quick-access-tree';
 import DSHome from './home';
 import RemoteContainer, { openProjectWithoutLogging } from './remote-container';
@@ -95,7 +96,17 @@ export class DevStarExtension {
       vscode.commands.registerCommand('devstar.showHome', (url: string) =>
         this.dsHome.toggle(url)
       ),
-      vscode.commands.registerCommand('devstar.logout', () => {
+      vscode.commands.registerCommand('devstar.clean', () => {
+        // 先清除ssh key
+        if (fs.existsSync(this.user.getUserPrivateKeyPath())) {
+          fs.unlinkSync(this.user.getUserPrivateKeyPath())
+        }
+
+        if (fs.existsSync(this.user.getUserPublicKeyPath())) {
+          fs.unlinkSync(this.user.getUserPublicKeyPath())
+        }
+        console.log("User's ssh key has been deleted!")
+        // 退出登录
         this.user.logout()
       })
     );