feat(stackable-versioned): Generate primitive Status struct

Author: Techassi

Cargo.lock

@@ -61,9 +61,10 @@ pub(crate) struct KubernetesSkipArguments {
 /// This struct contains crate overrides to be passed to `#[kube]`.
 #[derive(Clone, Debug, FromMeta)]
 pub(crate) struct KubernetesCrateArguments {
-    pub(crate) kube_core: Option<Path>,
-    pub(crate) k8s_openapi: Option<Path>,
-    pub(crate) schemars: Option<Path>,
-    pub(crate) serde: Option<Path>,
-    pub(crate) serde_json: Option<Path>,
+    pub kube_core: Option<Path>,
+    pub k8s_openapi: Option<Path>,
+    pub schemars: Option<Path>,
+    pub serde: Option<Path>,
+    pub serde_json: Option<Path>,
+    pub versioned: Option<Path>,
 }

crates/stackable-versioned-macros/src/attrs/k8s.rs

@@ -95,7 +95,7 @@ impl Container {
     ///
     /// This function only returns `Some` if it is a struct. Enums cannot be used to define
     /// Kubernetes custom resources.
-    pub(crate) fn generate_kubernetes_item(
+    pub fn generate_kubernetes_item(
         &self,
         version: &VersionDefinition,
     ) -> Option<(IdentString, String, TokenStream)> {
@@ -109,7 +109,7 @@ impl Container {
     ///
     /// This function only returns `Some` if it is a struct. Enums cannot be used to define
     /// Kubernetes custom resources.
-    pub(crate) fn generate_kubernetes_merge_crds(
+    pub fn generate_kubernetes_merge_crds(
         &self,
         enum_variant_idents: &[IdentString],
         enum_variant_strings: &[String],
@@ -129,7 +129,14 @@ impl Container {
         }
     }
 
-    pub(crate) fn get_original_ident(&self) -> &Ident {
+    pub fn generate_kubernetes_status_struct(&self, vis: &Visibility) -> Option<TokenStream> {
+        match self {
+            Container::Struct(s) => s.generate_kubernetes_status_struct(vis),
+            Container::Enum(_) => None,
+        }
+    }
+
+    pub fn get_original_ident(&self) -> &Ident {
         match &self {
             Container::Struct(s) => s.common.idents.original.as_ident(),
             Container::Enum(e) => e.common.idents.original.as_ident(),
@@ -252,6 +259,10 @@ impl StandaloneContainer {
             false,
         ));
 
+        if self.versions.len() > 1 {
+            tokens.extend(self.container.generate_kubernetes_status_struct(vis));
+        }
+
         tokens
     }
 }
@@ -344,11 +355,13 @@ pub struct KubernetesCrateOptions {
     pub schemars: Override<Path>,
     pub serde: Override<Path>,
     pub serde_json: Override<Path>,
+    pub versioned: Override<Path>,
 }
 
 impl Default for KubernetesCrateOptions {
     fn default() -> Self {
         Self {
+            versioned: Override::Default(parse_quote! { ::stackable_versioned }),
             k8s_openapi: Override::Default(parse_quote! { ::k8s_openapi }),
             serde_json: Override::Default(parse_quote! { ::serde_json }),
             kube_core: Override::Default(parse_quote! { ::kube::core }),
@@ -382,6 +395,10 @@ impl From<KubernetesCrateArguments> for KubernetesCrateOptions {
             crate_options.serde = Override::Overridden(serde);
         }
 
+        if let Some(versioned) = args.versioned {
+            crate_options.versioned = Override::Overridden(versioned);
+        }
+
         crate_options
     }
 }
@@ -396,6 +413,7 @@ impl ToTokens for KubernetesCrateOptions {
             kube_core,
             schemars,
             serde,
+            ..
         } = self;
 
         if let Override::Overridden(k8s_openapi) = k8s_openapi {

crates/stackable-versioned-macros/src/codegen/container/mod.rs

@@ -2,7 +2,7 @@ use std::ops::Not;
 
 use darling::{Error, FromAttributes, Result, util::IdentString};
 use proc_macro2::TokenStream;
-use quote::{ToTokens, quote};
+use quote::{ToTokens, format_ident, quote};
 use syn::{Generics, ItemStruct, Path, Visibility, parse_quote};
 
 use crate::{
@@ -305,10 +305,7 @@ impl Struct {
 
 // Kubernetes-specific token generation
 impl Struct {
-    pub(crate) fn generate_kube_attribute(
-        &self,
-        version: &VersionDefinition,
-    ) -> Option<TokenStream> {
+    pub fn generate_kube_attribute(&self, version: &VersionDefinition) -> Option<TokenStream> {
         match &self.common.options.kubernetes_options {
             Some(kubernetes_options) => {
                 // Required arguments
@@ -359,7 +356,7 @@ impl Struct {
         }
     }
 
-    pub(crate) fn generate_kubernetes_item(
+    pub fn generate_kubernetes_item(
         &self,
         version: &VersionDefinition,
     ) -> Option<(IdentString, String, TokenStream)> {
@@ -384,7 +381,7 @@ impl Struct {
         }
     }
 
-    pub(crate) fn generate_kubernetes_merge_crds(
+    pub fn generate_kubernetes_merge_crds(
         &self,
         enum_variant_idents: &[IdentString],
         enum_variant_strings: &[String],
@@ -434,4 +431,41 @@ impl Struct {
             _ => None,
         }
     }
+
+    pub fn generate_kubernetes_status_struct(&self, vis: &Visibility) -> Option<TokenStream> {
+        match &self.common.options.kubernetes_options {
+            Some(kubernetes_options) => {
+                let status_ident = format_ident!(
+                    "{struct_ident}Status",
+                    struct_ident = self.common.idents.kubernetes.as_ident()
+                );
+
+                let versioned_crate = &*kubernetes_options.crates.versioned;
+                let schemars_crate = &*kubernetes_options.crates.schemars;
+                let serde_crate = &*kubernetes_options.crates.serde;
+
+                // FIXME (@Techassi): This needs to be set in the #[kube] attribute
+                match &kubernetes_options.status {
+                    Some(status) => Some(quote! {
+                        #[derive(Clone, Debug, #serde_crate::Deserialize, #serde_crate::Serialize, #schemars_crate::JsonSchema)]
+                        #[serde(rename_all = "camelCase")]
+                        #vis struct #status_ident {
+                            pub crd_changes: #versioned_crate::CrdChanges,
+
+                            #[serde(flatten)]
+                            pub status: #status,
+                        }
+                    }),
+                    None => Some(quote! {
+                        #[derive(Clone, Debug, #serde_crate::Deserialize, #serde_crate::Serialize, #schemars_crate::JsonSchema)]
+                        #[serde(rename_all = "camelCase")]
+                        #vis struct #status_ident {
+                            pub crd_changes: #versioned_crate::CrdChanges,
+                        }
+                    }),
+                }
+            }
+            None => None,
+        }
+    }
 }

crates/stackable-versioned-macros/src/codegen/container/struct.rs

@@ -14,7 +14,11 @@ all-features = true
 full = ["k8s"]
 k8s = [
   "stackable-versioned-macros/k8s", # Forward the k8s feature to the underlying macro crate
+  "dep:schemars",
+  "dep:serde",
 ]
 
 [dependencies]
 stackable-versioned-macros = { path = "../stackable-versioned-macros" }
+schemars = { workspace = true, optional = true }
+serde = { workspace = true, optional = true }

crates/stackable-versioned/Cargo.toml

@@ -24,3 +24,9 @@ pub trait AsVersionStr {
         Self::VERSION
     }
 }
+
+#[derive(Clone, Debug, serde::Deserialize, serde::Serialize, schemars::JsonSchema)]
+pub struct CrdChanges {
+    pub downgrades: Vec<String>,
+    pub upgrades: Vec<String>,
+}