From d79a4a824323d1d6cca9b763e05e9d54ee1f2e10 Mon Sep 17 00:00:00 2001
From: Levi Yan <leviyanx@gmail.com>
Date: Mon, 28 Oct 2024 19:02:56 +0800
Subject: [PATCH] fix: now ssh connection use use private key instead of  the
 default

---
 src/home.ts             | 2 +-
 src/remote-container.ts | 8 +++++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/home.ts b/src/home.ts
index 092c9f7..3a5749a 100644
--- a/src/home.ts
+++ b/src/home.ts
@@ -12,8 +12,8 @@ export default class DSHome {
 
   constructor(context: vscode.ExtensionContext) {
     this.context = context;
-    this.remoteContainer = new RemoteContainer();
     this.user = new User(context);
+    this.remoteContainer = new RemoteContainer(this.user);
   }
 
   async toggle(url: string = DSHome.defaultUrl) {
diff --git a/src/remote-container.ts b/src/remote-container.ts
index f0403d9..4640b10 100644
--- a/src/remote-container.ts
+++ b/src/remote-container.ts
@@ -6,8 +6,14 @@ import * as rd from 'readline'
 const { NodeSSH } = require('node-ssh')
 
 import * as utils from './utils';
+import User from './user';
 
 export default class RemoteContainer {
+  private user:User;
+
+  constructor(user:User) {
+    this.user = user
+  }
 
   async firstConnect(host: string, username: string, port: number): Promise<string>; // connect with key
   // deprecated
@@ -29,7 +35,7 @@ export default class RemoteContainer {
             host: host,
             username: username,
             port: port,
-            privateKey: utils.getDefaultPrivateKey()
+            privateKeyPath: this.user.getUserPrivateKeyPath()
           });
           progress.report({ message: "连接成功，开始安装" });