Handling both EOF and EOL cases

Ackee-Blockchain · Aug 16, 2024 · 41066bd · 41066bd
1 parent bb4291e
commit 41066bd
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/crates/client/src/test_generator.rs b/crates/client/src/test_generator.rs
@@ -743,10 +743,20 @@ impl TestGenerator {
                     return;
                 }
             }
+            // Check if the file ends with a newline
+            let mut file = File::open(&gitignore_path)?;
+            let mut buf = [0; 1];
+            file.seek(io::SeekFrom::End(-1))?;
+            file.read_exact(&mut buf)?;
+
             let file = OpenOptions::new().append(true).open(gitignore_path);
 
             if let Ok(mut file) = file {
-                writeln!(file, "\n{}", ignored_path)?;
+                if buf[0] == b'\n' {
+                    writeln!(file, "{}", ignored_path)?;
+                } else {
+                    writeln!(file, "\n{}", ignored_path)?;
+                }
                 println!("{FINISH} [{GIT_IGNORE}] update with [{ignored_path}]");
             }
         } else {