use crate::profile::typedef::{self, FitBaseType};
use crate::proto::*;
use alloc::borrow::ToOwned;
use alloc::string::String;
use alloc::vec::Vec;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize, Serializer, ser::SerializeStruct};
#[cfg_attr(feature = "serde", derive(Deserialize), serde(from = "De"))]
#[derive(Debug, Clone)]
pub struct FileCapabilities {
pub message_index: typedef::MessageIndex,
pub r#type: typedef::File,
pub flags: typedef::FileFlags,
pub directory: String,
pub max_count: u16,
pub max_size: u32,
pub unknown_fields: Vec<Field>,
pub developer_fields: Vec<DeveloperField>,
}
impl FileCapabilities {
pub const MESSAGE_INDEX: u8 = 254;
pub const TYPE: u8 = 0;
pub const FLAGS: u8 = 1;
pub const DIRECTORY: u8 = 2;
pub const MAX_COUNT: u8 = 3;
pub const MAX_SIZE: u8 = 4;
pub const fn new() -> Self {
Self {
message_index: typedef::MessageIndex(u16::MAX),
r#type: typedef::File(u8::MAX),
flags: typedef::FileFlags(u8::MIN),
directory: String::new(),
max_count: u16::MAX,
max_size: u32::MAX,
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
fn count_valid_fields(&self) -> usize {
(self.message_index.0 != u16::MAX) as usize
+ (self.r#type.0 != u8::MAX) as usize
+ (self.flags.0 != u8::MIN) as usize
+ (!self.directory.is_empty()) as usize
+ (self.max_count != u16::MAX) as usize
+ (self.max_size != u32::MAX) as usize
}
}
impl Default for FileCapabilities {
fn default() -> Self {
Self::new()
}
}
impl From<&Message> for FileCapabilities {
fn from(mesg: &Message) -> Self {
const KNOWN_NUMS: [u64; 4] = [31, 0, 0, 4611686018427387904];
let mut n = 0u64;
for field in &mesg.fields {
n += (KNOWN_NUMS[field.num as usize >> 6] >> (field.num & 63)) & 1 ^ 1
}
let mut v = Self::new();
v.unknown_fields = Vec::<Field>::with_capacity(n as usize);
v.developer_fields = mesg.developer_fields.clone();
for field in &mesg.fields {
match field.num {
254 => v.message_index = typedef::MessageIndex(field.value.as_u16()),
0 => v.r#type = typedef::File(field.value.as_u8()),
1 => v.flags = typedef::FileFlags(field.value.as_u8z()),
2 => v.directory = field.value.as_str().to_owned(),
3 => v.max_count = field.value.as_u16(),
4 => v.max_size = field.value.as_u32(),
_ => v.unknown_fields.push(field.clone()),
};
}
v
}
}
impl From<FileCapabilities> for Message {
fn from(m: FileCapabilities) -> Self {
let mut fields =
Vec::<Field>::with_capacity(m.count_valid_fields() + m.unknown_fields.len());
if m.message_index.0 != u16::MAX {
fields.push(Field {
num: 254,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.message_index.0),
is_expanded: false,
});
};
if m.r#type.0 != u8::MAX {
fields.push(Field {
num: 0,
base_type: FitBaseType::ENUM,
value: Value::Uint8(m.r#type.0),
is_expanded: false,
});
};
if m.flags.0 != u8::MIN {
fields.push(Field {
num: 1,
base_type: FitBaseType::UINT8Z,
value: Value::Uint8(m.flags.0),
is_expanded: false,
});
};
if !m.directory.is_empty() {
fields.push(Field {
num: 2,
base_type: FitBaseType::STRING,
value: Value::String(m.directory),
is_expanded: false,
});
};
if m.max_count != u16::MAX {
fields.push(Field {
num: 3,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.max_count),
is_expanded: false,
});
};
if m.max_size != u32::MAX {
fields.push(Field {
num: 4,
base_type: FitBaseType::UINT32,
value: Value::Uint32(m.max_size),
is_expanded: false,
});
};
fields.extend_from_slice(&m.unknown_fields);
Self {
header: 0,
num: typedef::MesgNum::FILE_CAPABILITIES,
fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Serialize for FileCapabilities {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let n = self.count_valid_fields() + 2;
let mut state = serializer.serialize_struct("FileCapabilities", n)?;
if self.message_index.0 != u16::MAX {
state.serialize_field("message_index", &self.message_index)?;
}
if self.r#type.0 != u8::MAX {
state.serialize_field("type", &self.r#type)?;
}
if self.flags.0 != u8::MIN {
state.serialize_field("flags", &self.flags)?;
}
if !self.directory.is_empty() {
state.serialize_field("directory", &self.directory)?;
}
if self.max_count != u16::MAX {
state.serialize_field("max_count", &self.max_count)?;
}
if self.max_size != u32::MAX {
state.serialize_field("max_size", &self.max_size)?;
}
if !self.unknown_fields.is_empty() {
state.serialize_field("unknown_fields", &self.unknown_fields)?;
}
if !self.developer_fields.is_empty() {
state.serialize_field("developer_fields", &self.developer_fields)?;
}
state.end()
}
}
#[cfg(feature = "serde")]
#[cfg_attr(feature = "serde", derive(Deserialize), serde(default))]
struct De {
message_index: typedef::MessageIndex,
r#type: typedef::File,
flags: typedef::FileFlags,
directory: String,
max_count: u16,
max_size: u32,
unknown_fields: Vec<Field>,
developer_fields: Vec<DeveloperField>,
}
#[cfg(feature = "serde")]
impl From<De> for FileCapabilities {
fn from(m: De) -> Self {
Self {
message_index: m.message_index,
r#type: m.r#type,
flags: m.flags,
directory: m.directory,
max_count: m.max_count,
max_size: m.max_size,
unknown_fields: m.unknown_fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Default for De {
fn default() -> Self {
Self {
message_index: typedef::MessageIndex(u16::MAX),
r#type: typedef::File(u8::MAX),
flags: typedef::FileFlags(u8::MIN),
directory: String::new(),
max_count: u16::MAX,
max_size: u32::MAX,
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
}