use crate::{encode_section, ComponentSection, ComponentSectionId, Encode};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum CanonicalOption {
UTF8,
UTF16,
CompactUTF16,
Into(u32),
}
impl Encode for CanonicalOption {
fn encode(&self, sink: &mut Vec<u8>) {
match self {
Self::UTF8 => sink.push(0x00),
Self::UTF16 => sink.push(0x01),
Self::CompactUTF16 => sink.push(0x02),
Self::Into(index) => {
sink.push(0x03);
index.encode(sink);
}
}
}
}
#[derive(Clone, Debug, Default)]
pub struct ComponentFunctionSection {
bytes: Vec<u8>,
num_added: u32,
}
impl ComponentFunctionSection {
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 lift<O>(&mut self, type_index: u32, func_index: u32, options: O) -> &mut Self
where
O: IntoIterator<Item = CanonicalOption>,
O::IntoIter: ExactSizeIterator,
{
let options = options.into_iter();
self.bytes.push(0x00);
type_index.encode(&mut self.bytes);
options.len().encode(&mut self.bytes);
for option in options {
option.encode(&mut self.bytes);
}
func_index.encode(&mut self.bytes);
self.num_added += 1;
self
}
pub fn lower<O>(&mut self, func_index: u32, options: O) -> &mut Self
where
O: IntoIterator<Item = CanonicalOption>,
O::IntoIter: ExactSizeIterator,
{
let options = options.into_iter();
self.bytes.push(0x01);
options.len().encode(&mut self.bytes);
for option in options {
option.encode(&mut self.bytes);
}
func_index.encode(&mut self.bytes);
self.num_added += 1;
self
}
}
impl Encode for ComponentFunctionSection {
fn encode(&self, sink: &mut Vec<u8>) {
encode_section(
sink,
ComponentSectionId::Function,
self.num_added,
&self.bytes,
);
}
}
impl ComponentSection for ComponentFunctionSection {}