veilid_remote_api/

lib.rs

//! # veilid-remote-api
//!
//! This crate provides the ability to control a Veilid node remotely.
//!
//! The [JsonRequestProcessor] is a wrapper around the [VeilidAPI] and provides a way to process requests and send responses.

use veilid_core::tools::*;
use veilid_core::*;

use parking_lot::Mutex;
use schemars::{schema_for, JsonSchema};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fmt;
use tracing::*;

mod routing_context;
pub use routing_context::*;

mod table_db;
pub use table_db::*;

mod crypto_system;
pub use crypto_system::*;

mod dht_transaction;
pub use dht_transaction::*;

mod process;
pub use process::*;

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct Request {
    /// Operation Id (pairs with Response, or empty if unidirectional).
    #[serde(default)]
    pub id: u32,
    /// The request operation variant.
    #[serde(flatten)]
    pub op: RequestOp,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(tag = "type")]
pub enum RecvMessage {
    Response(Response),
    Update(VeilidUpdate),
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct Response {
    /// Operation Id (pairs with Request, or empty if unidirectional).
    #[serde(default)]
    pub id: u32,
    /// The response operation variant.
    #[serde(flatten)]
    pub op: ResponseOp,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(tag = "op")]
pub enum RequestOp {
    Control {
        args: Vec<String>,
    },
    GetState,
    IsShutdown,
    Attach,
    Detach,
    GenerateMemberId {
        #[schemars(with = "String")]
        writer_key: PublicKey,
    },
    GetDhtRecordKey {
        schema: DHTSchema,
        #[schemars(with = "String")]
        owner: PublicKey,
        #[schemars(with = "Option<String>")]
        encryption_key: Option<SharedSecret>,
    },
    NewPrivateRoute,
    NewCustomPrivateRoute {
        #[serde(default)]
        private_spec: PrivateSpec,
    },
    ImportRemotePrivateRoute {
        #[serde(with = "as_human_base64")]
        #[schemars(with = "String")]
        blob: Vec<u8>,
    },
    ReleasePrivateRoute {
        #[schemars(with = "String")]
        route_id: RouteId,
    },
    AppCallReply {
        #[schemars(with = "String")]
        call_id: OperationId,
        #[serde(with = "as_human_base64")]
        #[schemars(with = "String")]
        message: Vec<u8>,
    },
    // Routing Context
    NewRoutingContext,
    RoutingContext(RoutingContextRequest),
    // DHT Transaction
    TransactDhtRecords {
        #[schemars(with = "Vec<String>")]
        record_keys: Vec<RecordKey>,
        options: Option<TransactDHTRecordsOptions>,
    },
    DhtTransaction(DhtTransactionRequest),
    // TableDb
    OpenTableDb {
        name: String,
        column_count: u32,
    },
    DeleteTableDb {
        name: String,
    },
    TableDb(TableDbRequest),
    TableDbTransaction(TableDbTransactionRequest),
    // Crypto
    GetCryptoSystem {
        #[schemars(with = "String")]
        kind: CryptoKind,
    },
    CryptoSystem(CryptoSystemRequest),
    VerifySignatures {
        #[schemars(with = "Vec<String>")]
        node_ids: Vec<PublicKey>,
        #[serde(with = "as_human_base64")]
        #[schemars(with = "String")]
        data: Vec<u8>,
        #[schemars(with = "Vec<String>")]
        signatures: Vec<Signature>,
    },
    GenerateSignatures {
        #[serde(with = "as_human_base64")]
        #[schemars(with = "String")]
        data: Vec<u8>,
        #[schemars(with = "Vec<String>")]
        key_pairs: Vec<KeyPair>,
    },
    GenerateKeyPair {
        #[schemars(with = "String")]
        kind: CryptoKind,
    },
    // Misc
    Now,
    Debug {
        command: String,
    },
    VeilidVersionString,
    VeilidVersion,
    VeilidFeatures,
    DefaultVeilidConfig,
    ValidCryptoKinds,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(tag = "op")]
pub enum ResponseOp {
    Control {
        #[serde(flatten)]
        result: ApiResult<String>,
    },
    GetState {
        #[serde(flatten)]
        result: ApiResult<Box<VeilidState>>,
    },
    IsShutdown {
        value: bool,
    },
    Attach {
        #[serde(flatten)]
        result: ApiResult<()>,
    },
    Detach {
        #[serde(flatten)]
        result: ApiResult<()>,
    },
    GenerateMemberId {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<String>")]
        result: ApiResultWithString<MemberId>,
    },
    GetDhtRecordKey {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<String>")]
        result: ApiResultWithString<RecordKey>,
    },
    NewPrivateRoute {
        #[serde(flatten)]
        result: ApiResult<RouteBlob>,
    },
    NewCustomPrivateRoute {
        #[serde(flatten)]
        result: ApiResult<RouteBlob>,
    },
    ImportRemotePrivateRoute {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<String>")]
        result: ApiResultWithString<RouteId>,
    },
    ReleasePrivateRoute {
        #[serde(flatten)]
        result: ApiResult<()>,
    },
    AppCallReply {
        #[serde(flatten)]
        result: ApiResult<()>,
    },
    // Routing Context
    NewRoutingContext {
        #[serde(flatten)]
        result: ApiResult<u32>,
    },
    RoutingContext(Box<RoutingContextResponse>),
    // DHT Transaction
    TransactDhtRecords {
        #[serde(flatten)]
        result: ApiResult<u32>,
    },
    DhtTransaction(Box<DhtTransactionResponse>),
    // TableDb
    OpenTableDb {
        #[serde(flatten)]
        result: ApiResult<u32>,
    },
    DeleteTableDb {
        #[serde(flatten)]
        result: ApiResult<bool>,
    },
    TableDb(TableDbResponse),
    TableDbTransaction(TableDbTransactionResponse),
    // Crypto
    GetCryptoSystem {
        #[serde(flatten)]
        result: ApiResult<u32>,
    },
    BestCryptoSystem {
        #[serde(flatten)]
        result: ApiResult<u32>,
    },
    CryptoSystem(CryptoSystemResponse),
    VerifySignatures {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<Option<Vec<String>>>")]
        result: ApiResultWithOptVecString<Option<PublicKeyGroup>>,
    },
    GenerateSignatures {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<Vec<String>>")]
        result: ApiResultWithVecString<Vec<Signature>>,
    },
    GenerateKeyPair {
        #[serde(flatten)]
        #[schemars(with = "ApiResult<String>")]
        result: ApiResultWithString<KeyPair>,
    },
    // Misc
    Now {
        #[schemars(with = "String")]
        value: Timestamp,
    },
    Debug {
        #[serde(flatten)]
        result: ApiResult<String>,
    },
    VeilidVersionString {
        value: String,
    },
    VeilidVersion {
        major: u32,
        minor: u32,
        patch: u32,
    },
    DefaultVeilidConfig {
        value: String,
    },
    VeilidFeatures {
        value: Vec<String>,
    },
    ValidCryptoKinds {
        #[schemars(with = "Vec<String>")]
        value: Vec<CryptoKind>,
    },
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResult<T>
where
    T: Clone + fmt::Debug + JsonSchema,
{
    Ok { value: T },
    Err { error: VeilidAPIError },
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResultWithString<T>
where
    T: Clone + fmt::Debug,
{
    Ok {
        #[schemars(with = "String")]
        value: T,
    },
    Err {
        error: VeilidAPIError,
    },
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResultWithVecU8 {
    Ok {
        #[serde(with = "as_human_base64")]
        #[schemars(with = "String")]
        value: Vec<u8>,
    },
    Err {
        error: VeilidAPIError,
    },
}
#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(transparent)]
pub struct VecU8 {
    #[serde(with = "as_human_base64")]
    #[schemars(with = "String")]
    value: Vec<u8>,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResultWithVecVecU8 {
    Ok {
        #[schemars(with = "Vec<String>")]
        value: Vec<VecU8>,
    },
    Err {
        error: VeilidAPIError,
    },
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResultWithVecString<T>
where
    T: Clone + fmt::Debug,
{
    Ok {
        #[schemars(with = "Vec<String>")]
        value: T,
    },
    Err {
        error: VeilidAPIError,
    },
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
#[serde(untagged)]
pub enum ApiResultWithOptVecString<T>
where
    T: Clone + fmt::Debug,
{
    Ok {
        #[schemars(with = "Option<Vec<String>>")]
        value: T,
    },
    Err {
        error: VeilidAPIError,
    },
}

pub fn emit_schemas(out: &mut HashMap<String, String>) {
    let schema_request = schema_for!(Request);
    let schema_recv_message = schema_for!(RecvMessage);

    out.insert(
        "Request".to_owned(),
        serde_json::to_string_pretty(&schema_request).unwrap_or_log(),
    );

    out.insert(
        "RecvMessage".to_owned(),
        serde_json::to_string_pretty(&schema_recv_message).unwrap_or_log(),
    );
}