From 2937c81a8c8c8130f43a429dfc7b2d235af4f69c Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@skypack.dev>
Date: Mon, 22 Nov 2021 05:30:41 -0500
Subject: [PATCH] spike: experiment with automatic sitemap generation

---
 .../src/components/Nav.astro                  |  2 +
 packages/astro/src/core/create-vite.ts        |  2 +
 packages/astro/src/runtime/server/index.ts    |  4 ++
 .../astro/src/vite-plugin-sitemap/README.md   |  3 ++
 .../astro/src/vite-plugin-sitemap/index.ts    | 39 +++++++++++++++++++
 5 files changed, 50 insertions(+)
 create mode 100644 packages/astro/src/vite-plugin-sitemap/README.md
 create mode 100644 packages/astro/src/vite-plugin-sitemap/index.ts

diff --git a/examples/blog-multiple-authors/src/components/Nav.astro b/examples/blog-multiple-authors/src/components/Nav.astro
index 04a537f87..08f6fabc6 100644
--- a/examples/blog-multiple-authors/src/components/Nav.astro
+++ b/examples/blog-multiple-authors/src/components/Nav.astro
@@ -3,6 +3,8 @@ export interface Props {
   title: string;
 }
 const { title } = Astro.props;
+const routes = await Astro.generateRouteManifest();
+console.log(routes);
 ---
 
 <style lang="scss">
diff --git a/packages/astro/src/core/create-vite.ts b/packages/astro/src/core/create-vite.ts
index 8dd9d1663..24d3eb5e8 100644
--- a/packages/astro/src/core/create-vite.ts
+++ b/packages/astro/src/core/create-vite.ts
@@ -11,6 +11,7 @@ import configAliasVitePlugin from '../vite-plugin-config-alias/index.js';
 import markdownVitePlugin from '../vite-plugin-markdown/index.js';
 import jsxVitePlugin from '../vite-plugin-jsx/index.js';
 import fetchVitePlugin from '../vite-plugin-fetch/index.js';
+import sitemapVitePlugin from '../vite-plugin-sitemap/index.js';
 import { resolveDependency } from './util.js';
 
 // Some packages are just external, and that’s the way it goes.
@@ -56,6 +57,7 @@ export async function createVite(inlineConfig: ViteConfigWithSSR, { astroConfig,
       jsxVitePlugin({ config: astroConfig, logging }),
       astroPostprocessVitePlugin({ config: astroConfig, devServer }),
       fetchVitePlugin(),
+      sitemapVitePlugin({ config: astroConfig })
     ],
     publicDir: fileURLToPath(astroConfig.public),
     root: fileURLToPath(astroConfig.projectRoot),
diff --git a/packages/astro/src/runtime/server/index.ts b/packages/astro/src/runtime/server/index.ts
index 96e358266..2933336bd 100644
--- a/packages/astro/src/runtime/server/index.ts
+++ b/packages/astro/src/runtime/server/index.ts
@@ -284,6 +284,10 @@ export function createAstro(fileURLStr: string, site: string): AstroGlobalPartia
   return {
     site: new URL(site),
     fetchContent,
+    async generateRouteManifest() {
+      const { default: routes } = await import('virtual:@astrojs/sitemap');
+      return routes;
+    },
     // INVESTIGATE is there a use-case for multi args?
     resolve(...segments: string[]) {
       return segments.reduce((u, segment) => new URL(segment, u), url).pathname;
diff --git a/packages/astro/src/vite-plugin-sitemap/README.md b/packages/astro/src/vite-plugin-sitemap/README.md
new file mode 100644
index 000000000..28d65a095
--- /dev/null
+++ b/packages/astro/src/vite-plugin-sitemap/README.md
@@ -0,0 +1,3 @@
+# vite-plugin-fetch
+
+Adds fetch support in SSR contexts.
diff --git a/packages/astro/src/vite-plugin-sitemap/index.ts b/packages/astro/src/vite-plugin-sitemap/index.ts
new file mode 100644
index 000000000..8615ad168
--- /dev/null
+++ b/packages/astro/src/vite-plugin-sitemap/index.ts
@@ -0,0 +1,39 @@
+import type { Plugin } from '../core/vite';
+import type { AstroConfig } from '../@types/astro';
+
+import { createRouteManifest } from '../core/ssr/routing.js';
+
+export default function pluginSitemap({ config }: { config: AstroConfig }): Plugin {
+  const VIRTUAL_FILE_ID = `virtual:@astrojs/sitemap`;
+  
+  return {
+    name: '@astrojs/vite-plugin-sitemap',
+    resolveId(id) {
+      if (id === VIRTUAL_FILE_ID) {
+        return VIRTUAL_FILE_ID
+      }
+      return null
+    },
+    load(id) {
+      if (id === VIRTUAL_FILE_ID) {
+        const routeManifest = createRouteManifest({ config });
+        const sitemap: Record<string, any> = {};
+        for (const route of routeManifest.routes) {
+          if (route.pathname) {
+            sitemap[route.pathname] = {
+              component: '/' + route.component,
+              dynamic: false
+            }
+          } else {
+            const pathname = route.component.replace(/^src\/pages/, '').replace(/\.(md|astro)$/, '');
+            sitemap[pathname] = {
+              component: '/' + route.component,
+              dynamic: true
+            }
+          }
+        }
+        return `export default ${JSON.stringify(sitemap)}`
+      }
+    }
+  }
+}