libgm 0.5.1

A tool for modding, unpacking and decompiling GameMaker games
Documentation
use macros::num_enum;

use crate::prelude::*;
use crate::util::init::num_enum_from;
use crate::wad::deserialize::reader::DataReader;
use crate::wad::elements::GMElement;
use crate::wad::elements::extension::Kind;
use crate::wad::serialize::builder::DataBuilder;

#[derive(Debug, Clone, PartialEq)]
pub struct Function {
    pub name: String,
    pub id: u32,
    pub kind: Kind,
    pub return_type: ReturnType,
    pub ext_name: String,
    pub arguments: Vec<Argument>,
}

impl GMElement for Function {
    fn deserialize(reader: &mut DataReader) -> Result<Self> {
        let name: String = reader.read_gm_string()?;
        let id = reader.read_u32()?;
        let kind: Kind = num_enum_from(reader.read_i32()?)?; // Assumption; UTMT uses u32
        let return_type: ReturnType = num_enum_from(reader.read_i32()?)?;
        let ext_name: String = reader.read_gm_string()?;
        let arguments: Vec<Argument> = reader.read_simple_list()?;
        Ok(Self {
            name,
            id,
            kind,
            return_type,
            ext_name,
            arguments,
        })
    }

    fn serialize(&self, builder: &mut DataBuilder) -> Result<()> {
        builder.write_gm_string(&self.name);
        builder.write_u32(self.id);
        builder.write_i32(self.kind.into());
        builder.write_i32(self.return_type.into());
        builder.write_gm_string(&self.ext_name);
        builder.write_simple_list(&self.arguments)?;
        Ok(())
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct Argument {
    pub return_type: ReturnType,
}

impl GMElement for Argument {
    fn deserialize(reader: &mut DataReader) -> Result<Self> {
        let return_type: ReturnType = num_enum_from(reader.read_i32()?)?;
        Ok(Self { return_type })
    }

    fn serialize(&self, builder: &mut DataBuilder) -> Result<()> {
        builder.write_i32(self.return_type.into());
        Ok(())
    }
}

#[num_enum(i32)]
pub enum ReturnType {
    String = 1,
    Double = 2,
}