Make match ref/deref as needed

Much of this can be yanked out and made simpler
when match-modes lands in stable

1 files changed, 24 insertions, 1 deletions

diff --git a/askama_shared/src/generator.rs b/askama_shared/src/generator.rs
index 0c8b1e2..701de02 100644
--- a/askama_shared/src/generator.rs
+++ b/askama_shared/src/generator.rs
@@ -161,6 +161,7 @@ impl<'a> Generator<'a> {
         self.write_header(state, "::askama::Template", &[]);
         self.writeln("fn render_into(&self, writer: &mut ::std::fmt::Write) -> \
                       ::askama::Result<()> {");
+        self.writeln("#[allow(unused_imports)] use ::std::ops::Deref as HiddenDerefTrait;");
         self.handle(state, state.nodes, AstLevel::Top);
         self.flush_ws(&WS(false, false));
         self.writeln("Ok(())");
@@ -199,6 +200,7 @@ impl<'a> Generator<'a> {
             "fn render_trait_into(&self, timpl: &{}, writer: &mut ::std::fmt::Write) \
              -> ::askama::Result<()> {{",
             state.trait_name));
+        self.writeln("#[allow(unused_imports)] use ::std::ops::Deref as HiddenDerefTrait;");
 
         if let Some(nodes) = nodes {
             self.handle(state, nodes, AstLevel::Top);
@@ -439,14 +441,18 @@ impl<'a> Generator<'a> {
         }
 
         self.write("match ");
+        self.write("(&");
         self.visit_expr(expr);
+        self.write(").deref()");
         self.writeln(" {");
 
         for arm in arms {
             let &(ref ws, ref variant, ref params, ref body) = arm;
             self.locals.push();
             match *variant {
-                Some(ref param) => { self.visit_match_param(param); },
+                Some(ref param) => {
+                    self.visit_match_variant(param);
+                },
                 None => self.write("_"),
             };
             if params.len() > 0 {
@@ -629,11 +635,28 @@ impl<'a> Generator<'a> {
         }
     }
 
+    fn visit_match_variant(&mut self, param: &MatchParameter) -> DisplayWrap {
+        match *param {
+            MatchParameter::StrLit(s) => self.visit_str_lit(s),
+            MatchParameter::NumLit(s) => {
+                // Variants need to be references until match-modes land
+                self.write("&");
+                self.visit_num_lit(s)
+            },
+            MatchParameter::Name(s) => {
+                self.write("&");
+                self.write(s);
+                DisplayWrap::Unwrapped
+            }
+        }
+    }
+
     fn visit_match_param(&mut self, param: &MatchParameter) -> DisplayWrap {
         match *param {
             MatchParameter::NumLit(s) => self.visit_num_lit(s),
             MatchParameter::StrLit(s) => self.visit_str_lit(s),
             MatchParameter::Name(s) => {
+                self.write("ref ");
                 self.write(s);
                 DisplayWrap::Unwrapped
             }