package FlyWebSockets

import (
	"fmt"
	"sync"
	"time"

	"github.com/gorilla/websocket"
)

type FlyWSClient struct {
	lock    sync.Mutex
	Conn    *websocket.Conn
	StartAt time.Time
	IsClose bool
	// jsonWriteCh  chan interface{}
	// bytesWriteCh chan []byte
	// textWriteCh  chan []byte
}

func NewFlyWSClient(conn *websocket.Conn) *FlyWSClient {
	item := FlyWSClient{
		Conn:    conn,
		StartAt: time.Now(),
		IsClose: false,
	}
	// item.Start()
	return &item
}

// func (s *FlyWSClient) Start() {
// 	s.IsClose = false
// 	s.jsonWriteCh = make(chan interface{})
// 	s.bytesWriteCh = make(chan []byte)
// 	s.textWriteCh = make(chan []byte)
// 	go s.writeLoop()
// }

func (s *FlyWSClient) Close() {
	s.Conn.Close()
	s.IsClose = true
	// close(s.jsonWriteCh)
	// close(s.bytesWriteCh)
	// close(s.textWriteCh)
}

// func (s *FlyWSClient) writeLoop() {
// 	defer func() {
// 		if err := recover(); err != nil {
// 			fmt.Println(err)
// 		}
// 	}()
// 	for {
// 		select {
// 		case msg := <-s.jsonWriteCh:
// 			if err := s.Conn.WriteJSON(msg); err != nil {
// 				fmt.Println("Error writing JSON:", err)
// 			}
// 		case msg := <-s.bytesWriteCh:
// 			if err := s.Conn.WriteMessage(websocket.BinaryMessage, msg); err != nil {
// 				fmt.Println("Error writing bytes:", err)
// 			}
// 		case msg := <-s.textWriteCh:
// 			if err := s.Conn.WriteMessage(websocket.TextMessage, msg); err != nil {
// 				fmt.Println("Error writing bytes:", err)
// 			}
// 		}
// 	}
// }

func (s *FlyWSClient) ReadJSON(v interface{}) error {
	s.lock.Lock()
	defer s.lock.Unlock()
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()

	return s.Conn.ReadJSON(v)
}
func (s *FlyWSClient) WriteJSON(v interface{}) error {
	s.lock.Lock()
	defer s.lock.Unlock()
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()
	return s.Conn.WriteJSON(v)
}
func (s *FlyWSClient) WriteMessage(data []byte) error {
	s.lock.Lock()
	defer s.lock.Unlock()
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()
	return s.Conn.WriteMessage(websocket.TextMessage, data)
}

func (s *FlyWSClient) SendJSON(data WS_Resp) error {
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()
	return s.Conn.WriteJSON(data)
}

func (s *FlyWSClient) SendText(text string) error {
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()
	return s.Conn.WriteMessage(websocket.TextMessage, []byte(text))
}

func (s *FlyWSClient) SendBytes(data []byte) error {
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
	}()
	return s.Conn.WriteMessage(websocket.BinaryMessage, data)
}