From 89d44601930f4a9452fd40c2262652b1a847a8cf Mon Sep 17 00:00:00 2001
From: xengineering <me@xengineering.eu>
Date: Sun, 4 Jun 2023 17:12:04 +0200
Subject: Re-implement XML encoding

---
 xmpp/encoder_decoder.go | 35 ++++++++++++++++++++++++++++++++++-
 xmpp/session.go         |  4 ++--
 2 files changed, 36 insertions(+), 3 deletions(-)

(limited to 'xmpp')

diff --git a/xmpp/encoder_decoder.go b/xmpp/encoder_decoder.go
index c2d0423..14113a6 100644
--- a/xmpp/encoder_decoder.go
+++ b/xmpp/encoder_decoder.go
@@ -1,11 +1,44 @@
 package xmpp
 
+import (
+	"encoding/xml"
+	"log"
+	"os"
+)
+
 type encoderDecoder struct{
 	session *session
+	ct *xml.Encoder  // client transport encoder
+	cl *xml.Encoder  // client log encoder
 }
 
 func newEncoderDecoder(s *session) encoderDecoder {
-	ed := encoderDecoder{s}
+	ed := encoderDecoder{}
+
+	ed.session = s
+
+	ed.ct = xml.NewEncoder(s.transport)
+	ed.ct.Indent("", "    ")
+
+	ed.cl = xml.NewEncoder(os.Stdout)
+	ed.cl.Indent("C: ", "    ")
 
 	return ed
 }
+
+func (ed *encoderDecoder) encodeToken(t xml.Token) error {
+	var err error
+	defer func() { if err != nil { log.Println(err) } }()
+
+	err = ed.ct.EncodeToken(t)
+	if err != nil { return err }
+	err = ed.ct.Flush()
+	if err != nil { return err }
+
+	err = ed.cl.EncodeToken(t)
+	if err != nil { return err }
+	err = ed.cl.Flush()
+	if err != nil { return err }
+
+	return nil
+}
diff --git a/xmpp/session.go b/xmpp/session.go
index 71b3acf..db9a079 100644
--- a/xmpp/session.go
+++ b/xmpp/session.go
@@ -24,8 +24,6 @@ func StartSession(out chan any, jid string, pwd string) chan any {
 	s.in = make(chan any)
 	s.out = out
 
-	s.ed = newEncoderDecoder(&s)
-
 	go s.run()
 
 	return s.in
@@ -38,6 +36,8 @@ func (s *session) run() {
 	if err != nil { return }
 	defer s.transport.Close()
 
+	s.ed = newEncoderDecoder(s)
+
 	s.out <- SessionConnect{}
 
 	for {
-- 
cgit v1.2.3-70-g09d2