package api

import (
	"encoding/json"
	"log"
	"net"
	"net/http"
	"time"

	"github.com/leandrofars/oktopus/internal/db"
)

type StatusCount struct {
	Online  int
	Offline int
}

type GeneralInfo struct {
	MqttRtt           time.Duration
	ProductClassCount []db.ProductClassCount
	StatusCount       StatusCount
	VendorsCount      []db.VendorsCount
}

// TODO: fix when mqtt broker is not set don't break api
func (a *Api) generalInfo(w http.ResponseWriter, r *http.Request) {

	var result GeneralInfo

	productclasscount, err := a.Db.RetrieveProductsClassInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	vendorcount, err := a.Db.RetrieveVendorsInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	statuscount, err := a.Db.RetrieveStatusInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	for _, v := range statuscount {
		switch db.Status(v.Status) {
		case db.Online:
			result.StatusCount.Online = v.Count
		case db.Offline:
			result.StatusCount.Offline = v.Count
		}
	}

	result.VendorsCount = vendorcount
	result.ProductClassCount = productclasscount

	conn, err := net.Dial("tcp", a.Mqtt.Addr+":"+a.Mqtt.Port)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		json.NewEncoder(w).Encode("Error to connect to broker: " + err.Error())
		return
	}
	defer conn.Close()

	info, err := tcpInfo(conn.(*net.TCPConn))
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		json.NewEncoder(w).Encode("Error to get TCP socket info")
		return
	}
	rtt := time.Duration(info.Rtt) * time.Microsecond

	result.MqttRtt = rtt / 1000

	err = json.NewEncoder(w).Encode(result)
	if err != nil {
		log.Println(err)
	}
}

func (a *Api) vendorsInfo(w http.ResponseWriter, r *http.Request) {
	vendors, err := a.Db.RetrieveVendorsInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
	err = json.NewEncoder(w).Encode(vendors)
	if err != nil {
		log.Println(err)
	}
}

func (a *Api) productClassInfo(w http.ResponseWriter, r *http.Request) {
	vendors, err := a.Db.RetrieveProductsClassInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
	err = json.NewEncoder(w).Encode(vendors)
	if err != nil {
		log.Println(err)
	}
}

func (a *Api) statusInfo(w http.ResponseWriter, r *http.Request) {
	vendors, err := a.Db.RetrieveStatusInfo()
	if err != nil {
		log.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	var status StatusCount
	for _, v := range vendors {
		switch db.Status(v.Status) {
		case db.Online:
			status.Online = v.Count
		case db.Offline:
			status.Offline = v.Count
		}
	}

	err = json.NewEncoder(w).Encode(status)
	if err != nil {
		log.Println(err)
	}
}