fix(proto-compiler): std and no-std generated code conflict

proto-compiler/Cargo.toml

@@ -10,6 +10,7 @@ publish = false
 
 [dependencies]
 walkdir = { version = "2.3" }
+prost = { version = "0.12" }
 prost-build = { version = "0.12" }
 tempfile = { version = "3.2.0" }
 regex = { "version" = "1.7.1" }

proto-compiler/src/functions.rs

@@ -247,6 +247,7 @@
     tenderdash_lib_target: &Path,
     abci_ver: &str,
     td_ver: &str,
+    module_name: &str,
 ) {
     let mut file_names = WalkDir::new(prost_dir)
         .into_iter()
@@ -276,11 +277,7 @@
 
         let mut tab_count = parts.len();
 
-        let mut inner_content = format!(
-            "{}include!(\"prost/{}\");",
-            tab.repeat(tab_count),
-            file_name
-        );
+        let mut inner_content = format!("{}include!(\"./{}\");", tab.repeat(tab_count), file_name);
 
         for part in parts {
             tab_count -= 1;
@@ -304,13 +301,16 @@
     pub const ABCI_VERSION: &str = \"{}\";
     /// Version of Tenderdash server used to generate protobuf configs
     pub const TENDERDASH_VERSION: &str = \"{}\";
+    /// Name of module where generated files are stored; used to distinguish between std and no-std version
+    pub const TENDERDASH_MODULE_NAME: &str = \"{}\";
 }}
 ",
         content,
         crate::constants::TENDERDASH_REPO,
         tenderdash_commitish(),
         abci_ver,
         td_ver,
+        module_name,
     );
 
     let mut file =
@@ -348,10 +348,12 @@
 pub(crate) fn check_state(dir: &Path, commitish: &str) -> bool {
     let state_file = PathBuf::from(&dir).join("download.state");
 
-    match read_to_string(state_file) {
+    let expected = commitish.to_string();
+
+    match read_to_string(&state_file) {
         Ok(content) => {
-            println!("[info] => Detected Tenderdash version: {}.", content.trim());
-            content.eq(commitish)
+            println!("[info] => Detected Tenderdash version: {}.", content);
+            content == expected
         },
         Err(_) => false,
     }

proto-compiler/src/lib.rs

@@ -18,16 +18,15 @@ use crate::functions::{check_state, save_state};
 /// Checkouts tenderdash repository to ../target/tenderdash and generates
 /// Rust protobuf definitions in ../proto/src/prost/ and
 /// ../proto/src/tenderdash.rs
-pub fn proto_compile() {
+///
+/// # Arguments
+///
+/// * `module_name` - name of module to put generated files into
+pub fn proto_compile(module_name: &str) {
     let root = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
 
-    let tenderdash_lib_target = root
-        .join("..")
-        .join("proto")
-        .join("src")
-        .join("tenderdash.rs");
-
-    let prost_out_dir = root.join("..").join("proto").join("src").join("prost");
+    let prost_out_dir = root.join("..").join("proto").join("src").join(module_name);
+    let tenderdash_lib_target = prost_out_dir.join("mod.rs");
 
     let out_dir = var("OUT_DIR")
         .map(PathBuf::from)
@@ -121,7 +120,13 @@ pub fn proto_compile() {
     println!("[info] => Removing old structs and copying new structs.");
     copy_files(&out_dir, &prost_out_dir); // This panics if it fails.
 
-    generate_tenderdash_lib(&out_dir, &tenderdash_lib_target, &abci_ver, &tenderdash_ver);
+    generate_tenderdash_lib(
+        &out_dir,
+        &tenderdash_lib_target,
+        &abci_ver,
+        &tenderdash_ver,
+        module_name,
+    );
 
     save_state(&prost_out_dir, &commitish);
     println!("[info] => Done!");

proto/build.rs

@@ -11,7 +11,10 @@ fn main() {
         env::set_var("TENDERDASH_COMMITISH", DEFAULT_VERSION);
     }
 
-    tenderdash_proto_compiler::proto_compile();
+    #[cfg(feature = "std")]
+    tenderdash_proto_compiler::proto_compile("tenderdash_std");
+    #[cfg(not(feature = "std"))]
+    tenderdash_proto_compiler::proto_compile("tenderdash_nostd");
 
     println!("cargo:rerun-if-changed=../proto-compiler/src");
     println!("cargo:rerun-if-changed=Cargo.toml");

proto/src/.gitignore

@@ -1,2 +1,6 @@
+tenderdash_nostd/
+tenderdash_std/
+
+# prost/ and tenderdash.rs are deprecated and can be removed in the future
 prost/
 tenderdash.rs

proto/src/lib.rs

@@ -20,8 +20,6 @@ pub mod google {
 }
 
 mod error;
-#[allow(warnings)]
-mod tenderdash;
 
 #[cfg(not(feature = "std"))]
 use core::{
@@ -34,12 +32,22 @@ use std::fmt::Display;
 use bytes::{Buf, BufMut};
 pub use error::Error;
 use prost::{encoding::encoded_len_varint, Message};
-pub use tenderdash::*;
+#[cfg(not(feature = "std"))]
+pub mod tenderdash_nostd;
+#[allow(warnings)]
+#[cfg(not(feature = "std"))]
+pub use tenderdash_nostd::*;
+
+#[cfg(feature = "std")]
+#[allow(warnings)]
+pub mod tenderdash_std;
+#[cfg(feature = "std")]
+pub use tenderdash_std::*;
 
 pub mod serializers;
 
+pub use meta::ABCI_VERSION;
 use prelude::*;
-pub use tenderdash::meta::ABCI_VERSION;
 #[cfg(feature = "grpc")]
 pub use tonic;