package main

import (
	"database/sql"
	"fmt"
	"log"
	"net/http"
	"os"

	"code.springupsoftware.com/cmc/cmc-sales/internal/cmc/db"
	"code.springupsoftware.com/cmc/cmc-sales/internal/cmc/handlers"
	"code.springupsoftware.com/cmc/cmc-sales/internal/cmc/templates"
	_ "github.com/go-sql-driver/mysql"
	"github.com/gorilla/mux"
	"github.com/joho/godotenv"
)

func main() {
	// Load environment variables
	if err := godotenv.Load(); err != nil {
		log.Println("No .env file found")
	}

	// Database configuration
	dbHost := getEnv("DB_HOST", "localhost")
	dbPort := getEnv("DB_PORT", "3306")
	dbUser := getEnv("DB_USER", "cmc")
	dbPass := getEnv("DB_PASSWORD", "xVRQI&cA?7AU=hqJ!%au")
	dbName := getEnv("DB_NAME", "cmc")

	// Connect to database
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true", dbUser, dbPass, dbHost, dbPort, dbName)
	database, err := sql.Open("mysql", dsn)
	if err != nil {
		log.Fatal("Failed to connect to database:", err)
	}
	defer database.Close()

	// Test database connection
	if err := database.Ping(); err != nil {
		log.Fatal("Failed to ping database:", err)
	}

	log.Println("Connected to database successfully")

	// Create queries instance
	queries := db.New(database)

	// Initialize template manager
	tmpl, err := templates.NewTemplateManager("templates")
	if err != nil {
		log.Fatal("Failed to initialize templates:", err)
	}

	// Create handlers
	customerHandler := handlers.NewCustomerHandler(queries)
	productHandler := handlers.NewProductHandler(queries)
	purchaseOrderHandler := handlers.NewPurchaseOrderHandler(queries)
	enquiryHandler := handlers.NewEnquiryHandler(queries)
	pageHandler := handlers.NewPageHandler(queries, tmpl)

	// Setup routes
	r := mux.NewRouter()
	
	// Static files
	r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
	
	// API routes
	api := r.PathPrefix("/api/v1").Subrouter()

	// Customer routes
	api.HandleFunc("/customers", customerHandler.List).Methods("GET")
	api.HandleFunc("/customers", customerHandler.Create).Methods("POST")
	api.HandleFunc("/customers/{id}", customerHandler.Get).Methods("GET")
	api.HandleFunc("/customers/{id}", customerHandler.Update).Methods("PUT")
	api.HandleFunc("/customers/{id}", customerHandler.Delete).Methods("DELETE")
	api.HandleFunc("/customers/search", customerHandler.Search).Methods("GET")

	// Product routes
	api.HandleFunc("/products", productHandler.List).Methods("GET")
	api.HandleFunc("/products", productHandler.Create).Methods("POST")
	api.HandleFunc("/products/{id}", productHandler.Get).Methods("GET")
	api.HandleFunc("/products/{id}", productHandler.Update).Methods("PUT")
	api.HandleFunc("/products/{id}", productHandler.Delete).Methods("DELETE")
	api.HandleFunc("/products/search", productHandler.Search).Methods("GET")

	// Purchase Order routes
	api.HandleFunc("/purchase-orders", purchaseOrderHandler.List).Methods("GET")
	api.HandleFunc("/purchase-orders", purchaseOrderHandler.Create).Methods("POST")
	api.HandleFunc("/purchase-orders/{id}", purchaseOrderHandler.Get).Methods("GET")
	api.HandleFunc("/purchase-orders/{id}", purchaseOrderHandler.Update).Methods("PUT")
	api.HandleFunc("/purchase-orders/{id}", purchaseOrderHandler.Delete).Methods("DELETE")
	api.HandleFunc("/purchase-orders/search", purchaseOrderHandler.Search).Methods("GET")

	// Enquiry routes
	api.HandleFunc("/enquiries", enquiryHandler.List).Methods("GET")
	api.HandleFunc("/enquiries", enquiryHandler.Create).Methods("POST")
	api.HandleFunc("/enquiries/{id}", enquiryHandler.Get).Methods("GET")
	api.HandleFunc("/enquiries/{id}", enquiryHandler.Update).Methods("PUT")
	api.HandleFunc("/enquiries/{id}", enquiryHandler.Delete).Methods("DELETE")
	api.HandleFunc("/enquiries/{id}/undelete", enquiryHandler.Undelete).Methods("PUT")
	api.HandleFunc("/enquiries/{id}/status", enquiryHandler.UpdateStatus).Methods("PUT")
	api.HandleFunc("/enquiries/{id}/mark-submitted", enquiryHandler.MarkSubmitted).Methods("PUT")
	api.HandleFunc("/enquiries/search", enquiryHandler.Search).Methods("GET")

	// Health check
	api.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		w.Write([]byte(`{"status":"ok"}`))
	}).Methods("GET")

	// Page routes
	r.HandleFunc("/", pageHandler.Home).Methods("GET")
	
	// Customer pages
	r.HandleFunc("/customers", pageHandler.CustomersIndex).Methods("GET")
	r.HandleFunc("/customers/new", pageHandler.CustomersNew).Methods("GET")
	r.HandleFunc("/customers/search", pageHandler.CustomersSearch).Methods("GET")
	r.HandleFunc("/customers/{id}", pageHandler.CustomersShow).Methods("GET")
	r.HandleFunc("/customers/{id}/edit", pageHandler.CustomersEdit).Methods("GET")
	
	// Product pages
	r.HandleFunc("/products", pageHandler.ProductsIndex).Methods("GET")
	r.HandleFunc("/products/new", pageHandler.ProductsNew).Methods("GET")
	r.HandleFunc("/products/search", pageHandler.ProductsSearch).Methods("GET")
	r.HandleFunc("/products/{id}", pageHandler.ProductsShow).Methods("GET")
	r.HandleFunc("/products/{id}/edit", pageHandler.ProductsEdit).Methods("GET")
	
	// Purchase Order pages
	r.HandleFunc("/purchase-orders", pageHandler.PurchaseOrdersIndex).Methods("GET")
	r.HandleFunc("/purchase-orders/new", pageHandler.PurchaseOrdersNew).Methods("GET")
	r.HandleFunc("/purchase-orders/search", pageHandler.PurchaseOrdersSearch).Methods("GET")
	r.HandleFunc("/purchase-orders/{id}", pageHandler.PurchaseOrdersShow).Methods("GET")
	r.HandleFunc("/purchase-orders/{id}/edit", pageHandler.PurchaseOrdersEdit).Methods("GET")
	
	// Enquiry pages
	r.HandleFunc("/enquiries", pageHandler.EnquiriesIndex).Methods("GET")
	r.HandleFunc("/enquiries/new", pageHandler.EnquiriesNew).Methods("GET")
	r.HandleFunc("/enquiries/search", pageHandler.EnquiriesSearch).Methods("GET")
	r.HandleFunc("/enquiries/{id}", pageHandler.EnquiriesShow).Methods("GET")
	r.HandleFunc("/enquiries/{id}/edit", pageHandler.EnquiriesEdit).Methods("GET")

	// HTMX endpoints
	r.HandleFunc("/customers", customerHandler.Create).Methods("POST")
	r.HandleFunc("/customers/{id}", customerHandler.Update).Methods("PUT")
	r.HandleFunc("/customers/{id}", customerHandler.Delete).Methods("DELETE")
	
	r.HandleFunc("/products", productHandler.Create).Methods("POST")
	r.HandleFunc("/products/{id}", productHandler.Update).Methods("PUT")
	r.HandleFunc("/products/{id}", productHandler.Delete).Methods("DELETE")
	
	r.HandleFunc("/purchase-orders", purchaseOrderHandler.Create).Methods("POST")
	r.HandleFunc("/purchase-orders/{id}", purchaseOrderHandler.Update).Methods("PUT")
	r.HandleFunc("/purchase-orders/{id}", purchaseOrderHandler.Delete).Methods("DELETE")
	
	r.HandleFunc("/enquiries", enquiryHandler.Create).Methods("POST")
	r.HandleFunc("/enquiries/{id}", enquiryHandler.Update).Methods("PUT")
	r.HandleFunc("/enquiries/{id}", enquiryHandler.Delete).Methods("DELETE")
	r.HandleFunc("/enquiries/{id}/undelete", enquiryHandler.Undelete).Methods("PUT")
	r.HandleFunc("/enquiries/{id}/status", enquiryHandler.UpdateStatus).Methods("PUT")
	r.HandleFunc("/enquiries/{id}/mark-submitted", enquiryHandler.MarkSubmitted).Methods("PUT")

	// Start server
	port := getEnv("PORT", "8080")
	log.Printf("Starting server on port %s", port)
	if err := http.ListenAndServe(":"+port, r); err != nil {
		log.Fatal("Failed to start server:", err)
	}
}

func getEnv(key, defaultValue string) string {
	if value := os.Getenv(key); value != "" {
		return value
	}
	return defaultValue
}