package main

import (
	"log"
	"net/http"

	socketio "github.com/googollee/go-socket.io"
)

func withCORS(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Access-Control-Allow-Origin", "*")
		w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
		w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
		w.Header().Set("Access-Control-Max-Age", "3600")

		// Handle preflight requests
		if r.Method == http.MethodOptions {
			w.WriteHeader(http.StatusOK)
			return
		}

		next.ServeHTTP(w, r)
	})
}
func main() {
	log.Println("go...")
	server := socketio.NewServer(nil)
	server.OnConnect("/", func(s socketio.Conn) error {
		s.SetContext("")
		log.Println("Client connected:", s.ID())
		return nil
	})
	server.OnDisconnect("/", func(s socketio.Conn, reason string) {
		log.Println("Client disconnected:", s.ID(), "Reason:", reason)
	})
	server.OnError("/", func(s socketio.Conn, e error) {
		log.Println("Socket.IO error for client:", s.ID(), "Error:", e)
	})
	server.OnEvent("/", "message", func(s socketio.Conn, msg string) {
		log.Println("Received message from client:", msg)
		s.Emit("response", "Message received: "+msg)
	})
	go func() {
		if err := server.Serve(); err != nil {
			log.Fatalf("Socket.IO server failed: %v", err)
		}
	}()
	defer server.Close()
	http.Handle("/", withCORS(http.FileServer(http.Dir("./static"))))
	http.Handle("/socket.io/", withCORS(server))
	ipAddress := "192.168.0.25"
	port := "8877"
	address := ipAddress + ":" + port
	log.Printf("Serving at ws://%s...", address)
	if err := http.ListenAndServe(address, nil); err != nil {
		log.Fatalf("Failed to start HTTP server: %v", err)
	}
}