import { promises as fs } from "fs";
import path from "path";
import { getBlogPosts, getDocs } from "@/utils/blog";
import { seoPages } from "../lib/seo-pages";

async function getPagePaths(
	dir: string,
): Promise<{ path: string; lastModified: string }[]> {
	const entries = await fs.readdir(dir, { withFileTypes: true });
	const paths: { path: string; lastModified: string }[] = [];

	for (const entry of entries) {
		const fullPath = path.join(dir, entry.name);
		if (
			entry.isDirectory() &&
			entry.name !== "dashboard" &&
			entry.name !== "blog" &&
			!entry.name.startsWith("[")
		) {
			const subPaths = await getPagePaths(fullPath);
			paths.push(...subPaths);
		} else if (
			entry.isFile() &&
			(entry.name === "page.tsx" || entry.name === "page.mdx")
		) {
			const relativePath = path.relative(process.cwd(), dir);
			// Filter out route groups (directories wrapped in parentheses) and construct clean path
			const pathSegments = relativePath
				.split(path.sep)
				.slice(1)
				.filter(
					(segment) => !(segment.startsWith("(") && segment.endsWith(")")),
				);
			const routePath =
				pathSegments.length > 0 ? "/" + pathSegments.join("/") : "/";

			if (!routePath.includes("/dashboard") && !routePath.includes("[")) {
				const stats = await fs.stat(fullPath);
				paths.push({
					path: routePath === "/app" ? "/" : routePath,
					lastModified: stats.mtime.toISOString(),
				});
			}
		}
	}

	return paths;
}

export default async function sitemap() {
	const appDirectory = path.join(process.cwd(), "app");
	const pagePaths = await getPagePaths(appDirectory);

	// Add blog post routes
	const blogPosts = getBlogPosts();
	const blogRoutes = blogPosts.map((post) => {
		const publishedAt =
			"publishedAt" in post.metadata
				? post.metadata.publishedAt
				: new Date().toISOString();
		const publishDate = new Date(publishedAt);
		publishDate.setHours(9, 0, 0, 0); // Set time to 9:00 AM
		return {
			path: `/blog/${post.slug}`,
			lastModified: publishDate.toISOString(),
		};
	});

	// Add docs routes
	const docs = getDocs();
	const docsRoutes = docs.map((doc) => ({
		path: `/docs/${doc.slug}`,
		lastModified: new Date().toISOString(), // You might want to add a publishedAt to doc metadata
	}));

	// Add SEO pages
	const seoRoutes = Object.keys(seoPages).map((slug) => ({
		path: `/${slug}`,
		lastModified: new Date().toISOString(),
	}));

	// Combine routes and ensure '/' is first
	const allRoutes = [...pagePaths, ...blogRoutes, ...docsRoutes, ...seoRoutes];
	const homeRoute = allRoutes.find((route) => route.path === "/");
	const otherRoutes = allRoutes.filter((route) => route.path !== "/");

	const routes = [...(homeRoute ? [homeRoute] : []), ...otherRoutes].map(
		(route) => ({
			url: `https://cap.so${route.path}`,
			lastModified: route.lastModified,
		}),
	);

	return routes;
}
