nym/nym-node-status-api/nym-node-status-client/src/models.rs

use nym_crypto::asymmetric::ed25519::{PublicKey, Signature};
use nym_gateway_probe::types::AttachedTicketMaterials;
use serde::{Deserialize, Serialize};
use std::fmt::Debug;

pub mod get_testrun {
    use crate::auth::SignedRequest;

    use super::*;
    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct Payload {
        pub agent_public_key: PublicKey,
        pub timestamp: i64,
    }

    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct GetTestrunRequest {
        pub payload: Payload,
        pub signature: Signature,
    }

    impl SignedRequest for GetTestrunRequest {
        type Payload = Payload;

        fn public_key(&self) -> &PublicKey {
            &self.payload.agent_public_key
        }

        fn signature(&self) -> &Signature {
            &self.signature
        }

        fn payload(&self) -> &Self::Payload {
            &self.payload
        }
    }
}

#[derive(Deserialize, Serialize, Debug)]
pub struct TestrunAssignment {
    pub testrun_id: i32,
    pub assigned_at_utc: i64,
    pub gateway_identity_key: String,
}

impl TestrunAssignment {
    pub fn with_ticket_materials(
        self,
        materials: AttachedTicketMaterials,
    ) -> TestrunAssignmentWithTickets {
        TestrunAssignmentWithTickets {
            assignment: self,
            ticket_materials: materials,
        }
    }
}

#[derive(Deserialize, Serialize)]
pub struct TestrunAssignmentWithTickets {
    #[serde(flatten)]
    pub assignment: TestrunAssignment,

    pub ticket_materials: AttachedTicketMaterials,
}

impl Debug for TestrunAssignmentWithTickets {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        // no need to include full binary data behind the ticketbook data
        f.debug_struct("TestrunAssignmentWithTickets")
            .field("assignment", &self.assignment)
            .field("ticket_materials", &"attached")
            .finish()
    }
}

pub mod submit_results {
    use crate::auth::SignedRequest;

    use super::*;
    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct Payload {
        pub probe_result: nym_gateway_probe::ProbeResult,
        pub probe_log: String,
        pub agent_public_key: PublicKey,
        pub assigned_at_utc: i64,
    }

    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct SubmitResults {
        pub payload: Payload,
        pub signature: Signature,
    }

    impl SignedRequest for SubmitResults {
        type Payload = Payload;

        fn public_key(&self) -> &PublicKey {
            &self.payload.agent_public_key
        }

        fn signature(&self) -> &Signature {
            &self.signature
        }

        fn payload(&self) -> &Self::Payload {
            &self.payload
        }
    }
}

pub mod submit_results_v2 {
    use crate::auth::SignedRequest;

    use super::*;
    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct Payload {
        pub probe_result: nym_gateway_probe::ProbeResult,
        pub probe_log: String,
        pub agent_public_key: PublicKey,
        pub assigned_at_utc: i64,
        pub gateway_identity_key: String,
    }

    #[derive(Debug, Clone, Deserialize, Serialize)]
    pub struct SubmitResultsV2 {
        pub payload: Payload,
        pub signature: Signature,
    }

    impl SignedRequest for SubmitResultsV2 {
        type Payload = Payload;

        fn public_key(&self) -> &PublicKey {
            &self.payload.agent_public_key
        }

        fn signature(&self) -> &Signature {
            &self.signature
        }

        fn payload(&self) -> &Self::Payload {
            &self.payload
        }
    }
}