wasmer_deploy_schema/schema/
capability.rs

1use std::collections::HashMap;
2
3use serde::{Deserialize, Serialize};
4
5use super::{
6    CapabilityCpuV1, CapabilityFileSystemV1, CapabilityLoggingV1, CapabilityMemorySwapV1,
7    CapabilityMemoryV1, CapabilityNetworkDnsV1, CapabilityNetworkGatewayV1, CapabilityNetworkV1,
8    CapabilityPersistentMemoryV1, CapabilityWasiV1, Merge,
9};
10
11#[derive(Serialize, Deserialize, PartialEq, Eq, Clone, Debug, schemars::JsonSchema)]
12pub enum CapabilityV1 {
13    #[serde(rename = "cpu")]
14    Cpu(CapabilityCpuV1),
15    #[serde(rename = "fs")]
16    FileSystem(CapabilityFileSystemV1),
17    #[serde(rename = "memory_swap")]
18    MemorySwap(CapabilityMemorySwapV1),
19    #[serde(rename = "memory_persistent")]
20    MemoryPersistent(CapabilityPersistentMemoryV1),
21    #[serde(rename = "network")]
22    Network(CapabilityNetworkV1),
23    #[serde(rename = "network_dns")]
24    NetworkDns(CapabilityNetworkDnsV1),
25    #[serde(rename = "network_gateway")]
26    NetworkGateway(CapabilityNetworkGatewayV1),
27}
28
29#[derive(Serialize, Deserialize, Default, PartialEq, Eq, Clone, Debug, schemars::JsonSchema)]
30pub struct CapabilityMapV1 {
31    #[serde(skip_serializing_if = "Option::is_none")]
32    pub wasi: Option<CapabilityWasiV1>,
33    #[serde(skip_serializing_if = "Option::is_none")]
34    pub cpu: Option<CapabilityCpuV1>,
35    #[serde(skip_serializing_if = "Option::is_none")]
36    pub fs: Option<CapabilityFileSystemV1>,
37    #[serde(skip_serializing_if = "Option::is_none")]
38    pub memory: Option<CapabilityMemoryV1>,
39
40    #[serde(skip_serializing_if = "Option::is_none")]
41    pub network: Option<CapabilityNetworkV1>,
42    #[serde(skip_serializing_if = "Option::is_none")]
43    pub network_dns: Option<CapabilityNetworkDnsV1>,
44    #[serde(skip_serializing_if = "Option::is_none")]
45    pub network_gateway: Option<CapabilityNetworkGatewayV1>,
46    #[serde(skip_serializing_if = "Option::is_none")]
47    pub logging: Option<CapabilityLoggingV1>,
48
49    /// Additional unknown capabilities.
50    #[serde(flatten)]
51    pub other: HashMap<String, serde_json::Value>,
52}
53
54impl Merge for CapabilityMapV1 {
55    fn merge_extend(self, other: &Self) -> Self {
56        Self {
57            wasi: self.wasi.merge_extend(&other.wasi),
58            cpu: self.cpu.merge_extend(&other.cpu),
59            fs: self.fs.merge_extend(&other.fs),
60            memory: self.memory.merge_extend(&other.memory),
61            network: self.network.merge_extend(&other.network),
62            network_dns: self.network_dns.merge_extend(&other.network_dns),
63            network_gateway: self.network_gateway.merge_extend(&other.network_gateway),
64            logging: self.logging.merge_extend(&other.logging),
65            other: self.other,
66        }
67    }
68}