per-site sitemaps via beacon_site

lib/beacon/loader/routes.ex

@@ -49,6 +49,11 @@ defmodule Beacon.Loader.Routes do
           @endpoint.url() <> beacon_media_path(file_name)
         end
 
+        def beacon_page_url(%{path: path} = page) do
+          prefix = @router.__beacon_scoped_prefix_for_site__(@site)
+          Path.join([@endpoint.url(), prefix, path])
+        end
+
         defp sanitize_path(path) do
           String.replace(path, "//", "/")
         end

lib/beacon/router.ex

@@ -145,6 +145,8 @@ defmodule Beacon.Router do
           get "/__beacon_assets__/css-:md5", Beacon.Web.AssetsController, :css, assigns: %{site: opts[:site]}
           get "/__beacon_assets__/js-:md5", Beacon.Web.AssetsController, :js, assigns: %{site: opts[:site]}
 
+          get "/sitemap.xml", Beacon.Web.SitemapController, :show, as: :beacon_sitemap, assigns: %{site: opts[:site]}
+
           live "/*path", Beacon.Web.PageLive, :path
         end
       end

lib/beacon/web/controllers/sitemap_controller.ex

@@ -0,0 +1,32 @@
+defmodule Beacon.Web.SitemapController do
+  @moduledoc false
+  use Beacon.Web, :controller
+
+  def init(action) when action in [:index, :show], do: action
+
+  # def call(%{assigns: %{sites: _sites}} = _conn, :index) do
+  #   # TODO render embedded sitemap_index.xml.eex
+  # end
+
+  def call(%{assigns: %{site: site}} = conn, :show) do
+    conn
+    |> put_view(Beacon.Web.SitemapXML)
+    |> put_resp_content_type("text/xml")
+    # may need to adjust caching
+    |> put_resp_header("cache-control", "public max-age=300")
+    |> render(:sitemap, pages: get_pages(site))
+  end
+
+  defp get_pages(site) do
+    routes_module = Beacon.Loader.fetch_routes_module(site)
+
+    site
+    |> Beacon.Content.list_published_pages()
+    |> Enum.map(fn page ->
+      %{
+        loc: Beacon.apply_mfa(site, routes_module, :beacon_page_url, [page]),
+        lastmod: page.updated_at
+      }
+    end)
+  end
+end

lib/beacon/web/sitemap/sitemap.xml.eex

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
+    <%= for page <- @pages do %><url>
+        <loc><%= page.loc %></loc>
+        <lastmod><%= page.lastmod %></lastmod>
+    </url><% end %>
+</urlset>

lib/beacon/web/sitemap_xml.ex

@@ -0,0 +1,4 @@
+defmodule Beacon.Web.SitemapXML do
+  import Phoenix.Template, only: [embed_templates: 1]
+  embed_templates "sitemap/*.xml"
+end

test/beacon_web/controllers/media_library_controller_test.exs

@@ -1,12 +1,6 @@
 defmodule Beacon.Web.Controllers.MediaLibraryControllerTest do
   use Beacon.Web.ConnCase, async: true
 
-  setup do
-    Process.flag(:error_handler, Beacon.ErrorHandler)
-    Process.put(:__beacon_site__, :my_site)
-    :ok
-  end
-
   test "show", %{conn: conn} do
     %{file_name: file_name} = Beacon.Test.Fixtures.beacon_media_library_asset_fixture(site: :my_site)
     routes = Beacon.Loader.fetch_routes_module(:my_site)

test/beacon_web/controllers/sitemap_controller_test.exs

@@ -0,0 +1,33 @@
+defmodule Beacon.Web.SitemapControllerTest do
+  use Beacon.Web.ConnCase, async: false
+
+  test "show", %{conn: conn} do
+    site = :my_site
+
+    layout =
+      beacon_published_layout_fixture(
+        site: site,
+        template: """
+        <header>Page header</header>
+        <%= @inner_content %>
+        <footer>Page footer</footer>
+        """
+      )
+
+    page = beacon_published_page_fixture(site: site, path: "/foo", layout_id: layout.id)
+
+    routes = Beacon.Loader.fetch_routes_module(site)
+    conn = get(conn, "/sitemap.xml")
+
+    assert response(conn, 200) == """
+    &lt;?xml version=&quot;1.0&quot; encoding=&quot;UTF-8&quot;?&gt;
+    &lt;urlset xmlns=&quot;http://www.sitemaps.org/schemas/sitemap/0.9&quot;&gt;
+        &lt;url&gt;
+            &lt;loc&gt;#{routes.beacon_page_url(page)}&lt;/loc&gt;
+            &lt;lastmod&gt;#{DateTime.to_string(page.updated_at)}&lt;/lastmod&gt;
+        &lt;/url&gt;
+    &lt;/urlset&gt;
+    """
+    assert response_content_type(conn, :xml) =~ "charset=utf-8"
+  end
+end