ブログの画像管理をGitHubからCloudflare R2へ移行した

import fs from "node:fs";
import path from "node:path";
import process from "node:process";
import { fileURLToPath } from "node:url";
import { PutObjectCommand, S3Client } from "@aws-sdk/client-s3";
import sharp from "sharp";

const isDryRun = process.argv.includes("--dry-run");
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);
const targetDir = path.join(__dirname, "../src/assets/images/");
const ACCOUNT_ID = process.env.ACCOUNT_ID ?? "";
const ACCESS_KEY_ID = process.env.ACCESS_KEY_ID ?? "";
const SECRET_KEY_ID = process.env.SECRET_KEY_ID ?? "";
const BUCKET_NAME = process.env.BUCKET_NAME ?? "";
let fileCount = 0; // ファイル数カウント
const startTime = Date.now(); // 実行開始時間

if (
  ACCOUNT_ID === "" ||
  ACCESS_KEY_ID === "" ||
  SECRET_KEY_ID === "" ||
  BUCKET_NAME === ""
) {
  console.log("envの値でどこかが未指定");
  process.exit(1);
}

const r2Client = new S3Client({
  region: "auto",
  endpoint: `https://${ACCOUNT_ID}.r2.cloudflarestorage.com`,
  credentials: {
    accessKeyId: ACCESS_KEY_ID,
    secretAccessKey: SECRET_KEY_ID,
  },
});

async function processImageFile(filePath, parentDir) {
  const fileName = path.parse(filePath).name;
  const avifBuffer = await sharp(filePath).toFormat("avif").toBuffer();
  const key = `${parentDir}/${fileName}.avif`;
  await uploadToR2(avifBuffer, key, "image/avif");
}

async function processVideoFile(filePath, parentDir) {
  const fileName = path.parse(filePath).name;
  const key = `${parentDir}/${fileName}.mp4`;
  const fileData = fs.readFileSync(filePath);
  await uploadToR2(fileData, key, "video/mp4");
}

async function uploadToR2(data, key, contentType) {
  if (isDryRun) {
    console.log(`Dry run: Skipping actual upload for ${key}`);
    return;
  }
  try {
    const uploadParams = {
      Bucket: BUCKET_NAME,
      Key: key,
      Body: data,
      ContentType: contentType,
    };
    await r2Client.send(new PutObjectCommand(uploadParams));
    console.log(`Uploaded: ${key}`);
  } catch (err) {
    console.error(`Failed to upload ${key}:`, err);
  }
}

async function walkDir(dir, callback) {
  for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
    const fullPath = path.join(dir, entry.name);
    if (entry.isDirectory()) {
      await callback(fullPath, entry.name);
    }
  }
}

async function processFilesInDirectory(dir, parentDir) {
  for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
    if (entry.isFile()) {
      const ext = path.extname(entry.name).toLowerCase();
      const filePath = path.join(dir, entry.name);
      if (ext === ".jpg" || ext === ".png") {
        await processImageFile(filePath, parentDir);
      } else if (ext === ".mp4") {
        await processVideoFile(filePath, parentDir);
      } else {
        console.warn(`Unexpected file extension: ${entry.name}`);
      }
      fileCount++;
    }
  }
}

async function main() {
  console.log(
    isDryRun
      ? "Running in dry-run mode. No files will be uploaded."
      : "Starting actual upload.",
  );
  await walkDir(targetDir, async (subDir, parentDir) => {
    await processFilesInDirectory(subDir, parentDir);
  });

  const endTime = Date.now();
  const elapsedTime = (endTime - startTime) / 1000;
  const minutes = Math.floor(elapsedTime / 60);
  const seconds = Math.floor(elapsedTime % 60);
  console.log(`Processed ${fileCount} files in ${minutes}分${seconds}秒`);
}

main();