package api

import (
	"encoding/json"
	"fmt"
	"log"
	"net"

	"github.com/cilium/cilium/pkg/mac"
	"github.com/dustin/go-humanize"
	"sinanmohd.com/redq/bpf/usage"
)

type BandwidthStat struct {
	Ingress string `json:"ingress"`
	Egress  string `json:"egress"`
}

type BandwidthResp map[string]BandwidthStat

func handleBandwidth(conn net.Conn, u *usage.Usage) {
	resp := make(BandwidthResp)
	var ingressTotal, egressTotal uint64

	u.Mutex.RLock()
	for key, value := range u.Data {
		ingressTotal += value.BandwidthIngress
		egressTotal += value.BandwidthEgress

		m := mac.Uint64MAC(key)
		resp[m.String()] = BandwidthStat{
			Ingress: fmt.Sprintf("%s/s", humanize.Bytes(value.BandwidthIngress)),
			Egress:  fmt.Sprintf("%s/s", humanize.Bytes(value.BandwidthEgress)),
		}
	}
	u.Mutex.RUnlock()

	resp["total"] = BandwidthStat{
		Ingress: fmt.Sprintf("%s/s", humanize.Bytes(ingressTotal)),
		Egress:  fmt.Sprintf("%s/s", humanize.Bytes(egressTotal)),
	}

	buf, err := json.Marshal(resp)
	if err != nil {
		log.Printf("marshaling json: %s", err)
		return
	}

	conn.Write(buf)
}