package controllers

import (
	"context"
	"encoding/json"
	"log"
	"math/big"
	"net/http"
	"strconv"
	"time"

	"github.com/gin-gonic/gin"
	"gitlab.com/pactual1/backend/config"
	"gitlab.com/pactual1/backend/database/contract"
	"gitlab.com/pactual1/backend/database/device"
	"gitlab.com/pactual1/backend/models"
	"gitlab.com/pactual1/backend/services/blockchain"
	"gitlab.com/pactual1/backend/shared"
)

func SaveDeviceInfo(c *gin.Context) {
	var deviceInfo models.DeviceInfo
	rawData, _ := c.GetRawData()

	// Unmarshal to the important info structure
	err := json.Unmarshal(rawData, &deviceInfo)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid JSON payload"})
		log.Printf("Invalid json payload : %v", err)
		return
	}

	deviceInfo, currentDevice, err := device.SaveDeviceInfoToDB(deviceInfo, rawData)
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}

	if currentDevice.CurrentContractID != nil {
		deviceContract, _, err := contract.GetContractByID(*currentDevice.CurrentContractID)
		if err != nil {
			log.Printf("SaveDeviceInfo - GetContractByID error : %v", err)
			c.JSON(http.StatusInternalServerError, gin.H{"error": "Could not fetch device contract"})
			return
		}

		if deviceContract.Status == models.ContractStatusActive {
			deviceInfoBytes, _ := json.Marshal(deviceInfo)
			deviceInfoEncryptedStr, err := shared.NewEncryptionClient(config.AppConfig.Service.BlockchainSecret).Encrypt(string(deviceInfoBytes))
			if err != nil {
				log.Printf("SaveDeviceInfo - Enrypt error : %v", err)
				c.JSON(http.StatusInternalServerError, gin.H{"error": "Could not encrypt device info"})
				return
			}

			err = blockchain.NewService(config.AppConfig.Blockchain).AddIOTData(context.Background(), shared.CovertUintToByte32(deviceContract.ID), shared.CovertUintToByte32(currentDevice.ID), big.NewInt(time.Now().Unix()), []byte(deviceInfoEncryptedStr))
			if err != nil {
				log.Printf("SaveDeviceInfo CREATE -DeviceInfo Blockchain Error: %v", err)
				c.JSON(http.StatusInternalServerError, gin.H{"error": "Could not save device info in blockchain"})
				return
			}
		}
	}


	log.Printf("Successfully received and saved device info: %v", deviceInfo)
	c.JSON(http.StatusOK, gin.H{"message": "Successfully received and saved device info", "data": deviceInfo})
}

func GetDeviceData(c *gin.Context) {
	// Get the device ID and contract ID from query parameters
	deviceIDStr := c.DefaultQuery("device_id", "")
	contractIDStr := c.DefaultQuery("contract_id", "")

	if deviceIDStr == "" {
		log.Printf("GetDeviceData Error: Device ID is required")
		c.JSON(http.StatusBadRequest, gin.H{"error": "Device ID is required"})
		return
	}

	deviceID, err := strconv.ParseUint(deviceIDStr, 10, 32)
	if err != nil {
		log.Printf("GetDeviceData Error: Invalid Device ID: %v", err)
		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid Device ID"})
		return
	}

	contractID, err := strconv.ParseUint(contractIDStr, 10, 32)
	if err != nil {
		log.Printf("GetDeviceData Error: Invalid Contract ID: %v", err)
		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid Contract ID"})
		return
	}

	contract, st, err := contract.GetContractByID(uint(contractID))
	if err != nil {
		c.JSON(st, gin.H{"error": err.Error()})
		return
	}

	deviceConnectedToContract := false
	for _, contractDeviceID := range contract.DeviceIDs {
		if deviceID == uint64(contractDeviceID) {
			deviceConnectedToContract = true
		}
	}

	if !deviceConnectedToContract {
		log.Printf("Device %v is not connected to contract %v", deviceID, contractID)
		c.JSON(http.StatusInternalServerError, gin.H{"error": "Device is not present int his contract"})
		return
	}

	featureCollection, st, err := device.GetDeviceInfoForContract(deviceID, contract)

	if err != nil {
		c.JSON(st, gin.H{"error": err.Error()})
		return
	}

	// Respond with the GeoJSON feature collection
	c.JSON(http.StatusOK, gin.H{"data": featureCollection})
}

func GetDevicesByContract(c *gin.Context) {
	// Get the contract ID from query parameter
	contractIDStr := c.DefaultQuery("contract_id", "")
	if contractIDStr == "" {
		log.Printf("GetDevicesByContract Error: Contract ID is required")
		c.JSON(http.StatusBadRequest, gin.H{"error": "Contract ID is required"})
		return
	}

	// Convert string to uint
	contractID, err := strconv.ParseUint(contractIDStr, 10, 32)
	if err != nil {
		log.Printf("GetDevicesByContract Error: Invalid Contract ID: %v", err)
		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid Contract ID"})
		return
	}

	log.Printf("This is the  ID: %v", contractID)
	devices, st, err := device.GetDevicesForContract(contractID)

	if err != nil {
		c.JSON(st, gin.H{"error": err.Error()})
		return
	}
	// Respond with the devices
	c.JSON(http.StatusOK, gin.H{"data": models.ConvertDeviceToResponse(devices)})
}