keyfork/keyfork-derive-util/src/path.rs

86 lines
2.2 KiB
Rust
Raw Normal View History

use crate::index::DerivationIndex;
use serde::{Deserialize, Serialize};
use thiserror::Error;
/// Errors associated with creating a [`DerivationPath`].
#[derive(Error, Debug)]
pub enum Error {
/// A [`DerivationIndex`] was not able to be created.
#[error("Unable to create index: {0}")]
UnableToCreateIndex(#[from] super::index::Error),
/// The path could not be parsed due to a bad prefix. Paths must be in the format:
///
/// m [/ index [']]+
///
/// The prefix for the path must be `m`, and all indices must be integers between 0 and
/// 2^31.
#[error("Unable to parse path due to bad path prefix")]
UnknownPathPrefix,
}
2023-09-01 04:57:05 +00:00
type Result<T, E = Error> = std::result::Result<T, E>;
const PREFIX: &str = "m";
/// A fully qualified path to derive a key.
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
pub struct DerivationPath {
pub(crate) path: Vec<DerivationIndex>,
}
impl DerivationPath {
/// Returns an iterator over the [`DerivationPath`].
pub fn iter(&self) -> impl Iterator<Item = &DerivationIndex> {
self.path.iter()
}
}
impl std::str::FromStr for DerivationPath {
type Err = Error;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let mut iter = s.split('/');
if iter.next() != Some(PREFIX) {
return Err(Error::UnknownPathPrefix);
}
Ok(Self {
path: iter
.map(DerivationIndex::from_str)
.map(|maybe_err| maybe_err.map_err(From::from))
.collect::<Result<Vec<DerivationIndex>>>()?,
})
}
}
impl std::fmt::Display for DerivationPath {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{PREFIX}")?;
for index in self.iter() {
write!(f, "/{index}")?;
}
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::str::FromStr;
#[test]
#[should_panic]
fn requires_master_path() {
DerivationPath::from_str("1234/5678'").unwrap();
}
#[test]
fn equivalency() -> Result<()> {
let paths = ["m/1234'/5678", "m/44'/0'/0'", "m"];
for path in paths {
assert_eq!(&DerivationPath::from_str(path)?.to_string(), path);
}
Ok(())
}
}