From 1d46cedcbb92edf69f080bf3dca07509a14159e2 Mon Sep 17 00:00:00 2001
From: xengineering <me@xengineering.eu>
Date: Sun, 21 May 2023 13:12:23 +0200
Subject: Introduce xengineering.eu/xmpp.Conn

---
 xmpp/xmpp.go | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

(limited to 'xmpp')

diff --git a/xmpp/xmpp.go b/xmpp/xmpp.go
index 5193f02..1d00739 100644
--- a/xmpp/xmpp.go
+++ b/xmpp/xmpp.go
@@ -15,8 +15,19 @@ const (
 	ShouldDisconnectEvent
 )
 
-func Run(ch chan Event, jid string, pwd string) {
-	conn, err := setupConn(jid)
+type Conn struct {
+	ch chan Event
+	jid, pwd string
+}
+
+func NewConn(ch chan Event, jid string, pwd string) *Conn {
+	c := Conn{ch, jid, pwd}
+
+	return &c
+}
+
+func (c *Conn) Run() {
+	conn, err := setupConn(c.jid)
 	if err != nil {
 		log.Print(err)
 		return
@@ -32,15 +43,15 @@ func Run(ch chan Event, jid string, pwd string) {
 
 	tr := newTokenRouter(&enc)
 
-	end := sendStreamStart(&enc, jid)
+	end := sendStreamStart(&enc, c.jid)
 	defer sendStreamEnd(&enc, end)
 
-	ch <- ConnectEvent
-	defer func() { ch <- DisconnectEvent }()
+	c.ch <- ConnectEvent
+	defer func() { c.ch <- DisconnectEvent }()
 
 	for {
 		select {
-		case ev := <-ch:
+		case ev := <-c.ch:
 			switch ev {
 			case ShouldDisconnectEvent:
 				return
-- 
cgit v1.2.3-70-g09d2