use std::io::Write;
use syn;
use bindgen::annotation::*;
use bindgen::cdecl;
use bindgen::config::{Config, Layout};
use bindgen::ir::*;
use bindgen::library::*;
use bindgen::rename::*;
use bindgen::utilities::*;
use bindgen::writer::*;
#[derive(Debug, Clone)]
pub struct Function {
pub name: String,
pub annotations: AnnotationSet,
pub ret: Type,
pub args: Vec<(String, Type)>,
pub extern_decl: bool,
}
impl Function {
pub fn load(name: String,
annotations: AnnotationSet,
decl: &syn::FnDecl,
extern_decl: bool) -> Result<Function, String>
{
let args = decl.inputs.iter()
.try_skip_map(|x| x.as_ident_and_type())?;
let ret = decl.output.as_type()?;
Ok(Function {
name: name,
annotations: annotations,
ret: ret,
args: args,
extern_decl: extern_decl,
})
}
pub fn add_deps(&self, library: &Library, out: &mut DependencyList) {
self.ret.add_deps(library, out);
for &(_, ref ty) in &self.args {
ty.add_deps(library, out);
}
}
pub fn add_monomorphs(&self, library: &Library, out: &mut Monomorphs) {
self.ret.add_monomorphs(library, out);
for &(_, ref ty) in &self.args {
ty.add_monomorphs(library, out);
}
}
pub fn add_specializations(&self, library: &Library, out: &mut SpecializationList) {
self.ret.add_specializations(library, out);
for &(_, ref ty) in &self.args {
ty.add_specializations(library, out);
}
}
pub fn rename_args(&mut self, config: &Config) {
let rules = [self.annotations.parse_atom::<RenameRule>("rename-all"),
config.function.rename_args];
if let Some(r) = find_first_some(&rules) {
self.args = self.args.iter()
.map(|x| (r.apply_to_snake_case(&x.0,
IdentifierType::FunctionArg),
x.1.clone()))
.collect()
}
}
pub fn mangle_paths(&mut self, monomorphs: &Monomorphs) {
for &mut (_, ref mut ty) in &mut self.args {
ty.mangle_paths(monomorphs);
}
}
}
impl Source for Function {
fn write<F: Write>(&self, config: &Config, out: &mut SourceWriter<F>) {
fn write_1<W: Write>(func: &Function, config: &Config, out: &mut SourceWriter<W>) {
let prefix = config.function.prefix(&func.annotations);
let postfix = config.function.postfix(&func.annotations);
if let Some(ref prefix) = prefix {
out.write(prefix);
out.write(" ");
}
cdecl::write_func(out, &func, false);
if let Some(ref postfix) = postfix {
out.write(" ");
out.write(postfix);
}
out.write(";");
}
fn write_2<W: Write>(func: &Function, config: &Config, out: &mut SourceWriter<W>) {
let prefix = config.function.prefix(&func.annotations);
let postfix = config.function.postfix(&func.annotations);
if let Some(ref prefix) = prefix {
out.write(prefix);
out.new_line();
}
cdecl::write_func(out, &func, true);
if let Some(ref postfix) = postfix {
out.new_line();
out.write(postfix);
}
out.write(";");
};
let option_1 = out.measure(|out| write_1(self, config, out));
if (config.function.args == Layout::Auto && option_1 <= config.line_length) ||
config.function.args == Layout::Horizontal {
write_1(self, config, out);
} else {
write_2(self, config, out);
}
}
}
pub trait SynFnArgHelpers {
fn as_ident_and_type(&self) -> Result<Option<(String, Type)>, String>;
}
impl SynFnArgHelpers for syn::FnArg {
fn as_ident_and_type(&self) -> Result<Option<(String, Type)>, String> {
match self {
&syn::FnArg::Captured(syn::Pat::Ident(_, ref ident, _), ref ty) => {
if let Some(x) = Type::load(ty)? {
Ok(Some((ident.to_string(), x)))
} else {
Ok(None)
}
}
_ => Err(format!("parameter has unexpected type")),
}
}
}