refactor: convert js to ts

2024-07-02 12:15:39 +08:00
parent 4afa0ccccc
commit fd9fe52529
12 changed files with 207 additions and 234 deletions
--- a/src/main.ts
+++ b/src/main.ts
@@ -0,0 +1,119 @@
+import vscode from 'vscode';
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+import QuickAccessTreeProvider from './views/quick-access-tree';
+import SIHome from './home';
+
+const { NodeSSH } = require('node-ssh')
+
+export class SuperIDEExtension {
+  siHome: SIHome;
+
+  constructor(private context: vscode.ExtensionContext) {
+    this.siHome = new SIHome();
+
+    context.subscriptions.push(
+      vscode.window.registerTreeDataProvider(
+        'superide.quickAccess',
+        new QuickAccessTreeProvider()
+      )
+    );
+
+    this.registerGlobalCommands(context);
+
+    this.startSuperIDEHome();
+  }
+
+  async startSuperIDEHome() {
+    vscode.commands.executeCommand('superide.showHome');
+  }
+
+  // TODO: finish firstConnect (automatically)
+  async connectRemoteContainer() {
+    const host = 'mengning.com.cn';
+    const username = 'linux';
+    const password = 'a7%Xs&&TXG';
+    const port = 31874;
+
+    console.log('Connecting to remote container...');
+    // detect if local ssh config has the host registered
+    // TODO: use remote.SSH.configFile
+    const sshConfigPath = path.join(os.homedir(), '.ssh', 'config');
+    if (fs.existsSync(sshConfigPath)) {
+      const config = fs.readFileSync(sshConfigPath, 'utf8');
+      // TODO: more robust regex
+      if (config.includes(host)) {
+        console.log('Host already registered in local ssh config');
+        // TODO: connect the specified folder
+      } else {
+        // the host has not been registered in the local ssh config
+
+        // 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');
+
+        await this.firstConnect(
+          host,
+          username,
+          password,
+          port,
+          publicKey
+        )
+
+        // 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');
+
+        // connect the host with remote-ssh extension
+        try {
+          await vscode.commands.executeCommand('opensshremotes.openEmptyWindowInCurrentWindow', host)
+        } catch (error) {
+          console.log("Error connecting to host: ", error);
+        }
+      }
+    }
+  }
+
+  async firstConnect(host: string, username: string, password: string, port: number, publicKey: string): Promise<void> {
+    const ssh = new NodeSSH();
+
+    try {
+      await ssh.connect({
+        host: host,
+        username: username,
+        password: password,
+        port: port
+      });
+
+      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();
+    }
+  }
+
+  registerGlobalCommands(context: vscode.ExtensionContext) {
+    context.subscriptions.push(
+      vscode.commands.registerCommand('superide.showHome', (url) =>
+        this.siHome.toggle(url)
+      ),
+      vscode.commands.registerCommand('superide.connectRemoteContainer', () =>
+        this.connectRemoteContainer()
+      )
+    );
+  }
+}
+
+export function activate(context: vscode.ExtensionContext) {
+  return new SuperIDEExtension(context);
+}
+
+export function deactivate() {
+}