refactor: restructure session initialization to avoid circular references

session/lifecycle/lifecycle.go

@@ -0,0 +1,124 @@
+package lifecycle
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"log"
+	"net"
+	"time"
+	"tunnel_pls/session/slug"
+	"tunnel_pls/types"
+
+	"golang.org/x/crypto/ssh"
+)
+
+type Interaction interface {
+	SendMessage(string)
+}
+
+type Forwarder interface {
+	Close() error
+	GetTunnelType() types.TunnelType
+}
+
+type Lifecycle struct {
+	Status  types.Status
+	Conn    ssh.Conn
+	Channel ssh.Channel
+
+	Interaction Interaction
+	Forwarder   Forwarder
+	SlugManager slug.Manager
+}
+
+type SessionLifecycle interface {
+	Close() error
+	WaitForRunningStatus()
+	SetStatus(status types.Status)
+	GetConnection() ssh.Conn
+	GetChannel() ssh.Channel
+	SetChannel(channel ssh.Channel)
+}
+
+func (l *Lifecycle) GetChannel() ssh.Channel {
+	return l.Channel
+}
+
+func (l *Lifecycle) SetChannel(channel ssh.Channel) {
+	l.Channel = channel
+}
+func (l *Lifecycle) GetConnection() ssh.Conn {
+	return l.Conn
+}
+func (l *Lifecycle) SetStatus(status types.Status) {
+	l.Status = status
+}
+func (l *Lifecycle) WaitForRunningStatus() {
+	timeout := time.After(3 * time.Second)
+	ticker := time.NewTicker(150 * time.Millisecond)
+	defer ticker.Stop()
+	frames := []string{"-", "\\", "|", "/"}
+	i := 0
+	for {
+		select {
+		case <-ticker.C:
+			l.Interaction.SendMessage(fmt.Sprintf("\rLoading %s", frames[i]))
+			i = (i + 1) % len(frames)
+			if l.Status == types.RUNNING {
+				l.Interaction.SendMessage("\r\033[K")
+				return
+			}
+		case <-timeout:
+			l.Interaction.SendMessage("\r\033[K")
+			l.Interaction.SendMessage("TCP/IP request not received in time.\r\nCheck your internet connection and confirm the server responds within 3000ms.\r\nEnsure you ran the correct command. For more details, visit https://tunnl.live.\r\n\r\n")
+			err := l.Close()
+			if err != nil {
+				log.Printf("failed to close session: %v", err)
+			}
+			log.Println("Timeout waiting for session to start running")
+			return
+		}
+	}
+}
+
+func (l *Lifecycle) Close() error {
+	err := l.Forwarder.Close()
+	if err != nil {
+		return err
+	}
+	//if s.Forwarder.Listener != nil {
+	//	err := s.Forwarder.Listener.Close()
+	//	if err != nil && !errors.Is(err, net.ErrClosed) {
+	//		return err
+	//	}
+	//}
+
+	if l.Channel != nil {
+		err := l.Channel.Close()
+		if err != nil && !errors.Is(err, io.EOF) {
+			return err
+		}
+	}
+
+	if l.Conn != nil {
+		err := l.Conn.Close()
+		if err != nil && !errors.Is(err, net.ErrClosed) {
+			return err
+		}
+	}
+
+	//clientSlug := l.SlugManager.Get()
+	//if clientSlug != "" {
+	//	unregisterClient(clientSlug)
+	//}
+
+	//if l.Forwarder.GetType() == "TCP" && s.Forwarder.Listener != nil {
+	//	err := portUtil.Manager.SetPortStatus(s.Forwarder.ForwardedPort, false)
+	//	if err != nil {
+	//		return err
+	//	}
+	//}
+
+	return nil
+}