Initial commit

server/ssh_service.js

@@ -0,0 +1,65 @@
+const { Client } = require('ssh2');
+const fs = require('fs');
+require('dotenv').config();
+
+const getSSHConfig = () => {
+    const config = {
+        host: process.env.SSH_HOST,
+        username: process.env.SSH_USER,
+        port: 22,
+    };
+
+    let privateKeyPath = process.env.SSH_KEY_PATH;
+    if (privateKeyPath && privateKeyPath.endsWith('.pub')) {
+        privateKeyPath = privateKeyPath.replace('.pub', '');
+    }
+
+    if (privateKeyPath) {
+        config.privateKey = fs.readFileSync(privateKeyPath);
+    } else if (process.env.SSH_PASSWORD) {
+        config.password = process.env.SSH_PASSWORD;
+    }
+
+    return config;
+};
+
+const execCommand = (command) => {
+    return new Promise((resolve, reject) => {
+        const conn = new Client();
+        conn.on('ready', () => {
+            conn.exec(command, (err, stream) => {
+                if (err) {
+                    conn.end();
+                    return reject(err);
+                }
+                let output = '';
+                let errorOutput = '';
+                stream.on('close', (code, signal) => {
+                    conn.end();
+                    if (code !== 0) return reject(new Error(`Command failed with code ${code}: ${errorOutput}`));
+                    resolve(output.trim());
+                }).on('data', (data) => {
+                    output += data;
+                }).stderr.on('data', (data) => {
+                    errorOutput += data;
+                });
+            });
+        }).on('error', (err) => {
+            reject(err);
+        }).connect(getSSHConfig());
+    });
+};
+
+const readFile = (filePath) => {
+    return execCommand(`cat "${filePath}"`);
+};
+
+const writeFile = (filePath, content) => {
+    // Basic implementation: echo content to file.
+    // WARNING: Not safe for large files or complex headers, but start simple.
+    // Ideally use SFTP for robustness later.
+    const escapedContent = content.replace(/"/g, '\\"');
+    return execCommand(`echo "${escapedContent}" > "${filePath}"`);
+};
+
+module.exports = { execCommand, readFile, writeFile };