package common

import (
	"encoding/json"
)

// Constants

const (
	EchoReqID            = 1
	EchoResID            = 128 + EchoReqID
	ListPeersReqID       = 2
	ListPeersResID       = 128 + ListPeersReqID
	AuthReqID            = 3
	AuthResID            = 128 + AuthReqID
	StartChatAReqID      = 4
	StartChatBReqID      = 5
	StartChatCReqID      = 6
	StartChatDReqID      = 7
	StartChatFinishReqID = 8
)

// Requests & responses subtypes

type PeerInfo struct {
	ID          int    `json:"id"`
	Addr        string `json:"addr"`
	HasNickname bool   `json:"hasNickname"`
	Nickname    string `json:"nickname"`
}

// Requests & responses:

type RFrame struct {
	ID   int             `json:"id"`
	Rest json.RawMessage `json:"r"`
}

func (rf RFrame) IsRequest() bool {
	return rf.ID <= 128
}

func (rf RFrame) IsResponse() bool {
	return rf.ID > 128
}

func (rf RFrame) IsError() bool {
	return rf.ID > 256
}

func RequestFrameFrom(req Request) (RFrame, error) {
	jsonBytes, err := json.Marshal(req)

	if err != nil {
		return *new(RFrame), err
	}

	return RFrame{req.ID(), jsonBytes}, nil
}

func RequestFromFrame[T Request](reqFrame RFrame) (T, error) {
	var req T
	err := json.Unmarshal(reqFrame.Rest, &req)

	if err != nil {
		return *new(T), err
	}

	return req, nil
}

func ResponseFrameFrom(res Response) (RFrame, error) {
	jsonBytes, err := json.Marshal(res)

	if err != nil {
		return *new(RFrame), err
	}

	return RFrame{res.ID(), jsonBytes}, nil
}

func ResponseFromFrame[T Response](resFrame RFrame) (T, error) {
	var res T
	err := json.Unmarshal(resFrame.Rest, &res)

	if err != nil {
		return *new(T), err
	}

	return res, nil
}

type Request interface {
	ID() int
}

type Response Request

type EchoRequest struct {
	EchoByte byte `json:"echoByte"`
}

func (EchoRequest) ID() int {
	return EchoReqID
}

type EchoResponse struct {
	EchoByte byte `json:"echoByte"`
}

func (EchoResponse) ID() int {
	return EchoResID
}

type ListPeersRequest struct {
}

func (ListPeersRequest) ID() int {
	return ListPeersReqID
}

type ListPeersResponse struct {
	PeersInfo []PeerInfo `json:"peers"`
}

func (ListPeersResponse) ID() int {
	return ListPeersResID
}

type AuthRequest struct {
	Nickname string `json:"nickname"`
	Password string `json:"password"`
}

func (AuthRequest) ID() int {
	return AuthReqID
}

type AuthResponse struct {
	IsSuccess bool
}

func (AuthResponse) ID() int {
	return AuthResID
}

// "Stateful" requests like these need to have some information identifying
// what operation they are linked to
// There may be some errors if two requests are sent by one host to the same
// other host... may they?

type StartChatARequest struct {
	Nickname string `json:"nickname"`
}

func (StartChatARequest) ID() int {
	return StartChatAReqID
}

type StartChatBRequest struct {
	Nickname string `json:"nickname"`
}

func (StartChatBRequest) ID() int {
	return StartChatBReqID
}

type StartChatCRequest struct {
	Nickname string `json:"nickname"`
}

func (StartChatCRequest) ID() int {
	return StartChatCReqID
}

type StartChatDRequest struct {
	Nickname  string `json:"nickname"`
	PunchCode string `json:"punchCode"`
}

func (StartChatDRequest) ID() int {
	return StartChatDReqID
}

type StartChatFinishRequest struct {
	OtherSideNickname string `json:"otherSideNickname"`
	OtherSideAddress  string `json:"otherSideAddress"`
}

func (StartChatFinishRequest) ID() int {
	return StartChatFinishReqID
}

type Initiation struct {
	AbANick      string
	AbBNick      string
	Stage        int
	AbAPunchCode string
	AbBPunchCode string
	AbAAddress   string
	AbBAddress   string
}

const (
	InitiationStageA = 1
	InitiationStageB = 2
	InitiationStageC = 3
	InitiationStageD = 4
)

type PunchRequest struct {
	PunchCode string `json:"punchCode"`
}

type PunchResponse struct {
	IPAddr int32 `json:"ipAddr"`
}

type Settings struct {
	WsapiAddr string
	UdpAddr   string
}

type ServerSettings struct {
	Settings
}

type ClientSettings struct {
	Settings
}