use std::borrow::Cow;
use crate::http::HeaderName;
use anyhow::anyhow;
use indexmap::{IndexMap, IndexSet};
#[derive(Debug, serde::Serialize, serde::Deserialize)]
pub struct JwsHeader<'a> {
pub alg: Cow<'a, str>,
pub kid: Cow<'a, str>,
#[serde(default)]
pub tl_version: Cow<'a, str>,
#[serde(default)]
pub tl_headers: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub jku: Option<Cow<'a, str>>,
}
impl<'a> JwsHeader<'a> {
pub(crate) fn new_v2(
kid: &'a str,
headers: &IndexMap<HeaderName<'_>, &[u8]>,
jku: Option<&'a str>,
) -> Self {
let header_keys = headers.keys().fold(String::new(), |mut all, next| {
if !all.is_empty() {
all.push(',');
}
all.push_str(next.0);
all
});
Self {
alg: Cow::Borrowed("ES512"),
kid: Cow::Borrowed(kid),
tl_version: Cow::Borrowed("2"),
tl_headers: header_keys,
jku: jku.map(Cow::Borrowed),
}
}
pub(crate) fn filter_headers(
&'a self,
headers: &IndexMap<HeaderName<'_>, &'a [u8]>,
) -> anyhow::Result<IndexMap<HeaderName<'a>, &'a [u8]>> {
let required_headers: IndexSet<_> = self
.tl_headers
.split(',')
.filter(|h| !h.is_empty())
.map(HeaderName)
.collect();
let ordered_headers: IndexMap<_, _> = required_headers
.iter()
.map(|h| {
let hval = headers
.get(h)
.ok_or_else(|| anyhow!("Missing tl_header `{}` declared in signature", h))?;
Ok((*h, *hval))
})
.collect::<anyhow::Result<_>>()?;
Ok(ordered_headers)
}
}