From 2fade1039c1842f08b30da5c95b5542b57e38ec6 Mon Sep 17 00:00:00 2001
From: xengineering <me@xengineering.eu>
Date: Mon, 3 Jul 2023 22:17:15 +0200
Subject: Move xml.Encoder to session struct

The encoderDecoder sub-struct of the session struct should be removed in
little steps. This is the first one.
---
 xmpp/session.go | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'xmpp/session.go')

diff --git a/xmpp/session.go b/xmpp/session.go
index a43e4f4..b4a8fab 100644
--- a/xmpp/session.go
+++ b/xmpp/session.go
@@ -4,6 +4,7 @@ import (
 	"crypto/tls"
 	"crypto/x509"
 	"encoding/xml"
+	"io"
 	"log"
 )
 
@@ -18,6 +19,7 @@ type session struct {
 	out         chan<- any
 	transport   *tls.Conn
 	ed          encoderDecoder
+	tx          *xml.Encoder
 	rx          chan xml.Token
 	resourceReq string
 }
@@ -49,6 +51,10 @@ func (s *session) run() {
 	go s.ed.run()
 	defer func() { s.ed.terminator <- true }()
 
+	lw := logger{"[TX] "}
+	w := io.MultiWriter(s.transport, lw)
+	s.tx = xml.NewEncoder(w)
+
 	s.out <- SessionConnect{}
 
 	runStreamPair(s)
-- 
cgit v1.2.3-70-g09d2