use std::borrow::Cow;
use std::collections::HashMap;
use syn::{parse_quote, Stmt, Expr};
use proc_macro2::{Ident, Span};
use crate::parse::Fixture;
pub(crate) fn fixture_resolver<'a>(fixtures: impl Iterator<Item=&'a Fixture>) -> impl Resolver + 'a {
fixtures.map(|f|
( f.name.to_string(), extract_resolve_expression(f).into() )
).collect::<HashMap<_, Expr>>()
}
pub(crate) trait Resolver {
fn resolve(&self, ident: &Ident) -> Option<Cow<Expr>>;
}
pub(crate) fn arg_2_fixture(ident: &Ident, resolver: &impl Resolver) -> Stmt {
let fixture = resolver
.resolve(ident)
.map(|e| e.clone())
.unwrap_or_else(|| default_fixture_resolve(ident));
parse_quote! {
let #ident = #fixture;
}
}
impl<'a> Resolver for HashMap<String, &'a Expr> {
fn resolve(&self, ident: &Ident) -> Option<Cow<Expr>> {
let ident = ident.to_string();
self.get(&ident)
.map(|&c| Cow::Borrowed(c) )
}
}
impl<'a> Resolver for HashMap<String, Expr> {
fn resolve(&self, ident: &Ident) -> Option<Cow<Expr>> {
let ident = ident.to_string();
self.get(&ident)
.map(|c| Cow::Borrowed(c) )
}
}
impl<R1: Resolver, R2: Resolver> Resolver for (R1, R2) {
fn resolve(&self, ident: &Ident) -> Option<Cow<Expr>> {
self.0.resolve(ident).or_else(|| self.1.resolve(ident))
}
}
fn default_fixture_resolve(ident: &Ident) -> Cow<Expr> {
Cow::Owned(parse_quote! { #ident::default() } )
}
fn extract_resolve_expression(fixture: &Fixture) -> syn::Expr {
let name = &fixture.name;
let positional= &fixture.positional;
let pname = format!("partial_{}", positional.len());
let partial = Ident::new(&pname, Span::call_site());
parse_quote! { #name::#partial(#(#positional), *) }
}