optionally return tree

[html-element-library.git] / lib / HTML / Element / Library.pm

diff --git a/lib/HTML/Element/Library.pm b/lib/HTML/Element/Library.pm
index df9681f9d73b269e40d0ca65be585f931139f1b7..4c2d8793da93fa49dabc1c98df05aad4b09d621a 100644 (file)
--- a/lib/HTML/Element/Library.pm
+++ b/lib/HTML/Element/Library.pm
@@ -34,12 +34,20 @@ our $VERSION = '3.53';
 # https://rt.cpan.org/Ticket/Display.html?id=44105
 sub HTML::Element::fillinform {
 
-    my ($tree, $hashref)=@_;
+    my ($tree, $hashref, $return_tree)=@_;
+
+    (ref $hashref) eq 'HASH' or die 'hashref not supplied as argument' ;
 
     use HTML::FillInForm;
     my $html = $tree->as_HTML;
     my $new_html = HTML::FillInForm->fill(\$html, $hashref);
 
+    if ($return_tree) {
+       HTML::TreeBuilder->new_from_content($new_html);
+    } else {
+       $new_html;
+    }
+
 }
 
 sub HTML::Element::siblings {