style: tab -> 2 spaces in all typescript files
This commit is contained in:
Levi Yan
50
src/home.ts
50
src/home.ts
@@ -7,15 +7,15 @@ import * as utils from './utils'
|
||||
export default class DSHome {
|
||||
private context: vscode.ExtensionContext;
|
||||
private remoteContainer: RemoteContainer;
|
||||
private user:User;
|
||||
private devstarHomePageUrl:string;
|
||||
private user: User;
|
||||
private devstarHomePageUrl: string;
|
||||
|
||||
constructor(context: vscode.ExtensionContext) {
|
||||
this.context = context;
|
||||
this.user = new User(context);
|
||||
this.remoteContainer = new RemoteContainer(this.user);
|
||||
|
||||
const devstarHomePageUrl:string|undefined = vscode.workspace.getConfiguration('devstar').get('devstarHomePage')
|
||||
const devstarHomePageUrl: string | undefined = vscode.workspace.getConfiguration('devstar').get('devstarHomePage')
|
||||
if (undefined == devstarHomePageUrl || "" == devstarHomePageUrl) {
|
||||
this.devstarHomePageUrl = "https://devstar.cn/devstar-home"
|
||||
// this.devstarHomePageUrl = "http://localhost:3000/devstar-home"
|
||||
@@ -44,31 +44,31 @@ export default class DSHome {
|
||||
case 'getUserToken':
|
||||
const userToken = this.user.getUserTokenFromLocal()
|
||||
if (userToken === undefined) {
|
||||
panel.webview.postMessage({ command: 'getUserToken', data: {userToken: ''}})
|
||||
panel.webview.postMessage({ command: 'getUserToken', data: { userToken: '' } })
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({ command: 'getUserToken', data: {userToken: userToken}})
|
||||
panel.webview.postMessage({ command: 'getUserToken', data: { userToken: userToken } })
|
||||
break;
|
||||
}
|
||||
case 'getUsername':
|
||||
const username = this.user.getUsernameFromLocal()
|
||||
if (username === undefined) {
|
||||
panel.webview.postMessage({command: 'getUsername', data: {username: ''}})
|
||||
panel.webview.postMessage({ command: 'getUsername', data: { username: '' } })
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({command: 'getUsername', data: {username: username}})
|
||||
panel.webview.postMessage({ command: 'getUsername', data: { username: username } })
|
||||
break;
|
||||
}
|
||||
case 'firstOpenRemoteFolder':
|
||||
await this.remoteContainer.firstConnect(data.host, data.username, data.port)
|
||||
.then((_res) => {
|
||||
if (_res == 'success') {
|
||||
// only success then open folder
|
||||
this.remoteContainer.openRemoteFolder(data.host, data.username, data.port, data.path);
|
||||
}
|
||||
}).catch(error => {
|
||||
console.error(`Failed to connect ${data.host}: `, error)
|
||||
})
|
||||
.then((_res) => {
|
||||
if (_res == 'success') {
|
||||
// only success then open folder
|
||||
this.remoteContainer.openRemoteFolder(data.host, data.username, data.port, data.path);
|
||||
}
|
||||
}).catch(error => {
|
||||
console.error(`Failed to connect ${data.host}: `, error)
|
||||
})
|
||||
break;
|
||||
case 'openRemoteFolder':
|
||||
this.remoteContainer.openRemoteFolder(data.host, data.username, data.port, data.path);
|
||||
@@ -77,43 +77,43 @@ export default class DSHome {
|
||||
case 'setUserToken':
|
||||
this.user.setUserTokenToLocal(data.userToken)
|
||||
if (data.userToken === this.user.getUserTokenFromLocal()) {
|
||||
panel.webview.postMessage({ command: 'setUserToken', data: {ok: true}})
|
||||
panel.webview.postMessage({ command: 'setUserToken', data: { ok: true } })
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({ command: 'setUserToken', data: {ok: false}})
|
||||
panel.webview.postMessage({ command: 'setUserToken', data: { ok: false } })
|
||||
break;
|
||||
}
|
||||
case 'setUsername':
|
||||
this.user.setUsernameToLocal(data.username);
|
||||
if (data.username === this.user.getUsernameFromLocal()) {
|
||||
panel.webview.postMessage({command: 'setUsername', data: {ok: true}});
|
||||
panel.webview.postMessage({ command: 'setUsername', data: { ok: true } });
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({command: 'setUsername', data: {ok: false}});
|
||||
panel.webview.postMessage({ command: 'setUsername', data: { ok: false } });
|
||||
break;
|
||||
}
|
||||
case 'getUserPublicKey':
|
||||
var userPublicKey = '';
|
||||
if (this.user.existUserPrivateKey()) {
|
||||
userPublicKey = this.user.getUserPublicKey();
|
||||
panel.webview.postMessage({command: 'getUserPublicKey', data: {userPublicKey: userPublicKey}})
|
||||
panel.webview.postMessage({ command: 'getUserPublicKey', data: { userPublicKey: userPublicKey } })
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({command: 'getUserPublicKey', data: {userPublicKey: userPublicKey}})
|
||||
panel.webview.postMessage({ command: 'getUserPublicKey', data: { userPublicKey: userPublicKey } })
|
||||
break;
|
||||
}
|
||||
case 'createUserPublicKey':
|
||||
await this.user.createUserSSHKey();
|
||||
if (this.user.existUserPublicKey()) {
|
||||
panel.webview.postMessage({command: 'createUserPublicKey', data: {ok: true}})
|
||||
panel.webview.postMessage({ command: 'createUserPublicKey', data: { ok: true } })
|
||||
break;
|
||||
} else {
|
||||
panel.webview.postMessage({command: 'createUserPublicKey', data: {ok: false}})
|
||||
panel.webview.postMessage({ command: 'createUserPublicKey', data: { ok: false } })
|
||||
break;
|
||||
}
|
||||
case 'getMachineName':
|
||||
const machineName = os.hostname();
|
||||
panel.webview.postMessage({command: 'getMachineName', data: {machineName: machineName}})
|
||||
panel.webview.postMessage({ command: 'getMachineName', data: { machineName: machineName } })
|
||||
case 'getDefaultPublicKey':
|
||||
var defaultPublicKey;
|
||||
if (utils.existDefaultPublicKey()) {
|
||||
@@ -123,7 +123,7 @@ export default class DSHome {
|
||||
utils.createSSHKey();
|
||||
defaultPublicKey = utils.getDefaultPublicKey();
|
||||
}
|
||||
panel.webview.postMessage({ command: 'getDefaultPublicKey', data: {defaultPublicKey: defaultPublicKey}})
|
||||
panel.webview.postMessage({ command: 'getDefaultPublicKey', data: { defaultPublicKey: defaultPublicKey } })
|
||||
}
|
||||
},
|
||||
undefined,
|
||||
|
||||
@@ -9,40 +9,40 @@ import * as utils from './utils';
|
||||
import User from './user';
|
||||
|
||||
export default class RemoteContainer {
|
||||
private user:User;
|
||||
private user: User;
|
||||
|
||||
constructor(user:User) {
|
||||
constructor(user: User) {
|
||||
this.user = user
|
||||
}
|
||||
|
||||
async firstConnect(host: string, username: string, port: number): Promise<string>; // connect with key
|
||||
// deprecated
|
||||
async firstConnect(host: string, username: string, port: number, password: string): Promise<string>;
|
||||
async firstConnect(host: string, username: string, port: number, password: string): Promise<string>;
|
||||
|
||||
async firstConnect(host: string, username: string, port: number, password?: string): Promise<string> {
|
||||
return new Promise(async (resolve, reject) => {
|
||||
const ssh = new NodeSSH();
|
||||
vscode.window.withProgress({
|
||||
location: vscode.ProgressLocation.Notification,
|
||||
title: "正在容器中安装vscode-server及devstar插件",
|
||||
cancellable: false
|
||||
location: vscode.ProgressLocation.Notification,
|
||||
title: "正在容器中安装vscode-server及devstar插件",
|
||||
cancellable: false
|
||||
}, async (progress) => {
|
||||
try {
|
||||
try {
|
||||
|
||||
if (password === undefined) {
|
||||
// connect with key
|
||||
await ssh.connect({
|
||||
host: host,
|
||||
username: username,
|
||||
port: port,
|
||||
privateKeyPath: this.user.getUserPrivateKeyPath()
|
||||
});
|
||||
progress.report({ message: "连接成功，开始安装" });
|
||||
if (password === undefined) {
|
||||
// connect with key
|
||||
await ssh.connect({
|
||||
host: host,
|
||||
username: username,
|
||||
port: port,
|
||||
privateKeyPath: this.user.getUserPrivateKeyPath()
|
||||
});
|
||||
progress.report({ message: "连接成功，开始安装" });
|
||||
|
||||
// install vscode-server and devstar extension
|
||||
const vscodeCommitId = await utils.getVsCodeCommitId();
|
||||
const vscodeServerUrl = `https://vscode.download.prss.microsoft.com/dbazure/download/stable/${vscodeCommitId}/vscode-server-linux-x64.tar.gz`
|
||||
const installVscodeServerScript = `
|
||||
// install vscode-server and devstar extension
|
||||
const vscodeCommitId = await utils.getVsCodeCommitId();
|
||||
const vscodeServerUrl = `https://vscode.download.prss.microsoft.com/dbazure/download/stable/${vscodeCommitId}/vscode-server-linux-x64.tar.gz`
|
||||
const installVscodeServerScript = `
|
||||
mkdir -p ~/.vscode-server/bin/${vscodeCommitId} && \\
|
||||
if [ "$(ls -A ~/.vscode-server/bin/${vscodeCommitId})" ]; then
|
||||
~/.vscode-server/bin/${vscodeCommitId}/bin/code-server --install-extension mengning.devstar
|
||||
@@ -55,25 +55,25 @@ export default class RemoteContainer {
|
||||
~/.vscode-server/bin/${vscodeCommitId}/bin/code-server --install-extension mengning.devstar
|
||||
fi
|
||||
`;
|
||||
await ssh.execCommand(installVscodeServerScript);
|
||||
console.log("vscode-server and extension installed");
|
||||
vscode.window.showInformationMessage('安装完成！');
|
||||
await ssh.execCommand(installVscodeServerScript);
|
||||
console.log("vscode-server and extension installed");
|
||||
vscode.window.showInformationMessage('安装完成！');
|
||||
|
||||
await ssh.dispose();
|
||||
} else {
|
||||
// connect with password (deprecate in future)
|
||||
await ssh.connect({
|
||||
host: host,
|
||||
username: username,
|
||||
password: password,
|
||||
port: port
|
||||
});
|
||||
progress.report({ message: "连接成功，开始安装" });
|
||||
await ssh.dispose();
|
||||
} else {
|
||||
// connect with password (deprecate in future)
|
||||
await ssh.connect({
|
||||
host: host,
|
||||
username: username,
|
||||
password: password,
|
||||
port: port
|
||||
});
|
||||
progress.report({ message: "连接成功，开始安装" });
|
||||
|
||||
// install vscode-server and devstar extension
|
||||
const vscodeCommitId = await utils.getVsCodeCommitId();
|
||||
const vscodeServerUrl = `https://vscode.download.prss.microsoft.com/dbazure/download/stable/${vscodeCommitId}/vscode-server-linux-x64.tar.gz`
|
||||
const installVscodeServerScript = `
|
||||
// install vscode-server and devstar extension
|
||||
const vscodeCommitId = await utils.getVsCodeCommitId();
|
||||
const vscodeServerUrl = `https://vscode.download.prss.microsoft.com/dbazure/download/stable/${vscodeCommitId}/vscode-server-linux-x64.tar.gz`
|
||||
const installVscodeServerScript = `
|
||||
mkdir -p ~/.vscode-server/bin/${vscodeCommitId} && \\
|
||||
if [ "$(ls -A ~/.vscode-server/bin/${vscodeCommitId})" ]; then
|
||||
~/.vscode-server/bin/${vscodeCommitId}/bin/code-server --install-extension mengning.devstar
|
||||
@@ -86,63 +86,63 @@ export default class RemoteContainer {
|
||||
~/.vscode-server/bin/${vscodeCommitId}/bin/code-server --install-extension mengning.devstar
|
||||
fi
|
||||
`;
|
||||
await ssh.execCommand(installVscodeServerScript);
|
||||
console.log("vscode-server and extension installed");
|
||||
vscode.window.showInformationMessage('安装完成！');
|
||||
await ssh.execCommand(installVscodeServerScript);
|
||||
console.log("vscode-server and extension installed");
|
||||
vscode.window.showInformationMessage('安装完成！');
|
||||
|
||||
// add the public key to the remote authorized_keys
|
||||
if (!utils.existDefaultPublicKey() || !utils.existDefaultPrivateKey()) {
|
||||
// if there is no public key, generate one
|
||||
utils.createSSHKey()
|
||||
// add the public key to the remote authorized_keys
|
||||
if (!utils.existDefaultPublicKey() || !utils.existDefaultPrivateKey()) {
|
||||
// if there is no public key, generate one
|
||||
utils.createSSHKey()
|
||||
}
|
||||
|
||||
const publicKey = utils.getDefaultPublicKey();
|
||||
await ssh.execCommand(`mkdir -p ~/.ssh && echo '${publicKey}' >> ~/.ssh/authorized_keys`);
|
||||
console.log('Public key added to remote authorized_keys');
|
||||
|
||||
await ssh.dispose();
|
||||
}
|
||||
|
||||
const publicKey = utils.getDefaultPublicKey();
|
||||
await ssh.execCommand(`mkdir -p ~/.ssh && echo '${publicKey}' >> ~/.ssh/authorized_keys`);
|
||||
console.log('Public key added to remote authorized_keys');
|
||||
|
||||
} catch (error) {
|
||||
console.error('Failed to install vscode-server and extension: ', error);
|
||||
await ssh.dispose();
|
||||
reject(error);
|
||||
}
|
||||
|
||||
} catch (error) {
|
||||
console.error('Failed to install vscode-server and extension: ', error);
|
||||
await ssh.dispose();
|
||||
reject(error);
|
||||
}
|
||||
// only connect successfully then save the host info
|
||||
const sshConfigPath = path.join(os.homedir(), '.ssh', 'config');
|
||||
// check if the host and related info exist in local ssh config file before saving
|
||||
var canAppendSSHConfig = true
|
||||
if (fs.existsSync(sshConfigPath)) {
|
||||
var reader = rd.createInterface(fs.createReadStream(sshConfigPath))
|
||||
|
||||
// only connect successfully then save the host info
|
||||
const sshConfigPath = path.join(os.homedir(), '.ssh', 'config');
|
||||
// check if the host and related info exist in local ssh config file before saving
|
||||
var canAppendSSHConfig = true
|
||||
if (fs.existsSync(sshConfigPath)) {
|
||||
var reader = rd.createInterface(fs.createReadStream(sshConfigPath))
|
||||
|
||||
for await (const line of reader) {
|
||||
// host format: hostname:port
|
||||
if (line.includes(`Host ${host}-${port}`)) {
|
||||
// the container ssh info exists
|
||||
canAppendSSHConfig = false
|
||||
break;
|
||||
for await (const line of reader) {
|
||||
// host format: hostname:port
|
||||
if (line.includes(`Host ${host}-${port}`)) {
|
||||
// the container ssh info exists
|
||||
canAppendSSHConfig = false
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (canAppendSSHConfig) {
|
||||
// save the host to the local ssh config file
|
||||
const hostInConfig = `${host}-${port}` // host format: hostname-port
|
||||
const privateKeyPath = this.user.getUserPrivateKeyPath();
|
||||
const newSShConfigContent =
|
||||
`\nHost ${hostInConfig}\n HostName ${host}\n Port ${port}\n User ${username}\n PreferredAuthentications publickey\n IdentityFile ${privateKeyPath}\n `;
|
||||
fs.writeFileSync(sshConfigPath, newSShConfigContent, { encoding: 'utf8', flag: 'a' });
|
||||
console.log('Host registered in local ssh config');
|
||||
}
|
||||
if (canAppendSSHConfig) {
|
||||
// save the host to the local ssh config file
|
||||
const hostInConfig = `${host}-${port}` // host format: hostname-port
|
||||
const privateKeyPath = this.user.getUserPrivateKeyPath();
|
||||
const newSShConfigContent =
|
||||
`\nHost ${hostInConfig}\n HostName ${host}\n Port ${port}\n User ${username}\n PreferredAuthentications publickey\n IdentityFile ${privateKeyPath}\n `;
|
||||
fs.writeFileSync(sshConfigPath, newSShConfigContent, { encoding: 'utf8', flag: 'a' });
|
||||
console.log('Host registered in local ssh config');
|
||||
}
|
||||
|
||||
resolve("success");
|
||||
});
|
||||
resolve("success");
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
openRemoteFolder(host: string, username:string ,port:number, path: string): void {
|
||||
openRemoteFolder(host: string, username: string, port: number, path: string): void {
|
||||
var host = `${host}-${port}`
|
||||
const command = `code --remote ssh-remote+${username}@${host} ${path} --reuse-window`
|
||||
let terminal = vscode.window.activeTerminal || vscode.window.createTerminal(`Ext Terminal`);
|
||||
|
||||
216
src/user.ts
216
src/user.ts
@@ -9,123 +9,123 @@ const {
|
||||
const sshpk = require('sshpk');
|
||||
|
||||
export default class User {
|
||||
private context:vscode.ExtensionContext;
|
||||
private username:string|undefined;
|
||||
private userToken:string|undefined;
|
||||
private usernameKey:string = 'devstarUsername'
|
||||
private userTokenKey:string = 'devstarUserToken'
|
||||
private context: vscode.ExtensionContext;
|
||||
private username: string | undefined;
|
||||
private userToken: string | undefined;
|
||||
private usernameKey: string = 'devstarUsername'
|
||||
private userTokenKey: string = 'devstarUserToken'
|
||||
|
||||
constructor(context: vscode.ExtensionContext) {
|
||||
this.context = context;
|
||||
this.username = this.context.globalState.get(this.usernameKey);
|
||||
this.userToken = this.context.globalState.get(this.userTokenKey);
|
||||
constructor(context: vscode.ExtensionContext) {
|
||||
this.context = context;
|
||||
this.username = this.context.globalState.get(this.usernameKey);
|
||||
this.userToken = this.context.globalState.get(this.userTokenKey);
|
||||
}
|
||||
|
||||
private isLogged() {
|
||||
var existUsername = false;
|
||||
var existUserToken = false;
|
||||
if (this.username != undefined && this.username != '') {
|
||||
existUsername = true;
|
||||
}
|
||||
if (this.userToken != undefined && this.userToken != '') {
|
||||
existUserToken = true;
|
||||
}
|
||||
|
||||
private isLogged() {
|
||||
var existUsername = false;
|
||||
var existUserToken = false;
|
||||
if(this.username != undefined && this.username != '') {
|
||||
existUsername = true;
|
||||
}
|
||||
if(this.userToken != undefined && this.userToken != '') {
|
||||
existUserToken = true;
|
||||
}
|
||||
if (existUsername && existUserToken) {
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (existUsername && existUserToken) {
|
||||
return true;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
public getUsernameFromLocal(): string | undefined {
|
||||
return this.username;
|
||||
}
|
||||
|
||||
public getUserTokenFromLocal(): string | undefined {
|
||||
return this.userToken;
|
||||
}
|
||||
|
||||
public setUsernameToLocal(username: string) {
|
||||
this.context.globalState.update(this.usernameKey, username);
|
||||
this.username = username;
|
||||
}
|
||||
|
||||
public setUserTokenToLocal(userToken: string) {
|
||||
this.context.globalState.update(this.userTokenKey, userToken)
|
||||
this.userToken = userToken
|
||||
}
|
||||
|
||||
public getUserPrivateKeyPath(): string {
|
||||
if (!this.isLogged) {
|
||||
return '';
|
||||
}
|
||||
|
||||
public getUsernameFromLocal(): string|undefined {
|
||||
return this.username;
|
||||
return path.join(os.homedir(), '.ssh', `id_rsa_${this.username}`)
|
||||
}
|
||||
|
||||
public getUserPublicKeyPath(): string {
|
||||
if (!this.isLogged) {
|
||||
return '';
|
||||
}
|
||||
|
||||
public getUserTokenFromLocal(): string|undefined{
|
||||
return this.userToken;
|
||||
return path.join(os.homedir(), '.ssh', `id_rsa_${this.username}.pub`)
|
||||
}
|
||||
|
||||
public existUserPublicKey(): boolean {
|
||||
const userPublicKeyPath = this.getUserPublicKeyPath();
|
||||
return fs.existsSync(userPublicKeyPath)
|
||||
}
|
||||
|
||||
public existUserPrivateKey(): boolean {
|
||||
const userPrivateKeyPath = this.getUserPrivateKeyPath();
|
||||
return fs.existsSync(userPrivateKeyPath)
|
||||
}
|
||||
|
||||
public getUserPublicKey(): string {
|
||||
const userPublicKeyPath = this.getUserPublicKeyPath();
|
||||
const userPublicKey = fs.readFileSync(userPublicKeyPath, 'utf-8');
|
||||
// remove `\r` `\n`
|
||||
const trimmedDefaultPublicKey = userPublicKey.replace(/[\r\n]/g, "");
|
||||
return trimmedDefaultPublicKey;
|
||||
}
|
||||
|
||||
public getUserPrivateKey(): string {
|
||||
const userPrivateKey = this.getUserPrivateKeyPath();
|
||||
return fs.readFileSync(userPrivateKey, 'utf-8');
|
||||
}
|
||||
|
||||
public async createUserSSHKey() {
|
||||
if (this.existUserPublicKey() && this.existUserPrivateKey()) {
|
||||
// if both public and private key exists, stop
|
||||
return;
|
||||
}
|
||||
|
||||
public setUsernameToLocal(username:string) {
|
||||
this.context.globalState.update(this.usernameKey, username);
|
||||
this.username = username;
|
||||
}
|
||||
|
||||
public setUserTokenToLocal(userToken:string) {
|
||||
this.context.globalState.update(this.userTokenKey, userToken)
|
||||
this.userToken = userToken
|
||||
}
|
||||
|
||||
public getUserPrivateKeyPath() : string{
|
||||
if (!this.isLogged) {
|
||||
return '';
|
||||
}
|
||||
|
||||
return path.join(os.homedir(), '.ssh', `id_rsa_${this.username}`)
|
||||
}
|
||||
|
||||
public getUserPublicKeyPath() :string{
|
||||
if (!this.isLogged) {
|
||||
return '';
|
||||
}
|
||||
|
||||
return path.join(os.homedir(), '.ssh', `id_rsa_${this.username}.pub`)
|
||||
}
|
||||
|
||||
public existUserPublicKey() :boolean{
|
||||
const userPublicKeyPath = this.getUserPublicKeyPath();
|
||||
return fs.existsSync(userPublicKeyPath)
|
||||
}
|
||||
|
||||
public existUserPrivateKey() :boolean{
|
||||
const userPrivateKeyPath = this.getUserPrivateKeyPath();
|
||||
return fs.existsSync(userPrivateKeyPath)
|
||||
}
|
||||
|
||||
public getUserPublicKey(): string {
|
||||
const userPublicKeyPath = this.getUserPublicKeyPath();
|
||||
const userPublicKey = fs.readFileSync(userPublicKeyPath, 'utf-8');
|
||||
// remove `\r` `\n`
|
||||
const trimmedDefaultPublicKey = userPublicKey.replace(/[\r\n]/g, "");
|
||||
return trimmedDefaultPublicKey;
|
||||
}
|
||||
|
||||
public getUserPrivateKey(): string {
|
||||
const userPrivateKey = this.getUserPrivateKeyPath();
|
||||
return fs.readFileSync(userPrivateKey, 'utf-8');
|
||||
}
|
||||
|
||||
public async createUserSSHKey() {
|
||||
if (this.existUserPublicKey() && this.existUserPrivateKey()) {
|
||||
// if both public and private key exists, stop
|
||||
return;
|
||||
}
|
||||
|
||||
const {
|
||||
publicKey,
|
||||
privateKey,
|
||||
} = await generateKeyPairSync('rsa', {
|
||||
modulusLength: 4096,
|
||||
publicKeyEncoding: {
|
||||
type: 'pkcs1',
|
||||
format: 'pem',
|
||||
},
|
||||
privateKeyEncoding: {
|
||||
type: 'pkcs1',
|
||||
format: 'pem',
|
||||
},
|
||||
});
|
||||
const publicKeyFingerprint = sshpk.parseKey(publicKey, 'pem').toString('ssh');
|
||||
const publicKeyStr = publicKeyFingerprint; // public key is public key fingerprint
|
||||
const privateKeyStr = privateKey;
|
||||
|
||||
try {
|
||||
await fs.writeFileSync(this.getUserPublicKeyPath(), publicKeyStr);
|
||||
await fs.writeFileSync(this.getUserPrivateKeyPath(), privateKeyStr);
|
||||
// limit the permission of private key to prevent that the private key not works
|
||||
await fs.chmodSync(this.getUserPrivateKeyPath(), 0o600)
|
||||
} catch(error) {
|
||||
console.error("Failed to write public/private key into the default ssh public/key file: ", error);
|
||||
}
|
||||
const {
|
||||
publicKey,
|
||||
privateKey,
|
||||
} = await generateKeyPairSync('rsa', {
|
||||
modulusLength: 4096,
|
||||
publicKeyEncoding: {
|
||||
type: 'pkcs1',
|
||||
format: 'pem',
|
||||
},
|
||||
privateKeyEncoding: {
|
||||
type: 'pkcs1',
|
||||
format: 'pem',
|
||||
},
|
||||
});
|
||||
const publicKeyFingerprint = sshpk.parseKey(publicKey, 'pem').toString('ssh');
|
||||
const publicKeyStr = publicKeyFingerprint; // public key is public key fingerprint
|
||||
const privateKeyStr = privateKey;
|
||||
|
||||
try {
|
||||
await fs.writeFileSync(this.getUserPublicKeyPath(), publicKeyStr);
|
||||
await fs.writeFileSync(this.getUserPrivateKeyPath(), privateKeyStr);
|
||||
// limit the permission of private key to prevent that the private key not works
|
||||
await fs.chmodSync(this.getUserPrivateKeyPath(), 0o600)
|
||||
} catch (error) {
|
||||
console.error("Failed to write public/private key into the default ssh public/key file: ", error);
|
||||
}
|
||||
}
|
||||
}
|
||||
158
src/utils.ts
158
src/utils.ts
@@ -10,114 +10,114 @@ const {
|
||||
} = require('node:crypto')
|
||||
|
||||
export function fetch(url: string): Promise<string> {
|
||||
// determine the library to use (based on the url protocol)
|
||||
const lib = url.startsWith('https://') ? https : http;
|
||||
// determine the library to use (based on the url protocol)
|
||||
const lib = url.startsWith('https://') ? https : http;
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
lib.get(url, (response) => {
|
||||
// make sure the status code is 200
|
||||
if (response.statusCode !== 200) {
|
||||
reject(new Error(`Failed to load page, status code: ${response.statusCode}`));
|
||||
return;
|
||||
}
|
||||
return new Promise((resolve, reject) => {
|
||||
lib.get(url, (response) => {
|
||||
// make sure the status code is 200
|
||||
if (response.statusCode !== 200) {
|
||||
reject(new Error(`Failed to load page, status code: ${response.statusCode}`));
|
||||
return;
|
||||
}
|
||||
|
||||
let data = '';
|
||||
response.on('data', (chunk) => {
|
||||
data += chunk;
|
||||
});
|
||||
response.on('end', () => {
|
||||
resolve(data);
|
||||
});
|
||||
}).on('error', (err) => {
|
||||
reject(err);
|
||||
});
|
||||
let data = '';
|
||||
response.on('data', (chunk) => {
|
||||
data += chunk;
|
||||
});
|
||||
response.on('end', () => {
|
||||
resolve(data);
|
||||
});
|
||||
}).on('error', (err) => {
|
||||
reject(err);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
export const Sleep = (ms:number)=> {
|
||||
return new Promise(resolve=>setTimeout(resolve, ms))
|
||||
export const Sleep = (ms: number) => {
|
||||
return new Promise(resolve => setTimeout(resolve, ms))
|
||||
}
|
||||
|
||||
export function getVsCodeCommitId(): Promise<string> {
|
||||
return new Promise((resolve, reject) => {
|
||||
exec('code --version', (error, stdout, stderr) => {
|
||||
if (error) {
|
||||
reject('Error occurred:' + error.message);
|
||||
return;
|
||||
}
|
||||
if (stderr) {
|
||||
reject('Error output:' + stderr);
|
||||
return;
|
||||
}
|
||||
const lines = stdout.trim().split('\n');
|
||||
if (lines.length > 1) {
|
||||
const commitId = lines[1]; // 第二行是 commit ID
|
||||
resolve(commitId);
|
||||
} else {
|
||||
reject('Unexpected output format:' + stdout);
|
||||
}
|
||||
if (error) {
|
||||
reject('Error occurred:' + error.message);
|
||||
return;
|
||||
}
|
||||
if (stderr) {
|
||||
reject('Error output:' + stderr);
|
||||
return;
|
||||
}
|
||||
const lines = stdout.trim().split('\n');
|
||||
if (lines.length > 1) {
|
||||
const commitId = lines[1]; // 第二行是 commit ID
|
||||
resolve(commitId);
|
||||
} else {
|
||||
reject('Unexpected output format:' + stdout);
|
||||
}
|
||||
});
|
||||
})
|
||||
}
|
||||
|
||||
export function getDefaultPrivateKeyPath() : string{
|
||||
return path.join(os.homedir(), '.ssh', 'id_rsa')
|
||||
export function getDefaultPrivateKeyPath(): string {
|
||||
return path.join(os.homedir(), '.ssh', 'id_rsa')
|
||||
}
|
||||
|
||||
export function getDefaultPublicKeyPath() :string{
|
||||
return path.join(os.homedir(), '.ssh', 'id_rsa.pub')
|
||||
export function getDefaultPublicKeyPath(): string {
|
||||
return path.join(os.homedir(), '.ssh', 'id_rsa.pub')
|
||||
}
|
||||
|
||||
export function existDefaultPublicKey() :boolean{
|
||||
const defaultPublicKeyPath = getDefaultPublicKeyPath();
|
||||
return fs.existsSync(defaultPublicKeyPath)
|
||||
export function existDefaultPublicKey(): boolean {
|
||||
const defaultPublicKeyPath = getDefaultPublicKeyPath();
|
||||
return fs.existsSync(defaultPublicKeyPath)
|
||||
}
|
||||
|
||||
export function existDefaultPrivateKey() :boolean{
|
||||
const defaultPrivateKeyPath = getDefaultPrivateKeyPath();
|
||||
return fs.existsSync(defaultPrivateKeyPath)
|
||||
export function existDefaultPrivateKey(): boolean {
|
||||
const defaultPrivateKeyPath = getDefaultPrivateKeyPath();
|
||||
return fs.existsSync(defaultPrivateKeyPath)
|
||||
}
|
||||
|
||||
export function getDefaultPublicKey(): string {
|
||||
const defaultPublicKeyPath = getDefaultPublicKeyPath();
|
||||
const defaultPublicKey = fs.readFileSync(defaultPublicKeyPath, 'utf-8');
|
||||
// remove `\r` `\n`
|
||||
const trimmedDefaultPublicKey = defaultPublicKey.replace(/[\r\n]/g, "");
|
||||
return trimmedDefaultPublicKey;
|
||||
const defaultPublicKeyPath = getDefaultPublicKeyPath();
|
||||
const defaultPublicKey = fs.readFileSync(defaultPublicKeyPath, 'utf-8');
|
||||
// remove `\r` `\n`
|
||||
const trimmedDefaultPublicKey = defaultPublicKey.replace(/[\r\n]/g, "");
|
||||
return trimmedDefaultPublicKey;
|
||||
}
|
||||
|
||||
export function getDefaultPrivateKey(): string {
|
||||
const defaultPrivateKey = getDefaultPrivateKeyPath();
|
||||
return fs.readFileSync(defaultPrivateKey, 'utf-8');
|
||||
const defaultPrivateKey = getDefaultPrivateKeyPath();
|
||||
return fs.readFileSync(defaultPrivateKey, 'utf-8');
|
||||
}
|
||||
|
||||
export function createSSHKey() {
|
||||
if (existDefaultPublicKey() && existDefaultPrivateKey()) {
|
||||
// if both public and private key exists, stop
|
||||
return;
|
||||
}
|
||||
if (existDefaultPublicKey() && existDefaultPrivateKey()) {
|
||||
// if both public and private key exists, stop
|
||||
return;
|
||||
}
|
||||
|
||||
const {
|
||||
publicKey,
|
||||
privateKey,
|
||||
} = generateKeyPairSync('rsa', {
|
||||
modulusLength: 4096,
|
||||
publicKeyEncoding: {
|
||||
type: 'spki',
|
||||
format: 'pem',
|
||||
},
|
||||
privateKeyEncoding: {
|
||||
type: 'pkcs8',
|
||||
format: 'pem',
|
||||
cipher: 'aes-256-cbc',
|
||||
passphrase: 'devstar'
|
||||
},
|
||||
});
|
||||
const {
|
||||
publicKey,
|
||||
privateKey,
|
||||
} = generateKeyPairSync('rsa', {
|
||||
modulusLength: 4096,
|
||||
publicKeyEncoding: {
|
||||
type: 'spki',
|
||||
format: 'pem',
|
||||
},
|
||||
privateKeyEncoding: {
|
||||
type: 'pkcs8',
|
||||
format: 'pem',
|
||||
cipher: 'aes-256-cbc',
|
||||
passphrase: 'devstar'
|
||||
},
|
||||
});
|
||||
|
||||
try {
|
||||
fs.writeFileSync(getDefaultPublicKeyPath(), publicKey);
|
||||
fs.writeFileSync(getDefaultPrivateKeyPath(), privateKey);
|
||||
} catch(error) {
|
||||
console.error("Failed to write public/private key into the default ssh public/key file: ", error);
|
||||
}
|
||||
try {
|
||||
fs.writeFileSync(getDefaultPublicKeyPath(), publicKey);
|
||||
fs.writeFileSync(getDefaultPrivateKeyPath(), privateKey);
|
||||
} catch (error) {
|
||||
console.error("Failed to write public/private key into the default ssh public/key file: ", error);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user