From 68a1f00dfcceee94f847807842e04bf3a6635875 Mon Sep 17 00:00:00 2001
From: Javier Godoy <11554739+javier-godoy@users.noreply.github.com>
Date: Thu, 30 Apr 2026 12:23:12 -0300
Subject: [PATCH] fix: cleanup multiline DemoSource

Close #85
---
 .../META-INF/resources/frontend/code-viewer.ts         | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/base/src/main/resources/META-INF/resources/frontend/code-viewer.ts b/base/src/main/resources/META-INF/resources/frontend/code-viewer.ts
index 709cc93..034d98c 100644
--- a/base/src/main/resources/META-INF/resources/frontend/code-viewer.ts
+++ b/base/src/main/resources/META-INF/resources/frontend/code-viewer.ts
@@ -324,6 +324,15 @@ pre[class*="language-"] {
         if (!guards.every(x=>x=='istrue')) {
             lines[i] = null;
         }
+
+        if (lines[i] !== null && lines[i]!.startsWith('@DemoSource')) {
+            let open = lines[i]!.includes('(');
+            do {
+                open &&= !(lines[i] || '').includes(')');
+                lines[i++] = null;
+            } while (open && i < lines.length);
+            i--;
+        }
     }
     
     return lines.filter(line=>line!==null)
@@ -332,7 +341,6 @@ pre[class*="language-"] {
        !line.match("//\\s*hide-source(\\s|$)")
     && !line.startsWith('@Route')
     && !line.startsWith('@PageTitle')
-    && !line.startsWith('@DemoSource')
     && !line.startsWith('@SuppressWarnings')
     && !line.startsWith('@Ignore')
     && !line.startsWith('package ')