#!/usr/bin/env bun

import { readdirSync, statSync } from "node:fs";

import { join } from "node:path";

const distDir = "./docs/dist";

const ogImageUrl = "https://sequoia.pub/og.png";

// Function to recursively find all HTML files

function findHtmlFiles(dir: string): string[] {

	const files: string[] = [];

	const entries = readdirSync(dir);

	for (const entry of entries) {

		const fullPath = join(dir, entry);

		const stat = statSync(fullPath);

		if (stat.isDirectory()) {

			files.push(...findHtmlFiles(fullPath));

		} else if (entry.endsWith(".html")) {

			files.push(fullPath);

		}

	}

	return files;

}

// Function to inject OG image meta tags

async function injectOgImageTags(filePath: string) {

	const file = Bun.file(filePath);

	const content = await file.text();

	// Check if og:image already exists

	if (content.includes('property="og:image"')) {

		console.log(`⏭️  Skipping ${filePath} - og:image already exists`);

		return;

	}

	// Find the position to inject the meta tag

	// We'll insert it after og:description if it exists, or before twitter:card

	const ogDescriptionMatch = content.match(

		/<meta property="og:description"[^>]*>/,

	);

	const twitterCardMatch = content.match(/<meta name="twitter:card"[^>]*>/);

	let insertPosition: number;

	if (ogDescriptionMatch && ogDescriptionMatch.index !== undefined) {

		insertPosition = ogDescriptionMatch.index + ogDescriptionMatch[0].length;

	} else if (twitterCardMatch && twitterCardMatch.index !== undefined) {

		insertPosition = twitterCardMatch.index;

	} else {

		// Fallback: insert before </head>

		const headCloseMatch = content.indexOf("</head>");

		if (headCloseMatch === -1) {

			console.log(`⚠️  Warning: Could not find insertion point in ${filePath}`);

			return;

		}

		insertPosition = headCloseMatch;

	}

	// Inject the og:image and twitter:image meta tags

	const ogImageTag = `<meta property="og:image" content="${ogImageUrl}"/>`;

	const twitterImageTag = `<meta name="twitter:image" content="${ogImageUrl}"/>`;

	const newContent =

		content.slice(0, insertPosition) +

		ogImageTag +

		twitterImageTag +

		content.slice(insertPosition);

	// Write the modified content back to the file

	await Bun.write(filePath, newContent);

	console.log(`✅ Injected og:image tags into ${filePath}`);

}

// Main execution

async function main() {

	console.log("🔍 Finding HTML files in dist directory...");

	const htmlFiles = findHtmlFiles(distDir);

	console.log(`📄 Found ${htmlFiles.length} HTML files`);

	for (const file of htmlFiles) {

		await injectOgImageTags(file);

	}

	console.log("\n✨ Done! All HTML files have been processed.");

}

main();