brdb 0.6.4

use std::{
    fmt::Display,
    io::{Read, Write},
    sync::Arc,
};

use indexmap::IndexMap;
use lalrpop_util::lalrpop_mod;
use rmp::{Marker, decode::RmpRead};

mod global_data;
mod intern;
pub mod read;
mod value;
pub mod write;
pub use global_data::*;
pub use intern::*;
pub use value::*;
pub mod as_brdb;

pub(crate) type PlaintextEnumBody = Vec<(String, i32)>;
pub(crate) type PlaintextStructBody = Vec<(String, BrdbStructPropRaw)>;
pub(crate) type PlaintextVariantBody = Vec<String>;

lalrpop_mod!(plaintext);

use crate::{
    errors::BrdbSchemaError,
    schema::{
        as_brdb::AsBrdbValue,
        read::{read_owned_str, read_str_from_len},
    },
};

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum BrdbSchemaStructProperty {
    Type(BrdbInterned),
    Array(BrdbInterned),
    FlatArray(BrdbInterned),
    Map(BrdbInterned, BrdbInterned),
}
pub type BrdbSchemaStruct = IndexMap<BrdbInterned, BrdbSchemaStructProperty>;
pub type BrdbSchemaEnum = IndexMap<BrdbInterned, i32>;

pub type BrdbSchemaMetaEnum = (String, Vec<(String, i32)>);
pub type BrdbSchemaMetaStruct = (String, Vec<(String, BrdbStructPropRaw)>);
pub type BrdbSchemaMeta = (
    Vec<BrdbSchemaMetaEnum>,
    Vec<BrdbSchemaMetaVariant>,
    Vec<BrdbSchemaMetaStruct>,
);
/// A tagged-union/variant: a name and the ordered list of member type names it
/// can hold. The encoded value is a uint tag (the member index) followed by the
/// value of that member type. Newer schemas use these for wire graph variants
/// (e.g. `WireGraphVariant`, `WireGraphPrimMathVariant`, `WireGraphArrayVariant`).
pub type BrdbSchemaMetaVariant = (String, Vec<String>);

impl BrdbSchemaStructProperty {
    pub fn as_string(&self, schema: &BrdbSchema) -> String {
        match self {
            BrdbSchemaStructProperty::Type(t) => {
                schema.intern.lookup(*t).unwrap_or("UnknownType".to_owned())
            }
            BrdbSchemaStructProperty::Array(t) => {
                format!(
                    "{}[]",
                    schema
                        .intern
                        .lookup(*t)
                        .unwrap_or("UnknownArrayType".to_owned())
                )
            }
            BrdbSchemaStructProperty::FlatArray(t) => format!(
                "{}[flat]",
                schema
                    .intern
                    .lookup(*t)
                    .unwrap_or("UnknownFlatArrayType".to_owned())
            ),
            BrdbSchemaStructProperty::Map(k, v) => {
                let key = schema
                    .intern
                    .lookup(*k)
                    .unwrap_or("UnknownMapKeyType".to_owned());
                let value = schema
                    .intern
                    .lookup(*v)
                    .unwrap_or("UnknownMapValueType".to_owned());
                format!("{{{key}: {value}}}")
            }
        }
    }
}

#[derive(Default, Debug, Clone)]
pub struct BrdbSchema {
    pub intern: BrdbIntern,
    pub(crate) global_data: Arc<BrdbSchemaGlobalData>,
    pub enums: IndexMap<BrdbInterned, BrdbSchemaEnum>,
    pub structs: IndexMap<BrdbInterned, BrdbSchemaStruct>,
    /// Tagged-union/variant definitions: name -> ordered member types. Only
    /// populated by newer (3-element) binary schemas; empty otherwise.
    pub variants: IndexMap<BrdbInterned, Vec<BrdbInterned>>,
}

impl Display for BrdbSchema {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        for (name, values) in &self.enums {
            let name = self
                .intern
                .lookup(*name)
                .unwrap_or("UnknownEnum".to_owned());
            writeln!(f, "enum {name} {{")?;
            for (key, value) in values {
                let key = self.intern.lookup(*key).unwrap_or("UnknownKey".to_owned());
                writeln!(f, "    {key} = {value},")?;
            }
            writeln!(f, "}}")?;
        }
        for (name, members) in &self.variants {
            let name = self
                .intern
                .lookup(*name)
                .unwrap_or("UnknownVariant".to_owned());
            writeln!(f, "variant {name} {{")?;
            for member in members {
                let member = self
                    .intern
                    .lookup(*member)
                    .unwrap_or("UnknownMember".to_owned());
                writeln!(f, "    {member},")?;
            }
            writeln!(f, "}}")?;
        }
        for (name, properties) in &self.structs {
            let name = self
                .intern
                .lookup(*name)
                .unwrap_or("UnknownStruct".to_owned());
            writeln!(f, "struct {name} {{")?;
            for (prop_name, prop_type) in properties {
                let prop_name = self
                    .intern
                    .lookup(*prop_name)
                    .unwrap_or("UnknownProperty".to_owned());
                writeln!(f, "    {prop_name}: {},", prop_type.as_string(self))?;
            }
            writeln!(f, "}}")?;
        }
        Ok(())
    }
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum BrdbStructPropRaw {
    Type(String),
    Array(String),
    FlatArray(String),
    Map(String, String),
}

impl BrdbSchema {
    pub fn get_struct(&self, name: &str) -> Option<&BrdbSchemaStruct> {
        self.structs.get(&self.intern.get(name)?)
    }

    pub fn get_struct_interned(&self, id: BrdbInterned) -> Option<&BrdbSchemaStruct> {
        self.structs.get(&id)
    }

    pub fn get_enum(&self, name: &str) -> Option<&BrdbSchemaEnum> {
        self.enums.get(&self.intern.get(name)?)
    }

    pub fn get_enum_interned(&self, id: BrdbInterned) -> Option<&BrdbSchemaEnum> {
        self.enums.get(&id)
    }

    pub fn get_variant(&self, name: &str) -> Option<&Vec<BrdbInterned>> {
        self.variants.get(&self.intern.get(name)?)
    }

    pub fn get_variant_interned(&self, id: BrdbInterned) -> Option<&Vec<BrdbInterned>> {
        self.variants.get(&id)
    }

    pub fn parse_to_meta(input: &str) -> Result<BrdbSchemaMeta, String> {
        plaintext::MetaParser::new()
            .parse(input)
            .map_err(|e| e.to_string())
    }

    /// Parse a schema from a plaintext input string into a `BrdbSchema`
    pub fn new_parsed(input: &str) -> Result<BrdbSchema, BrdbSchemaError> {
        let (enums, variants, structs) =
            Self::parse_to_meta(input).map_err(BrdbSchemaError::ParseError)?;
        let mut schema = BrdbSchema::default();
        schema.add_meta(enums, structs);
        schema.add_variants(variants);
        Ok(schema)
    }

    /// Read a schema from a msgpack .schema file into a human readable format
    pub fn read_to_meta(
        mut buf: impl Read,
    ) -> Result<
        (
            Vec<BrdbSchemaMetaEnum>,
            Vec<BrdbSchemaMetaVariant>,
            Vec<BrdbSchemaMetaStruct>,
        ),
        BrdbSchemaError,
    > {
        // Older saves use a 2-element schema: [enums, structs].
        // Newer saves use a 3-element schema: [enums, variants, structs], where
        // the middle element is a tagged-union/variant table.
        let header = rmp::decode::read_array_len(&mut buf)?;
        if header != 2 && header != 3 {
            return Err(BrdbSchemaError::InvalidHeader(header));
        }
        let has_variants = header == 3;

        // Read enums
        let mut enums = vec![];
        let num_enums = rmp::decode::read_map_len(&mut buf)? as usize;
        for _ in 0..num_enums {
            let enum_name = read_owned_str(&mut buf)?;
            let value_count = rmp::decode::read_map_len(&mut buf)? as usize;
            let mut values = Vec::with_capacity(value_count as usize);
            for _ in 0..value_count {
                let key = read_owned_str(&mut buf)?;
                let value = read::read_int(&mut buf)? as i32;
                values.push((key, value));
            }
            enums.push((enum_name, values));
        }

        // Read the variant table (newer schemas only). Each entry is a variant
        // name mapped to the ordered list of member type names it can hold.
        let mut variants = vec![];
        if has_variants {
            let num_variants = rmp::decode::read_map_len(&mut buf)? as usize;
            for _ in 0..num_variants {
                let variant_name = read_owned_str(&mut buf)?;
                let member_count = rmp::decode::read_array_len(&mut buf)? as usize;
                let mut members = Vec::with_capacity(member_count);
                for _ in 0..member_count {
                    members.push(read_owned_str(&mut buf)?);
                }
                variants.push((variant_name, members));
            }
        }

        // Read structs
        let mut structs = vec![];
        let num_structs = rmp::decode::read_map_len(&mut buf)? as usize;
        for _ in 0..num_structs {
            let struct_name = read_owned_str(&mut buf)?;

            let num_props = rmp::decode::read_map_len(&mut buf)? as usize;
            let mut properties = Vec::with_capacity(num_props);
            for _ in 0..num_props {
                let prop_name = read_owned_str(&mut buf)?;
                let prop_type_marker = rmp::decode::read_marker(&mut buf)
                    .map_err(|e| BrdbSchemaError::RmpMarkerReadError(e.0))?;
                let property = match prop_type_marker {
                    // Basic types
                    Marker::FixStr(size) => {
                        BrdbStructPropRaw::Type(read_str_from_len(&mut buf, size as usize)?)
                    }
                    Marker::Str8 => {
                        let len = buf.read_data_u8()? as usize;
                        BrdbStructPropRaw::Type(read_str_from_len(&mut buf, len)?)
                    }
                    Marker::Str16 => {
                        let len = buf.read_data_u16()? as usize;
                        BrdbStructPropRaw::Type(read_str_from_len(&mut buf, len)?)
                    }

                    Marker::FixArray(len) if len == 0 => {
                        return Err(BrdbSchemaError::InvalidSchema(
                            "0 length FixArray marker not supported".to_string(),
                        ));
                    }
                    // Array type
                    Marker::FixArray(len) if len == 1 => {
                        let array_type = read_owned_str(&mut buf)?;
                        BrdbStructPropRaw::Array(array_type)
                    }
                    // Flat array has a specific format: [type, nil]
                    Marker::FixArray(len) if len == 2 => {
                        let array_type = read_owned_str(&mut buf)?;
                        // Ensure the second element is nil
                        rmp::decode::read_nil(&mut buf)
                            .map_err(|e| BrdbSchemaError::RmpValueReadError(e))?;
                        BrdbStructPropRaw::FlatArray(array_type)
                    }

                    Marker::FixMap(len) if len != 1 => {
                        return Err(BrdbSchemaError::InvalidSchema(
                            "FixMap with length != 1 is not supported".to_string(),
                        ));
                    }
                    Marker::FixMap(len) if len == 1 => {
                        let key_type = read_owned_str(&mut buf)?;
                        let value_type = read_owned_str(&mut buf)?;
                        BrdbStructPropRaw::Map(key_type, value_type)
                    }
                    marker => {
                        return Err(BrdbSchemaError::InvalidSchema(format!(
                            "Unsupported property type marker: {marker:?}",
                        )));
                    }
                };

                properties.push((prop_name, property));
            }
            structs.push((struct_name, properties));
        }

        Ok((enums, variants, structs))
    }

    pub fn from_meta(
        enums: impl IntoIterator<Item = (String, Vec<(String, i32)>)>,
        variants: impl IntoIterator<Item = (String, Vec<String>)>,
        structs: impl IntoIterator<Item = (String, Vec<(String, BrdbStructPropRaw)>)>,
    ) -> Self {
        let mut schema = BrdbSchema::default();
        schema.add_meta(enums, structs);
        schema.add_variants(variants);
        schema
    }

    /// Bulk insert enums and structs from metadata into this schema
    pub fn add_meta(
        &mut self,
        enums: impl IntoIterator<Item = (String, Vec<(String, i32)>)>,
        structs: impl IntoIterator<Item = (String, Vec<(String, BrdbStructPropRaw)>)>,
    ) {
        for (name, values) in enums {
            self.add_enum(name, values);
        }
        for (name, props) in structs {
            self.add_struct(name, props);
        }
    }

    /// Read from a msgpack .schema buffer into a populated `BrdbSchema` struct
    pub fn read(buf: impl Read) -> Result<BrdbSchema, BrdbSchemaError> {
        let (enums, variants, structs) = Self::read_to_meta(buf)?;
        let mut schema = BrdbSchema::default();
        schema.add_meta(enums, structs);
        schema.add_variants(variants);
        Ok(schema)
    }

    /// Add a single enum to the schema
    pub fn add_enum(&mut self, name: String, values: Vec<(String, i32)>) {
        let name = self.intern.get_or_insert(name);
        let values = values
            .into_iter()
            .map(|(k, v)| (self.intern.get_or_insert(k), v))
            .collect::<BrdbSchemaEnum>();
        self.enums.insert(name, values);
    }

    /// Add a single struct to the schema
    pub fn add_struct(&mut self, name: String, props: Vec<(String, BrdbStructPropRaw)>) {
        let name = self.intern.get_or_insert(name);
        let props = props
            .into_iter()
            .map(|(prop, prop_ty)| {
                (
                    self.intern.get_or_insert(prop),
                    match prop_ty {
                        BrdbStructPropRaw::Type(ty) => {
                            BrdbSchemaStructProperty::Type(self.intern.get_or_insert(ty))
                        }
                        BrdbStructPropRaw::Array(ty) => {
                            BrdbSchemaStructProperty::Array(self.intern.get_or_insert(ty))
                        }
                        BrdbStructPropRaw::FlatArray(ty) => {
                            BrdbSchemaStructProperty::FlatArray(self.intern.get_or_insert(ty))
                        }
                        BrdbStructPropRaw::Map(k_ty, v_ty) => BrdbSchemaStructProperty::Map(
                            self.intern.get_or_insert(k_ty),
                            self.intern.get_or_insert(v_ty),
                        ),
                    },
                )
            })
            .collect::<BrdbSchemaStruct>();
        self.structs.insert(name, props);
    }

    /// Add tagged-union/variant definitions to the schema. Each entry maps a
    /// variant name to the ordered list of member type names it can hold.
    pub fn add_variants(&mut self, variants: impl IntoIterator<Item = (String, Vec<String>)>) {
        for (name, members) in variants {
            let name = self.intern.get_or_insert(name);
            let members = members
                .into_iter()
                .map(|m| self.intern.get_or_insert(m))
                .collect::<Vec<_>>();
            self.variants.insert(name, members);
        }
    }

    /// Iterate all struct names in this schema.
    pub fn struct_names(&self) -> Vec<String> {
        self.structs
            .keys()
            .filter_map(|k| self.intern.lookup(*k))
            .collect()
    }

    /// Extract a single struct's metadata (and any variant types it references)
    /// back into `BrdbSchemaMeta` form so it can be inserted into another
    /// schema via `add_meta`/`add_variants`.
    pub fn extract_struct_meta(&self, struct_name: &str) -> Option<BrdbSchemaMeta> {
        let id = self.intern.get(struct_name)?;
        let props = self.structs.get(&id)?;
        let enums = Vec::new();
        let mut variants: Vec<BrdbSchemaMetaVariant> = Vec::new();
        let mut structs = Vec::new();

        // Pull in any variant definitions referenced by this struct's
        // properties so the extracted meta is self-contained.
        for prop in props.values() {
            let referenced = match prop {
                BrdbSchemaStructProperty::Type(t)
                | BrdbSchemaStructProperty::Array(t)
                | BrdbSchemaStructProperty::FlatArray(t) => *t,
                BrdbSchemaStructProperty::Map(_, v) => *v,
            };
            if let Some(members) = self.get_variant_interned(referenced) {
                let name = self.intern.lookup(referenced).unwrap();
                if !variants.iter().any(|(n, _)| n == &name) {
                    let member_names = members
                        .iter()
                        .map(|m| self.intern.lookup(*m).unwrap())
                        .collect();
                    variants.push((name, member_names));
                }
            }
        }

        let raw_props: Vec<(String, BrdbStructPropRaw)> = props
            .iter()
            .map(|(k, v)| {
                let key = self.intern.lookup(*k).unwrap();
                let raw = match v {
                    BrdbSchemaStructProperty::Type(t) => {
                        BrdbStructPropRaw::Type(self.intern.lookup(*t).unwrap())
                    }
                    BrdbSchemaStructProperty::Array(t) => {
                        BrdbStructPropRaw::Array(self.intern.lookup(*t).unwrap())
                    }
                    BrdbSchemaStructProperty::FlatArray(t) => {
                        BrdbStructPropRaw::FlatArray(self.intern.lookup(*t).unwrap())
                    }
                    BrdbSchemaStructProperty::Map(k, v) => BrdbStructPropRaw::Map(
                        self.intern.lookup(*k).unwrap(),
                        self.intern.lookup(*v).unwrap(),
                    ),
                };
                (key, raw)
            })
            .collect();
        structs.push((struct_name.to_owned(), raw_props));
        Some((enums, variants, structs))
    }

    /// Extract the named structs plus every type they transitively reference
    /// (sub-structs, enums, variants, and variant member types) into
    /// self-contained meta. Primitive/unknown type names are skipped.
    ///
    /// Used to build a minimal component schema that embeds only what a world
    /// actually uses, mirroring how the game writes `ComponentsShared.schema`
    /// (only the data structs that appear, not the full catalog).
    pub fn extract_structs_transitive<'a>(
        &self,
        seeds: impl IntoIterator<Item = &'a str>,
    ) -> BrdbSchemaMeta {
        let mut enums: Vec<BrdbSchemaMetaEnum> = Vec::new();
        let mut variants: Vec<BrdbSchemaMetaVariant> = Vec::new();
        let mut structs: Vec<BrdbSchemaMetaStruct> = Vec::new();
        let mut seen: std::collections::HashSet<String> = std::collections::HashSet::new();
        let mut work: Vec<String> = seeds.into_iter().map(str::to_owned).collect();

        while let Some(name) = work.pop() {
            if !seen.insert(name.clone()) {
                continue;
            }
            if let Some(members) = self.get_variant(&name) {
                let member_names: Vec<String> =
                    members.iter().filter_map(|m| self.intern.lookup(*m)).collect();
                work.extend(member_names.iter().cloned());
                variants.push((name, member_names));
            } else if let Some(values) = self.get_enum(&name) {
                let vals = values
                    .iter()
                    .filter_map(|(k, v)| self.intern.lookup(*k).map(|k| (k, *v)))
                    .collect();
                enums.push((name, vals));
            } else if let Some(props) = self.get_struct(&name) {
                let mut raw_props = Vec::with_capacity(props.len());
                for (k, v) in props {
                    let key = self.intern.lookup(*k).unwrap();
                    let raw = match v {
                        BrdbSchemaStructProperty::Type(t) => {
                            let n = self.intern.lookup(*t).unwrap();
                            work.push(n.clone());
                            BrdbStructPropRaw::Type(n)
                        }
                        BrdbSchemaStructProperty::Array(t) => {
                            let n = self.intern.lookup(*t).unwrap();
                            work.push(n.clone());
                            BrdbStructPropRaw::Array(n)
                        }
                        BrdbSchemaStructProperty::FlatArray(t) => {
                            let n = self.intern.lookup(*t).unwrap();
                            work.push(n.clone());
                            BrdbStructPropRaw::FlatArray(n)
                        }
                        BrdbSchemaStructProperty::Map(k2, v2) => {
                            let kn = self.intern.lookup(*k2).unwrap();
                            let vn = self.intern.lookup(*v2).unwrap();
                            work.push(kn.clone());
                            work.push(vn.clone());
                            BrdbStructPropRaw::Map(kn, vn)
                        }
                    };
                    raw_props.push((key, raw));
                }
                structs.push((name, raw_props));
            }
            // else: primitive/unknown type — nothing to extract
        }
        (enums, variants, structs)
    }

    /// Struct keys in dependency-first order: a struct is emitted only after
    /// every struct it references through its fields. The game's schema
    /// deserializer builds serializers top-to-bottom and requires a referenced
    /// struct to be defined "higher up" than the struct using it, so the
    /// written struct order must be topological. Stable where unconstrained
    /// (preserves insertion order); cycles (shouldn't occur) break arbitrarily.
    ///
    /// `O(structs + references)`. Runs once per schema serialization on a small
    /// struct set, so it's not a hot path. Recursion depth is the longest
    /// dependency chain (a few levels in practice).
    pub(crate) fn topo_struct_order(&self) -> Vec<BrdbInterned> {
        let mut visited = std::collections::HashSet::with_capacity(self.structs.len());
        let mut order = Vec::with_capacity(self.structs.len());
        for &root in self.structs.keys() {
            self.visit_struct_deps(root, &mut visited, &mut order);
        }
        order
    }

    /// Depth-first post-order: append `id`'s struct deps before `id` itself.
    /// The `visited` set makes each struct land once and breaks any cycle.
    fn visit_struct_deps(
        &self,
        id: BrdbInterned,
        visited: &mut std::collections::HashSet<BrdbInterned>,
        order: &mut Vec<BrdbInterned>,
    ) {
        if !visited.insert(id) {
            return;
        }
        if let Some(props) = self.structs.get(&id) {
            for prop in props.values() {
                let deps: [Option<BrdbInterned>; 2] = match prop {
                    BrdbSchemaStructProperty::Type(t)
                    | BrdbSchemaStructProperty::Array(t)
                    | BrdbSchemaStructProperty::FlatArray(t) => [Some(*t), None],
                    BrdbSchemaStructProperty::Map(k, v) => [Some(*k), Some(*v)],
                };
                for dep in deps.into_iter().flatten() {
                    if self.structs.contains_key(&dep) {
                        self.visit_struct_deps(dep, visited, order);
                    }
                }
            }
        }
        order.push(id);
    }

    /// Attach global data to the schema
    pub fn with_global_data(mut self, global_data: Arc<BrdbSchemaGlobalData>) -> Self {
        self.global_data = global_data;
        self
    }

    /// Attach global data to the schema
    pub fn set_global_data(&mut self, global_data: Arc<BrdbSchemaGlobalData>) {
        self.global_data = global_data;
    }

    /// Serialize the schema as msgpack
    pub fn write(&self, mut buf: impl Write) -> Result<(), BrdbSchemaError> {
        // Always emit the 3-element form ([enums, variants, structs]). Current
        // game builds expect it, and an empty variants map is valid; the reader
        // still accepts the legacy 2-element form for old saves.
        rmp::encode::write_array_len(&mut buf, 3)?;

        let lookup = |interned: BrdbInterned| {
            interned
                .get(self)
                .ok_or(BrdbSchemaError::StringNotInterned(interned.0))
        };

        rmp::encode::write_map_len(&mut buf, self.enums.len() as u32)?;
        for (enum_name, values) in &self.enums {
            rmp::encode::write_str(&mut buf, lookup(*enum_name)?)?;
            rmp::encode::write_map_len(&mut buf, values.len() as u32)?;
            for (key, value) in values {
                rmp::encode::write_str(&mut buf, lookup(*key)?)?;
                write::write_int(&mut buf, *value as i64)?;
            }
        }

        rmp::encode::write_map_len(&mut buf, self.variants.len() as u32)?;
        for (variant_name, members) in &self.variants {
            rmp::encode::write_str(&mut buf, lookup(*variant_name)?)?;
            rmp::encode::write_array_len(&mut buf, members.len() as u32)?;
            for member in members {
                rmp::encode::write_str(&mut buf, lookup(*member)?)?;
            }
        }

        // Structs must be written in dependency-first order (the game requires
        // a referenced struct to appear before the struct that uses it).
        let struct_order = self.topo_struct_order();
        rmp::encode::write_map_len(&mut buf, struct_order.len() as u32)?;
        for struct_name in struct_order {
            let properties = self
                .structs
                .get(&struct_name)
                .expect("topo order only yields known struct keys");
            rmp::encode::write_str(&mut buf, lookup(struct_name)?)?;
            rmp::encode::write_map_len(&mut buf, properties.len() as u32)?;
            for (prop_name, prop_type) in properties {
                rmp::encode::write_str(&mut buf, lookup(*prop_name)?)?;
                match prop_type {
                    BrdbSchemaStructProperty::Type(t) => {
                        rmp::encode::write_str(&mut buf, lookup(*t)?)?
                    }
                    BrdbSchemaStructProperty::Array(t) => {
                        rmp::encode::write_array_len(&mut buf, 1)?;
                        rmp::encode::write_str(&mut buf, lookup(*t)?)?;
                    }
                    BrdbSchemaStructProperty::FlatArray(t) => {
                        rmp::encode::write_array_len(&mut buf, 2)?;
                        rmp::encode::write_str(&mut buf, lookup(*t)?)?;
                        rmp::encode::write_nil(&mut buf)?;
                    }
                    BrdbSchemaStructProperty::Map(key_type, value_type) => {
                        rmp::encode::write_map_len(&mut buf, 1)?;
                        rmp::encode::write_str(&mut buf, lookup(*key_type)?)?;
                        rmp::encode::write_str(&mut buf, lookup(*value_type)?)?;
                    }
                }
            }
        }

        Ok(())
    }

    /// Serialize the schema to a vector of bytes
    pub fn to_bytes(&self) -> Result<Vec<u8>, BrdbSchemaError> {
        let mut buf = Vec::new();
        self.write(&mut buf)?;
        Ok(buf)
    }

    pub fn write_brdb(
        &self,
        ty: &str,
        value: &impl AsBrdbValue,
    ) -> Result<Vec<u8>, BrdbSchemaError> {
        let mut buf = Vec::new();
        write::write_brdb(self, &mut buf, ty, value)?;
        Ok(buf)
    }
}

pub trait ReadBrdbSchema {
    fn read_brdb_schema(&mut self) -> Result<Arc<BrdbSchema>, BrdbSchemaError>;
    fn read_brdb_schema_with_data(
        &mut self,
        data: Arc<BrdbSchemaGlobalData>,
    ) -> Result<Arc<BrdbSchema>, BrdbSchemaError>;
    fn read_brdb(
        &mut self,
        schema: &Arc<BrdbSchema>,
        ty: &str,
    ) -> Result<BrdbValue, BrdbSchemaError>;
}

impl<R: Read> ReadBrdbSchema for R {
    fn read_brdb_schema(&mut self) -> Result<Arc<BrdbSchema>, BrdbSchemaError> {
        BrdbSchema::read(self).map(Arc::new)
    }
    fn read_brdb_schema_with_data(
        &mut self,
        data: Arc<BrdbSchemaGlobalData>,
    ) -> Result<Arc<BrdbSchema>, BrdbSchemaError> {
        BrdbSchema::read(self).map(|schema| Arc::new(schema.with_global_data(data)))
    }
    fn read_brdb(
        &mut self,
        schema: &Arc<BrdbSchema>,
        ty: &str,
    ) -> Result<BrdbValue, BrdbSchemaError> {
        read::read_type(schema, ty, self)
    }
}

#[cfg(test)]
mod schema_tests {
    #[test]
    fn test_plaintext() {
        let input = "enum Foo {
    A = 0,
    B = 1,
}
struct Bar {
    value: i32,
    arr: str[],
    flat_arr: str[flat],
    map: {str: i32},
}
";
        let (enums, _variants, structs) = super::BrdbSchema::parse_to_meta(input).unwrap();

        // When inserting all the enums and structs into a schema it should
        // produce the same displayed output as the input
        let mut schema = super::BrdbSchema::default();
        schema.add_meta(enums.clone(), structs.clone());
        assert_eq!(schema.to_string(), input,);
    }

    #[test]
    fn test_plaintext_variants() {
        // Display order is enums, then variants, then structs.
        let input = "enum Foo {
    A = 0,
}
variant WireGraphVariant {
    f64,
    i64,
    bool,
    Vector,
}
struct Bar {
    value: WireGraphVariant,
}
";
        let schema = super::BrdbSchema::new_parsed(input).unwrap();

        // The variant table should be populated and resolvable by name.
        let members = schema.get_variant("WireGraphVariant").unwrap();
        assert_eq!(members.len(), 4);

        // Parsing then displaying should round-trip exactly.
        assert_eq!(schema.to_string(), input);
    }
}