package api

import (
	"database/sql"
	"net/http"
	"os"
	"path/filepath"
)

type Server struct {
	db      *sql.DB
	dataDir string
	secret  string
}

func NewRouter(db *sql.DB, dataDir, secret string) http.Handler {
	s := &Server{db: db, dataDir: dataDir, secret: secret}
	mux := http.NewServeMux()

	// Auth
	mux.HandleFunc("POST /api/auth/login", s.handleLogin)
	mux.HandleFunc("POST /api/auth/logout", s.handleLogout)

	// Users (admin)
	mux.HandleFunc("POST /api/users/create", s.requireAdmin(s.handleCreateUser))
	mux.HandleFunc("POST /api/users/list", s.requireAdmin(s.handleListUsers))

	// Files
	mux.HandleFunc("POST /api/files/list", s.requireAuth(s.handleListFiles))
	mux.HandleFunc("POST /api/files/read", s.requireAuth(s.handleReadFile))
	mux.HandleFunc("POST /api/files/write", s.requireAuth(s.handleWriteFile))
	mux.HandleFunc("POST /api/files/create", s.requireAuth(s.handleCreateFile))
	mux.HandleFunc("POST /api/files/create-folder", s.requireAuth(s.handleCreateFolder))
	mux.HandleFunc("POST /api/files/delete", s.requireAuth(s.handleDeleteFile))
	mux.HandleFunc("POST /api/files/search", s.requireAuth(s.handleSearchFiles))
	mux.HandleFunc("POST /api/files/shared", s.requireAuth(s.handleSharedFiles))

	// Build jobs
	mux.HandleFunc("POST /api/build/submit", s.requireAuth(s.handleBuildSubmit))
	mux.HandleFunc("POST /api/build/jobs", s.requireAuth(s.handleBuildJobs))
	mux.HandleFunc("POST /api/build/status", s.requireAuth(s.handleBuildStatus))
	mux.HandleFunc("POST /api/build/cancel", s.requireAuth(s.handleBuildCancel))

	// Daemon endpoints
	mux.HandleFunc("POST /api/daemon/poll", s.requireAuth(s.handleDaemonPoll))
	mux.HandleFunc("POST /api/daemon/heartbeat", s.requireAuth(s.handleDaemonHeartbeat))
	mux.HandleFunc("POST /api/daemon/report", s.requireAuth(s.handleDaemonReport))

	// Static frontend
	frontendDir := filepath.Join(dataDir, "..", "frontend", "dist")
	if _, err := os.Stat(frontendDir); err != nil {
		frontendDir = "./frontend/dist"
	}
	fs := http.FileServer(http.Dir(frontendDir))
	mux.Handle("/", fs)

	return mux
}