lief 1.0.0

Official Rust bindings for LIEF
use super::Command;
use lief_ffi as ffi;

use crate::common::FromFFI;
use crate::declare_iterator;
use std::marker::PhantomData;

/// Structure which represents a library dependency
pub struct Dylib<'a> {
    ptr: cxx::UniquePtr<ffi::MachO_Dylib>,
    _owner: PhantomData<&'a ffi::MachO_Binary>,
}

impl Dylib<'_> {
    /// Library name
    pub fn name(&self) -> String {
        self.ptr.name().to_string()
    }

    pub fn set_name(&mut self, name: &str) {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        self.ptr.pin_mut().set_name(&__cxx_s);
    }

    /// Create a new ID_DYLIB command
    pub fn id_dylib(name: &str, timestamp: u32, current_version: u32, compat_version: u32) -> Self {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        Self::from_ffi(ffi::MachO_Dylib::id_dylib(
            &__cxx_s,
            timestamp,
            current_version,
            compat_version,
        ))
    }

    /// Create a new LOAD_DYLIB command
    pub fn load_dylib(
        name: &str,
        timestamp: u32,
        current_version: u32,
        compat_version: u32,
    ) -> Self {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        Self::from_ffi(ffi::MachO_Dylib::load_dylib(
            &__cxx_s,
            timestamp,
            current_version,
            compat_version,
        ))
    }

    /// Create a new REEXPORT_DYLIB command
    pub fn reexport_dylib(
        name: &str,
        timestamp: u32,
        current_version: u32,
        compat_version: u32,
    ) -> Self {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        Self::from_ffi(ffi::MachO_Dylib::reexport_dylib(
            &__cxx_s,
            timestamp,
            current_version,
            compat_version,
        ))
    }

    /// Create a new WEAK_DYLIB command
    pub fn weak_dylib(
        name: &str,
        timestamp: u32,
        current_version: u32,
        compat_version: u32,
    ) -> Self {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        Self::from_ffi(ffi::MachO_Dylib::weak_dylib(
            &__cxx_s,
            timestamp,
            current_version,
            compat_version,
        ))
    }

    /// Create a new LAZY_LOAD_DYLIB command
    pub fn lazy_load_dylib(
        name: &str,
        timestamp: u32,
        current_version: u32,
        compat_version: u32,
    ) -> Self {
        cxx::let_cxx_string!(__cxx_s = name.to_string());
        Self::from_ffi(ffi::MachO_Dylib::lazy_load_dylib(
            &__cxx_s,
            timestamp,
            current_version,
            compat_version,
        ))
    }

    /// Original string offset of the name
    pub fn name_offset(&self) -> u32 {
        self.ptr.name_offset()
    }

    /// Date and Time when the shared library was built
    pub fn timestamp(&self) -> u32 {
        self.ptr.timestamp()
    }

    /// Current version of the shared library
    pub fn current_version(&self) -> (u64, u64, u64) {
        let vec = Vec::from(self.ptr.current_version().as_slice());
        if vec.len() != 3 {
            return (0, 0, 0);
        }
        (vec[0], vec[1], vec[2])
    }
    /// Compatibility version of the shared library
    pub fn compatibility_version(&self) -> (u64, u64, u64) {
        let vec = Vec::from(self.ptr.compatibility_version().as_slice());
        if vec.len() != 3 {
            return (0, 0, 0);
        }
        (vec[0], vec[1], vec[2])
    }
}

impl std::fmt::Debug for Dylib<'_> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let base = self as &dyn Command;
        f.debug_struct("Dylib")
            .field("base", &base)
            .field("name", &self.name())
            .field("timestamp", &self.timestamp())
            .field("current_version", &self.current_version())
            .field("compatibility_version", &self.compatibility_version())
            .finish()
    }
}

impl FromFFI<ffi::MachO_Dylib> for Dylib<'_> {
    fn from_ffi(cmd: cxx::UniquePtr<ffi::MachO_Dylib>) -> Self {
        Self {
            ptr: cmd,
            _owner: PhantomData,
        }
    }
}

impl Command for Dylib<'_> {
    fn get_base(&self) -> &ffi::MachO_Command {
        self.ptr.as_ref().unwrap().as_ref()
    }
}

declare_iterator!(
    Libraries,
    Dylib<'a>,
    ffi::MachO_Dylib,
    ffi::MachO_Binary,
    ffi::MachO_Binary_it_libraries
);