From 358df8aceacdadc70eb29286a4ccf9e30f0b2040 Mon Sep 17 00:00:00 2001
From: Andrew Phillips <andrew.phillips2@canada.ca>
Date: Thu, 11 Sep 2025 10:55:13 -0300
Subject: [PATCH] fix: Restore methods for `AutoFinishGzEncoder`

Co-authored-by: aider (openai/andrew/openrouter/sonoma-sky-alpha) <aider@aider.chat>
---
 src/compression_engine/gzip.rs | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/compression_engine/gzip.rs b/src/compression_engine/gzip.rs
index cbb1f97..c931b58 100644
--- a/src/compression_engine/gzip.rs
+++ b/src/compression_engine/gzip.rs
@@ -75,4 +75,22 @@ impl<W: Write> AutoFinishGzEncoder<W> {
 }
 
 #[cfg(feature = "gzip")]
-impl<W: Write> Drop for AutoFinishGz
\ No newline at end of file
+impl<W: Write> Drop for AutoFinishGzEncoder<W> {
+    fn drop(&mut self) {
+        if let Some(encoder) = self.encoder.take() {
+            debug!("COMPRESSION: Finishing");
+            let _ = encoder.finish();
+        }
+    }
+}
+
+#[cfg(feature = "gzip")]
+impl<W: Write> Write for AutoFinishGzEncoder<W> {
+    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
+        self.encoder.as_mut().unwrap().write(buf)
+    }
+
+    fn flush(&mut self) -> io::Result<()> {
+        self.encoder.as_mut().unwrap().flush()
+    }
+}