From 4c56d5a12febec96e819c7a165e5098f2c693deb Mon Sep 17 00:00:00 2001
From: Runxi Yu
Date: Sat, 7 Mar 2026 21:55:01 +0800
Subject: protocol/v0v1/server: Add ProgessWriter and ErrorWriter

---
 protocol/v0v1/server/session.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'protocol')

diff --git a/protocol/v0v1/server/session.go b/protocol/v0v1/server/session.go
index 98ed199c..b27ad8cd 100644
--- a/protocol/v0v1/server/session.go
+++ b/protocol/v0v1/server/session.go
@@ -86,3 +86,25 @@ func (session *Session) WriteFlush() error {
 
 	return session.enc.WriteFlush()
 }
+
+// ProgressWriter returns one chunking writer for sideband progress output.
+//
+// When side-band-64k was not negotiated, writes are discarded.
+func (session *Session) ProgressWriter() io.Writer {
+	if !session.useSideBand {
+		return io.Discard
+	}
+
+	return sideband64k.NewChunkWriter(session.sideband, sideband64k.BandProgress)
+}
+
+// ErrorWriter returns one chunking writer for sideband error output.
+//
+// When side-band-64k was not negotiated, writes are discarded.
+func (session *Session) ErrorWriter() io.Writer {
+	if !session.useSideBand {
+		return io.Discard
+	}
+
+	return sideband64k.NewChunkWriter(session.sideband, sideband64k.BandError)
+}
-- 
cgit v1.3.1-10-gc9f91