aboutsummaryrefslogtreecommitdiffstats
path: root/askama_derive
diff options
context:
space:
mode:
Diffstat (limited to 'askama_derive')
-rw-r--r--askama_derive/src/generator.rs2
-rw-r--r--askama_derive/src/input.rs221
-rw-r--r--askama_derive/src/lib.rs3
3 files changed, 2 insertions, 224 deletions
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index 4f5b495..a12b1f0 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -1,6 +1,6 @@
use super::{get_template_source, Context, Heritage};
-use crate::input::{Source, TemplateInput};
use askama_shared::filters;
+use askama_shared::input::{Source, TemplateInput};
use askama_shared::parser::{
parse, Cond, Expr, MatchParameter, MatchParameters, MatchVariant, Node, Target, When, WS,
};
diff --git a/askama_derive/src/input.rs b/askama_derive/src/input.rs
deleted file mode 100644
index 6a54a99..0000000
--- a/askama_derive/src/input.rs
+++ /dev/null
@@ -1,221 +0,0 @@
-use askama_shared::{Config, Syntax};
-
-use std::path::PathBuf;
-
-use quote::ToTokens;
-use syn;
-
-pub struct TemplateInput<'a> {
- pub ast: &'a syn::DeriveInput,
- pub config: &'a Config<'a>,
- pub syntax: &'a Syntax<'a>,
- pub source: Source,
- pub print: Print,
- pub escaper: &'a str,
- pub ext: Option<String>,
- pub parent: Option<&'a syn::Type>,
- pub path: PathBuf,
-}
-
-impl<'a> TemplateInput<'a> {
- /// Extract the template metadata from the `DeriveInput` structure. This
- /// mostly recovers the data for the `TemplateInput` fields from the
- /// `template()` attribute list fields; it also finds the of the `_parent`
- /// field, if any.
- pub fn new<'n>(ast: &'n syn::DeriveInput, config: &'n Config) -> TemplateInput<'n> {
- // Check that an attribute called `template()` exists and that it is
- // the proper type (list).
- let meta = ast
- .attrs
- .iter()
- .find_map(|attr| match attr.parse_meta() {
- Ok(m) => {
- if m.path().is_ident("template") {
- Some(m)
- } else {
- None
- }
- }
- Err(e) => panic!("unable to parse attribute: {}", e),
- })
- .expect("no attribute 'template' found");
-
- let meta_list = match meta {
- syn::Meta::List(inner) => inner,
- _ => panic!("attribute 'template' has incorrect type"),
- };
-
- // Loop over the meta attributes and find everything that we
- // understand. Raise panics if something is not right.
- // `source` contains an enum that can represent `path` or `source`.
- let mut source = None;
- let mut print = Print::None;
- let mut escaping = None;
- let mut ext = None;
- let mut syntax = None;
- for item in meta_list.nested {
- let pair = match item {
- syn::NestedMeta::Meta(syn::Meta::NameValue(ref pair)) => pair,
- _ => panic!(
- "unsupported attribute argument {:?}",
- item.to_token_stream()
- ),
- };
-
- if pair.path.is_ident("path") {
- if let syn::Lit::Str(ref s) = pair.lit {
- if source.is_some() {
- panic!("must specify 'source' or 'path', not both");
- }
- source = Some(Source::Path(s.value()));
- } else {
- panic!("template path must be string literal");
- }
- } else if pair.path.is_ident("source") {
- if let syn::Lit::Str(ref s) = pair.lit {
- if source.is_some() {
- panic!("must specify 'source' or 'path', not both");
- }
- source = Some(Source::Source(s.value()));
- } else {
- panic!("template source must be string literal");
- }
- } else if pair.path.is_ident("print") {
- if let syn::Lit::Str(ref s) = pair.lit {
- print = s.value().into();
- } else {
- panic!("print value must be string literal");
- }
- } else if pair.path.is_ident("escape") {
- if let syn::Lit::Str(ref s) = pair.lit {
- escaping = Some(s.value());
- } else {
- panic!("escape value must be string literal");
- }
- } else if pair.path.is_ident("ext") {
- if let syn::Lit::Str(ref s) = pair.lit {
- ext = Some(s.value());
- } else {
- panic!("ext value must be string literal");
- }
- } else if pair.path.is_ident("syntax") {
- if let syn::Lit::Str(ref s) = pair.lit {
- syntax = Some(s.value())
- } else {
- panic!("syntax value must be string literal");
- }
- } else {
- panic!(
- "unsupported attribute key '{}' found",
- pair.path.to_token_stream()
- )
- }
- }
-
- // Validate the `source` and `ext` value together, since they are
- // related. In case `source` was used instead of `path`, the value
- // of `ext` is merged into a synthetic `path` value here.
- let source = source.expect("template path or source not found in attributes");
- let path = match (&source, &ext) {
- (&Source::Path(ref path), None) => config.find_template(path, None),
- (&Source::Source(_), Some(ext)) => PathBuf::from(format!("{}.{}", ast.ident, ext)),
- (&Source::Path(_), Some(_)) => {
- panic!("'ext' attribute cannot be used with 'path' attribute")
- }
- (&Source::Source(_), None) => {
- panic!("must include 'ext' attribute when using 'source' attribute")
- }
- };
-
- // Check to see if a `_parent` field was defined on the context
- // struct, and store the type for it for use in the code generator.
- let parent = match ast.data {
- syn::Data::Struct(syn::DataStruct {
- fields: syn::Fields::Named(ref fields),
- ..
- }) => fields
- .named
- .iter()
- .find(|f| f.ident.as_ref().filter(|name| *name == "_parent").is_some())
- .map(|f| &f.ty),
- _ => None,
- };
-
- if parent.is_some() {
- eprint!(
- " --> in struct {}\n = use of deprecated field '_parent'\n",
- ast.ident
- );
- }
-
- // Validate syntax
- let syntax = syntax.map_or_else(
- || config.syntaxes.get(config.default_syntax).unwrap(),
- |s| {
- config
- .syntaxes
- .get(&s)
- .unwrap_or_else(|| panic!("attribute syntax {} not exist", s))
- },
- );
-
- // Match extension against defined output formats
-
- let extension = escaping.unwrap_or_else(|| {
- path.extension()
- .map(|s| s.to_str().unwrap())
- .unwrap_or("")
- .to_string()
- });
-
- let mut escaper = None;
- for (extensions, path) in &config.escapers {
- if extensions.contains(&extension) {
- escaper = Some(path);
- break;
- }
- }
-
- let escaper = escaper.unwrap_or_else(|| {
- panic!("no escaper defined for extension '{}'", extension);
- });
-
- TemplateInput {
- ast,
- config,
- source,
- print,
- escaper,
- ext,
- parent,
- path,
- syntax,
- }
- }
-}
-
-pub enum Source {
- Path(String),
- Source(String),
-}
-
-#[derive(PartialEq)]
-pub enum Print {
- All,
- Ast,
- Code,
- None,
-}
-
-impl From<String> for Print {
- fn from(s: String) -> Print {
- use self::Print::*;
- match s.as_ref() {
- "all" => All,
- "ast" => Ast,
- "code" => Code,
- "none" => None,
- v => panic!("invalid value for print option: {}", v),
- }
- }
-}
diff --git a/askama_derive/src/lib.rs b/askama_derive/src/lib.rs
index 2189458..0b64057 100644
--- a/askama_derive/src/lib.rs
+++ b/askama_derive/src/lib.rs
@@ -3,9 +3,8 @@ extern crate proc_macro;
extern crate quote;
mod generator;
-mod input;
-use crate::input::{Print, Source, TemplateInput};
+use askama_shared::input::{Print, Source, TemplateInput};
use askama_shared::parser::{parse, Expr, Macro, Node};
use askama_shared::{read_config_file, Config};
use proc_macro::TokenStream;
generated by cgit (git 2.34.1) at 2024-11-22 04:18:17 +0000