common/iowrap: Add io wrappers

4 files changed, 33 insertions, 0 deletions

diff --git a/common/doc.go b/common/doc.go
new file mode 100644
index 00000000..1f685411
--- /dev/null
+++ b/common/doc.go
@@ -0,0 +1,2 @@
+// Package common encapsulates various helper packages not directly related to Git.
+package common
diff --git a/common/iowrap/doc.go b/common/iowrap/doc.go
new file mode 100644
index 00000000..8e25db16
--- /dev/null
+++ b/common/iowrap/doc.go
@@ -0,0 +1,2 @@
+// Package iowrap provides small public I/O wrapper interfaces and adapters.
+package iowrap
diff --git a/common/iowrap/nop_flush.go b/common/iowrap/nop_flush.go
new file mode 100644
index 00000000..38fc5f55
--- /dev/null
+++ b/common/iowrap/nop_flush.go
@@ -0,0 +1,20 @@
+package iowrap
+
+import "io"
+
+type nopFlusher struct {
+	io.Writer
+}
+
+// NopFlush adapts writer into a WriteFlusher with a no-op Flush.
+func NopFlush(writer io.Writer) WriteFlusher {
+	if writer == nil {
+		return nil
+	}
+
+	return nopFlusher{Writer: writer}
+}
+
+func (nopFlusher) Flush() error {
+	return nil
+}
diff --git a/common/iowrap/write_flusher.go b/common/iowrap/write_flusher.go
new file mode 100644
index 00000000..aaac8724
--- /dev/null
+++ b/common/iowrap/write_flusher.go
@@ -0,0 +1,9 @@
+package iowrap
+
+import "io"
+
+// WriteFlusher writes bytes and flushes buffered output state.
+type WriteFlusher interface {
+	io.Writer
+	Flush() error
+}