server/src/state/models.rs

use bytes::{BufMut, Bytes, BytesMut};
use iggy::bytes_serializable::BytesSerializable;
use iggy::error::IggyError;
use iggy::personal_access_tokens::create_personal_access_token::CreatePersonalAccessToken;
use serde::{Deserialize, Serialize};
use std::str::from_utf8;

#[derive(Debug, PartialEq, Serialize, Deserialize)]
pub struct CreatePersonalAccessTokenWithHash {
    pub command: CreatePersonalAccessToken,
    pub hash: String,
}

impl BytesSerializable for CreatePersonalAccessTokenWithHash {
    fn as_bytes(&self) -> Bytes {
        let mut bytes = BytesMut::new();
        let command_bytes = self.command.as_bytes();
        bytes.put_u32_le(command_bytes.len() as u32);
        bytes.put_slice(&command_bytes);
        bytes.put_u32_le(self.hash.len() as u32);
        bytes.put_slice(self.hash.as_bytes());
        bytes.freeze()
    }

    fn from_bytes(bytes: Bytes) -> Result<Self, IggyError>
    where
        Self: Sized,
    {
        let mut position = 0;
        let command_length = u32::from_le_bytes(bytes[position..position + 4].try_into()?);
        position += 4;
        let command_bytes = bytes.slice(position..position + command_length as usize);
        position += command_length as usize;
        let command = CreatePersonalAccessToken::from_bytes(command_bytes)?;
        let hash_length = u32::from_le_bytes(bytes[position..position + 4].try_into()?);
        position += 4;
        let hash = from_utf8(&bytes[position..position + hash_length as usize])?.to_string();
        Ok(Self { command, hash })
    }
}