Node.js 小工具

說明：把指定資料夾內的檔案、資料夾使用 7-Zip 壓縮成一個 .zip 檔，再把 .zip 檔上傳到 FTP Server，上傳成功後會使用 Outlook 2013 建立新 Email。

有用到 ftp package。

const fs = require('fs');
const path = require('path');
const { execSync } = require('child_process');

const ftpClient = require('ftp');

if (process.argv.length > 3) {
  console.log('檔名請勿包含空格！');
  return;
}

// 取得 .zip 檔名，預設為：YYYYMMDD-HHMMSS
const zipFileName = process.argv[2] || getDateTime();

// app
const z7 = '"C:\\Program Files\\7-Zip\\7z.exe"';
const outlook2013 = '"C:\\Program Files (x86)\\Microsoft Office\\Office15\\OUTLOOK.EXE"';

// 程式執行資料夾
const currentFolder = path.join(__dirname);

// 要被壓縮檔案、資料夾的路徑
const ftpFolder = path.join(currentFolder, 'FTP');

const fileList = fs.readdirSync(ftpFolder);

if (!fileList.length) {
  console.log('FTP 資料夾內沒有檔案');
  return;
}

// zip 檔輸出資料夾
const zipOutputFolder = path.join(currentFolder, 'OUTPUT');

// 如果 OUTPUT 資料夾存在，刪除
if (fs.existsSync(zipOutputFolder)) {
  try {
    fs.rmSync(zipOutputFolder, { recursive: true, force: true });
  } catch (err) {
    console.log(`錯誤：${err.message}`);
  }
}

// 建立 OUTPUT 資料夾
try {
  fs.mkdirSync(zipOutputFolder);
} catch (err) {
  console.log(`錯誤：${err.message}`);
}

const zipFile = path.join(zipOutputFolder, `${zipFileName}.zip`);

// 壓縮檔案
compressFiles();

// --- function ---
function formatBytes(bytes) {
  if (bytes === 0) {
    return '0 Bytes';
  }

  const k = 1024;
  const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB'];

  const i = Math.floor(Math.log(bytes) / Math.log(k));

  return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + sizes[i];
}

function getDateTime() {
  let today = new Date();

  let year = today.getFullYear();
  let month = ('0' + (today.getMonth() + 1)).slice(-2);
  let date = ('0' + today.getDate()).slice(-2);

  let hours = ('0' + today.getHours()).slice(-2);
  let minutes = ('0' + today.getMinutes()).slice(-2);
  let seconds = ('0' + today.getSeconds()).slice(-2);

  return `${year}${month}${date}-${hours}${minutes}${seconds}`;
}

function compressFiles() {
  fileList.forEach((f) => {
    const file = path.join(ftpFolder, f);
    const stats = fs.statSync(file);
    const type = stats.isDirectory() ? '資料夾' : '檔案';

    console.log(`${type} [${f}] 加入壓縮檔`);

    execSync(`${z7} a ${zipFile} "${file}"`, (error, stdout, stderr) => {
      if (error) {
        console.log(`[Error]: ${error.message}`);
        return;
      }

      if (stderr) {
        console.log(`[stderr]: ${stderr}`);
        return;
      }
    });
  });

  console.log('檔案壓縮 .... OK');

  // 上傳 FTP
  uploadFileToFtp();
}

function uploadFileToFtp() {
  const client = new ftpClient();

  client.on('ready', () => {
    client.put(
      zipFile,
      `/downloads/${zipFileName}.zip`,
      (err) => {
        if (err) {
          throw err;
        }

        console.log('檔案上傳 .... OK');

        client.end();

        // 建立新 email
        launchOutlook();
      }
    );
  });

  client.on('error', (err) => {
    console.log('FTP 連接錯誤：', err);
  });

  // 連接到 FTP 伺服器
  client.connect({
    host: '1.1.1.1',
    user: 'user',
    password: 'pwd',
  });
}

function launchOutlook() {
  const fileName = `${zipFileName}.zip`;
  const stats = fs.statSync(zipFile);
  const fileSize = formatBytes(stats.size);
  const downloadUrl = `https://www.abc.com.tw/downloads/${fileName}`;
  const htmlBody = `File Size: ${fileSize}\n\nDownload URL: ${downloadUrl}`;
  const mailtoLink = `mailto:?subject=${encodeURIComponent(fileName)}&body=${encodeURIComponent(
    htmlBody
  )}`;

  execSync(`start ${outlook2013} "${mailtoLink}"`, (error, stdout, stderr) => {
    if (error) {
      console.log(`[Error]: ${error.message}`);
      return;
    }

    if (stderr) {
      console.log(`[stderr]: ${stderr}`);
      return;
    }
  });
}