Add scope to file header

red-blox · Dec 19, 2023 · b3c8bbd · b3c8bbd
1 parent 0bc82cf
commit b3c8bbd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/zap/src/output/client.rs b/zap/src/output/client.rs
@@ -370,7 +370,7 @@ impl<'a> ClientOutput<'a> {
 	}
 
 	pub fn output(mut self) -> String {
-		self.push_file_header();
+		self.push_file_header("Client");
 
 		self.push(include_str!("client.luau"));
 

diff --git a/zap/src/output/mod.rs b/zap/src/output/mod.rs
@@ -147,12 +147,12 @@ pub trait Output {
 		}
 	}
 
-	fn push_file_header(&mut self) {
+	fn push_file_header(&mut self, scope: &str) {
 		self.push_line("--!native");
 		self.push_line("--!optimize 2");
 
 		self.push_line(&format!(
-			"-- Generated by Zap v{} (https://github.com/red-blox/zap)",
+			"-- {scope} generated by Zap v{} (https://github.com/red-blox/zap)",
 			env!("CARGO_PKG_VERSION")
 		));
 

diff --git a/zap/src/output/server.rs b/zap/src/output/server.rs
@@ -529,7 +529,7 @@ impl<'a> ServerOutput<'a> {
 	}
 
 	pub fn output(mut self) -> String {
-		self.push_file_header();
+		self.push_file_header("Server");
 
 		self.push(include_str!("server.luau"));