Enable minisite to use a custom template when overriding left menu

yupiik · Oct 13, 2022 · b53c478 · b53c478
1 parent 7e2c976
commit b53c478
Showing 1 changed file with 17 additions and 5 deletions.
diff --git a/minisite-core/src/main/java/io/yupiik/tools/minisite/MiniSite.java b/minisite-core/src/main/java/io/yupiik/tools/minisite/MiniSite.java
@@ -237,7 +237,11 @@ protected String leftMenu(final Map<Page, Path> files) {
                 final Path output = configuration.getTarget();
                 return findIndexPages(files).map(it -> toMenuItem(output, it, itemTemplate)).collect(joining());
             }
-            throw new IllegalArgumentException("Unknown key '" + key + "'");
+            try {
+                return findPageTemplate(templatesDir, key);
+            } catch (final RuntimeException re) {
+                throw new IllegalArgumentException("Unknown key '" + key + "'");
+            }
         }).replace(template);
     }
 
@@ -346,7 +350,11 @@ protected String generateIndex(final Map<Page, Path> htmls, final Function<Page,
                                 case "content":
                                     return indexContent;
                                 default:
-                                    throw new IllegalArgumentException("Unknown key '" + key + "'");
+                                    try {
+                                        return findPageTemplate(templatesDir, key);
+                                    } catch (final RuntimeException re) {
+                                        throw new IllegalArgumentException("Unknown key '" + key + "'");
+                                    }
                             }
                         }).replace(contentTemplate))));
 
@@ -459,10 +467,14 @@ protected String getDefaultInterpolation(final String key, final Page page,
             case "xyz": // passthrough
                 return "{{xyz}}";
             default:
-                if (emptyIfMissing) {
-                    return "";
+                try {
+                    return findPageTemplate(getTemplatesDir(), key);
+                } catch (final RuntimeException re) {
+                    if (emptyIfMissing) {
+                        return "";
+                    }
+                    throw new IllegalArgumentException("Unknown template key '" + key + "'");
                 }
-                throw new IllegalArgumentException("Unknown template key '" + key + "'");
         }
     }