nodejs

系統

Node.js 小工具

•

[UPDATE] 之前的版本不能壓縮資料夾，用 ChatGPT 再修改一下….

const fs = require('fs');
const path = require('path');

const archiver = require('archiver');
const ftpClient = require('ftp');
const nodemailer = require('nodemailer');

const toUser = '[email protected]';
let fileSize;

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

if (!checkFileName(zipFileName)) {
  console.error('檔名輸入有誤！！');
  return false;
}

// input_files 資料夾
const inputDir = path.join(__dirname, 'input_files');

// 確認 output_zip 資料夾是否存在，若不存在則建立
const outputDir = path.join(__dirname, 'output_zip');

if (!fs.existsSync(outputDir)) {
  fs.mkdirSync(outputDir);
}

// 建立一個輸出檔案流，指定存放在 output_zip 資料夾下
const outputFilePath = path.join(outputDir, `${zipFileName}.zip`);

// 壓縮
zipDirectory(inputDir, outputFilePath);

// --- 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 checkFileName(fileName) {
  const regex = /^[a-zA-Z0-9_-]+$/;
  return regex.test(fileName);
}

function zipDirectory(sourceDir, outPath) {
  const output = fs.createWriteStream(outPath);

  const archive = archiver('zip', {
    zlib: { level: 9 } // 設定壓縮等級 (1-9)
  });

  // 開始壓縮
  archive.pipe(output);
  archive.directory(sourceDir, false); // 將資料夾的內容添加至壓縮檔
  archive.finalize();

  output.on('close', () => {
    fileSize = formatBytes(archive.pointer());

    console.log(`[${zipFileName}.zip] 壓縮完成，總共 ${fileSize}`);

    // 壓縮完成後上傳到 FTP 伺服器
    uploadFileToFtp();
  });

  archive.on('error', (err) => {
    throw err;
  });
}

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

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

        console.log('檔案上傳完成 OK');
        client.end();

        // 發送電子郵件通知
        sendEmail();
      }
    );
  });

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

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

function sendEmail() {
  const transporter = nodemailer.createTransport({
    host: 'smtp.office365.com',
    port: 587,
    secure: false, // true for 465, false for other ports
    auth: {
      user: 'user',
      pass: 'pass',
    },
  });

  const mailOptions = {
    from: '"SYSTEM" <[email protected]>',
    to: toUser,
    subject: `${zipFileName}.zip`,
    text: `File Size: ${fileSize}\nhttps://www.abc.com.tw/downloads/${zipFileName}.zip`,
  };

  transporter.sendMail(mailOptions, (err) => {
    if (err) {
      console.error('電子郵件發送失敗：', err);
      return false;
    }

    console.log('電子郵件發送成功 OK');
  });
}

這個小工具可以把特定資料夾內的檔案壓縮成一個 .zip 檔，再把 .zip 檔上傳到 FTP Server，上傳成功後會發送通知 Email。

有用到 archiver、ftp、nodemailer 三個 package。

const fs = require('fs');
const path = require('path');

const archiver = require('archiver');
const ftpClient = require('ftp');
const nodemailer = require('nodemailer');

const toUser = '[email protected]'; // 要收到通知的 user
let fileSize;

const archive = archiver('zip', {
  zlib: { level: 9 }, // 設定壓縮等級
});

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

if (!checkFileName(zipFileName)) {
  console.error('檔名輸入有誤！！');
  return false;
}

// 確認 output_zip 資料夾是否存在，若不存在則建立
const outputDir = path.join(__dirname, 'output_zip');

if (!fs.existsSync(outputDir)) {
  fs.mkdirSync(outputDir);
}

// 建立一個輸出檔案流，指定存放在 output_zip 資料夾下
const outputFilePath = path.join(outputDir, `${zipFileName}.zip`);

const output = fs.createWriteStream(outputFilePath);

// 連結檔案流至壓縮檔
archive.pipe(output);

// 獲取 input_files 資料夾中的所有檔案
const inputDir = path.join(__dirname, 'input_files');

fs.readdir(inputDir, (err, files) => {
  if (err) {
    throw err;
  }

  let count = 0;

  // 使用 for 迴圈添加每個檔案至壓縮檔
  for (const file of files) {
    const filePath = path.join(inputDir, file);

    archive.file(filePath, { name: file });
    console.log(`[${file}] 加入壓縮檔....`);

    count++;
  }

  console.log(`共 [${count}] 個檔案加入`);

  // 完成壓縮檔案
  archive.finalize();
});

output.on('close', () => {
  fileSize = formatBytes(archive.pointer());

  console.log(`[${zipFileName}.zip] 壓縮完成，總共 ${fileSize}`);

  // 壓縮完成後上傳到 FTP 伺服器
  uploadFileToFtp();
});

archive.on('warning', (err) => {
  if (err.code !== 'ENOENT') {
    throw err;
  }
});

archive.on('error', (err) => {
  throw err;
});

// --- 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 checkFileName(fileName) {
  const regex = /^[a-zA-Z0-9_-]+$/;
  return regex.test(fileName);
}

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

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

        console.log('檔案上傳完成 OK');
        client.end();

        // 發送電子郵件通知
        sendEmail();
      }
    );
  });

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

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

function sendEmail() {
  const transporter = nodemailer.createTransport({
    host: 'smtp.office365.com',
    port: 587,
    secure: false, // true for 465, false for other ports
    auth: {
      user: 'user',
      pass: 'pass',
    },
  });

  const mailOptions = {
    from: '"SYSTEM" <[email protected]>',
    to: toUser,
    subject: `${zipFileName}.zip`,
    text: `File Size: ${fileSize}\nhttps://www.abc.com.tw/downloads/${zipFileName}.zip`,
  };

  transporter.sendMail(mailOptions, (err) => {
    if (err) {
      console.error('電子郵件發送失敗：', err);
      return false;
    }

    console.log('電子郵件發送成功 OK');
  });
}

Download

系統

檢查 Email 格式

•

為了發 EDM 整理網站的會員 email。

發現好多奇怪的 email….

到底是怎麼通過驗證的呢？！

問了 ChatGPT，修改一下，下面是用 Node.js 驗證 email 格式的 code，有用到 isemail 這個 package。

email.txt 裡的 email 為一行一個。

const fs = require('fs');

const isEmail = require('isemail');

let emails;

fs.readFile('./email.txt', 'utf8', (err, data) => {
  if (err) {
    console.error('Err:', err);
    return;
  }

  emails = data.split(/\r?\n|\r/).filter((line) => line.length > 0);

  console.log('emails.length:', emails.length);

  emails.forEach((email) => {
    if (!isEmail.validate(email)) {
      console.log(email);
    }

    // if (isEmail.validate(email, { errorLevel: true }) !== 0) {
    //  console.log(email);
    // }
  });
});
系統

監控網路連線狀態

•

同事有監控網路連線狀態並開關軟體的需求，請我做個小工具。

運作方式：每五秒會 ping 三個 DNS，三個 DNS 都沒回應就判斷網路有問題，會把斷線時間寫入檔案，並關掉指定軟體。三個 DNS 都有回應代表正常。網路有問題又回復連線的話，會等待三十秒再把指定軟體打開。

下面是 Node.js 的 code，有用到 ping 這個 package。

const { execSync, execFile } = require('child_process');
const fs = require('fs');
const ping = require('ping');

const hosts = ['168.95.1.1', '8.8.8.8', '1.1.1.1'];
const appName = 'app.exe';
const appPath = `C:\\Program Files (x86)\\App\\${appName}`;

let status;
let isClosed = false;
let intervalId;
let countdown = 30;

setInterval(() => {
  let onCount = 0;
  let offCount = 0;

  hosts.forEach((host) => {
    ping.sys.probe(host, (isAlive) => {
      if (isAlive) {
        onCount++;
        console.log(`${host} (on)`);

        // 連線正常
        if (onCount >= hosts.length) {
          console.log('--- Status: Online ---');

          // 連線回復
          if (status === 'OFFLINE') {
            status = '';
            console.log('Status: Network Back....');

            // 啟動 app
            launchApp();
          }
        }
      } else {
        offCount++;
        console.log(`${host} (off)`);

        // 連線有問題
        if (offCount >= hosts.length) {
          status = 'OFFLINE';
          console.log(`==== Status: ${status} [${getDateTime()}] ====`);

          // 寫入斷線時間
          logOffline();

          // 關閉 app
          if (!isClosed) {
            closeApp();
          }
        }
      }
    });
  });
}, 5000);

function closeApp() {
  isClosed = true;
  // console.log('closeApp()-isClosed:', isClosed);

  setTimeout(() => {
    console.log('Close App');
    execSync(`taskkill /f /im ${appName} /t`);
  }, 2000);
}

function launchApp() {
  isClosed = false;
  // console.log('launchApp()-isClosed:', isClosed);

  intervalId = setInterval(() => {
    console.log(`Countdown: ${countdown--}`);

    if (countdown === 0) {
      // console.log('clear-intervalId:', intervalId);
      clearInterval(intervalId);

      countdown = 30;

      console.log('Launch App');
      execFile(appPath);
    }
  }, 1000);

  // console.log('set-intervalId:', intervalId);
}

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 logOffline() {
  const data = `OFFLINE ${getDateTime()}\n`;

  fs.appendFile('offline_log.txt', data, 'utf8', (err) => {
    if (err) {
      console.log('Err:', err);
    }
  });
}


參考連結：

返回頂端