import { Client } from "ssh2";
import SftpClient from "ssh2-sftp-client";
import { fileURLToPath } from "url";
import { dirname } from "path";

const sftp = new SftpClient();

// SSH 連接配置
const sshConfig = {
  host: "xxx",
  port: 22,
  username: "xx",
  password: "xxxx",
};

// 獲取當(dāng)前模塊的 URL
const __filename = fileURLToPath(import.meta.url);
// 獲取當(dāng)前模塊的目錄路徑
const __dirname = dirname(__filename);

const localFolderPath = `${__dirname}/../dist`; // 本地 dist 文件夾路徑
const remoteFolderPath = "/home/xxx"; // 遠(yuǎn)程服務(wù)器目標(biāo)文件夾路徑

/* 下面的可以不用動(dòng) */
// 封裝 SSH 連接函數(shù)為 Promise
const connectSSH = (sshConfig) => {
  return new Promise((resolve, reject) => {
    const conn = new Client();
    conn
      .on("ready", () => {
        resolve(conn);
      })
      .connect(sshConfig);
    conn.on("error", (err) => {
      reject(err);
    });
  });
};

// 封裝連接 SSH 并執(zhí)行命令的函數(shù)為 Promise
const executeSSHCommand = (conn, command) => {
  return new Promise((resolve, reject) => {
    conn.exec(command, (err, stream) => {
      if (err) {
        conn.end();
        reject(err);
        return;
      }
      let stdout = "";
      let stderr = "";
      stream.on("data", (data) => (stdout += data.toString()));
      stream.stderr.on("data", (data) => (stderr += data.toString()));
      stream.on("close", (code, signal) => {
        // conn.end();
        if (code === 0) {
          resolve(stdout);
        } else {
          reject(stderr || `Command failed with code ${code}`);
        }
      });
    });
  });
};

// 封裝上傳文件夾函數(shù)為 Promise
const uploadFolder = async (localFolderPath, remoteFolderPath) => {
  await sftp
    .connect(sshConfig)
    .then(() => {
      console.log("正在上傳中...");
      return sftp.uploadDir(localFolderPath, remoteFolderPath);
    })
    .then(() => {
      console.log("File uploaded successfully");
      sftp.end();
    })
    .catch((err) => {
      console.error(err.message);
      sftp.end();
    });
};

// 主函數(shù)
const main = async () => {
  try {
    // 連接 SSH
    const conn = await connectSSH(sshConfig);
    console.log("SSH 連接已建立");

    // 刪除遠(yuǎn)程文件夾
    await executeSSHCommand(conn, `rm -rf ${remoteFolderPath}/*`);

    // 上傳文件夾
    await uploadFolder(localFolderPath, remoteFolderPath);

    // 關(guān)閉 SSH 連接
    conn.end();
  } catch (err) {
    console.error("發(fā)生錯(cuò)誤:", err);
    conn.end();
  }
};

// 執(zhí)行主函數(shù)
main();