package server

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

	"tangled.org/atscan.net/plcbundle-go/bundle"
)

// New creates a new HTTP server
func New(manager *bundle.Manager, config *Config) *Server {
	if config.Version == "" {
		config.Version = "dev"
	}

	s := &Server{
		manager:   manager,
		addr:      config.Addr,
		config:    config,
		startTime: time.Now(),
	}

	handler := s.createHandler()

	s.httpServer = &http.Server{
		Addr:    config.Addr,
		Handler: handler,
	}

	return s
}

// createHandler creates the HTTP handler with all routes
func (s *Server) createHandler() http.Handler {
	mux := http.NewServeMux()

	// Specific routes first
	mux.HandleFunc("GET /index.json", s.handleIndexJSON())
	mux.HandleFunc("GET /bundle/{number}", s.handleBundle())
	mux.HandleFunc("GET /data/{number}", s.handleBundleData())
	mux.HandleFunc("GET /jsonl/{number}", s.handleBundleJSONL())
	mux.HandleFunc("GET /op/{pointer}", s.handleOperation())
	mux.HandleFunc("GET /status", s.handleStatus())
	mux.HandleFunc("GET /debug/memory", s.handleDebugMemory())
	mux.HandleFunc("GET /debug/didindex", s.handleDebugDIDIndex())
	mux.HandleFunc("GET /debug/resolver", s.handleDebugResolver())

	// WebSocket
	if s.config.EnableWebSocket {
		mux.HandleFunc("GET /ws", s.handleWebSocket())
	}

	// Sync mode endpoints
	if s.config.SyncMode {
		mux.HandleFunc("GET /mempool", s.handleMempool())
	}

	// Root and DID resolver
	mux.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) {
		path := r.URL.Path

		if path == "/" {
			s.handleRoot()(w, r)
			return
		}

		if s.config.EnableResolver {
			s.handleDIDRouting(w, r)
			return
		}

		sendJSON(w, 404, map[string]string{"error": "not found"})
	})

	// Apply middleware in correct order:
	handler := corsMiddleware(mux)

	return handler
}

// ListenAndServe starts the HTTP server
func (s *Server) ListenAndServe() error {
	return s.httpServer.ListenAndServe()
}

// Shutdown gracefully shuts down the server
func (s *Server) Shutdown(ctx context.Context) error {
	return s.httpServer.Shutdown(ctx)
}

// GetStartTime returns when the server started
func (s *Server) GetStartTime() time.Time {
	return s.startTime
}

// Handler returns the configured HTTP handler
func (s *Server) Handler() http.Handler {
	return s.createHandler()
}