package entranceserver

import (
	"encoding/binary"
	"encoding/hex"
	"erupe-ce/common/stringsupport"
	_config "erupe-ce/config"
	"fmt"
	"net"

	"erupe-ce/common/byteframe"
	"erupe-ce/server/channelserver"
)

// Server Channels
var currentplayers uint16

func encodeServerInfo(config *_config.Config, s *Server, local bool) []byte {
	serverInfos := config.Entrance.Entries
	bf := byteframe.NewByteFrame()

	for serverIdx, si := range serverInfos {
		// Prevent MezFes Worlds displaying on Z1
		if config.RealClientMode <= _config.Z1 {
			if si.Type == 6 {
				continue
			}
		}
		if config.RealClientMode <= _config.G6 {
			if si.Type == 5 {
				continue
			}
		}

		sid := (4096 + serverIdx*256) * 6000
		if si.IP == "" {
			si.IP = config.Host
		}
		if local {
			bf.WriteUint32(0x0100007F) // 127.0.0.1
		} else {
			bf.WriteUint32(binary.LittleEndian.Uint32(net.ParseIP(si.IP).To4()))
		}
		bf.WriteUint16(16 + uint16(serverIdx))
		bf.WriteUint16(0x0000)
		bf.WriteUint16(uint16(len(si.Channels)))
		bf.WriteUint8(si.Type)
		bf.WriteUint8(uint8(((channelserver.TimeAdjusted().Unix() / 86400) + int64(serverIdx)) % 3))
		bf.WriteUint8(si.Recommended)

		if s.erupeConfig.RealClientMode <= _config.GG {
			combined := append(stringsupport.UTF8ToSJIS(si.Name), []byte{0x00}...)
			combined = append(combined, stringsupport.UTF8ToSJIS(si.Description)...)
			bf.WriteUint8(uint8(len(combined)))
			bf.WriteBytes(combined)
		} else {
			bf.WriteUint8(0) // Prevents malformed server name
			combined := append(stringsupport.UTF8ToSJIS(si.Name), []byte{0x00}...)
			combined = append(combined, stringsupport.UTF8ToSJIS(si.Description)...)
			bf.WriteBytes(stringsupport.PaddedString(string(combined), 65, false))
		}

		if s.erupeConfig.RealClientMode >= _config.GG {
			bf.WriteUint32(si.AllowedClientFlags)
		}

		for channelIdx, ci := range si.Channels {
			sid = (4096 + serverIdx*256) + (16 + channelIdx)
			bf.WriteUint16(ci.Port)
			bf.WriteUint16(16 + uint16(channelIdx))
			bf.WriteUint16(ci.MaxPlayers)
			err := s.db.QueryRow("SELECT current_players FROM servers WHERE server_id=$1", sid).Scan(&currentplayers)
			if err != nil {
				currentplayers = 0
			}
			bf.WriteUint16(currentplayers)
			bf.WriteUint32(0)
			bf.WriteUint32(0)
			bf.WriteUint32(0)
			bf.WriteUint16(319) // Unk
			bf.WriteUint16(252) // Unk
			bf.WriteUint16(248) // Unk
			bf.WriteUint16(0x3039)
		}
	}
	bf.WriteUint32(uint32(channelserver.TimeAdjusted().Unix()))
	bf.WriteUint32(0x0000003C)
	return bf.Data()
}

func makeHeader(data []byte, respType string, entryCount uint16, key byte) []byte {
	bf := byteframe.NewByteFrame()
	bf.WriteBytes([]byte(respType))
	bf.WriteUint16(entryCount)
	bf.WriteUint16(uint16(len(data)))
	if len(data) > 0 {
		bf.WriteUint32(CalcSum32(data))
		bf.WriteBytes(data)
	}

	dataToEncrypt := bf.Data()

	bf = byteframe.NewByteFrame()
	bf.WriteUint8(key)
	bf.WriteBytes(EncryptBin8(dataToEncrypt, key))
	return bf.Data()
}

func makeSv2Resp(config *_config.Config, s *Server, local bool) []byte {
	serverInfos := config.Entrance.Entries
	// Decrease by the number of MezFes Worlds
	var mf int
	if config.RealClientMode <= _config.Z1 {
		for _, si := range serverInfos {
			if si.Type == 6 {
				mf++
			}
		}
	}
	// and Return Worlds
	var ret int
	if config.RealClientMode <= _config.G6 {
		for _, si := range serverInfos {
			if si.Type == 5 {
				ret++
			}
		}
	}
	rawServerData := encodeServerInfo(config, s, local)

	if s.erupeConfig.DevMode && s.erupeConfig.DevModeOptions.LogOutboundMessages {
		fmt.Printf("[Server] -> [Client]\nData [%d bytes]:\n%s\n", len(rawServerData), hex.Dump(rawServerData))
	}

	respType := "SV2"
	if config.RealClientMode <= _config.G32 {
		respType = "SVR"
	}

	bf := byteframe.NewByteFrame()
	bf.WriteBytes(makeHeader(rawServerData, respType, uint16(len(serverInfos)-(mf+ret)), 0x00))
	return bf.Data()
}

func makeUsrResp(pkt []byte, s *Server) []byte {
	bf := byteframe.NewByteFrameFromBytes(pkt)
	_ = bf.ReadUint32() // ALL+
	_ = bf.ReadUint8()  // 0x00
	userEntries := bf.ReadUint16()
	resp := byteframe.NewByteFrame()
	for i := 0; i < int(userEntries); i++ {
		cid := bf.ReadUint32()
		var sid uint16
		err := s.db.QueryRow("SELECT(SELECT server_id FROM sign_sessions WHERE char_id=$1) AS _", cid).Scan(&sid)
		if err != nil {
			resp.WriteBytes(make([]byte, 4))
		} else {
			resp.WriteUint16(sid)
			resp.WriteUint16(0)
		}
	}

	if s.erupeConfig.DevMode && s.erupeConfig.DevModeOptions.LogOutboundMessages {
		fmt.Printf("[Server] -> [Client]\nData [%d bytes]:\n%s\n", len(resp.Data()), hex.Dump(resp.Data()))
	}

	return makeHeader(resp.Data(), "USR", userEntries, 0x00)
}