wative 1.0.37 → 1.0.39

package/src/wative.ts

@@ -0,0 +1,127 @@
+import * as path from 'path';
+import { Command } from 'commander';
+import { execSync } from 'child_process';
+import { getAndGenerateConfig } from './ssh/utils';
+
+const os = require('os');
+const program = new Command();
+const packageJson = require('../package.json');
+
+const hasSystemd = () => {
+    try {
+        execSync('systemctl --version', { stdio: 'ignore' });
+        return true;
+    } catch (error) {
+        return false;
+    }
+};
+
+program
+    .version(packageJson.version) // 版本信息
+    .description('Wative CLI Tool');
+
+// 定义 wative version 命令
+program
+    .command('version')
+    .description('Show the version of wative')
+    .action(() => {
+        console.log(`Wative CLI Tool v${packageJson.version}`);
+    });
+
+
+const sshCommand = program.command('ssh').description('Manage SSH service');
+
+sshCommand
+    .command('start')
+    .description('Start SSH service')
+    .action(() => {
+        if (!hasSystemd()) {
+            console.log('Systemd is not available.');
+            return;
+        }
+
+        try {
+            execSync('wative-ssh', { stdio: 'inherit' });
+        } catch (error) { }
+
+        console.log('Starting SSH service...');
+    });
+
+// sshCommand
+//     .command('stop')
+//     .description('Stop SSH service')
+//     .action(() => {
+//         if (!hasSystemd()) {
+//             console.log('Systemd is not available.');
+//             return;
+//         }
+//         console.log('Stopping SSH service...');
+//     });
+
+
+const sshServiceCommand = sshCommand.command('service').description('Manage SSH service');
+sshServiceCommand
+    .command('add')
+    .description('Add SSH service')
+    .action(async () => {
+        if (!hasSystemd()) {
+            console.log('Systemd is not available.');
+            return;
+        }
+
+        const userHomeDirectory = os.homedir();
+        const keystorePath = path.join(userHomeDirectory, '.wative');
+
+        await getAndGenerateConfig(keystorePath);
+
+        const serviceFilePath = '/etc/systemd/system/wative.service';
+        const serviceContent = `[Unit]
+            Description=Wative SSH Server
+            After=network.target
+
+            [Service]
+            ExecStart=/usr/local/bin/wative-ssh
+            Restart=always
+            User=root
+            Group=root
+            StandardOutput=append:/var/log/wative.log
+            StandardError=append:/var/log/wative.err
+
+            [Install]
+            WantedBy=multi-user.target
+        `;
+
+        console.log('Adding SSH service...');
+        execSync(`echo '${serviceContent}' > ${serviceFilePath}`, { stdio: 'inherit' });
+        execSync('systemctl daemon-reload', { stdio: 'inherit' });
+        execSync('systemctl enable wative', { stdio: 'inherit' });
+        execSync('systemctl start wative', { stdio: 'inherit' });
+    });
+
+sshServiceCommand
+    .command('remove')
+    .description('Remove SSH service')
+    .action(() => {
+        if (!hasSystemd()) {
+            console.log('Systemd is not available.');
+            return;
+        }
+        console.log('Removing SSH service...');
+        execSync('systemctl stop wative', { stdio: 'inherit' });
+        execSync('systemctl disable wative', { stdio: 'inherit' });
+        execSync('rm /etc/systemd/system/wative.service', { stdio: 'inherit' });
+        execSync('systemctl daemon-reload', { stdio: 'inherit' });
+    });
+
+
+// 定义 wative cmd 命令
+program
+    .command('cmd')
+    .description('Start wallet-cli shell')
+    .action(async () => {
+        try {
+            execSync('wallet-cli', { stdio: 'inherit' });
+        } catch (error) { }
+    });
+
+program.parse(process.argv);