lat5211/devstar_plugin
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: only successfully finished first connect then open remote folder

Browse Source
This commit is contained in:
Levi Yan
2024-07-30 23:35:08 +08:00
parent 9b95f6376c
commit 3b7cf8af57
2 changed files with 44 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/home.ts
View File

@@ -31,8 +31,13 @@ export default class DSHome {
async (message) => {
switch (message.command) {
case 'firstOpenRemoteFolder':
await this.remoteContainer.firstConnect(message.host, message.username, message.password, message.port);
this.remoteContainer.openRemoteFolder(message.host, message.path);
await this.remoteContainer.firstConnect(message.host, message.username, message.password, message.port)
.then((result) => {
if (result === 'success') {
// only success then open folder
this.remoteContainer.openRemoteFolder(message.host, message.path);
}
})
return;
case 'openRemoteFolder':
this.remoteContainer.openRemoteFolder(message.host, message.path);

70
src/remote-container.ts
View File

@@ -3,45 +3,49 @@ import * as path from 'path';
import * as os from 'os';
import * as vscode from 'vscode';
const { NodeSSH } = require('node-ssh')
import * as utils from './utils';
export default class RemoteContainer {
async firstConnect(host: string, username: string, password: string, port: number): Promise<void> {
// connect the host and add the public key to the remote authorized_keys
const defaultPublicKeyPath = path.join(os.homedir(), '.ssh', 'id_rsa.pub');
// TODO: if there is no public key, generate one
const publicKey = fs.readFileSync(defaultPublicKeyPath, 'utf8');
async firstConnect(host: string, username: string, password: string, port: number): Promise<string> {
return new Promise(async (resolve, reject) => {
// connect the host and add the public key to the remote authorized_keys
const defaultPublicKeyPath = path.join(os.homedir(), '.ssh', 'id_rsa.pub');
// TODO: if there is no public key, generate one
const publicKey = fs.readFileSync(defaultPublicKeyPath, 'utf8');
// first connect to the remote host using password
const ssh = new NodeSSH();
try {
await ssh.connect({
host: host,
username: username,
password: password,
port: port
});
// first connect to the remote host using password
const ssh = new NodeSSH();
try {
await ssh.connect({
host: host,
username: username,
password: password,
port: port
});
// add the public key to the remote authorized_keys
await ssh.execCommand(`mkdir -p ~/.ssh && echo '${publicKey}' >> ~/.ssh/authorized_keys`);
console.log('Public key added to remote authorized_keys');
await ssh.dispose();
// add the public key to the remote authorized_keys
await ssh.execCommand(`mkdir -p ~/.ssh && echo '${publicKey}' >> ~/.ssh/authorized_keys`);
console.log('Public key added to remote authorized_keys');
} catch (error) {
console.error('Error adding public key: ', error);
} finally {
await ssh.dispose();
}
// TODO: request change password
// TODO: use remote.SSH.configFile
const sshConfigPath = path.join(os.homedir(), '.ssh', 'config');
// append the host to the local ssh config file
const sshConfigContent =
`\nHost ${host}\n HostName ${host}\n Port ${port}\n User ${username}\n PreferredAuthentications publickey\n IdentityFile ~/.ssh/id_rsa\n `;
// append the host to the local ssh config file
fs.writeFileSync(sshConfigPath, sshConfigContent, { encoding: 'utf8', flag: 'a' });
console.log('Host registered in local ssh config');
// only connect successfully then save the host info
// TODO: request change password
// TODO: use remote.SSH.configFile
const sshConfigPath = path.join(os.homedir(), '.ssh', 'config');
// append the host to the local ssh config file
const sshConfigContent =
`\nHost ${host}\n HostName ${host}\n Port ${port}\n User ${username}\n PreferredAuthentications publickey\n IdentityFile ~/.ssh/id_rsa\n `;
// append the host to the local ssh config file
fs.writeFileSync(sshConfigPath, sshConfigContent, { encoding: 'utf8', flag: 'a' });
console.log('Host registered in local ssh config');
resolve("success");
} catch (error) {
console.error('Error adding public key: ', error);
await ssh.dispose();
reject("failed");
}
});
}