Merge pull request #199 from pepe/temple-capture

Add capture* functions to temple module
janet-lang · Sep 24, 2024 · a6f7fe9 · a6f7fe9
2 parents d015a35 + 8168ba7
commit a6f7fe9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/spork/temple.janet b/spork/temple.janet
@@ -127,10 +127,25 @@
   [path &]
   (with-dyns [:current-file path]
     (let [tmpl (create (slurp path) path)]
-      @{'render @{:doc "Main template function."
-                  :value (fn render [&keys args] (tmpl args))}
-        'render-dict @{:doc "Template function, but pass arguments as a dictionary."
-                       :value tmpl}})))
+      @{'render
+        @{:doc "Main template function."
+          :value (fn render [&keys args] (tmpl args))}
+        'render-dict
+        @{:doc "Template function, but pass arguments as a dictionary."
+          :value tmpl}
+        'capture
+        @{:doc "Template function that returns buffer of rendered template."
+          :value (fn capture [&keys args]
+                   (def b @"")
+                   (with-dyns [:out b] (tmpl args))
+                   b)}
+        'capture-dict
+        @{:doc "Template function that returns buffer of rendered template,
+                              but pass arguments as a dictionary."
+          :value (fn capture-dict [args]
+                   (def b @"")
+                   (with-dyns [:out b] (tmpl args))
+                   b)}})))
 
 (defn add-loader
   "Adds the custom template loader to Janet's module/loaders and

diff --git a/test/suite-temple.janet b/test/suite-temple.janet
@@ -33,6 +33,9 @@
                 </html>
                 ```)
 
+(assert (deep= (hi/capture-dict {:a 1 :b 2})
+               (hi/capture :a 1 :b 2)))
+
 (def str-template "<html>hello {{ (args :arg) }}</html>")
 (def render (temple/compile str-template))
 (def out (render :arg "world"))