package logingate

import (
	"context"
	"fmt"
	"io"
	"log"
	"net"
	"sync"

	"ananta-ps/internal/config"
	"ananta-ps/pkg/crypto/chacha8"
	"ananta-ps/pkg/protocol"
)

type Server struct {
	cfg        *config.Config
	listeners  []net.Listener
	cancelFunc context.CancelFunc
	wg         sync.WaitGroup
}

func New(cfg *config.Config) *Server {
	return &Server{
		cfg: cfg,
	}
}

func (s *Server) Start() error {
	ctx, cancel := context.WithCancel(context.Background())
	s.cancelFunc = cancel

	for _, port := range s.cfg.Network.LoginPorts {
		addr := fmt.Sprintf("%s:%d", s.cfg.Network.BindHost, port)
		ln, err := net.Listen("tcp", addr)
		if err != nil {
			s.Stop()
			return fmt.Errorf("failed to listen on login gate %s: %w", addr, err)
		}
		s.listeners = append(s.listeners, ln)
		log.Printf("[LOGIN-GATE] Listening on %s:%d", s.cfg.Network.AdvertisedHost, port)

		s.wg.Add(1)
		go s.acceptLoop(ctx, ln, port)
	}

	return nil
}

func (s *Server) Stop() error {
	if s.cancelFunc != nil {
		s.cancelFunc()
	}
	for _, ln := range s.listeners {
		_ = ln.Close()
	}
	s.wg.Wait()
	return nil
}

func (s *Server) acceptLoop(ctx context.Context, ln net.Listener, port int) {
	defer s.wg.Done()
	for {
		conn, err := ln.Accept()
		if err != nil {
			select {
			case <-ctx.Done():
				return
			default:
				log.Printf("[LOGIN-GATE] Accept error on port %d: %v", port, err)
				return
			}
		}

		go s.handleConnection(ctx, conn, port)
	}
}

type Session struct {
	conn       net.Conn
	readCipher *chacha8.UxChaCha8
	sendCipher *chacha8.UxChaCha8
	mu         sync.Mutex
}

func (sess *Session) SendPacket(packet *protocol.RpcPacket) error {
	sess.mu.Lock()
	defer sess.mu.Unlock()

	var payload []byte
	switch packet.Kind {
	case protocol.PacketKindInvoke:
		payload = protocol.BuildInvoke(packet.MethodID, packet.InvokeID, packet.Body)
	case protocol.PacketKindReturn:
		payload = protocol.BuildReturn(packet.MethodID, packet.InvokeID, packet.Error, packet.Body)
	case protocol.PacketKindNotify:
		payload = protocol.BuildNotify(packet.MethodID, packet.Body)
	}

	return protocol.WriteFrame(sess.conn, 0x09, payload, sess.sendCipher)
}

func (sess *Session) ReturnEmptyOk(invokeID int32, methodID uint32) error {
	return sess.SendPacket(&protocol.RpcPacket{
		Kind:     protocol.PacketKindReturn,
		InvokeID: invokeID,
		MethodID: methodID,
		Error:    0,
		Body:     nil,
	})
}

func (sess *Session) ReturnEmpty(invokeID int32, methodID uint32, errCode int32) error {
	return sess.SendPacket(&protocol.RpcPacket{
		Kind:     protocol.PacketKindReturn,
		InvokeID: invokeID,
		MethodID: methodID,
		Error:    errCode,
		Body:     nil,
	})
}

func (sess *Session) ReturnData(invokeID int32, methodID uint32, body []byte) error {
	return sess.SendPacket(&protocol.RpcPacket{
		Kind:     protocol.PacketKindReturn,
		InvokeID: invokeID,
		MethodID: methodID,
		Error:    0,
		Body:     body,
	})
}

func (sess *Session) NotifyData(methodID uint32, body []byte) error {
	return sess.SendPacket(&protocol.RpcPacket{
		Kind:     protocol.PacketKindNotify,
		MethodID: methodID,
		Body:     body,
	})
}

func (s *Server) handleConnection(ctx context.Context, conn net.Conn, port int) {
	defer conn.Close()

	log.Printf("[LOGIN-GATE] Client connected from %s on port %d", conn.RemoteAddr(), port)

	readCipher, writeCipher, err := protocol.HandleHandshake(conn)
	if err != nil {
		log.Printf("[LOGIN-GATE] Handshake error from %s: %v", conn.RemoteAddr(), err)
		return
	}

	sess := &Session{
		conn:       conn,
		readCipher: readCipher,
		sendCipher: writeCipher,
	}

	for {
		select {
		case <-ctx.Done():
			return
		default:
		}

		frame, err := protocol.ReadFrame(conn, sess.readCipher)
		if err != nil {
			if err != io.EOF {
				log.Printf("[LOGIN-GATE] Connection %s read error: %v", conn.RemoteAddr(), err)
			}
			return
		}

		// Frame mode 0x03 is heartbeat
		if frame.Mode == 0x03 {
			// Echo heartbeat back
			_ = protocol.WriteFrame(conn, 0x03, frame.Payload, sess.sendCipher)
			continue
		}

		packet := protocol.ParseRpcPacket(frame.Payload)
		if err := s.dispatch(sess, &packet); err != nil {
			log.Printf("[LOGIN-GATE] Dispatch error for method %d: %v", packet.MethodID, err)
		}
	}
}

func (s *Server) dispatch(sess *Session, pkt *protocol.RpcPacket) error {
	switch pkt.MethodID {
	case protocol.MethodCheckVersion, protocol.MethodHasOnlinePlayer:
		return sess.ReturnEmptyOk(pkt.InvokeID, pkt.MethodID)

	case protocol.MethodAskNewHotFixPatchLogin:
		return sess.ReturnEmpty(pkt.InvokeID, pkt.MethodID, int32(s.cfg.Client.NoMoreHotfixPatchError))

	case protocol.MethodRequestPatchesCheckDataFromLogin:
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, protocol.EncodePatchCheck())

	case protocol.MethodRequestPatchesFromLogin:
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, protocol.EncodeEmptyList())

	case protocol.MethodAskUniSdkShareToken_Login, protocol.MethodAskUniSdkShareToken_Gate:
		w := protocol.NewUxWriter()
		w.UxString(s.cfg.Player.ShareToken)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, w.Bytes())

	case protocol.MethodRequestFpPassToken:
		w := protocol.NewUxWriter()
		w.UxString(s.cfg.Player.FpPassToken)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, w.Bytes())

	case protocol.MethodCheckAccount, protocol.MethodCheckAccountPassBy, protocol.MethodCheckAccountOpenId:
		body := protocol.EncodeCheckAccount(
			s.cfg.Player.AccountId,
			s.cfg.Player.UserName,
			s.cfg.Player.LoginToken,
			s.cfg.Client.Aid,
			s.cfg.Client.ServerID,
			s.cfg.Player.Pid,
		)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, body)

	case protocol.MethodTryLogin:
		if err := sess.ReturnEmptyOk(pkt.InvokeID, pkt.MethodID); err != nil {
			return err
		}
		// Send SyncRoleList (35167428)
		w := protocol.NewUxWriter()
		w.U64(s.cfg.Player.Pid)
		return sess.NotifyData(protocol.MethodSyncRoleList, w.Bytes())

	case protocol.MethodRequestCreateRoleEx:
		w := protocol.NewUxWriter()
		w.U64(s.cfg.Player.Pid)
		if err := sess.ReturnData(pkt.InvokeID, pkt.MethodID, w.Bytes()); err != nil {
			return err
		}
		return sess.NotifyData(protocol.MethodSyncRoleList, w.Bytes())

	case protocol.MethodRequestEnterGame, protocol.MethodDebugRequestEnterGame:
		body := protocol.EncodeEnterGame(
			s.cfg.Client.Aid,
			s.cfg.Player.Pid,
			s.cfg.Network.AdvertisedHost,
			s.cfg.Network.LoginPorts[0],
			s.cfg.Player.LoginToken,
			s.cfg.Client.ServerID,
			s.cfg.Player.AccountId,
		)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, body)

	case protocol.MethodAskCloseConnectionToGate:
		w := protocol.NewUxWriter()
		w.U32(0)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, w.Bytes())

	case protocol.MethodGetServerTime:
		if pkt.Kind == protocol.PacketKindInvoke {
			if err := sess.ReturnEmptyOk(pkt.InvokeID, pkt.MethodID); err != nil {
				return err
			}
		}
		return sess.NotifyData(protocol.MethodSendServerTime, protocol.EncodeServerTime(0))

	case protocol.MethodQuerySkey:
		w := protocol.NewUxWriter()
		w.UxString(s.cfg.Player.Skey)
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, w.Bytes())

	case protocol.MethodGateLogin:
		log.Printf("[GATE] Authenticated Gate session registered for IMasterToClient S2C")
		if err := sess.ReturnEmptyOk(pkt.InvokeID, pkt.MethodID); err != nil {
			return err
		}
		// 1. SendServerTime
		_ = sess.NotifyData(protocol.MethodSendServerTime, protocol.EncodeServerTime(0))

		// 2. SyncGameSwitchToClient_3 (53940944) - Crucial for vehicles, NPCs, gadgets
		gameSwitches := protocol.BuildGameSwitchPayload()
		log.Printf("[GATE] SyncGameSwitchToClient_3 entries=%d body=%dB", len(protocol.GameSwitchCatalog), len(gameSwitches))
		_ = sess.NotifyData(protocol.MethodSyncGameSwitchToClient_3, gameSwitches)

		// 3. SyncPlayerGameServerInfo (52737039) - Handoff to Game Server
		gameInfo := protocol.EncodeGameServerInfo(s.cfg.Network.AdvertisedHost, s.cfg.Network.GamePort, s.cfg.Player.GameToken)
		return sess.NotifyData(protocol.MethodSyncPlayerGameServerInfo, gameInfo)

	case protocol.MethodGetFriendApplicationListToMe,
		protocol.MethodGetAllChatGroupLatestMessage,
		protocol.MethodGetSimplePlayerInfoByPidList:
		return sess.ReturnData(pkt.InvokeID, pkt.MethodID, protocol.EncodeEmptyList())

	case protocol.MethodReportNetworkQuality, protocol.MethodDavinciCode:
		// S2C Notify ignore
		return nil

	default:
		if pkt.Kind == protocol.PacketKindInvoke {
			log.Printf("[LOGIN-GATE] Unhandled Invoke MethodID %d, returning empty OK", pkt.MethodID)
			return sess.ReturnEmptyOk(pkt.InvokeID, pkt.MethodID)
		}
	}

	return nil
}