package socket

import (
	"JISQueueing/common"
	"JISQueueing/db"
	"encoding/json"
	"strconv"

	"github.com/gorilla/websocket"
)

var tickets = make(map[common.Ticket]bool)

func newTicket(ticket common.Ticket, who *websocket.Conn) {
	tickets[ticket] = false
	jsonTicket, _ := json.Marshal(ticket)

	sendStaffMessage("new " + string(jsonTicket))
	who.WriteMessage(websocket.TextMessage, []byte("success new "+string(jsonTicket)))
}

func claimedTicket(ticket common.Ticket, staff common.Staff, table int, who *websocket.Conn) {
	if !db.SetClaimer(ticket.ID, staff.Username) {
		who.WriteMessage(websocket.TextMessage, []byte("error accept the ticket has already been completed or cancelled"))
		return
	}
	delete(tickets, ticket)
	ticket.Staff = staff.Username
	tickets[ticket] = true

	jsonTicket, _ := json.Marshal(ticket)
	sendDisplayMessage("claimed " + strconv.Itoa(table) + " " + strconv.Itoa(ticket.ID))
	who.WriteMessage(websocket.TextMessage, []byte("success claimed "+string(jsonTicket)))
	for conn, s := range connToStaff {
		if s.Username != staff.Username {
			conn.WriteMessage(websocket.TextMessage, []byte("claimed "+string(jsonTicket)))
		}
	}
}

func finishedTicket(ticket common.Ticket, id int, table int, who *websocket.Conn) {
	delete(tickets, ticket)

	if success := db.FinishTicket(id); !success {
		who.WriteMessage(websocket.TextMessage, []byte("error complete the ticket has already been completed or cancelled"))
		return
	}
	who.WriteMessage(websocket.TextMessage, []byte("success complete "+strconv.Itoa(id)))
	sendDisplayMessage("complete " + strconv.Itoa(table))
}

func cancelTicket(ticket common.Ticket, id int, table int, who *websocket.Conn) {
	delete(tickets, ticket)

	if success := db.CancelTicket(id); !success {
		who.WriteMessage(websocket.TextMessage, []byte("error cancel the ticket has already been completed or cancelled"))
		return
	}
	who.WriteMessage(websocket.TextMessage, []byte("success cancel "+strconv.Itoa(id)))
	sendDisplayMessage("cancel " + strconv.Itoa(table))
}

func newStaff(table int, who *websocket.Conn) {
	who.WriteMessage(websocket.TextMessage, []byte("success pick "+strconv.Itoa(table)))
	for member, num := range connToTable {
		if num == -1 {
			sendTaken(member)
		}
	}
	sendDisplayMessage("pick " + strconv.Itoa(table))
}

func leaveStaff(table int) {
	sendDisplayMessage("unpick " + strconv.Itoa(table))
	for member, num := range connToTable {
		if num == -1 {
			sendTaken(member)
		}
	}
}