use std::io::Write;
use syn;
use bindgen::config::{Config, Language};
use bindgen::dependencies::Dependencies;
use bindgen::ir::{AnnotationSet, Cfg, CfgWrite, Documentation, GenericParams, Item, ItemContainer,
Repr, Type};
use bindgen::library::Library;
use bindgen::mangle;
use bindgen::monomorph::Monomorphs;
use bindgen::rename::{IdentifierType, RenameRule};
use bindgen::utilities::{find_first_some, IterHelpers};
use bindgen::writer::{ListType, Source, SourceWriter};
#[derive(Debug, Clone)]
pub struct Struct {
pub name: String,
pub generic_params: GenericParams,
pub fields: Vec<(String, Type, Documentation)>,
pub is_tagged: bool,
pub tuple_struct: bool,
pub cfg: Option<Cfg>,
pub annotations: AnnotationSet,
pub documentation: Documentation,
}
impl Struct {
pub fn load(
name: String,
decl: &syn::VariantData,
generics: &syn::Generics,
attrs: &Vec<syn::Attribute>,
mod_cfg: &Option<Cfg>,
) -> Result<Struct, String> {
if Repr::load(attrs) != Repr::C {
return Err("Struct is not marked #[repr(C)].".to_owned());
}
let (fields, tuple_struct) = match decl {
&syn::VariantData::Struct(ref fields) => {
let out = fields.iter().try_skip_map(|x| x.as_ident_and_type())?;
(out, false)
}
&syn::VariantData::Tuple(ref fields) => {
let mut out = Vec::new();
let mut current = 0;
for field in fields {
if let Some(x) = Type::load(&field.ty)? {
out.push((format!("{}", current), x, Documentation::load(&field.attrs)));
current += 1;
}
}
(out, true)
}
&syn::VariantData::Unit => (vec![], false),
};
Ok(Struct {
name: name,
generic_params: GenericParams::new(generics),
fields: fields,
is_tagged: false,
tuple_struct: tuple_struct,
cfg: Cfg::append(mod_cfg, Cfg::load(attrs)),
annotations: AnnotationSet::load(attrs)?,
documentation: Documentation::load(attrs),
})
}
pub fn simplify_option_to_ptr(&mut self) {
for &mut (_, ref mut ty, _) in &mut self.fields {
ty.simplify_option_to_ptr();
}
}
pub fn is_generic(&self) -> bool {
self.generic_params.len() > 0
}
pub fn add_monomorphs(&self, library: &Library, out: &mut Monomorphs) {
if self.is_generic() {
return;
}
for &(_, ref ty, _) in &self.fields {
ty.add_monomorphs(library, out);
}
}
pub fn mangle_paths(&mut self, monomorphs: &Monomorphs) {
for &mut (_, ref mut ty, _) in &mut self.fields {
ty.mangle_paths(monomorphs);
}
}
}
impl Item for Struct {
fn name(&self) -> &str {
&self.name
}
fn cfg(&self) -> &Option<Cfg> {
&self.cfg
}
fn annotations(&self) -> &AnnotationSet {
&self.annotations
}
fn annotations_mut(&mut self) -> &mut AnnotationSet {
&mut self.annotations
}
fn container(&self) -> ItemContainer {
ItemContainer::Struct(self.clone())
}
fn rename_for_config(&mut self, config: &Config) {
config.export.rename(&mut self.name);
for &mut (_, ref mut ty, _) in &mut self.fields {
ty.rename_for_config(config);
}
let field_rules = [
self.annotations.parse_atom::<RenameRule>("rename-all"),
config.structure.rename_fields,
];
let mut names = self.fields.iter_mut().map(|field| &mut field.0);
if let Some(o) = self.annotations.list("field-names") {
for (dest, src) in names.zip(o) {
*dest = src;
}
} else if let Some(r) = find_first_some(&field_rules) {
for name in names {
*name = r.apply_to_snake_case(name, IdentifierType::StructMember);
}
} else if self.tuple_struct {
if self.is_tagged {
names.next();
}
for name in names {
name.insert(0, '_');
}
}
}
fn add_dependencies(&self, library: &Library, out: &mut Dependencies) {
for &(_, ref ty, _) in &self.fields {
ty.add_dependencies_ignoring_generics(&self.generic_params, library, out);
}
}
fn instantiate_monomorph(
&self,
generic_values: &Vec<Type>,
library: &Library,
out: &mut Monomorphs,
) {
assert!(self.generic_params.len() > 0 && self.generic_params.len() == generic_values.len());
let mappings = self.generic_params
.iter()
.zip(generic_values.iter())
.collect::<Vec<_>>();
let monomorph = Struct {
name: mangle::mangle_path(&self.name, generic_values),
generic_params: GenericParams::default(),
fields: self.fields
.iter()
.map(|x| (x.0.clone(), x.1.specialize(&mappings), x.2.clone()))
.collect(),
is_tagged: self.is_tagged,
tuple_struct: self.tuple_struct,
cfg: self.cfg.clone(),
annotations: self.annotations.clone(),
documentation: self.documentation.clone(),
};
monomorph.add_monomorphs(library, out);
out.insert_struct(self, monomorph, generic_values.clone());
}
}
impl Source for Struct {
fn write<F: Write>(&self, config: &Config, out: &mut SourceWriter<F>) {
self.cfg.write_before(config, out);
self.documentation.write(config, out);
self.generic_params.write(config, out);
if config.language == Language::C {
out.write("typedef struct");
} else {
write!(out, "struct {}", self.name);
}
out.open_brace();
if config.documentation {
out.write_vertical_source_list(&self.fields, ListType::Cap(";"));
} else {
out.write_vertical_source_list(
&self.fields
.iter()
.map(|&(ref name, ref ty, _)| (name.clone(), ty.clone()))
.collect(),
ListType::Cap(";"),
);
}
if config.language == Language::Cxx {
let mut wrote_start_newline = false;
if config.structure.derive_constructor(&self.annotations)
&& !self.fields.is_empty()
{
if !wrote_start_newline {
wrote_start_newline = true;
out.new_line();
}
out.new_line();
let arg_renamer = |name: &str| {
config.function.rename_args
.as_ref()
.unwrap_or(&RenameRule::GeckoCase)
.apply_to_snake_case(name, IdentifierType::FunctionArg)
};
write!(out, "{}(", self.name);
out.write_vertical_source_list(
&self.fields
.iter()
.map(|&(ref name, ref ty, _)| {
(format!("const& {}", arg_renamer(name)), ty.clone())
})
.collect(),
ListType::Join(","),
);
write!(out, ")");
out.new_line();
write!(out, " : ");
out.write_vertical_source_list(
&self.fields
.iter()
.map(|x| format!("{}({})", x.0, arg_renamer(&x.0)))
.collect(),
ListType::Join(","),
);
out.new_line();
write!(out, "{{}}");
out.new_line();
}
let other = if let Some(r) = config.function.rename_args {
r.apply_to_snake_case("other", IdentifierType::FunctionArg)
} else {
String::from("other")
};
let mut emit_op = |op, conjuc| {
if !wrote_start_newline {
wrote_start_newline = true;
out.new_line();
}
out.new_line();
write!(
out,
"bool operator{}(const {}& {}) const",
op, self.name, other
);
out.open_brace();
out.write("return ");
out.write_vertical_source_list(
&self.fields
.iter()
.map(|x| format!("{} {} {}.{}", x.0, op, other, x.0))
.collect(),
ListType::Join(&format!(" {}", conjuc)),
);
out.write(";");
out.close_brace(false);
};
if config.structure.derive_eq(&self.annotations) && !self.fields.is_empty()
&& self.fields.iter().all(|x| x.1.can_cmp_eq())
{
emit_op("==", "&&");
}
if config.structure.derive_neq(&self.annotations) && !self.fields.is_empty()
&& self.fields.iter().all(|x| x.1.can_cmp_eq())
{
emit_op("!=", "||");
}
if config.structure.derive_lt(&self.annotations) && self.fields.len() == 1
&& self.fields[0].1.can_cmp_order()
{
emit_op("<", "&&");
}
if config.structure.derive_lte(&self.annotations) && self.fields.len() == 1
&& self.fields[0].1.can_cmp_order()
{
emit_op("<=", "&&");
}
if config.structure.derive_gt(&self.annotations) && self.fields.len() == 1
&& self.fields[0].1.can_cmp_order()
{
emit_op(">", "&&");
}
if config.structure.derive_gte(&self.annotations) && self.fields.len() == 1
&& self.fields[0].1.can_cmp_order()
{
emit_op(">=", "&&");
}
}
if config.language == Language::C {
out.close_brace(false);
write!(out, " {};", self.name);
} else {
out.close_brace(true);
}
self.cfg.write_after(config, out);
}
}
pub trait SynFieldHelpers {
fn as_ident_and_type(&self) -> Result<Option<(String, Type, Documentation)>, String>;
}
impl SynFieldHelpers for syn::Field {
fn as_ident_and_type(&self) -> Result<Option<(String, Type, Documentation)>, String> {
let ident = self.ident
.as_ref()
.ok_or(format!("field is missing identifier"))?
.clone();
let converted_ty = Type::load(&self.ty)?;
if let Some(x) = converted_ty {
Ok(Some((
ident.to_string(),
x,
Documentation::load(&self.attrs),
)))
} else {
Ok(None)
}
}
}