use lief_ffi as ffi;
use crate::common::FromFFI;
use crate::pdb::types::PdbType;
use crate::pdb::types::attribute::Attributes;
use crate::pdb::types::method::Methods;
use std::marker::PhantomData;
pub trait ClassLike {
#[doc(hidden)]
fn get_classlike(&self) -> &ffi::PDB_types_ClassLike;
fn unique_name(&self) -> String {
self.get_classlike().unique_name().to_string()
}
fn attributes(&self) -> Attributes<'_> {
Attributes::new(self.get_classlike().attributes())
}
fn methods(&self) -> Methods<'_> {
Methods::new(self.get_classlike().methods())
}
}
pub struct Class<'a> {
ptr: cxx::UniquePtr<ffi::PDB_types_Class>,
_owner: PhantomData<&'a ()>,
}
impl FromFFI<ffi::PDB_types_Class> for Class<'_> {
fn from_ffi(cmd: cxx::UniquePtr<ffi::PDB_types_Class>) -> Self {
Self {
ptr: cmd,
_owner: PhantomData,
}
}
}
impl ClassLike for Class<'_> {
fn get_classlike(&self) -> &ffi::PDB_types_ClassLike {
self.ptr.as_ref().unwrap().as_ref()
}
}
impl PdbType for Class<'_> {
fn get_base(&self) -> &ffi::PDB_Type {
self.ptr.as_ref().unwrap().as_ref().as_ref()
}
}
pub struct Structure<'a> {
ptr: cxx::UniquePtr<ffi::PDB_types_Structure>,
_owner: PhantomData<&'a ()>,
}
impl FromFFI<ffi::PDB_types_Structure> for Structure<'_> {
fn from_ffi(cmd: cxx::UniquePtr<ffi::PDB_types_Structure>) -> Self {
Self {
ptr: cmd,
_owner: PhantomData,
}
}
}
impl ClassLike for Structure<'_> {
fn get_classlike(&self) -> &ffi::PDB_types_ClassLike {
self.ptr.as_ref().unwrap().as_ref()
}
}
impl PdbType for Structure<'_> {
fn get_base(&self) -> &ffi::PDB_Type {
self.ptr.as_ref().unwrap().as_ref().as_ref()
}
}
pub struct Interface<'a> {
ptr: cxx::UniquePtr<ffi::PDB_types_Interface>,
_owner: PhantomData<&'a ()>,
}
impl FromFFI<ffi::PDB_types_Interface> for Interface<'_> {
fn from_ffi(cmd: cxx::UniquePtr<ffi::PDB_types_Interface>) -> Self {
Self {
ptr: cmd,
_owner: PhantomData,
}
}
}
impl ClassLike for Interface<'_> {
fn get_classlike(&self) -> &ffi::PDB_types_ClassLike {
self.ptr.as_ref().unwrap().as_ref()
}
}
impl PdbType for Interface<'_> {
fn get_base(&self) -> &ffi::PDB_Type {
self.ptr.as_ref().unwrap().as_ref().as_ref()
}
}