use crate::{encode_section, Encode, Section, SectionId};
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum Export {
Function(u32),
Table(u32),
Memory(u32),
Global(u32),
Tag(u32),
}
impl Encode for Export {
fn encode(&self, sink: &mut Vec<u8>) {
let (kind, index) = match self {
Self::Function(i) => (0x00, *i),
Self::Table(i) => (0x01, *i),
Self::Memory(i) => (0x02, *i),
Self::Global(i) => (0x03, *i),
Self::Tag(i) => (0x04, *i),
};
sink.push(kind);
index.encode(sink);
}
}
#[derive(Clone, Debug, Default)]
pub struct ExportSection {
bytes: Vec<u8>,
num_added: u32,
}
impl ExportSection {
pub fn new() -> Self {
Self::default()
}
pub fn len(&self) -> u32 {
self.num_added
}
pub fn is_empty(&self) -> bool {
self.num_added == 0
}
pub fn export(&mut self, name: &str, export: Export) -> &mut Self {
name.encode(&mut self.bytes);
export.encode(&mut self.bytes);
self.num_added += 1;
self
}
}
impl Encode for ExportSection {
fn encode(&self, sink: &mut Vec<u8>) {
encode_section(sink, SectionId::Export, self.num_added, &self.bytes);
}
}
impl Section for ExportSection {}