wirm 4.0.1

//! Intermediate Representation of a Module's Exports

use crate::ir::id::{ExportsID, FunctionID};
use crate::ir::types::{InjectTag, Tag, TagUtils};
use wasmparser::ExternalKind;

#[derive(Debug, Clone)]
/// Represents an export in a WebAssembly module.
pub struct Export {
    /// The name of the exported item.
    pub name: String,
    /// The kind of the export.
    pub kind: ExternalKind,
    /// The index of the exported item.
    pub index: u32,
    /// Marked for deletion
    pub(crate) deleted: bool,
    pub tag: InjectTag,
}
impl TagUtils for Export {
    fn get_or_create_tag(&mut self) -> &mut Tag {
        self.tag.get_or_insert_default()
    }

    fn get_tag(&self) -> &Option<Tag> {
        &self.tag
    }
}

impl<'a> From<wasmparser::Export<'a>> for Export {
    fn from(export: wasmparser::Export<'a>) -> Self {
        Export {
            name: export.name.to_string(),
            kind: export.kind,
            index: export.index,
            deleted: false,
            tag: None,
        }
    }
}

#[derive(Clone, Debug, Default)]
/// Represents the Exports Section of a WASM Module
pub struct ModuleExports {
    exports: Vec<Export>,
}

impl ModuleExports {
    /// Creates a new `ModuleExports` struct from a Vector of `Exports`
    pub fn new(exports: Vec<Export>) -> Self {
        ModuleExports { exports }
    }

    /// Iterate over all the exports in the module.
    ///     
    /// Note that exports may have been deleted.
    pub fn iter(&self) -> std::slice::Iter<'_, Export> {
        self.exports.iter()
    }

    /// Iterate over the exports in the module.
    ///     
    /// Note that exports may have been deleted.
    pub fn iter_mut(&mut self) -> std::slice::IterMut<'_, Export> {
        self.exports.iter_mut()
    }

    /// Checks if there are no exports
    pub fn is_empty(&self) -> bool {
        self.exports.is_empty()
    }

    /// Add an exported function
    pub fn add_export_func(&mut self, name: String, exp_id: u32) {
        self.add_export_func_with_tag(name, exp_id, Tag::default())
    }
    pub fn add_export_func_with_tag(&mut self, name: String, exp_id: u32, tag: Tag) {
        self.add_export_func_internal(name, exp_id, Some(tag))
    }
    pub(crate) fn add_export_func_internal(&mut self, name: String, exp_id: u32, tag: InjectTag) {
        let export = Export {
            name,
            kind: ExternalKind::Func,
            index: exp_id,
            deleted: false,
            tag,
        };
        self.exports.push(export);
    }

    /// Add an exported memory
    pub fn add_export_mem(&mut self, name: String, exp_id: u32) {
        self.add_export_mem_with_tag(name, exp_id, Tag::default())
    }
    pub fn add_export_mem_with_tag(&mut self, name: String, exp_id: u32, tag: Tag) {
        self.add_export_mem_internal(name, exp_id, Some(tag))
    }
    pub(crate) fn add_export_mem_internal(&mut self, name: String, exp_id: u32, tag: InjectTag) {
        let export = Export {
            name,
            kind: ExternalKind::Memory,
            index: exp_id,
            deleted: false,
            tag,
        };
        self.exports.push(export)
    }

    /// Add an exported table
    pub fn add_export_table(&mut self, name: String, exp_id: u32) {
        self.add_export_table_with_tag(name, exp_id, Tag::default())
    }
    pub fn add_export_table_with_tag(&mut self, name: String, exp_id: u32, tag: Tag) {
        self.add_export_table_internal(name, exp_id, Some(tag))
    }
    pub(crate) fn add_export_table_internal(&mut self, name: String, exp_id: u32, tag: InjectTag) {
        let export = Export {
            name,
            kind: ExternalKind::Table,
            index: exp_id,
            deleted: false,
            tag,
        };
        self.exports.push(export)
    }

    /// Add an exported global
    pub fn add_export_global(&mut self, name: String, exp_id: u32) {
        self.add_export_global_with_tag(name, exp_id, Tag::default())
    }
    pub fn add_export_global_with_tag(&mut self, name: String, exp_id: u32, tag: Tag) {
        self.add_export_global_internal(name, exp_id, Some(tag))
    }
    pub(crate) fn add_export_global_internal(&mut self, name: String, exp_id: u32, tag: InjectTag) {
        let export = Export {
            name,
            kind: ExternalKind::Global,
            index: exp_id,
            deleted: false,
            tag,
        };
        self.exports.push(export)
    }

    /// Get export by name and return if present
    pub fn get_by_name(&self, name: String) -> Option<Export> {
        for exp in self.exports.iter() {
            if exp.name == name {
                return Some(exp.clone());
            }
        }
        None
    }

    /// Get the export ID by name
    pub fn get_export_id_by_name(&self, name: String) -> Option<ExportsID> {
        for (idx, exp) in self.exports.iter().enumerate() {
            if exp.name == name {
                return Some(ExportsID(idx as u32));
            }
        }
        None
    }

    /// Get the export by ID
    pub fn get_by_id(&self, id: ExportsID) -> Option<Export> {
        if *id < self.exports.len() as u32 {
            Some(self.exports[*id as usize].clone())
        } else {
            None
        }
    }

    /// Get the Export ID from its function ID
    pub fn get_func_by_id(&self, id: FunctionID) -> Option<ExportsID> {
        for (export_id, exp) in self.exports.iter().enumerate() {
            if matches!(exp.kind, ExternalKind::Func) && exp.index == *id {
                return Some(ExportsID(export_id as u32));
            }
        }
        None
    }

    /// Get the function ID of an exported function
    pub fn get_func_by_name(&self, name: String) -> Option<FunctionID> {
        for exp in self.exports.iter() {
            if matches!(exp.kind, ExternalKind::Func) && exp.name == name {
                return Some(FunctionID(exp.index));
            }
        }
        None
    }

    /// Delete an export by its exports ID
    pub fn delete(&mut self, id: ExportsID) {
        // Must just mark for deletion as or else will result in indicies getting messed up
        self.exports[*id as usize].deleted = true;
    }
}