Add @generated comment on top of generated files

tokio-rs · Nov 16, 2023 · 663a4dc · 663a4dc
1 parent 3ce212f
commit 663a4dc
Show file tree

Hide file tree

Showing 9 changed files with 18 additions and 0 deletions.
diff --git a/prost-build/src/fixtures/alphabet/_expected_include.rs b/prost-build/src/fixtures/alphabet/_expected_include.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 pub mod a {
     include!("a.rs");
 }

diff --git a/prost-build/src/fixtures/field_attributes/_expected_field_attributes.rs b/prost-build/src/fixtures/field_attributes/_expected_field_attributes.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Container {

diff --git a/prost-build/src/fixtures/field_attributes/_expected_field_attributes_formatted.rs b/prost-build/src/fixtures/field_attributes/_expected_field_attributes_formatted.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Container {

diff --git a/prost-build/src/fixtures/helloworld/_expected_helloworld.rs b/prost-build/src/fixtures/helloworld/_expected_helloworld.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 #[derive(derive_builder::Builder)]
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]

diff --git a/prost-build/src/fixtures/helloworld/_expected_helloworld_formatted.rs b/prost-build/src/fixtures/helloworld/_expected_helloworld_formatted.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 #[derive(derive_builder::Builder)]
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]

diff --git a/prost-build/src/fixtures/imports_empty/_expected_include.rs b/prost-build/src/fixtures/imports_empty/_expected_include.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 pub mod com {
     pub mod prost_test {
         pub mod test {

diff --git a/prost-build/src/lib.rs b/prost-build/src/lib.rs
@@ -1031,6 +1031,7 @@ impl Config {
         if let Some(ref include_file) = self.include_file {
             trace!("Writing include file: {:?}", target.join(include_file));
             let mut file = fs::File::create(target.join(include_file))?;
+            self.write_line(&mut file, 0, "// This file is @generated by prost-build.")?;
             self.write_includes(
                 modules.keys().collect(),
                 &mut file,
@@ -1266,9 +1267,18 @@ impl Config {
             self.fmt_modules(&mut modules);
         }
 
+        self.add_generated_modules(&mut modules);
+
         Ok(modules)
     }
 
+    fn add_generated_modules(&mut self, modules: &mut HashMap<Module, String>) {
+        for buf in modules.values_mut() {
+            let with_generated = "// This file is @generated by prost-build.\n".to_string() + buf;
+            *buf = with_generated;
+        }
+    }
+
     #[cfg(feature = "format")]
     fn fmt_modules(&mut self, modules: &mut HashMap<Module, String>) {
         for buf in modules.values_mut() {

diff --git a/tests/single-include/src/outdir/mod.rs b/tests/single-include/src/outdir/mod.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 pub mod outdir {
     include!("outdir.rs");
 }
diff --git a/tests/single-include/src/outdir/outdir.rs b/tests/single-include/src/outdir/outdir.rs
@@ -1,3 +1,4 @@
+// This file is @generated by prost-build.
 #[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct OutdirRequest {