drop servicegenerater and use fds to inject into files

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

Author: clux

.gitignore

@@ -1,2 +1,3 @@
 /target
 Cargo.lock
+protos.fds

Cargo.toml

@@ -7,7 +7,8 @@ edition = "2018"
 [dependencies]
 bytes = "1.0.1"
 prost = "0.8.0"
-prost-types = "0.8"
 
 [build-dependencies]
 prost-build = "0.8.0"
+prost-types = "0.8.0"
+prost = "0.8.0"

README.md

@@ -1,58 +1,33 @@
-use std::cell::RefCell;
-use std::rc::Rc;
-
-#[derive(Default)]
-struct GeneratorState {
-    service_names: Vec<String>,
-    finalized: usize,
-    generated: usize
-}
-
-struct KubeGenerator {
-    data: String,
-    state: Rc<RefCell<GeneratorState>>,
-}
-impl KubeGenerator {
-    fn new(state: Rc<RefCell<GeneratorState>>) -> Self {
-        let data = std::fs::read_to_string("./openapi/api-resources.json").unwrap();
-        Self { data, state }
-    }
-}
-
-impl prost_build::ServiceGenerator for KubeGenerator {
-    fn generate(&mut self, service: prost_build::Service, buf: &mut String) {
-        let mut state = self.state.borrow_mut();
-        state.service_names.push(service.name);
-        state.generated += 1;
-        // TODO: THIS doesn't work? never called by prost_build, bug?
-        let generics = format!("// TODO: generate\n");
-        buf.push_str(&generics);
-    }
-
-    fn finalize(&mut self, buf: &mut String) {
-        let mut state = self.state.borrow_mut();
-        state.finalized += 1;
-        // NB: THIS works, but we need a name here before it's useful
-        //let generics = format!("// TODO: finalize\n");
-        //buf.push_str(&generics);
-    }
-}
+use prost_types::{FileDescriptorProto, FileDescriptorSet};
+use prost::Message;
 
 fn main() -> std::io::Result<()> {
-    let protos: Vec<&str> = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/protos.list"))
+    let protos = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/protos.list"))
         .lines()
-        .collect();
+        .collect::<Vec<&str>>();
 
-    let state = Rc::new(RefCell::new(GeneratorState::default()));
     prost_build::Config::new()
-        .service_generator(Box::new(KubeGenerator::new(Rc::clone(&state))))
+        // should probably switch to this
+        //.btree_map(&["."])
         .out_dir("./out")
         .compile_protos(protos.as_slice(), &["protos/"])?;
 
-    // sanity
-    let state = state.borrow();
-    assert_eq!(state.finalized, protos.len());
-    assert_eq!(state.generated, protos.len()); // TODO: why does generate not trigger
+    let apis = std::fs::read_to_string("./openapi/api-resources.json")?;
+
+    let buf = std::fs::read("./protos.fds").unwrap();
+    let fds = FileDescriptorSet::decode(&*buf).unwrap(); // pulls in proto::Message
+
+    // NB: FDS fields: https://github.com/tokio-rs/prost/blob/32bc87cd0b7301f6af1a338e9afd7717d0f42ca9/prost-types/src/protobuf.rs#L1-L7
+    // FDS usage: https://github.com/tokio-rs/prost/blob/32bc87cd0b7301f6af1a338e9afd7717d0f42ca9/prost-build/src/lib.rs#L765-L825
+    for f in fds.file {
+        use std::io::Write;
+        if let Some(pkg) = f.package {
+            let pkgpath = std::path::Path::new("./out").join(format!("{}.rs", pkg));
+            let generics = format!("// TODO genericsfor {}\n", pkg);
+            let mut file = std::fs::OpenOptions::new().write(true).append(true).open(&pkgpath)?;
+            file.write(generics.as_bytes())?;
+        }
+    }
 
     // Generate code in `src/` by reading files in `OUT_DIR`?
     // Need to create `mod.rs` file for each level based on the filename, and write generated code in correct file.

build.rs

@@ -61,8 +61,20 @@ swagger-transform:
 # Download and generate all swagger dependent files
 swagger: swagger-dl swagger-patch swagger-transform
 
+# Build a FileDescriptorSet for custom code generation
+build-fds:
+    #!/usr/bin/env bash
+    set -exuo pipefail
+    shopt -s globstar
+    protoc \
+        --include_imports \
+        --include_source_info \
+        --descriptor_set_out=protos.fds \
+        --proto_path=./protos \
+        ./protos/**/*.proto
+
 # Generate the library code from completed swagger and protos
-build:
+build: build-fds
     #!/usr/bin/env bash
     set -exuo pipefail
     rm -rf out/ && mkdir out

justfile

@@ -137,3 +137,4 @@ pub struct AdmissionReview {
     #[prost(message, optional, tag="2")]
     pub response: ::core::option::Option<AdmissionResponse>,
 }
+// TODO genericsfor api.admission.v1

out/api.admission.v1.rs

@@ -137,3 +137,4 @@ pub struct AdmissionReview {
     #[prost(message, optional, tag="2")]
     pub response: ::core::option::Option<AdmissionResponse>,
 }
+// TODO genericsfor api.admission.v1beta1

out/api.admission.v1beta1.rs

@@ -457,3 +457,4 @@ pub struct WebhookClientConfig {
     #[prost(bytes="vec", optional, tag="2")]
     pub ca_bundle: ::core::option::Option<::prost::alloc::vec::Vec<u8>>,
 }
+// TODO genericsfor api.admissionregistration.v1

out/api.admissionregistration.v1.rs

@@ -465,3 +465,4 @@ pub struct WebhookClientConfig {
     #[prost(bytes="vec", optional, tag="2")]
     pub ca_bundle: ::core::option::Option<::prost::alloc::vec::Vec<u8>>,
 }
+// TODO genericsfor api.admissionregistration.v1beta1

out/api.admissionregistration.v1beta1.rs

@@ -97,3 +97,4 @@ pub struct StorageVersionStatus {
     #[prost(message, repeated, tag="3")]
     pub conditions: ::prost::alloc::vec::Vec<StorageVersionCondition>,
 }
+// TODO genericsfor api.apiserverinternal.v1alpha1

out/api.apiserverinternal.v1alpha1.rs

@@ -737,3 +737,4 @@ pub struct StatefulSetUpdateStrategy {
     #[prost(message, optional, tag="2")]
     pub rolling_update: ::core::option::Option<RollingUpdateStatefulSetStrategy>,
 }
+// TODO genericsfor api.apps.v1

out/api.apps.v1.rs

@@ -484,3 +484,4 @@ pub struct StatefulSetUpdateStrategy {
     #[prost(message, optional, tag="2")]
     pub rolling_update: ::core::option::Option<RollingUpdateStatefulSetStrategy>,
 }
+// TODO genericsfor api.apps.v1beta1

out/api.apps.v1beta1.rs

@@ -785,3 +785,4 @@ pub struct StatefulSetUpdateStrategy {
     #[prost(message, optional, tag="2")]
     pub rolling_update: ::core::option::Option<RollingUpdateStatefulSetStrategy>,
 }
+// TODO genericsfor api.apps.v1beta2

out/api.apps.v1beta2.rs

@@ -165,3 +165,4 @@ pub struct UserInfo {
     #[prost(map="string, message", tag="4")]
     pub extra: ::std::collections::HashMap<::prost::alloc::string::String, ExtraValue>,
 }
+// TODO genericsfor api.authentication.v1

out/api.authentication.v1.rs

@@ -93,3 +93,4 @@ pub struct UserInfo {
     #[prost(map="string, message", tag="4")]
     pub extra: ::std::collections::HashMap<::prost::alloc::string::String, ExtraValue>,
 }
+// TODO genericsfor api.authentication.v1beta1

out/api.authentication.v1beta1.rs

@@ -263,3 +263,4 @@ pub struct SubjectRulesReviewStatus {
     #[prost(string, optional, tag="4")]
     pub evaluation_error: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.authorization.v1

out/api.authorization.v1.rs

@@ -263,3 +263,4 @@ pub struct SubjectRulesReviewStatus {
     #[prost(string, optional, tag="4")]
     pub evaluation_error: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.authorization.v1beta1

out/api.authorization.v1beta1.rs

@@ -482,3 +482,4 @@ pub struct ScaleStatus {
     #[prost(string, optional, tag="2")]
     pub selector: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.autoscaling.v1

out/api.autoscaling.v1.rs

@@ -459,3 +459,4 @@ pub struct ResourceMetricStatus {
     #[prost(message, optional, tag="3")]
     pub current_average_value: ::core::option::Option<super::super::super::apimachinery::pkg::api::resource::Quantity>,
 }
+// TODO genericsfor api.autoscaling.v2beta1

out/api.autoscaling.v2beta1.rs

@@ -483,3 +483,4 @@ pub struct ResourceMetricStatus {
     #[prost(message, optional, tag="2")]
     pub current: ::core::option::Option<MetricValueStatus>,
 }
+// TODO genericsfor api.autoscaling.v2beta2

out/api.autoscaling.v2beta2.rs

@@ -348,3 +348,4 @@ pub struct UncountedTerminatedPods {
     #[prost(string, repeated, tag="2")]
     pub failed: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.batch.v1

out/api.batch.v1.rs

@@ -114,3 +114,4 @@ pub struct JobTemplateSpec {
     #[prost(message, optional, tag="2")]
     pub spec: ::core::option::Option<super::v1::JobSpec>,
 }
+// TODO genericsfor api.batch.v1beta1

out/api.batch.v1beta1.rs

@@ -223,3 +223,4 @@ pub struct ExtraValue {
     #[prost(string, repeated, tag="1")]
     pub items: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.certificates.v1

out/api.certificates.v1.rs

@@ -174,3 +174,4 @@ pub struct ExtraValue {
     #[prost(string, repeated, tag="1")]
     pub items: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.certificates.v1beta1

out/api.certificates.v1beta1.rs

@@ -51,3 +51,4 @@ pub struct LeaseSpec {
     #[prost(int32, optional, tag="5")]
     pub lease_transitions: ::core::option::Option<i32>,
 }
+// TODO genericsfor api.coordination.v1

out/api.coordination.v1.rs

@@ -51,3 +51,4 @@ pub struct LeaseSpec {
     #[prost(int32, optional, tag="5")]
     pub lease_transitions: ::core::option::Option<i32>,
 }
+// TODO genericsfor api.coordination.v1beta1

out/api.coordination.v1beta1.rs

@@ -5824,3 +5824,4 @@ pub struct WindowsSecurityContextOptions {
     #[prost(bool, optional, tag="4")]
     pub host_process: ::core::option::Option<bool>,
 }
+// TODO genericsfor api.core.v1

out/api.core.v1.rs

@@ -171,3 +171,4 @@ pub struct ForZone {
     #[prost(string, optional, tag="1")]
     pub name: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.discovery.v1

out/api.discovery.v1.rs

@@ -174,3 +174,4 @@ pub struct ForZone {
     #[prost(string, optional, tag="1")]
     pub name: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.discovery.v1beta1

out/api.discovery.v1beta1.rs

@@ -98,3 +98,4 @@ pub struct EventSeries {
     #[prost(message, optional, tag="2")]
     pub last_observed_time: ::core::option::Option<super::super::super::apimachinery::pkg::apis::meta::v1::MicroTime>,
 }
+// TODO genericsfor api.events.v1

out/api.events.v1.rs

@@ -100,3 +100,4 @@ pub struct EventSeries {
     #[prost(message, optional, tag="2")]
     pub last_observed_time: ::core::option::Option<super::super::super::apimachinery::pkg::apis::meta::v1::MicroTime>,
 }
+// TODO genericsfor api.events.v1beta1

out/api.events.v1beta1.rs

@@ -1295,3 +1295,4 @@ pub struct SupplementalGroupsStrategyOptions {
     #[prost(message, repeated, tag="2")]
     pub ranges: ::prost::alloc::vec::Vec<IdRange>,
 }
+// TODO genericsfor api.extensions.v1beta1

out/api.extensions.v1beta1.rs

@@ -428,3 +428,4 @@ pub struct UserSubject {
     #[prost(string, optional, tag="1")]
     pub name: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.flowcontrol.v1alpha1

out/api.flowcontrol.v1alpha1.rs

@@ -428,3 +428,4 @@ pub struct UserSubject {
     #[prost(string, optional, tag="1")]
     pub name: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.flowcontrol.v1beta1

out/api.flowcontrol.v1beta1.rs

@@ -60,3 +60,4 @@ pub struct ImageReviewStatus {
     #[prost(map="string, string", tag="3")]
     pub audit_annotations: ::std::collections::HashMap<::prost::alloc::string::String, ::prost::alloc::string::String>,
 }
+// TODO genericsfor api.imagepolicy.v1alpha1

out/api.imagepolicy.v1alpha1.rs

@@ -494,3 +494,4 @@ pub struct ServiceBackendPort {
     #[prost(int32, optional, tag="2")]
     pub number: ::core::option::Option<i32>,
 }
+// TODO genericsfor api.networking.v1

out/api.networking.v1.rs

@@ -285,3 +285,4 @@ pub struct IngressTls {
     #[prost(string, optional, tag="2")]
     pub secret_name: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.networking.v1beta1

out/api.networking.v1beta1.rs

@@ -80,3 +80,4 @@ pub struct Scheduling {
     #[prost(message, repeated, tag="2")]
     pub tolerations: ::prost::alloc::vec::Vec<super::super::core::v1::Toleration>,
 }
+// TODO genericsfor api.node.v1

out/api.node.v1.rs

@@ -90,3 +90,4 @@ pub struct Scheduling {
     #[prost(message, repeated, tag="2")]
     pub tolerations: ::prost::alloc::vec::Vec<super::super::core::v1::Toleration>,
 }
+// TODO genericsfor api.node.v1alpha1

out/api.node.v1alpha1.rs

@@ -79,3 +79,4 @@ pub struct Scheduling {
     #[prost(message, repeated, tag="2")]
     pub tolerations: ::prost::alloc::vec::Vec<super::super::core::v1::Toleration>,
 }
+// TODO genericsfor api.node.v1beta1

out/api.node.v1beta1.rs

@@ -123,3 +123,4 @@ pub struct PodDisruptionBudgetStatus {
     #[prost(message, repeated, tag="7")]
     pub conditions: ::prost::alloc::vec::Vec<super::super::super::apimachinery::pkg::apis::meta::v1::Condition>,
 }
+// TODO genericsfor api.policy.v1

out/api.policy.v1.rs

@@ -415,3 +415,4 @@ pub struct SupplementalGroupsStrategyOptions {
     #[prost(message, repeated, tag="2")]
     pub ranges: ::prost::alloc::vec::Vec<IdRange>,
 }
+// TODO genericsfor api.policy.v1beta1

out/api.policy.v1beta1.rs

@@ -181,3 +181,4 @@ pub struct Subject {
     #[prost(string, optional, tag="4")]
     pub namespace: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.rbac.v1

out/api.rbac.v1.rs

@@ -188,3 +188,4 @@ pub struct Subject {
     #[prost(string, optional, tag="4")]
     pub namespace: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.rbac.v1alpha1

out/api.rbac.v1alpha1.rs

@@ -188,3 +188,4 @@ pub struct Subject {
     #[prost(string, optional, tag="4")]
     pub namespace: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.rbac.v1beta1

out/api.rbac.v1beta1.rs

@@ -44,3 +44,4 @@ pub struct PriorityClassList {
     #[prost(message, repeated, tag="2")]
     pub items: ::prost::alloc::vec::Vec<PriorityClass>,
 }
+// TODO genericsfor api.scheduling.v1

out/api.scheduling.v1.rs

@@ -45,3 +45,4 @@ pub struct PriorityClassList {
     #[prost(message, repeated, tag="2")]
     pub items: ::prost::alloc::vec::Vec<PriorityClass>,
 }
+// TODO genericsfor api.scheduling.v1alpha1

out/api.scheduling.v1alpha1.rs

@@ -45,3 +45,4 @@ pub struct PriorityClassList {
     #[prost(message, repeated, tag="2")]
     pub items: ::prost::alloc::vec::Vec<PriorityClass>,
 }
+// TODO genericsfor api.scheduling.v1beta1

out/api.scheduling.v1beta1.rs

@@ -446,3 +446,4 @@ pub struct VolumeNodeResources {
     #[prost(int32, optional, tag="1")]
     pub count: ::core::option::Option<i32>,
 }
+// TODO genericsfor api.storage.v1

out/api.storage.v1.rs

@@ -201,3 +201,4 @@ pub struct VolumeError {
     #[prost(string, optional, tag="2")]
     pub message: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor api.storage.v1alpha1

out/api.storage.v1alpha1.rs

@@ -540,3 +540,4 @@ pub struct VolumeNodeResources {
     #[prost(int32, optional, tag="1")]
     pub count: ::core::option::Option<i32>,
 }
+// TODO genericsfor api.storage.v1beta1

out/api.storage.v1beta1.rs

@@ -649,3 +649,4 @@ pub struct WebhookConversion {
     #[prost(string, repeated, tag="3")]
     pub conversion_review_versions: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
 }
+// TODO genericsfor apiextensions_apiserver.pkg.apis.apiextensions.v1

out/apiextensions_apiserver.pkg.apis.apiextensions.v1.rs

@@ -685,3 +685,4 @@ pub struct WebhookClientConfig {
     #[prost(bytes="vec", optional, tag="2")]
     pub ca_bundle: ::core::option::Option<::prost::alloc::vec::Vec<u8>>,
 }
+// TODO genericsfor apiextensions_apiserver.pkg.apis.apiextensions.v1beta1

out/apiextensions_apiserver.pkg.apis.apiextensions.v1beta1.rs

@@ -61,3 +61,4 @@ pub struct Quantity {
     #[prost(string, optional, tag="1")]
     pub string: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor apimachinery.pkg.api.resource

out/apimachinery.pkg.api.resource.rs

@@ -1137,3 +1137,4 @@ pub struct WatchEvent {
     #[prost(message, optional, tag="2")]
     pub object: ::core::option::Option<super::super::super::runtime::RawExtension>,
 }
+// TODO genericsfor apimachinery.pkg.apis.meta.v1

out/apimachinery.pkg.apis.meta.v1.rs

@@ -11,3 +11,4 @@ pub struct PartialObjectMetadataList {
     #[prost(message, repeated, tag="1")]
     pub items: ::prost::alloc::vec::Vec<super::v1::PartialObjectMetadata>,
 }
+// TODO genericsfor apimachinery.pkg.apis.meta.v1beta1

out/apimachinery.pkg.apis.meta.v1beta1.rs

@@ -184,3 +184,4 @@ pub struct CarpStatus {
     #[prost(message, optional, tag="7")]
     pub start_time: ::core::option::Option<super::super::meta::v1::Time>,
 }
+// TODO genericsfor apimachinery.pkg.apis.testapigroup.v1

out/apimachinery.pkg.apis.testapigroup.v1.rs

@@ -102,3 +102,4 @@ pub struct Unknown {
     #[prost(string, optional, tag="4")]
     pub content_type: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor apimachinery.pkg.runtime

out/apimachinery.pkg.runtime.rs

@@ -0,0 +1 @@
+// TODO genericsfor apimachinery.pkg.runtime.schema

out/apimachinery.pkg.runtime.schema.rs

@@ -16,3 +16,4 @@ pub struct IntOrString {
     #[prost(string, optional, tag="3")]
     pub str_val: ::core::option::Option<::prost::alloc::string::String>,
 }
+// TODO genericsfor apimachinery.pkg.util.intstr