use pubky_common::{auth::jws::ClientId, capabilities::Capabilities, crypto::PublicKey};
use url::Url;
use super::{
DeepLinkParseError, GrantApprovalFormat, GrantRelayChannel,
query_params::{
append_grant_approval_format, append_grant_params, append_grant_relay_channel,
append_relay_params, append_signup_details, optional_query, parse_capabilities,
parse_client_id, parse_client_pk, parse_grant_approval_format, parse_grant_relay_channel,
parse_homeserver, parse_relay,
},
typed_deep_link::{DeepLinkIntent, DeepLinkParams, TypedDeepLink},
};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct SignupGrantIntent;
impl DeepLinkIntent for SignupGrantIntent {
const NAME: &'static str = "signup_grant";
}
#[derive(Clone, PartialEq, Eq)]
#[non_exhaustive]
pub struct SignupGrantParams {
pub capabilities: Capabilities,
pub relay: Url,
pub relay_channel: GrantRelayChannel,
pub homeserver: PublicKey,
pub signup_token: Option<String>,
pub client_id: ClientId,
pub client_pk: PublicKey,
pub approval_format: GrantApprovalFormat,
}
impl SignupGrantParams {
#[must_use]
pub fn new(
capabilities: Capabilities,
relay: Url,
secret: [u8; 32],
homeserver: PublicKey,
client_id: ClientId,
client_pk: PublicKey,
) -> Self {
Self {
capabilities,
relay,
relay_channel: GrantRelayChannel::SharedSecret(secret),
homeserver,
signup_token: None,
client_id,
client_pk,
approval_format: GrantApprovalFormat::BareGrant,
}
}
}
impl std::fmt::Debug for SignupGrantParams {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("SignupGrantParams")
.field("capabilities", &self.capabilities)
.field("relay", &self.relay)
.field("relay_channel", &self.relay_channel)
.field("homeserver", &self.homeserver)
.field("signup_token", &"<redacted>")
.field("client_id", &self.client_id)
.field("client_pk", &self.client_pk)
.field("approval_format", &self.approval_format)
.finish()
}
}
impl DeepLinkParams for SignupGrantParams {
fn parse(url: &Url) -> Result<Self, DeepLinkParseError> {
let approval_format = parse_grant_approval_format(url)?;
let capabilities = parse_capabilities(url)?;
approval_format
.validate_capabilities(&capabilities)
.map_err(|error| DeepLinkParseError::InvalidQueryParameter("caps", Box::new(error)))?;
Ok(Self {
capabilities,
relay: parse_relay(url)?,
relay_channel: parse_grant_relay_channel(url, approval_format)?,
homeserver: parse_homeserver(url)?,
signup_token: optional_query(url, "st"),
client_id: parse_client_id(url)?,
client_pk: parse_client_pk(url)?,
approval_format,
})
}
fn append_query_pairs(&self, url: &mut Url) {
append_relay_params(url, &self.capabilities, &self.relay);
append_signup_details(url, &self.homeserver, self.signup_token.as_deref());
append_grant_params(url, &self.client_id, &self.client_pk);
append_grant_approval_format(url, self.approval_format);
append_grant_relay_channel(url, self.relay_channel);
}
}
pub type SignupGrantDeepLink = TypedDeepLink<SignupGrantIntent, SignupGrantParams>;
#[cfg(test)]
mod tests {
use std::str::FromStr;
use pubky_common::crypto::Keypair;
use super::*;
use crate::actors::auth::deep_links::DeepLinkScheme;
const HOMESERVER: &str = "5jsjx1o6fzu6aeeo697r3i5rx15zq41kikcye8wtwdqm4nb4tryo";
#[test]
fn parses_signup_grant_deep_link_without_signup_token() {
let client_pk = Keypair::random().public_key();
let deep_link: SignupGrantDeepLink = format!(
"pubkyauth://signup_grant?caps=/pub/pubky.app/:rw&relay=https://httprelay.pubky.app/inbox/&secret=kqnceEMgrNQM_xi06oQXjA3cJHX_RQmw1BY6JE1bse8&hs={HOMESERVER}&cid=franky.pubky.app&cpk={}",
client_pk.z32()
)
.parse()
.unwrap();
assert_eq!(deep_link.scheme(), DeepLinkScheme::PubkyAuth);
assert_eq!(deep_link.intent(), "signup_grant");
assert_eq!(deep_link.params().signup_token, None);
assert_eq!(deep_link.params().client_pk.z32(), client_pk.z32());
assert_eq!(
deep_link.params().approval_format,
GrantApprovalFormat::BareGrant
);
}
#[test]
fn parses_signup_grant_deep_link_with_signup_token() {
let client_pk = Keypair::random().public_key();
let deep_link: SignupGrantDeepLink = format!(
"pubkyauth://signup_grant?caps=/pub/pubky.app/:rw&relay=https://httprelay.pubky.app/inbox/&secret=kqnceEMgrNQM_xi06oQXjA3cJHX_RQmw1BY6JE1bse8&hs={HOMESERVER}&st=123&cid=franky.pubky.app&cpk={}",
client_pk.z32()
)
.parse()
.unwrap();
assert_eq!(deep_link.params().signup_token, Some("123".into()));
}
#[test]
fn creates_signup_grant_deep_link_from_params() {
let capabilities = Capabilities::builder().read_write("/").unwrap().finish();
let relay = Url::parse("https://httprelay.pubky.app/inbox/").unwrap();
let homeserver = PublicKey::from_str(HOMESERVER).unwrap();
let client_id = ClientId::new("franky.pubky.app").unwrap();
let client_pk = Keypair::random().public_key();
let mut params = SignupGrantParams::new(
capabilities,
relay,
[42; 32],
homeserver,
client_id,
client_pk,
);
assert_eq!(params.approval_format, GrantApprovalFormat::BareGrant);
assert!(params.signup_token.is_none());
params.signup_token = Some("1234567890".into());
let deep_link = SignupGrantDeepLink::new(DeepLinkScheme::PubkyAuth, params);
let parsed_again = SignupGrantDeepLink::parse_url(&deep_link.to_url()).unwrap();
assert_eq!(parsed_again, deep_link);
}
#[test]
fn rejects_missing_cid() {
let pk = Keypair::random().public_key();
let url = format!(
"pubkyauth://signup_grant?caps=/:rw&relay=https://httprelay.pubky.app/inbox/&secret=kqnceEMgrNQM_xi06oQXjA3cJHX_RQmw1BY6JE1bse8&hs={HOMESERVER}&cpk={}",
pk.z32()
);
let err = url.parse::<SignupGrantDeepLink>().unwrap_err();
assert!(matches!(
err,
DeepLinkParseError::MissingQueryParameter("cid")
));
}
#[test]
fn signed_approval_format_round_trips() {
let client_pk = Keypair::random().public_key();
let epk =
base64::Engine::encode(&base64::engine::general_purpose::URL_SAFE_NO_PAD, [7; 32]);
let link: SignupGrantDeepLink = format!(
"pubkyauth://signup_grant?caps=/:rw&relay=http://localhost/inbox&hs={HOMESERVER}&cid=test.app&cpk={}&af=v1&epk={epk}",
client_pk.z32(),
).parse().unwrap();
assert_eq!(
link.params().approval_format,
GrantApprovalFormat::SignedApprovalV1
);
assert_eq!(
SignupGrantDeepLink::parse_url(&link.to_url()).unwrap(),
link
);
}
}