58 lines
1.4 KiB
Rust
58 lines
1.4 KiB
Rust
use crate::error::MangoError;
|
|
use anchor_lang::prelude::*;
|
|
|
|
#[macro_export]
|
|
macro_rules! zip {
|
|
($x: expr) => ($x);
|
|
($x: expr, $($y: expr), +) => (
|
|
$x.zip(
|
|
zip!($($y), +))
|
|
)
|
|
}
|
|
#[allow(unused_imports)]
|
|
pub(crate) use zip;
|
|
|
|
pub fn fill_from_str<const N: usize>(name: &str) -> Result<[u8; N]> {
|
|
let name_bytes = name.as_bytes();
|
|
require!(name_bytes.len() <= N, MangoError::SomeError);
|
|
let mut name_ = [0u8; N];
|
|
name_[..name_bytes.len()].copy_from_slice(name_bytes);
|
|
Ok(name_)
|
|
}
|
|
|
|
pub fn format_zero_terminated_utf8_bytes(
|
|
name: &[u8],
|
|
fmt: &mut std::fmt::Formatter,
|
|
) -> std::result::Result<(), std::fmt::Error> {
|
|
fmt.write_str(
|
|
std::str::from_utf8(name)
|
|
.unwrap()
|
|
.trim_matches(char::from(0)),
|
|
)
|
|
}
|
|
|
|
// Returns (now_ts, now_slot)
|
|
pub fn clock_now() -> (u64, u64) {
|
|
let clock = Clock::get().unwrap();
|
|
(clock.unix_timestamp.try_into().unwrap(), clock.slot)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_fill_from_str() {
|
|
assert_eq!(fill_from_str::<4>(""), Ok([0, 0, 0, 0]));
|
|
assert_eq!(
|
|
fill_from_str::<4>("abc"),
|
|
Ok(['a' as u8, 'b' as u8, 'c' as u8, 0])
|
|
);
|
|
assert_eq!(
|
|
fill_from_str::<4>("abcd"),
|
|
Ok(['a' as u8, 'b' as u8, 'c' as u8, 'd' as u8])
|
|
);
|
|
assert!(fill_from_str::<4>("abcde").is_err());
|
|
}
|
|
}
|