From 273963c6ee47469032f2e74803daf0b40f396ce6 Mon Sep 17 00:00:00 2001
From: Runxi Yu
Date: Sun, 8 Mar 2026 01:24:48 +0800
Subject: protocol/v0v1/server: Add PrimaryDataWriter

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

diff --git a/protocol/v0v1/server/session.go b/protocol/v0v1/server/session.go
index b27ad8cd..75884806 100644
--- a/protocol/v0v1/server/session.go
+++ b/protocol/v0v1/server/session.go
@@ -108,3 +108,15 @@ func (session *Session) ErrorWriter() io.Writer {
 
 	return sideband64k.NewChunkWriter(session.sideband, sideband64k.BandError)
 }
+
+// PrimaryDataWriter returns one chunking writer for primary output bytes.
+//
+// When side-band-64k is enabled, writes are chunked into band-1 sideband
+// frames. Otherwise writes are chunked into direct pkt-line data frames.
+func (session *Session) PrimaryDataWriter() io.Writer {
+	if session.useSideBand {
+		return sideband64k.NewChunkWriter(session.sideband, sideband64k.BandData)
+	}
+
+	return pktline.NewChunkWriter(session.enc)
+}
-- 
cgit v1.3.1-10-gc9f91