1 files changed, 75 insertions, 0 deletions
diff --git a/perllib/FixMyStreet/Template/Stash.pm b/perllib/FixMyStreet/Template/Stash.pm
new file mode 100644
index 000000000..dd027400e
--- /dev/null
+++ b/perllib/FixMyStreet/Template/Stash.pm
@@ -0,0 +1,75 @@
+package FixMyStreet::Template::Stash;
+
+use strict;
+use warnings;
+use base qw(Template::Stash);
+use FixMyStreet::Template::Variable;
+use Scalar::Util qw(blessed);
+
+sub get {
+    my $self = shift;
+
+    my $value = $self->SUPER::get(@_);
+
+    $value = FixMyStreet::Template::Variable->new($value) unless ref $value;
+
+    return $value;
+}
+
+# To deal with being able to call var.upper or var.match
+sub _dotop {
+    my $self = shift;
+    my ($root, $item, $args, $lvalue) = @_;
+
+    $args ||= [ ];
+    $lvalue ||= 0;
+
+    return undef unless defined($root) and defined($item);
+    return undef if $item =~ /^[_.]/;
+
+    if (blessed($root) && $root->isa('FixMyStreet::Template::Variable')) {
+        if ((my $value = $Template::Stash::SCALAR_OPS->{ $item }) && ! $lvalue) {
+            my @result = &$value($root->{value}, @$args);
+            if (defined $result[0]) {
+                return scalar @result > 1 ? [ @result ] : $result[0];
+            }
+            return undef;
+        }
+    }
+
+    return $self->SUPER::_dotop(@_);
+}
+
+1;
+__END__
+
+=head1 NAME
+
+FixMyStreet::Template::Stash - The same as Template::HTML::Stash, but
+additionally copes with scalar operations on stash items.
+
+=head1 FUNCTIONS
+
+=head2 get()
+
+An overridden function from Template::Stash that calls the parent class's get
+method, and returns a FixMyStreet::Template::Variable instead of a raw string.
+
+=head2 _dotop()
+
+An overridden function from Template::Stash so that scalar operations on
+wrapped FixMyStreet::Template::Variable strings still function correctly.
+
+=head1 AUTHOR
+
+Martyn Smith, E<lt>msmith@cpan.orgE<gt>
+
+Matthew Somerville, E<lt>matthew@mysociety.orgE<gt>
+
+=head1 LICENSE
+
+This library is free software; you can redistribute it and/or modify
+it under the same terms as Perl itself, either Perl version 5.8.8 or,
+at your option, any later version of Perl 5 you may have available.
+
+=cut