import canvas from 'canvas';
import fs from 'fs';

/**
 * 判斷png圖片是否存在透明像素
 *
 * @param {*} pngPath png圖片路徑
 * @param {number} [limit=255] 透明像素點(diǎn)限度，默認(rèn)小與255視為透明像素
 * @param {boolean} [isToJpg=false] 如果沒有透明像素是否轉(zhuǎn)化為jpg圖片
 * @returns
 */
function hasOpacity(pngPath, limit = 255, isToJpg = false) {
  // 獲取圖片的buffer
  const buffer = fs.readFileSync(pngPath);

  // 圖片的寬度存在buffer的第17到20個(gè)字節(jié)
  const width = buffer.readUInt32BE(16);
  // 圖片的寬度存在buffer的第21到24個(gè)字節(jié)
  const height = buffer.readUInt32BE(20);

  // 創(chuàng)建一個(gè)畫布
  const pngCanvas = canvas.createCanvas(width, height);
  // 拿到畫筆
  const context = pngCanvas.getContext('2d');
  // 創(chuàng)建一張圖片
  const img = new canvas.Image();
  // 記載圖片
  img.src = buffer;
  // 使用畫筆把圖片畫在畫布上
  context.drawImage(img, 0, 0, width, height);

  // 獲取png圖片的數(shù)據(jù)的像素?cái)?shù)據(jù)
  let res = context.getImageData(0, 0, width, height);
  let imgData = res.data;
  // 每個(gè)像素占用4個(gè)字節(jié)，計(jì)算出一共有多少像素
  // [r, g, b, a]
  let piexCount = imgData.length / 4;

  // 是否已經(jīng)找到透明像素
  let isOpacity = false;

  for (let i = 0; i < piexCount; i++) {
    // 遍歷每個(gè)像素點(diǎn)，找透明通道
    let opacity = imgData[i * 4 + 3];
    if (opacity < limit) {
        // 如果小于limit，則存在透明像素，退出
      isOpacity = true;
      break;
    }
  }

  // 如果不存在透明像素且isToJpg為true，則轉(zhuǎn)化為jpg格式的圖拍呢
  if (!isOpacity && isToJpg) {
    const out = fs.createWriteStream(pngPath.split('.')[0] + '.jpg');
    pngCanvas.createJPEGStream().pipe(out);
    out.on('finish', () => console.log(pngPath, '轉(zhuǎn)成jpg成功'));
  }

  // 返回
  return isOpacity;
}

console.log(hasOpacity('hh.png', 254, true));
console.log(hasOpacity('jj.png'));