Initial commit

server/server.go

@@ -0,0 +1,13 @@
+package server
+
+import (
+	"net/http"
+)
+
+func CreateServer(addr string) *http.Server {
+	server := &http.Server{
+		Addr: addr,
+	}
+
+	return server
+}

server/websocket.go

@@ -0,0 +1,51 @@
+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()
+}