k8s_openapi_ext/
ext.rs

1use std::collections::BTreeMap;
2use std::fmt;
3
4use super::*;
5
6pub use cluster_role::ClusterRoleExt;
7pub use cluster_role_binding::ClusterRoleBindingExt;
8pub use configmap::ConfigMapExt;
9pub use configmap_volume_source::ConfigMapVolumeSourceExt;
10pub use container::ContainerExt;
11pub use container_port::ContainerPortExt;
12pub use cronjob::CronJobExt;
13pub use daemon_set::DaemonSetExt;
14pub use deployment::DeploymentExt;
15pub use env::EnvVarExt;
16pub use env::ToEnvFrom;
17pub use env::ToEnvVar;
18pub use hpa::HorizontalPodAutoscalerExt;
19pub use job::JobExt;
20pub use label_selector::LabelSelectorExt;
21pub use local_object_reference::LocalObjectReferenceExt;
22pub use metric::MetricExt;
23pub use namespace::NamespaceExt;
24pub use node::NodeExt;
25pub use owner_reference::OwnerReferenceExt;
26pub use pod::PodExt;
27pub use pod_spec::PodSpecExt;
28pub use pod_template_spec::PodTemplateSpecExt;
29pub use policy_rule::PolicyRuleExt;
30pub use probe::ProbeExt;
31pub use replica_set::ReplicaSetExt;
32pub use role::RoleExt;
33pub use role_binding::RoleBindingExt;
34pub use role_ref::RoleRefExt;
35pub use secret::SecretExt;
36pub use secret::SecretExt2;
37pub use secret_env_source::SecretEnvSourceExt;
38pub use secret_reference::SecretReferenceExt;
39pub use secret_volume_source::SecretVolumeSourceExt;
40pub use security_context::SecurityContextExt;
41pub use service::ServiceExt;
42pub use service_account::ServiceAccountExt;
43pub use service_port::ServicePortExt;
44pub use storage_class::StorageClassExt;
45pub use subject::SubjectExt;
46pub use taint::TaintExt;
47pub use time::TimeExt;
48pub use toleration::TolerationBuilder;
49pub use toleration::TolerationExt;
50pub use typed_object_reference::TypedObjectReferenceExt;
51pub use volume::VolumeExt;
52pub use volume_mount::VolumeMountExt;
53
54use effect::Effect;
55
56mod cluster_role;
57mod cluster_role_binding;
58mod configmap;
59mod configmap_volume_source;
60mod container;
61mod container_port;
62mod cronjob;
63mod daemon_set;
64mod deployment;
65mod effect;
66mod env;
67mod hpa;
68mod job;
69mod label_selector;
70mod local_object_reference;
71mod metric;
72mod namespace;
73mod node;
74mod owner_reference;
75mod pod;
76mod pod_spec;
77mod pod_template_spec;
78mod policy_rule;
79mod probe;
80mod replica_set;
81mod role;
82mod role_binding;
83mod role_ref;
84mod secret;
85mod secret_env_source;
86mod secret_reference;
87mod secret_volume_source;
88mod security_context;
89mod service;
90mod service_account;
91mod service_port;
92mod storage_class;
93mod subject;
94mod taint;
95mod time;
96mod toleration;
97mod typed_object_reference;
98mod volume;
99mod volume_mount;
100
101pub trait ResourceBuilder: Sized {
102    fn metadata(name: impl ToString) -> metav1::ObjectMeta {
103        metadata(name)
104    }
105
106    /// Set namespace for this object
107    ///
108    fn namespace(self, namespace: impl ToString) -> Self;
109
110    /// Set the owner for this object
111    ///
112    fn owner(self, owner: metav1::OwnerReference) -> Self;
113
114    /// Set one label for this object.
115    /// For settins multiple lables at once prefer `labels()`
116    ///
117    fn label(self, key: impl ToString, value: impl ToString) -> Self;
118
119    /// Set labels for this object
120    ///
121    fn labels(self, labels: impl IntoIterator<Item = (impl ToString, impl ToString)>) -> Self;
122
123    fn with_resource_version(self, resource_version: String) -> Self;
124
125    /// Set recommended label 'app.kubernetes.io/name'
126    ///
127    fn app_name(self, name: impl ToString) -> Self {
128        self.labels([(label::APP_NAME, name)])
129    }
130
131    /// Set recommended label 'app.kubernetes.io/instance'
132    ///
133    fn app_instance(self, instance: impl ToString) -> Self {
134        self.labels([(label::APP_INSTANCE, instance)])
135    }
136
137    /// Set recommended label 'app.kubernetes.io/version'
138    ///
139    fn app_version(self, version: impl ToString) -> Self {
140        self.labels([(label::APP_VERSION, version)])
141    }
142
143    /// Set recommended label 'app.kubernetes.io/component'
144    ///
145    fn app_component(self, component: impl ToString) -> Self {
146        self.labels([(label::APP_COMPONENT, component)])
147    }
148
149    /// Set recommended label 'app.kubernetes.io/part-of'
150    ///
151    fn app_part_of(self, part_of: impl ToString) -> Self {
152        self.labels([(label::APP_PART_OF, part_of)])
153    }
154
155    /// Set recommended label 'app.kubernetes.io/managed-by'
156    ///
157    fn app_managed_by(self, managed_by: impl ToString) -> Self {
158        self.labels([(label::APP_MANAGED_BY, managed_by)])
159    }
160}
161
162impl<T> ResourceBuilder for T
163where
164    T: openapi::Metadata<Ty = metav1::ObjectMeta>,
165{
166    fn namespace(mut self, namespace: impl ToString) -> Self {
167        let namespace = Some(namespace.to_string());
168        self.metadata_mut().namespace = namespace;
169        self
170    }
171
172    fn owner(mut self, owner: metav1::OwnerReference) -> Self {
173        self.metadata_mut()
174            .owner_references
175            .get_or_insert_default()
176            .push(owner);
177        self
178    }
179
180    fn label(mut self, key: impl ToString, value: impl ToString) -> Self {
181        self.metadata_mut()
182            .labels
183            .get_or_insert_default()
184            .insert(key.to_string(), value.to_string());
185        self
186    }
187
188    fn labels(mut self, labels: impl IntoIterator<Item = (impl ToString, impl ToString)>) -> Self {
189        let labels = labels
190            .into_iter()
191            .map(|(key, value)| (key.to_string(), value.to_string()));
192        self.metadata_mut()
193            .labels
194            .get_or_insert_default()
195            .extend(labels);
196        self
197    }
198
199    fn with_resource_version(mut self, resource_version: String) -> Self {
200        self.metadata_mut().resource_version = Some(resource_version);
201        self
202    }
203}
204
205fn metadata(name: impl ToString) -> metav1::ObjectMeta {
206    let name = Some(name.to_string());
207    metav1::ObjectMeta {
208        name,
209        // annotations: todo!(),
210        // creation_timestamp: todo!(),
211        // deletion_grace_period_seconds: todo!(),
212        // deletion_timestamp: todo!(),
213        // finalizers: todo!(),
214        // generate_name: todo!(),
215        // generation: todo!(),
216        // labels: todo!(),
217        // managed_fields: todo!(),
218        // namespace: todo!(),
219        // owner_references: todo!(),
220        // resource_version: todo!(),
221        // self_link: todo!(),
222        // uid: todo!(),
223        ..default()
224    }
225}
226
227trait HasSpec {
228    type Spec;
229    fn spec_mut(&mut self) -> &mut Self::Spec;
230}