fix: allow using endpoints under src/routes

astrolicious · Apr 18, 2024 · 6159cd0 · 6159cd0
1 parent 1248d50
commit 6159cd0
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 33 deletions.
diff --git a/.changeset/pink-rice-attack.md b/.changeset/pink-rice-attack.md
@@ -0,0 +1,5 @@
+---
+"@astrolicious/i18n": patch
+---
+
+Allows using endpoints under `src/routes`
diff --git a/package/src/routing/register.ts b/package/src/routing/register.ts
@@ -7,6 +7,7 @@ import type {
 	InjectedRoute,
 } from "astro";
 import { addPageDir } from "astro-pages";
+import { AstroError } from "astro/errors";
 import { withLeadingSlash } from "ufo";
 import { normalizePath } from "vite";
 import type { Options } from "../options.js";
@@ -76,39 +77,49 @@ const generateRoute = (
 
 		let content = readFileSync(page.entrypoint, "utf-8");
 
-		content = content
-			.replaceAll("getLocalePlaceholder()", `"${locale}"`)
-			.replaceAll(
-				"getLocalesPlaceholder()",
-				`[${locales.map((locale) => `"${locale}"`).join(", ")}]`,
-			)
-			.replaceAll("getDefaultLocalePlaceholder()", `"${defaultLocale}"`);
-
-		let [, frontmatter, ...body] = content.split("---");
-		// Handle static imports
-		frontmatter = frontmatter.replace(
-			/import\s+([\s\S]*?)\s+from\s+['"](.+?)['"]/g,
-			(_match, p1: string, p2: string) => {
-				const updatedPath =
-					p2.startsWith("./") || p2.startsWith("../")
-						? updateRelativeImports(p2, page.entrypoint, entrypoint)
-						: p2;
-				return `import ${p1} from '${updatedPath}'`;
-			},
-		);
-		// Handle dynamic imports
-		frontmatter = frontmatter.replace(
-			/import\s*\(\s*['"](.+?)['"]\s*\)/g,
-			(_match, p1: string) => {
-				const updatedPath =
-					p1.startsWith("./") || p1.startsWith("../")
-						? updateRelativeImports(p1, page.entrypoint, entrypoint)
-						: p1;
-				return `import('${updatedPath}')`;
-			},
-		);
+		if (page.entrypoint.endsWith(".astro")) {
+			try {
+				content = content
+					.replaceAll("getLocalePlaceholder()", `"${locale}"`)
+					.replaceAll(
+						"getLocalesPlaceholder()",
+						`[${locales.map((locale) => `"${locale}"`).join(", ")}]`,
+					)
+					.replaceAll("getDefaultLocalePlaceholder()", `"${defaultLocale}"`);
+
+				let [, frontmatter, ...body] = content.split("---");
+				// Handle static imports
+				frontmatter = frontmatter.replace(
+					/import\s+([\s\S]*?)\s+from\s+['"](.+?)['"]/g,
+					(_match, p1: string, p2: string) => {
+						const updatedPath =
+							p2.startsWith("./") || p2.startsWith("../")
+								? updateRelativeImports(p2, page.entrypoint, entrypoint)
+								: p2;
+						return `import ${p1} from '${updatedPath}'`;
+					},
+				);
+				// Handle dynamic imports
+				frontmatter = frontmatter.replace(
+					/import\s*\(\s*['"](.+?)['"]\s*\)/g,
+					(_match, p1: string) => {
+						const updatedPath =
+							p1.startsWith("./") || p1.startsWith("../")
+								? updateRelativeImports(p1, page.entrypoint, entrypoint)
+								: p1;
+						return `import('${updatedPath}')`;
+					},
+				);
+
+				content = `---${frontmatter}---${body.join("---")}`;
+			} catch (err) {
+				throw new AstroError(
+					`An error occured while transforming "${page.entrypoint}".`,
+					"Make sure it has a valid frontmatter, even empty",
+				);
+			}
+		}
 
-		content = `---${frontmatter}---${body.join("---")}`;
 		writeFileSync(entrypoint, content, "utf-8");
 
 		return {
@@ -145,7 +156,7 @@ const generateRoute = (
 		injectedRoute: {
 			pattern,
 			entrypoint,
-			prerender,
+			...(prerender ? { prerender } : {}),
 		},
 	};
 };

diff --git a/playground/src/routes/data.json.ts b/playground/src/routes/data.json.ts
@@ -0,0 +1,6 @@
+import { getLocale } from "i18n:astro";
+import type { APIRoute } from "astro";
+
+export const GET: APIRoute = () => {
+	return Response.json({ locale: getLocale(), foo: "bar" });
+};