package server

import (
	"github.com/Mindgamesnl/socketio"
	"github.com/ParadoxPixel/ThemePark-Websocket/handlers"
	"log"
	"net/http"
	"strings"
	"time"
)

var runningServer *socketio.Server

type WrappedServer struct {
	OriginalHandler http.Handler
}

func (wrapper WrappedServer) ServeHTTP(res http.ResponseWriter, req *http.Request) {
	res.Header().Set("Access-Control-Allow-Origin", "*")
	res.Header().Set("Access-Control-Allow-Methods", "GET, HEAD, POST, OPTIONS")
	res.Header().Set("Access-Control-Allow-Headers", "Content-Type, Access-Control-Allow-Origin")

	if !strings.HasPrefix(req.URL.Path, "/socket.io/") {
		_, _ = res.Write([]byte("{\"errors\":[{\"message\":\"something went wrong\",\"code\":\"BAD_REQUEST\"}],\"response\":{}}"))
		return
	}

	wrapper.OriginalHandler.ServeHTTP(res, req)
}

func Start(namespace string, address string, debug bool) {
	log.Println("Starting socket server")
	server, _ := socketio.NewServer(time.Second*25, time.Second*5, socketio.DefaultParser)
	runningServer = server

	handlers.Load(server.Namespace(namespace), debug)

	actualServer := CreateServer(address)
	wrapper := WrappedServer{
		OriginalHandler: server,
	}
	actualServer.Handler = wrapper

	log.Println("Starting web server on ["+address+"]")
	log.Fatal(actualServer.ListenAndServe())
}

func Stop() {
	log.Println("Stopping web server")
	_ = runningServer.Close()
}