feat: add sitemap

cptchloroplast · Dec 6, 2024 · b3b85ef · b3b85ef
1 parent 09127b2
commit b3b85ef
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git a/src/pages/sitemap.xml.ts b/src/pages/sitemap.xml.ts
@@ -0,0 +1,37 @@
+import type { Metadata, Navigation, Post, Tag } from "@schemas"
+import { PostService } from "@services"
+import type { APIContext } from "astro"
+
+export async function GET(context: APIContext) {
+  const metadata = context.locals.metadata
+  const service = PostService(context.locals.runtime.env.DB)
+  const posts = await service.list()
+  const tags = await service.listTags()
+  return new Response(renderSitemap(context.url.origen, metadata, posts), {
+    headers: {
+      "Content-Type": "application/xml"
+    }
+  })
+}
+
+function renderSitemap(origen: string, metadata: Metadata, posts: Post[]) {
+  const map: Record<string, Date> = {}
+  const latest = posts[0].published!
+  const first = posts[posts.length - 1].published!
+  map[`${origen}`] = latest
+  map[`${origen}/rss.xml`] = latest
+  map[`${origen}/contact`] = first
+  map[`${origen}/posts`] = latest
+  for (const post of posts) {
+    map[`${origen}/posts/${post.slug}`] = post.updated ?? post.published!
+  }
+  const entries = Object.entries(map)
+  entries.sort(([loc1], [loc2]) => loc1 < loc2 ? -1 : 1)
+  return '<?xml version="1.0" encoding="UTF-8"?><urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">' 
+    + entries.map(([loc, lastmod]) => url(loc, lastmod)).join("")
+    + "</urlset>"
+}
+
+function url(loc: string, lastmod: Date) {
+  return `<url><loc>${loc}</loc><lastmod>${lastmod.toISOString().split("T")[0]}</lastmod></url>`
+}
diff --git a/src/services/PostService.ts b/src/services/PostService.ts
@@ -1,5 +1,5 @@
 import type { D1Database } from "@cloudflare/workers-types"
-import { type Post, PostsTable, schema, TagsTable } from "@schemas"
+import { type Post, PostsTable, schema, type Tag, TagsTable } from "@schemas"
 import { conflictUpdateAllExcept } from "@utils"
 import { asc, count, desc, eq, inArray, type InferSelectModel, and, isNotNull } from "drizzle-orm"
 import { drizzle } from "drizzle-orm/d1"
@@ -10,7 +10,7 @@ type PostService = {
     getLatest(): Promise<Post | undefined>
     list(): Promise<Post[]>
     listByTag(tag: string): Promise<Post[]>
-    listTags(): Promise<{ name: string, count: number }[]>
+    listTags(): Promise<{ name: Tag, count: number }[]>
     upsert(value: Post): Promise<Post>
 }
 type PostRecord = InferSelectModel<typeof PostsTable> & { tags: InferSelectModel<typeof TagsTable>[] }