Fix sitemap URLs when serving locally

Author: Paritoshcodes

src/render_engine_cli/cli.py

@@ -192,11 +192,14 @@ def serve(module_site: str, clean: bool, reload: bool, port: int):
     module, site_name = split_module_site(module_site)
     site = get_site(module, site_name)
 
+    server_address = ("127.0.0.1", port)
+
     if clean:
         remove_output_folder(Path(site.output_path), console=console)
-    site.render()
 
-    server_address = ("127.0.0.1", port)
+    local_url = f"http://{server_address[0]}:{server_address[1]}"
+    site.update_site_vars(SITE_URL=local_url)
+    site.render(site_url=local_url)
 
     handler = ServerEventHandler(
         import_path=module,

src/render_engine_cli/event.py

@@ -93,7 +93,9 @@ def rebuild(self) -> None:
         if self.clean:
             remove_output_folder(Path(site.output_path), console=self.console)
         try:
-            site.render()
+            local_url = f"http://{self.server_address[0]}:{self.server_address[1]}"
+            site.update_site_vars(SITE_URL=local_url)
+            site.render(site_url=local_url)
         except Exception:
             self.console.print("[bold red]Failed to render site[/bold red]")
             self.console.print(traceback.format_exc())