// SPDX-License-Identifier: GPL-3.0-or-later
use crate::FBError;
use base64::prelude::{BASE64_STANDARD, Engine};
use bincode::{Options, DefaultOptions};

/// A key object that is specific to a message.
pub struct FBKey {
	pub(crate) indices: Vec<Vec<(usize, usize)>>,
}

impl FBKey {

	/// Returns the byte representation of the key.
	pub fn as_bytes(&self) -> Vec<u8> {
		let binc = DefaultOptions::new();
		binc.serialize(&self.indices)
			.expect("Should be fine")
	}

	/// Returns the base64 encoded representation of the key.
	pub fn as_base64(&self) -> String {
		BASE64_STANDARD.encode(&self.as_bytes())
	}

	/// Constructs the key from the provided bytes.
	/// # Errors
	/// [DecodeError](FBError::DecodeError)
	pub fn from_bytes(fbkey: &[u8]) -> Result<FBKey, FBError> {
		let binc = DefaultOptions::new();
		let indices: Vec<_> = binc.deserialize(&fbkey)
			.map_err(|_| FBError::DecodeError)?;
		if indices.len() < 2 {
			return Err(FBError::DecodeError);
		}

		Ok (FBKey {indices})
	}

	/// Constructs the key from the provided base64 encoded form.
	/// # Errors
	/// [DecodeError](FBError::DecodeError)
	pub fn from_base64(key_str: &str) -> Result<FBKey, FBError> {
		let indice_bytes = BASE64_STANDARD.decode(key_str)
			.map_err(|_| FBError::DecodeError)?;

		Self::from_bytes(&indice_bytes)
	}
}