fix: load visual editor on new pages

`Beacon.Content.get_page/2` fails since there's no page saved yet.

Author: leandrocp

lib/beacon/web/beacon_assigns.ex

@@ -74,7 +74,7 @@ defmodule Beacon.Web.BeaconAssigns do
     page_module = Beacon.Loader.Page.module_name(site, page.id)
     live_data = Beacon.Web.DataSource.live_data(site, path_info, Map.drop(query_params, ["path"]))
     path_params = Beacon.Router.path_params(page.path, path_info)
-    page_title = Beacon.Web.DataSource.page_title(site, page.id, live_data, source)
+    page_title = Beacon.Web.DataSource.page_title(site, page, live_data, source)
     components_module = Beacon.Loader.Components.module_name(site)
     info_handlers_module = Beacon.Loader.InfoHandlers.module_name(site)
     event_handlers_module = Beacon.Loader.EventHandlers.module_name(site)

lib/beacon/web/data_source.ex

@@ -7,8 +7,8 @@ defmodule Beacon.Web.DataSource do
     Beacon.apply_mfa(site, Beacon.Loader.fetch_live_data_module(site), :live_data, [path_info, params])
   end
 
-  def page_title(site, page_id, live_data, source) do
-    %{path: path, title: title} = page_assigns = page_assigns(site, page_id, source)
+  def page_title(site, page, live_data, source) do
+    %{path: path, title: title} = page_assigns = page_assigns(site, page, source)
 
     with {:ok, page_title} <- Beacon.Content.render_snippet(title, %{page: page_assigns, live_data: live_data}) do
       page_title
@@ -58,13 +58,14 @@ defmodule Beacon.Web.DataSource do
   end
 
   # only published pages will have the title evaluated for beacon
-  defp page_assigns(site, page_id, :beacon) do
-    Beacon.apply_mfa(site, Beacon.Loader.fetch_page_module(site, page_id), :page_assigns, [])
+  defp page_assigns(site, page, :beacon) do
+    Beacon.apply_mfa(site, Beacon.Loader.fetch_page_module(site, page.id), :page_assigns, [])
   end
 
   # beacon_live_admin needs the title for unpublished pages also
-  defp page_assigns(site, page_id, :admin) do
-    page = Beacon.Content.get_page(site, page_id)
+  defp page_assigns(site, page, :admin) do
+    # new pages are not yet in the database
+    page = if page.id, do: Beacon.Content.get_page(site, page.id), else: page
 
     %{
       id: page.id,