package main

import (
	"context"
	"errors"
	"log"
	"net/http"
	"time"

	"github.com/coder/websocket"
	"github.com/coder/websocket/wsjson"
)

func ServeWsConnection(responseWriter http.ResponseWriter, request *http.Request) {
	connection, err := websocket.Accept(responseWriter, request, &websocket.AcceptOptions{
		InsecureSkipVerify: true,
	})
	if err != nil {
		log.Printf("websocket accept error: %v", err)
		return
	}

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
	defer cancel()

	var client = Client{WsConn: connection}
	var isAuthenticated bool

	defer closeConnection(&client)
	for {
		var clientMessage map[string]any
		err := wsjson.Read(ctx, connection, &clientMessage)
		if err != nil {
			log.Printf("read error: %v", err)
			return
		}

		if len(clientMessage) > 0 {
			if isAuthenticated {
				if !handleAuthenticatedMessage(&client, &clientMessage) {
					return
				}
			} else {
				if !handleUnauthenticatedMessage(&client, &clientMessage) {
					return
				}
				isAuthenticated = true
			}
		}
	}
}

func sendMessageCloseIfTimeout(client *Client, message *map[string]any) {
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()

	err := wsjson.Write(ctx, client.WsConn, message)
	if err != nil {
		if errors.Is(err, context.DeadlineExceeded) {
			closeConnection(client)
		}
		log.Printf("write error: %v", err)
	}
}

func sendToAllMessageCloseIfTimeout(message *map[string]any) {
	mu.RLock()
	defer mu.RUnlock()
	for _, client := range CacheClients {
		sendMessageCloseIfTimeout(client, message)
	}
}

func handleAuthenticatedMessage(client *Client, clientMessage *map[string]any) bool {
	sendToAllMessageCloseIfTimeout(clientMessage)
	return true
}

func handleUnauthenticatedMessage(client *Client, clientMessage *map[string]any) bool {
	token, ok := (*clientMessage)["token"].(string)
	if !ok {
		var msg = map[string]any{
			"from":  "server",
			"error": "no token in message",
		}
		sendMessageCloseIfTimeout(client, &msg)
		return false
	}

	clientId, err := TokenValidateGetId(token)
	if err != nil {
		var msg = map[string]any{
			"from":  "server",
			"error": "invalid token",
		}
		sendMessageCloseIfTimeout(client, &msg)
		return false
	}

	clientFromCache, err := CacheGetClientById(clientId)
	if err != nil {
		var msg = map[string]any{
			"from":  "server",
			"error": "invalid token",
		}
		sendMessageCloseIfTimeout(client, &msg)
		return false
	}

	*client = *clientFromCache
	return true
}

func closeConnection(client *Client) {
	CacheDeleteClient(client.Id)
	client.WsConn.CloseNow()
}