Simplify code to find template path

1 files changed, 10 insertions, 23 deletions

diff --git a/askama_derive/src/lib.rs b/askama_derive/src/lib.rs
index 59d1ae1..244c43b 100644
--- a/askama_derive/src/lib.rs
+++ b/askama_derive/src/lib.rs
@@ -4,32 +4,19 @@ extern crate syn;
 
 use proc_macro::TokenStream;
 
-fn get_path_from_attrs(attrs: &Vec<syn::Attribute>) -> String {
-    for attr in attrs {
-        if attr.name() == "template" {
-            match attr.value {
-                syn::MetaItem::List(_, ref inner) => {
-                    match inner[0] {
-                        syn::NestedMetaItem::MetaItem(ref item) => {
-                            match item {
-                                &syn::MetaItem::NameValue(ref key, ref val) => {
-                                    assert_eq!(key.as_ref(), "path");
-                                    match val {
-                                        &syn::Lit::Str(ref s, _) => { return s.clone(); },
-                                        _ => panic!("template path must be a string"),
-                                    }
-                                },
-                                _ => panic!("template annotation must contain key/value pair"),
-                            }
-                        },
-                        _ => panic!("template annotation must contain item"),
-                    }
-                },
-                _ => panic!("template annotation must be of List type"),
+fn get_path_from_attrs(attrs: &[syn::Attribute]) -> String {
+    let attr = attrs.iter().find(|a| a.name() == "template").unwrap();
+    if let syn::MetaItem::List(_, ref inner) = attr.value {
+        if let syn::NestedMetaItem::MetaItem(ref item) = inner[0] {
+            if let &syn::MetaItem::NameValue(ref key, ref val) = item {
+                assert_eq!(key.as_ref(), "path");
+                if let &syn::Lit::Str(ref s, _) = val {
+                    return s.clone();
+                }
             }
         }
     }
-    panic!("template annotation not found");
+    panic!("template path not found in struct attributes");
 }
 
 #[proc_macro_derive(Template, attributes(template))]