use std::collections::HashMap;
use crate::{
conversion::{
api::{Api, ApiName, StructDetails, TypeKind},
apivec::ApiVec,
convert_error::ConvertErrorWithContext,
error_reporter::convert_apis,
},
types::QualifiedName,
};
use super::fun::{
FnAnalysis, FnKind, FnPhase, FnPrePhase2, PodAndConstructorAnalysis, PodAndDepAnalysis,
TraitMethodKind,
};
pub(crate) fn decorate_types_with_constructor_deps(apis: ApiVec<FnPrePhase2>) -> ApiVec<FnPhase> {
let mut constructors_and_allocators_by_type = find_important_constructors(&apis);
let mut results = ApiVec::new();
convert_apis(
apis,
&mut results,
Api::fun_unchanged,
|name, details, pod| {
decorate_struct(name, details, pod, &mut constructors_and_allocators_by_type)
},
Api::enum_unchanged,
Api::typedef_unchanged,
);
results
}
fn decorate_struct(
name: ApiName,
details: Box<StructDetails>,
fn_struct: PodAndConstructorAnalysis,
constructors_and_allocators_by_type: &mut HashMap<QualifiedName, Vec<QualifiedName>>,
) -> Result<Box<dyn Iterator<Item = Api<FnPhase>>>, ConvertErrorWithContext> {
let pod = fn_struct.pod;
let is_abstract = matches!(pod.kind, TypeKind::Abstract);
let constructor_and_allocator_deps = if is_abstract || pod.is_generic {
Vec::new()
} else {
constructors_and_allocators_by_type
.remove(&name.name)
.unwrap_or_default()
};
Ok(Box::new(std::iter::once(Api::Struct {
name,
details,
analysis: PodAndDepAnalysis {
pod,
constructor_and_allocator_deps,
constructors: fn_struct.constructors,
},
})))
}
fn find_important_constructors(
apis: &ApiVec<FnPrePhase2>,
) -> HashMap<QualifiedName, Vec<QualifiedName>> {
let mut results: HashMap<QualifiedName, Vec<QualifiedName>> = HashMap::new();
for api in apis.iter() {
if let Api::Function {
name,
analysis:
FnAnalysis {
kind:
FnKind::TraitMethod {
kind:
TraitMethodKind::Alloc
| TraitMethodKind::Dealloc
| TraitMethodKind::CopyConstructor
| TraitMethodKind::MoveConstructor,
impl_for,
..
},
ignore_reason: Ok(_),
..
},
..
} = api
{
results
.entry(impl_for.clone())
.or_default()
.push(name.name.clone())
}
}
results
}