Wrap os.Stdout in CountingWriter

Author: ashmckenzie

cmd/check/main.go

@@ -23,7 +23,7 @@ func main() {
 	command.CheckForVersionFlag(os.Args, Version, BuildTime)
 
 	readWriter := &readwriter.ReadWriter{
-		Out:    os.Stdout,
+		Out:    &readwriter.CountingWriter{W: os.Stdout},
 		In:     os.Stdin,
 		ErrOut: os.Stderr,
 	}

cmd/gitlab-shell-authorized-keys-check/main.go

@@ -24,7 +24,7 @@ func main() {
 	command.CheckForVersionFlag(os.Args, Version, BuildTime)
 
 	readWriter := &readwriter.ReadWriter{
-		Out:    os.Stdout,
+		Out:    &readwriter.CountingWriter{W: os.Stdout},
 		In:     os.Stdin,
 		ErrOut: os.Stderr,
 	}

cmd/gitlab-shell-authorized-principals-check/main.go

@@ -24,7 +24,7 @@ func main() {
 	command.CheckForVersionFlag(os.Args, Version, BuildTime)
 
 	readWriter := &readwriter.ReadWriter{
-		Out:    os.Stdout,
+		Out:    &readwriter.CountingWriter{W: os.Stdout},
 		In:     os.Stdin,
 		ErrOut: os.Stderr,
 	}

cmd/gitlab-shell/main.go

@@ -32,7 +32,7 @@ func main() {
 	command.CheckForVersionFlag(os.Args, Version, BuildTime)
 
 	readWriter := &readwriter.ReadWriter{
-		Out:    os.Stdout,
+		Out:    &readwriter.CountingWriter{W: os.Stdout},
 		In:     os.Stdin,
 		ErrOut: os.Stderr,
 	}

internal/sshd/session.go

@@ -168,7 +168,7 @@ func (s *session) handleShell(ctx context.Context, req *ssh.Request) (context.Co
 	}
 
 	rw := &readwriter.ReadWriter{
-		Out:    s.channel,
+		Out:    &readwriter.CountingWriter{W: s.channel},
 		In:     s.channel,
 		ErrOut: s.channel.Stderr(),
 	}