Load transformed JSON in codegen

Signed-off-by: kazk <[email protected]>

Author: kazk

justfile

@@ -56,7 +56,7 @@ swagger-transform:
     #!/usr/bin/env bash
     set -exuo pipefail
     cd k8s-pb-codegen/openapi
-    jq -f list-resources.jq < swagger.json > api-resources.json
+    jq -f transform.jq < swagger.json > transformed.json
 
 # Download and generate all swagger dependent files
 swagger: swagger-dl swagger-patch swagger-transform

k8s-pb-codegen/src/lib.rs

@@ -0,0 +1,50 @@
+#[derive(Clone, Debug, serde::Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Resource {
+    /// Name of the resource used in URL.
+    pub name: String,
+    /// True if the resource is namespaced. May still have `all` verbs like `list`.
+    pub namespaced: bool,
+    pub api_group_version: String,
+    pub group: String,
+    pub version: String,
+    pub kind: String,
+    /// Protobuf type path.
+    pub proto: String,
+    /// Rust type path.
+    pub rust: String,
+    /// Metadata type.
+    pub metadata: Option<String>,
+    /// Spec type if any.
+    pub spec: Option<String>,
+    /// Status type if any.
+    pub status: Option<String>,
+    /// Condition type if the resource has `.status.conditions`.
+    pub condition: Option<String>,
+    /// Verbs grouped by scope. `all` or `namespaced`.
+    pub scoped_verbs: ScopedVerbs,
+    /// All paths associated with this resource.
+    pub paths: Vec<String>,
+    // TODO `Option<Vec<Subresource>>`
+    /// Any subresources.
+    pub subresources: Vec<Subresource>,
+}
+
+#[derive(Clone, Debug, serde::Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct ScopedVerbs {
+    pub all: Option<Vec<String>>,
+    /// Namespaced actions.
+    pub namespaced: Option<Vec<String>>,
+}
+
+#[derive(Clone, Debug, serde::Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Subresource {
+    /// Name of the subresouce used in URL.
+    pub name: String,
+    /// Verbs grouped by scope. `all` or `namespaced`.
+    pub scoped_verbs: ScopedVerbs,
+    /// All paths associated with this subresource.
+    pub paths: Vec<String>,
+}

k8s-pb-codegen/src/main.rs

@@ -3,7 +3,9 @@ use prost_types::FileDescriptorSet;
 #[macro_use]
 extern crate log;
 use anyhow::{Context, Result};
-use std::{fs, path::Path};
+use std::{collections::HashMap, fs, path::Path};
+
+use k8s_pb_codegen::Resource;
 
 fn read_file<P: AsRef<Path>>(path: P) -> Result<String> {
     fs::read_to_string(&path).with_context(|| format!("read {}", path.as_ref().display()))
@@ -26,6 +28,11 @@ fn main() -> Result<()> {
         .out_dir("./out")
         .compile_protos(protos.as_slice(), &["protos/"])?;
 
+    info!("loading json");
+    let apif: String = read_file(root.join("openapi/transformed.json"))?;
+    let resources: HashMap<String, Resource> =
+        serde_json::from_str(&apif).with_context(|| "parse transformed.json".to_string())?;
+
     let buf =
         std::fs::read(root.join("protos.fds")).with_context(|| "read protos.fds".to_string())?;
     let fds = FileDescriptorSet::decode(&*buf).unwrap(); // pulls in proto::Message
@@ -34,7 +41,15 @@ fn main() -> Result<()> {
     // FDS usage: https://github.com/tokio-rs/prost/blob/32bc87cd0b7301f6af1a338e9afd7717d0f42ca9/prost-build/src/lib.rs#L765-L825
     for f in fds.file {
         if let Some(pkg) = f.package {
-            info!("generating {}", pkg);
+            for m in f.message_type {
+                if let Some(name) = m.name {
+                    let path = format!("{}.{}", pkg, name);
+                    if let Some(resource) = resources.get(&path) {
+                        info!("generating resource {}", path);
+                        debug!("{:?}", resource);
+                    }
+                }
+            }
         }
     }