package modules

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/admin-else/strom/event"
	"github.com/admin-else/strom/proto"
	"github.com/admin-else/strom/proto_base"

	"github.com/admin-else/strom/data"
	"github.com/admin-else/strom/proto_generated/v1_21_8"
)

type StatusClient struct {
	*proto.Conn
	Status string
}

func (s *StatusClient) Default(event any) (err error) {
	err = fmt.Errorf("unexpected event: %v", event)
	return
}

func (s *StatusClient) OnStart(_ event.OnStart) (err error) {
	parts := strings.Split(s.RemoteAddr().String(), ":")
	versionData, err := data.LookUpProtocolVersionByName(s.Version)
	if err != nil {
		return
	}
	port, err := strconv.Atoi(parts[1])
	if err != nil {
		return
	}
	err = s.Send(&v1_21_8.HandshakingToServerPacketSetProtocol{
		ProtocolVersion: int32(versionData.Version),
		ServerHost:      parts[0],
		ServerPort:      uint16(port),
		NextState:       int32(proto_base.Status),
	})
	if err != nil {
		return
	}
	s.SetState(proto_base.Status)
	err = s.Send(&v1_21_8.StatusToServerPacketPingStart{})
	return
}

func (s *StatusClient) OnStatus(p v1_21_8.StatusToClientPacketServerInfo) (err error) {
	s.Status = p.Response
	err = event.ErrHandlerDone
	return
}