wative 1.1.15 → 1.1.17

package/src/ssh/index.ts

@@ -1,52 +1,65 @@
-import WativeServer from './wative_server';
-import * as path from 'path';
-import { getAndGenerateConfig } from './utils';
+import { SSHServiceManager } from './service_manager';
+
+// Export the new SSH client
+export { WativeSSHClient, SSHConfig, createWativeSSHClient } from './client';
+export { WativeWielderServer } from './wative_server';
+export * from './types';
+export * from './utils';
 
 require('dotenv').config();
-const os = require('os');
 
-const getKeystorePasswords = async (allowedKeystores: string[]) => {
-    const passwords: any = [];
-    for (let i = 0; i < allowedKeystores.length; i++) {
-        const keystore = allowedKeystores[i].replace(/-/g, '_');
-        passwords.push(process.env[keystore]);
-    }
-    return passwords;
-}
+// 新的主函数，支持多配置文件
+const mainWithConfig = async (configName?: string) => {
+    const serviceManager = new SSHServiceManager();
 
+    try {
+        // 配置名称是必需的
+        if (!configName) {
+            throw new Error('Configuration name is required. Please specify a configuration name.');
+        }
 
-const main = async () => {
-    const userHomeDirectory = os.homedir();
-    const keystorePath = path.join(userHomeDirectory, '.wative');
+        console.log(`Starting SSH service with configuration: ${configName}`);
+        await serviceManager.startService(configName);
 
-    let config = await getAndGenerateConfig(keystorePath);
+        console.log('SSH service started successfully.');
 
-    const export_server_listen = config.ssh.listen;
-    const export_server_port = config.ssh.port;
-    const id_rsa_private_key_path = "/etc/wative/id_rsa_server";
-    const id_rsa_password = "wative_server";
+        // 保持进程运行
+        process.on('SIGINT', async () => {
+            console.log('\nShutting down SSH services...');
+            await serviceManager.stopAllServices();
+            process.exit(0);
+        });
 
-    const client_name = 'detake';
-    const client_id_rsa_public_key_path = "/etc/wative/id_rsa_detake.pub";
-    const client_keystores = config.keystore.allowed_keystores;
-    const client_keystore_passwords = await getKeystorePasswords(client_keystores);
+        process.on('SIGTERM', async () => {
+            console.log('\nShutting down SSH services...');
+            await serviceManager.stopAllServices();
+            process.exit(0);
+        });
+
+    } catch (error: any) {
+        console.error(`Failed to start SSH service: ${error.message}`);
+        process.exit(1);
+    }
+};
 
-    const wativeServer = new WativeServer(
-        id_rsa_private_key_path,
-        id_rsa_password,
-        [client_name],
-        [client_id_rsa_public_key_path],
-        [client_keystores],
-        keystorePath,
-        client_keystores,
-        client_keystore_passwords,
-    );
+// 检查命令行参数
+const args = process.argv.slice(2);
+const isDaemon = args.includes('--daemon');
+const configIndex = args.indexOf('--config');
+const configName = configIndex !== -1 && args[configIndex + 1] ? args[configIndex + 1] : undefined;
 
-    wativeServer.listen(export_server_listen, export_server_port);
+// 根据参数选择启动方式
+if (isDaemon || configName) {
+    // 使用新的多配置文件方式
+    mainWithConfig(configName).catch(error => {
+        console.error('Failed to start SSH service:', error);
+        process.exit(1);
+    });
 }
 
 process.on('unhandledRejection', (reason, p) => {
     console.log('Unhandled Rejection at:', p, 'reason:', reason);
 });
 
-main()
+// 导出函数供其他模块使用
+export { mainWithConfig };