//! [POST /_matrix/key/v2/query](https://matrix.org/docs/spec/server_server/r0.1.4#post-matrix-key-v2-query)

use std::{collections::BTreeMap, time::SystemTime};

use crate::discovery::ServerKey;
use ruma_api::ruma_api;
use ruma_identifiers::{ServerKeyId, ServerName};
use serde::{Deserialize, Serialize};

ruma_api! {
    metadata: {
        description: "Query for keys from multiple servers in a batch format.",
        method: POST,
        name: "get_remote_server_keys_batch",
        path: "/_matrix/key/v2/query",
        rate_limited: false,
        requires_authentication: false,
    }

    request: {
        /// The query criteria. The outer string key on the object is the server
        /// name (eg: matrix.org). The inner string key is the Key ID to query
        /// for the particular server. If no key IDs are given to be queried,
        /// the notary server should query for all keys. If no servers are
        /// given, the notary server must return an empty server_keys array in
        /// the response.
        ///
        /// The notary server may return multiple keys regardless of the Key IDs
        /// given.
        #[ruma_api(body)]
        pub server_keys: BTreeMap<Box<ServerName>, BTreeMap<ServerKeyId, QueryCriteria>>,

        /// A millisecond POSIX timestamp in milliseconds indicating when the
        /// returned certificates will need to be valid until to be useful to
        /// the requesting server.
        ///
        /// If not supplied, the current time as determined by the notary server
        /// is used.
        #[ruma_api(query)]
        #[serde(with = "ruma_serde::time::ms_since_unix_epoch")]
        pub minimum_valid_until_ts: SystemTime,
    }

    response: {
        /// The queried server's keys, signed by the notary server.
        pub server_keys: Vec<ServerKey>,
    }
}

/// The query criteria.
#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct QueryCriteria {
    /// A millisecond POSIX timestamp in milliseconds indicating when the
    /// returned certificates will need to be valid until to be useful to the
    /// requesting server.
    ///
    /// If not supplied, the current time as determined by the notary server is
    /// used.
    #[serde(
        default,
        skip_serializing_if = "Option::is_none",
        with = "ruma_serde::time::opt_ms_since_unix_epoch"
    )]
    pub minimum_valid_until_ts: Option<SystemTime>,
}