attempts to make servicegenerator work

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

Author: clux

build.rs

@@ -4,8 +4,8 @@ use std::rc::Rc;
 #[derive(Default)]
 struct GeneratorState {
     service_names: Vec<String>,
-    package_names: Vec<String>,
     finalized: usize,
+    generated: usize
 }
 
 struct KubeGenerator {
@@ -20,23 +20,19 @@ impl KubeGenerator {
 }
 
 impl prost_build::ServiceGenerator for KubeGenerator {
-    fn generate(&mut self, service: prost_build::Service, _buf: &mut String) {
+    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: generate generics from self.data here
+        let generics = format!("// TODO: generics\n");
+        buf.push_str(&generics);
     }
 
-    fn finalize(&mut self, _buf: &mut String) {
+    fn finalize(&mut self, buf: &mut String) {
         let mut state = self.state.borrow_mut();
         state.finalized += 1;
     }
-
-    fn finalize_package(&mut self, package: &str, buf: &mut String) {
-        let mut state = self.state.borrow_mut();
-        state.package_names.push(package.to_string());
-        // TODO: generate generics for pkg here using self.data
-        let pkg_generics = format!("// blahtest");
-        buf.push_str(&pkg_generics);
-    }
 }
 
 fn main() -> std::io::Result<()> {
@@ -51,8 +47,8 @@ fn main() -> std::io::Result<()> {
 
     // sanity
     let state = state.borrow();
-    //assert_eq!(state.service_names.len(), protos.len()); zero atm..
     assert_eq!(state.finalized, protos.len());
+    //assert_eq!(state.generated, protos.len()); // TODO: why does generate not trigger
 
     // 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.