polyjose 0.1.1

//! Issuing a signed JWT (RFC 7519 §7.1; ADR 0027 §8).

use zeroize::Zeroizing;

use crate::error::{Error, Result};
use crate::header::Header;
use crate::json::{Object, Value};
use crate::jwa::SigningAlg;
use crate::jwk::Jwk;
use crate::jws::{ProtectedHeader, SignOptions, sign_compact};
use crate::util::b64url_encode;

use super::Claims;
use super::typ::{AT_JWT, JWT};

/// Header parameters a JWT issuer must not add through
/// [`IssueOptions::header_param`]:
///
/// - `b64`: "JSON Web Tokens MUST NOT use "b64" with a "false" value"
///   (RFC 7797 §7), and `true` is the default (RFC 7797 §3); ADR 0027 §6.
/// - `crit`: the only extension polyjose understands is `b64`, which a JWT
///   never uses, so no JWT needs `crit` (RFC 7515 §4.1.11; ADR 0027 §6).
/// - `cty`: belongs to nesting, where the JWE layer sets `"JWT"`
///   (RFC 7519 §5.2, §7.1 step 5); on a JWS, `cty: JWT` would be JWS-in-JWS
///   nesting, which is not supported (ADR 0027 §5).
/// - `zip`, `enc`: JWE parameters (RFC 7516 §4.1.2, §4.1.3), meaningless
///   in a JWS (ADR 0027 §8).
const FORBIDDEN_PARAMS: [&str; 5] = ["b64", "crit", "cty", "zip", "enc"];

/// Header parameters [`sign`] sets itself: `alg` from
/// [`IssueOptions::new`] (RFC 7515 §4.1.1), `typ` from the typ setters
/// (RFC 7519 §5.1) and `kid` from the key (RFC 7515 §4.1.4, ADR 0027 §8).
/// A second source would make the header ambiguous, and its names must be
/// unique (RFC 7515 §4).
const OWNED_PARAMS: [&str; 3] = ["alg", "typ", "kid"];

/// JWK members that hold secret or private key material: `k`, the
/// symmetric key (RFC 7518 §6.4.1); `d`, the EC and OKP private key
/// (RFC 7518 §6.2.2.1, RFC 8037 §2) and the RSA private exponent; and `p`,
/// `q`, `dp`, `dq`, `qi`, `oth`, the other RSA private members (RFC 7518
/// §6.3.2). [`check_public_jwk`] refuses each of them whatever the `kty`,
/// so one that the key type does not define is not let through as an
/// unknown member (RFC 7517 §4).
const SECRET_JWK_MEMBERS: [&str; 8] = ["k", "d", "p", "q", "dp", "dq", "qi", "oth"];

/// Checks the value of a `jwk` header parameter: "the public key that
/// corresponds to the key used to digitally sign the JWS" (RFC 7515
/// §4.1.3), so it must be a valid JWK (RFC 7517 §4) with no secret or
/// private member.
///
/// A JWT header is readable by anyone who sees the token, so a secret or
/// private member there would publish the key (CLAUDE.md secrets rule,
/// ADR 0007).
///
/// # Errors
///
/// [`Error::InvalidHeader`]`("jwk")`, with no part of the value, when `v`
/// is not a JSON object, has any of [`SECRET_JWK_MEMBERS`], is not a valid
/// JWK ([`Jwk::from_object`], which also refuses an RSA private JWK,
/// ADR 0021), or is not a public key ([`Jwk::is_public`]; an `oct` key
/// never is, RFC 7518 §6.4).
fn check_public_jwk(v: &Value) -> Result<()> {
    let refused = || Error::InvalidHeader("jwk");
    let object = v.as_object().ok_or_else(refused)?;
    if SECRET_JWK_MEMBERS.iter().any(|n| object.contains(n)) {
        return Err(refused());
    }
    match Jwk::from_object(object) {
        Ok(key) if key.is_public() => Ok(()),
        _ => Err(refused()),
    }
}

/// How [`sign`] builds the JWS Protected Header of a JWT (RFC 7519 §5,
/// §7.1).
///
/// The header is `alg`, then `typ` (default [`JWT`], RFC 7519 §5.1),
/// then `kid` copied from the key when it has one (ADR 0027 §8), then any
/// extra members.
///
/// ```
/// use polyjose::jwa::SigningAlg;
/// use polyjose::jwt::IssueOptions;
///
/// let generic = IssueOptions::new(SigningAlg::ES256);
/// let access_token = IssueOptions::new(SigningAlg::ES256).access_token();
/// let untyped = IssueOptions::new(SigningAlg::EdDSA).no_typ();
/// # let _ = (generic, access_token, untyped);
/// ```
#[derive(Debug, Clone)]
pub struct IssueOptions {
    alg: SigningAlg,
    typ: Option<String>,
    params: Object,
}

impl IssueOptions {
    /// Options signing with `alg` (RFC 7515 §4.1.1) and `typ` set to
    /// [`JWT`] (RFC 7519 §5.1).
    pub fn new(alg: SigningAlg) -> Self {
        Self {
            alg,
            typ: Some(JWT.to_owned()),
            params: Object::new(),
        }
    }

    /// Sets `typ` to `t`, an explicit type such as `secevent+jwt`
    /// (RFC 7515 §4.1.9, RFC 8725 §3.11).
    ///
    /// `t` is written as given. `""` and [`JWT`] (the default) give no
    /// explicit typing: they do not tell this kind of JWT apart from any
    /// other, so they do not prevent the cross-JWT confusion RFC 8725
    /// §3.11 and §3.12 describe (draft-ietf-oauth-rfc8725bis §3.11). Use a
    /// media type specific to the application, as `access_token` does
    /// with `at+jwt` (RFC 9068 §2.1).
    pub fn typ(mut self, t: &str) -> Self {
        self.typ = Some(t.to_owned());
        self
    }

    /// Omits `typ` (it is OPTIONAL, RFC 7519 §5.1).
    pub fn no_typ(mut self) -> Self {
        self.typ = None;
        self
    }

    /// Sets `typ` to [`at+jwt`](super::typ::AT_JWT), for a JWT access
    /// token (RFC 9068 §2.1).
    pub fn access_token(self) -> Self {
        self.typ(AT_JWT)
    }

    /// Adds the header parameter `name` with the value `v` (RFC 7515
    /// §4.1–§4.3), after `alg`, `typ` and `kid`, in the order added.
    /// Setting a name again replaces its value and keeps its place.
    ///
    /// A registered parameter must have its registered JSON type, for
    /// example `jku` a string (RFC 7515 §4.1.2) and `x5c` an array of
    /// strings (§4.1.6).
    ///
    /// ```
    /// use polyjose::jwa::SigningAlg;
    /// use polyjose::jwt::IssueOptions;
    ///
    /// let opts = IssueOptions::new(SigningAlg::ES256).header_param("x-tenant", "a")?;
    /// assert!(IssueOptions::new(SigningAlg::ES256).header_param("crit", "b64").is_err());
    /// # let _ = opts;
    /// # Ok::<(), polyjose::Error>(())
    /// ```
    ///
    /// # Errors
    ///
    /// [`Error::InvalidHeader`]:
    ///
    /// - for a parameter a JWT must not carry or its issuer controls: `b64`
    ///   (RFC 7797 §7), `crit` (RFC 7515 §4.1.11), `cty` (RFC 7519 §5.2),
    ///   `zip` and `enc` (RFC 7516 §4.1.2–§4.1.3; ADR 0027 §6, §8);
    /// - for `alg`, `typ` and `kid`, which come from [`IssueOptions::new`],
    ///   the typ setters and the key (ADR 0027 §8);
    /// - for a registered parameter of the wrong JSON type (RFC 7515
    ///   §4.1);
    /// - `InvalidHeader("jwk")` for a `jwk` that is not a valid public JWK
    ///   (RFC 7515 §4.1.3, RFC 7517 §4) or that has any secret or private
    ///   member, whatever its `kty`: `k`, `d`, `p`, `q`, `dp`, `dq`, `qi`
    ///   or `oth` (RFC 7518 §6.2.2, §6.3.2, §6.4.1; RFC 8037 §2). The
    ///   header is public, so such a member would publish the key. The
    ///   error carries no part of the value.
    pub fn header_param(mut self, name: &str, v: impl Into<Value>) -> Result<Self> {
        if let Some(n) = FORBIDDEN_PARAMS.iter().find(|n| **n == name) {
            return Err(Error::InvalidHeader(n));
        }
        if let Some(n) = OWNED_PARAMS.iter().find(|n| **n == name) {
            return Err(Error::InvalidHeader(n));
        }
        let v = v.into();
        if name == "jwk" {
            check_public_jwk(&v)?;
        }
        // RFC 7515 §4.1: type-check a registered parameter now, with the
        // header parser's own rules, so the error points at this call.
        let mut probe = Object::new();
        probe.insert(name, v.clone());
        Header::from_object(probe)?;
        self.params.insert(name, v);
        Ok(self)
    }

    /// The JWS Protected Header for `key` (RFC 7519 §7.1 step 3–4): `alg`,
    /// `typ`, `kid` copied from the key, then the extra members, written
    /// in exactly that order.
    ///
    /// [`ProtectedHeader::new`] would serialize in [`Header`]'s fixed
    /// order (`kid` before `typ`), so the JSON is written here and kept
    /// verbatim through [`ProtectedHeader::from_encoded`], which also
    /// re-checks every registered parameter (RFC 7515 §5.1 steps 3–4).
    fn protected_header(&self, kid: Option<&str>) -> Result<ProtectedHeader> {
        let mut header = Object::new();
        header.insert("alg", self.alg.as_str());
        if let Some(typ) = &self.typ {
            header.insert("typ", typ.as_str());
        }
        if let Some(kid) = kid {
            header.insert("kid", kid);
        }
        // `params` never holds `alg`, `typ` or `kid` (`header_param`
        // refuses them), so no member above is replaced.
        for (name, value) in &self.params {
            header.insert(name, value.clone());
        }
        let json = Zeroizing::new(header.to_json());
        ProtectedHeader::from_encoded(&b64url_encode(json.as_bytes()))
    }
}

/// Whether polyjose issues JWTs with `alg`, decided from the algorithm
/// alone, before the key is read (ADR 0027 §8).
///
/// Fail-closed: only the algorithms listed here are issued.
///
/// - HS256/384/512 (RFC 7518 §3.2), ES256/384/512 (RFC 7518 §3.4),
///   ES256K (RFC 8812 §3.2), EdDSA (RFC 8037 §3.1) and Ed25519
///   (RFC 9864 §2.2) → `Ok`.
/// - `none` → [`Error::AlgorithmNotAllowed`] (RFC 8725 §3.2, ADR 0007).
/// - RS256/384/512 and PS256/384/512 → [`Error::UnsupportedAlgorithm`]:
///   RSA private-key operations are not shipped (ADR 0021).
/// - Anything else → [`Error::UnsupportedAlgorithm`] (RFC 7515 §4.1.1).
fn check_issuable(alg: &SigningAlg) -> Result<()> {
    use SigningAlg as A;
    match alg {
        A::HS256
        | A::HS384
        | A::HS512
        | A::ES256
        | A::ES384
        | A::ES512
        | A::ES256K
        | A::EdDSA
        | A::Ed25519 => Ok(()),
        A::None => Err(Error::AlgorithmNotAllowed),
        _ => Err(Error::UnsupportedAlgorithm),
    }
}

/// Signs `claims` as a compact JWS with `key` (RFC 7519 §7.1), the
/// payload being the Claims Set's JSON (ADR 0027 §1).
///
/// - The JWS Protected Header is built by `opts` (see [`IssueOptions`]):
///   `alg`, `typ` (default `JWT`, RFC 7519 §5.1), `kid` copied from `key`
///   when it has one (ADR 0027 §8), then the extra members.
/// - The payload octets are exactly [`Claims::to_json`] of `claims`
///   (RFC 7519 §7.1 steps 1–2). No claim is added: `sign` never sets
///   `iat`, `exp` or `jti` itself (ADR 0027 §8).
/// - The signature comes from [`sign_compact`] (RFC 7515 §5.1, §7.1),
///   including its key binding (ADR 0026 §2 step 8).
///
/// ```
/// use polyjose::jwa::SigningAlg;
/// use polyjose::jwk::Jwk;
/// use polyjose::jwt::{self, Claims, IssueOptions, NumericDate};
///
/// // RFC 7515 Appendix A.1.1's HMAC key.
/// let key = Jwk::from_json(br#"{"kty":"oct","k":"AyM1SysPpbyDfgZld3umj1qzKObwVMkoqQ-EstJQLr_T-1qS0gZH75aKtMN3Yj0iPS4hcgUuTwjAzZr1Z9CAow"}"#)?;
/// let mut claims = Claims::new();
/// claims.iss = Some("joe".into());
/// claims.exp = NumericDate::from_secs(1300819380);
/// let token = jwt::sign(&claims, &IssueOptions::new(SigningAlg::HS256), &key)?;
/// assert!(token.starts_with("eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9."));
/// # Ok::<(), polyjose::Error>(())
/// ```
///
/// **Secrets:** the Claims Set may be secret (ADR 0049 §4). Its JSON is
/// held in a wiping buffer while signing, but the returned token carries
/// it base64url-encoded, and wiping that `String` is the caller's job.
///
/// # Errors
///
/// In this order:
///
/// - [`Error::UnsupportedAlgorithm`] for RS256/384/512 and PS256/384/512,
///   before `key` is read: polyjose holds no RSA private keys (ADR 0021).
///   Also for an unknown algorithm (RFC 7515 §4.1.1).
/// - [`Error::AlgorithmNotAllowed`] for `none` (RFC 8725 §3.2).
/// - [`Error::KeyAlgorithmMismatch`] or [`Error::InvalidKey`] when `key`
///   cannot sign with the algorithm, including a public-only key
///   (ADR 0026 §2 step 8).
/// - [`Error::RandomnessUnavailable`] from hedged ECDSA signing
///   (ADR 0005, ADR 0026 §5).
pub fn sign(claims: &Claims, opts: &IssueOptions, key: &Jwk) -> Result<String> {
    // ADR 0021, ADR 0027 §8: decided by the algorithm alone, before any
    // key use (sign_compact would reject these too, after reading the key).
    check_issuable(&opts.alg)?;
    let protected = opts.protected_header(key.kid())?;
    // RFC 7519 §7.1 steps 1-2: the Claims Set's octets are the JWS Payload.
    // ADR 0049 §4: they may be secret, so they are wiped on drop.
    let payload = Zeroizing::new(claims.to_json());
    sign_compact(&protected, payload.as_bytes(), key, &SignOptions::new())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::jws::{Verified, verify_compact};
    use crate::jwt::typ::{Layer, check_jwt_header};
    use crate::jwt::validate::validate_claims;
    use crate::jwt::{Audience, FixedClock, NumericDate, Policy};

    /// RFC 7515 Appendix A.3.1: the P-256 key.
    const P256: &str = r#"{"kty":"EC","crv":"P-256","x":"f83OJ3D2xF1Bg8vub9tLe1gHMzV76e8Tus9uPHvRVEU","y":"x_FEzRu9m36HLN_tue659LNpXW6pCyStikYjKIWI5a0","d":"jpsQnnGQmL-YBIffH1136cspYG6-0iY7X1fCE9-E9LI"}"#;
    const ISSUER: &str = "https://issuer.example";
    const AUDIENCE: &str = "https://api.example";
    const ES256: [SigningAlg; 1] = [SigningAlg::ES256];

    fn key() -> Jwk {
        Jwk::from_json(P256.as_bytes()).expect("RFC 7515 A.3.1 key")
    }

    /// The seven claims RFC 9068 §2.2 requires of a JWT access token.
    fn access_token_claims() -> Claims {
        let mut c = Claims::new();
        c.iss = Some(ISSUER.into());
        c.exp = NumericDate::from_secs(1_300_819_380);
        c.aud = Some(Audience::One(AUDIENCE.into()));
        c.sub = Some("5ba552d67".into());
        c.insert("client_id", "s6BhdRkqt3").expect("client_id");
        c.iat = NumericDate::from_secs(1_300_818_000);
        c.jti = Some("dbe39bf3a3ba4238a513f51d6e1691c4".into());
        c
    }

    /// Verifies `token` with the JWS layer (ADR 0026), as T07's pipeline
    /// will before the JWT header and claim checks.
    fn verify(token: &str, policy: &Policy<'_>) -> Verified {
        let public = key().to_public().expect("public half");
        verify_compact(token, &public, &policy.verify_options()).expect("JWS verifies")
    }

    /// T04's header rules on the verified token's protected header.
    fn check_header(v: &Verified, policy: &Policy<'_>) -> Result<()> {
        let protected = v
            .protected
            .as_ref()
            .expect("compact JWS has a protected header");
        check_jwt_header(protected, v.unprotected.as_ref(), policy, Layer::Signed)
    }

    #[test]
    fn access_token_passes_the_rfc9068_header_and_claim_checks() {
        // RFC 9068 §2.1 (typ at+jwt), §2.2 (the seven claims), §4
        // (validation), through T04 and T06 (ADR 0027 §5).
        let clock = FixedClock(1_300_819_000);
        let policy = Policy::access_token(&clock, &ES256, &[ISSUER], &[AUDIENCE]).expect("policy");
        let opts = IssueOptions::new(SigningAlg::ES256).access_token();
        let token = sign(&access_token_claims(), &opts, &key()).expect("issued");

        let verified = verify(&token, &policy);
        assert_eq!(check_header(&verified, &policy), Ok(()));
        let claims = Claims::from_json(&verified.payload).expect("claims parse");
        assert_eq!(claims, access_token_claims());
        assert_eq!(validate_claims(&claims, &policy), Ok(()));
    }

    #[test]
    fn the_same_claims_with_the_default_typ_fail_the_access_token_header_check() {
        // RFC 8725 §3.12: a generic JWT must not pass as an access token.
        let clock = FixedClock(1_300_819_000);
        let policy = Policy::access_token(&clock, &ES256, &[ISSUER], &[AUDIENCE]).expect("policy");
        let token = sign(
            &access_token_claims(),
            &IssueOptions::new(SigningAlg::ES256),
            &key(),
        )
        .expect("issued");
        let err = check_header(&verify(&token, &policy), &policy).expect_err("typ JWT");
        assert!(matches!(err, Error::InvalidHeader(_)), "{err:?}");

        let untyped = sign(
            &access_token_claims(),
            &IssueOptions::new(SigningAlg::ES256).no_typ(),
            &key(),
        )
        .expect("issued");
        let err = check_header(&verify(&untyped, &policy), &policy).expect_err("no typ");
        assert!(matches!(err, Error::InvalidHeader(_)), "{err:?}");
    }

    #[test]
    fn an_access_token_fails_the_default_header_check() {
        // RFC 8725 §3.12, ADR 0027 §5: the other direction; an explicitly
        // typed token stays out of generic validation.
        let clock = FixedClock(1_300_819_000);
        let policy = Policy::new(&clock, &ES256).expect("policy");
        let opts = IssueOptions::new(SigningAlg::ES256).access_token();
        let token = sign(&access_token_claims(), &opts, &key()).expect("issued");
        let err = check_header(&verify(&token, &policy), &policy).expect_err("typ at+jwt");
        assert!(matches!(err, Error::InvalidHeader(_)), "{err:?}");
    }

    #[test]
    fn issuable_algorithms_are_an_explicit_list() {
        // ADR 0027 §8, ADR 0021: fail-closed on anything not listed.
        for alg in SigningAlg::KNOWN {
            let expected = match alg {
                SigningAlg::None => Err(Error::AlgorithmNotAllowed),
                SigningAlg::RS256
                | SigningAlg::RS384
                | SigningAlg::RS512
                | SigningAlg::PS256
                | SigningAlg::PS384
                | SigningAlg::PS512 => Err(Error::UnsupportedAlgorithm),
                _ => Ok(()),
            };
            assert_eq!(check_issuable(alg), expected, "{alg:?}");
        }
        assert_eq!(
            check_issuable(&SigningAlg::from("ES999")),
            Err(Error::UnsupportedAlgorithm)
        );
    }

    #[test]
    fn protected_header_order_is_alg_typ_kid_then_extras() {
        let opts = IssueOptions::new(SigningAlg::HS256)
            .header_param("x5t#S256", "abc")
            .expect("x5t#S256 is a string");
        let protected = opts.protected_header(Some("k1")).expect("header");
        let json = crate::util::b64url_decode(protected.encoded()).expect("base64url");
        assert_eq!(
            json,
            br#"{"alg":"HS256","typ":"JWT","kid":"k1","x5t#S256":"abc"}"#
        );
        assert_eq!(protected.header().kid.as_deref(), Some("k1"));
        assert_eq!(protected.header().x5t_s256.as_deref(), Some("abc"));
    }
}