package tester

import (
	"crypto/tls"

	"dev.sum7.eu/genofire/yaja/client"
	"dev.sum7.eu/genofire/yaja/model"
)

type Status struct {
	client               *client.Client
	password             string
	JID                  *model.JID        `json:"jid"`
	Domain               string            `json:"domain"`
	Login                bool              `json:"is_online"`
	MessageForConnection map[string]string `json:"-"`
	Connections          map[string]bool   `json:"-"`
	TLSVersion           string            `json:"tls_version"`
}

func NewStatus(jid *model.JID, password string) *Status {
	return &Status{
		JID:                  jid,
		Domain:               jid.Domain,
		MessageForConnection: make(map[string]string),
		Connections:          make(map[string]bool),
	}
}

func (s *Status) Update() {
	if s.client == nil || !s.Login {
		s.Login = false
		s.TLSVersion = ""
		return
	}
	if tlsstate := s.client.TLSConnectionState(); tlsstate != nil {
		switch tlsstate.Version {
		case tls.VersionSSL30:
			s.TLSVersion = "SSL 3.0"
		case tls.VersionTLS10:
			s.TLSVersion = "TLS 1.0"
		case tls.VersionTLS11:
			s.TLSVersion = "TLS 1.1"
		case tls.VersionTLS12:
			s.TLSVersion = "TLS 1.2"
		default:
			s.TLSVersion = "unknown " + string(tlsstate.Version)
		}
	} else {
		s.TLSVersion = ""
	}
}