package op

import (
	"io"

	json "github.com/json-iterator/go"

	"github.com/gorilla/websocket"
	pb "github.com/synctv-org/synctv/proto"
	"google.golang.org/protobuf/proto"
)

type Message interface {
	MessageType() int
	String() string
	Encode(w io.Writer) error
}

type ElementJsonMessage struct {
	*pb.ElementMessage
}

func (em *ElementJsonMessage) MessageType() int {
	return websocket.TextMessage
}

func (em *ElementJsonMessage) String() string {
	return em.ElementMessage.String()
}

func (em *ElementJsonMessage) Encode(w io.Writer) error {
	return json.NewEncoder(w).Encode(em)
}

type ElementMessage struct {
	*pb.ElementMessage
}

func (em *ElementMessage) MessageType() int {
	return websocket.BinaryMessage
}

func (em *ElementMessage) String() string {
	return em.ElementMessage.String()
}

func (em *ElementMessage) Encode(w io.Writer) error {
	b, err := proto.Marshal(em)
	if err != nil {
		return err
	}
	_, err = w.Write(b)
	return err
}

type PingMessage struct{}

func (pm *PingMessage) MessageType() int {
	return websocket.PingMessage
}

func (pm *PingMessage) String() string {
	return "Ping"
}

func (pm *PingMessage) Encode(w io.Writer) error {
	return nil
}