assetpack-transform-precomp2 0.3.0

use std::io::{BufRead, Write};

use assetpack_core::{
  TRANSFORM_ID_PRECOMP2, TRANSFORM_VERSION_PRECOMP2, TransformDecoder,
  file_transform::{FileHint, FileTransform, FileTransformConfig, TransformStats},
};

use crate::common::{
  TransformGate, precomp2_decode_bytes, precomp2_decode_config, precomp2_encode_bytes, precomp2_encode_config, read_all,
};

#[derive(Debug, Clone)]
pub struct Precomp2Transform {
  gate: TransformGate,
  encode_config: precomp2::EncodeConfig,
  decode_config: precomp2::DecodeConfig,
}

impl Precomp2Transform {
  pub fn new(config: &FileTransformConfig) -> Self {
    Self {
      gate: TransformGate::new(config),
      encode_config: precomp2_encode_config(&config.precomp2),
      decode_config: precomp2_decode_config(&config.precomp2),
    }
  }
}

impl TransformDecoder for Precomp2Transform {
  fn id(&self) -> u16 {
    TRANSFORM_ID_PRECOMP2
  }

  fn version(&self) -> u16 {
    TRANSFORM_VERSION_PRECOMP2
  }

  fn decode(&self, input: &mut dyn BufRead, out: &mut dyn Write) -> assetpack_core::Result<()> {
    let encoded = read_all(input)?;
    let decoded = precomp2_decode_bytes(&encoded, &self.decode_config)?;
    out.write_all(&decoded)?;
    Ok(())
  }
}

impl FileTransform for Precomp2Transform {
  fn quick_check(&self, hint: &FileHint) -> bool {
    self.gate.quick_check(hint)
  }

  fn encode(&self, input: &mut dyn BufRead, out: &mut dyn Write) -> assetpack_core::Result<TransformStats> {
    let input_bytes = read_all(input)?;
    let input_len = input_bytes.len() as u64;
    let encoded = precomp2_encode_bytes(&input_bytes, &self.encode_config)?;
    out.write_all(&encoded)?;
    Ok(TransformStats {
      input_bytes: input_len,
      output_bytes: encoded.len() as u64,
    })
  }
}