package controllers

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"
	"os"
	"time"

	"s-class-backend/config" // Import database

	mqtt "github.com/eclipse/paho.mqtt.golang"
	"github.com/gin-gonic/gin"
)

// --- STRUKTUR DATA REQUEST ---
type DeviceControlRequest struct {
	Device string `json:"device"`
	Action string `json:"action"`
}

type VerifyRequest struct {
	Token string `json:"token"`
}

// =========================================================================
// CACHE STATUS HARDWARE (Mengingat status terakhir perangkat untuk Web Admin)
// =========================================================================
var DeviceStatusCache = map[string]string{
	"lampu1":    "off",
	"lampu2":    "off",
	"ac":        "off",
	"projector": "off",
}

// =========================================================================
// FUNGSI 1: VERIFIKASI HARDWARE & HITUNG SISA WAKTU DINAMIS
// =========================================================================
func VerifyHardwareCode(c *gin.Context) {
	var req VerifyRequest

	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": "Format token tidak valid"})
		return
	}

	tokenInput := req.Token
	sekarang := time.Now()
	var jamSelesai time.Time
	isTokenValid := false

	type ResultTime struct {
		JamSelesai time.Time `gorm:"column:end_time"`
	}
	var result ResultTime

	errBooking := config.DB.Table("bookings").
		Select("end_time").
		Where("redeem_code = ?", tokenInput).
		Scan(&result).Error

	if errBooking == nil && !result.JamSelesai.IsZero() {
		jamSelesai = result.JamSelesai
		isTokenValid = true
	} else {
		errSchedule := config.DB.Table("class_schedules").
			Select("end_time").
			Where("kode_mk = ?", tokenInput).
			Scan(&result).Error

		if errSchedule == nil && !result.JamSelesai.IsZero() {
			jamSelesai = result.JamSelesai
			isTokenValid = true
		}
	}

	if !isTokenValid {
		c.JSON(http.StatusUnauthorized, gin.H{"error": "Token salah atau tidak ditemukan"})
		return
	}

	jamSelesaiHariIni := time.Date(sekarang.Year(), sekarang.Month(), sekarang.Day(),
		jamSelesai.Hour(), jamSelesai.Minute(), jamSelesai.Second(), 0, sekarang.Location())

	selisihWaktu := jamSelesaiHariIni.Sub(sekarang)
	sisaMenit := int(selisihWaktu.Minutes())

	if sisaMenit <= 0 {
		c.JSON(http.StatusForbidden, gin.H{"error": "Waktu peminjaman sudah habis"})
		return
	}

	// -------------------------------------------------------------
	// OPTIONAL: Jika token benar, paksa otomatis catat Lampu ON di Web
	// -------------------------------------------------------------
	DeviceStatusCache["lampu1"] = "on"
	DeviceStatusCache["lampu2"] = "on"

	c.JSON(http.StatusOK, gin.H{
		"status":           "success",
		"message":          "Token Valid",
		"duration_minutes": sisaMenit,
	})
}

// =========================================================================
// FUNGSI 2: KONTROL DEVICE VIA HOME ASSISTANT & MQTT
// =========================================================================
func ControlHardware(c *gin.Context) {
	var req DeviceControlRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": "Format data tidak valid"})
		return
	}

	// --- BLOK KONTROL LAMPU ---
	if req.Device == "lampu1" || req.Device == "lampu2" {
		broker := os.Getenv("MQTT_BROKER")
		user := os.Getenv("MQTT_USER")
		pass := os.Getenv("MQTT_PASSWORD")

		opts := mqtt.NewClientOptions()
		opts.AddBroker(broker)
		opts.SetUsername(user)
		opts.SetPassword(pass)
		opts.SetClientID(fmt.Sprintf("Golang-SCLASS-%d", time.Now().Unix()))

		client := mqtt.NewClient(opts)
		if token := client.Connect(); token.Wait() && token.Error() != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Golang gagal terhubung ke MQTT Broker"})
			return
		}
		defer client.Disconnect(250)

		topic := fmt.Sprintf("sclass/d101/%s", req.Device)
		token := client.Publish(topic, 0, false, req.Action)
		token.Wait()

		// ⭐ SIMPAN STATUS TERBARU LAMPU KE MEMORI GOLANG
		DeviceStatusCache[req.Device] = req.Action

		c.JSON(http.StatusOK, gin.H{
			"status":  "success",
			"message": fmt.Sprintf("Berhasil mengirim perintah %s ke %s", req.Action, req.Device),
		})
		return
	}

	// --- BLOK KONTROL AC & PROYEKTOR ---
	if req.Device == "ac" || req.Device == "projector" {
		haURL := os.Getenv("HA_URL")
		haToken := os.Getenv("HA_TOKEN")
		var entityID string

switch req.Device {
		case "ac":
			switch req.Action {
			case "on":
				entityID = "scene.ac_d101_on"
			default:
				entityID = "scene.ac_d101_off"
			}
		case "projector":
			switch req.Action {
			case "on":
				entityID = "scene.projector_d101_on"
			case "off":
				entityID = "scene.projector_d101_off"
			}
		}

		apiURL := fmt.Sprintf("%s/api/services/scene/turn_on", haURL)
		payload := map[string]string{"entity_id": entityID}
		jsonPayload, _ := json.Marshal(payload)

		reqHA, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(jsonPayload))
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Gagal membuat request ke HA"})
			return
		}

		reqHA.Header.Set("Authorization", "Bearer "+haToken)
		reqHA.Header.Set("Content-Type", "application/json")

		httpClient := &http.Client{Timeout: 5 * time.Second}
		resp, err := httpClient.Do(reqHA)
		if err != nil {
			c.JSON(http.StatusGatewayTimeout, gin.H{"error": "Gagal menghubungi Home Assistant"})
			return
		}
		defer resp.Body.Close()

		// ⭐ SIMPAN STATUS TERBARU AC/PROYEKTOR KE MEMORI GOLANG
		DeviceStatusCache[req.Device] = req.Action

		c.JSON(http.StatusOK, gin.H{
			"status":  "success",
			"message": fmt.Sprintf("Berhasil memicu scene %s", entityID),
		})
		return
	}

	c.JSON(http.StatusBadRequest, gin.H{"error": "Device tidak dikenali sistem"})
}

// =========================================================================
// FUNGSI 3: MENDAPATKAN STATUS DAYA DARI HOME ASSISTANT
// =========================================================================
func GetPowerStatus(c *gin.Context) {
	haURL := os.Getenv("HA_URL")
	haToken := os.Getenv("HA_TOKEN")

	entityID := "sensor.kwh_meter_power"
	apiURL := fmt.Sprintf("%s/api/states/%s", haURL, entityID)

	req, _ := http.NewRequest("GET", apiURL, nil)
	req.Header.Set("Authorization", "Bearer "+haToken)
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"power": 0})
		return
	}
	defer resp.Body.Close()

	var result map[string]interface{}
	json.NewDecoder(resp.Body).Decode(&result)

	powerStr, ok := result["state"].(string)
	if !ok {
		c.JSON(http.StatusOK, gin.H{"power": 0})
		return
	}

	c.JSON(http.StatusOK, gin.H{"power": powerStr})
}

// =========================================================================
// FUNGSI 4: MENGIRIM STATUS REAL-TIME KE WEB FRONTEND
// =========================================================================
func GetHardwareStatus(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{
		"status": "success",
		"data":   DeviceStatusCache,
	})
}