Update page router to inject live edit scripts when enabled

hydephp · Nov 13, 2023 · 9166d4e · 9166d4e
1 parent 3e4e39c
commit 9166d4e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/packages/realtime-compiler/src/Routing/PageRouter.php b/packages/realtime-compiler/src/Routing/PageRouter.php
@@ -5,6 +5,7 @@
 use Desilva\Microserve\Request;
 use Desilva\Microserve\Response;
 use Hyde\Foundation\Facades\Routes;
+use Hyde\Pages\Concerns\BaseMarkdownPage;
 use Hyde\Framework\Actions\StaticPageBuilder;
 use Hyde\RealtimeCompiler\Http\LiveEditController;
 use Hyde\Framework\Features\Documentation\DocumentationSearchPage;
@@ -75,6 +76,10 @@ protected function getHtml(HydePage $page): string
             $contents = $page->compile();
         }
 
+        if ($page instanceof BaseMarkdownPage && LiveEditController::enabled()) {
+            $contents = LiveEditController::injectLiveEditScript($contents);
+        }
+
         return $contents;
     }
 

diff --git a/packages/realtime-compiler/tests/RealtimeCompilerTest.php b/packages/realtime-compiler/tests/RealtimeCompilerTest.php
@@ -17,6 +17,10 @@
 
 ob_start();
 
+beforeEach(function () {
+    putenv('SERVER_LIVE_EDIT=false');
+});
+
 test('handle routes index page', function () {
     putenv('SERVER_DASHBOARD=false');
     mockRoute('');