package influxunifi

import (
	"time"

	influx "github.com/influxdata/influxdb1-client/v2"
	"golift.io/unifi"
)

// ClientPoints generates Unifi Client datapoints for InfluxDB.
// These points can be passed directly to influx.
func ClientPoints(c *unifi.Client, now time.Time) ([]*influx.Point, error) {
	tags := map[string]string{
		"id":          c.ID,
		"mac":         c.Mac,
		"user_id":     c.UserID,
		"site_id":     c.SiteID,
		"site_name":   c.SiteName,
		"ap_mac":      c.ApMac,
		"gw_mac":      c.GwMac,
		"sw_mac":      c.SwMac,
		"ap_name":     c.ApName,
		"gw_name":     c.GwName,
		"sw_name":     c.SwName,
		"oui":         c.Oui,
		"radio_name":  c.RadioName,
		"radio":       c.Radio,
		"radio_proto": c.RadioProto,
		"name":        c.Name,
		"fixed_ip":    c.FixedIP,
		"sw_port":     c.SwPort.Txt,
		"os_class":    c.OsClass.Txt,
		"os_name":     c.OsName.Txt,
		"dev_cat":     c.DevCat.Txt,
		"dev_id":      c.DevID.Txt,
		"dev_vendor":  c.DevVendor.Txt,
		"dev_family":  c.DevFamily.Txt,
		"is_wired":    c.IsWired.Txt,
		"is_guest":    c.IsGuest.Txt,
		"use_fixedip": c.UseFixedIP.Txt,
		"channel":     c.Channel.Txt,
		"vlan":        c.Vlan.Txt,
	}
	fields := map[string]interface{}{
		"anomalies":        c.Anomalies,
		"ip":               c.IP,
		"essid":            c.Essid,
		"bssid":            c.Bssid,
		"radio_desc":       c.RadioDescription,
		"satisfaction":     c.Satisfaction.Val,
		"hostname":         c.Hostname,
		"bytes_r":          c.BytesR,
		"ccq":              c.Ccq,
		"first_seen":       c.FirstSeen,
		"idle_time":        c.IdleTime,
		"last_seen":        c.LastSeen,
		"network":          c.Network,
		"noise":            c.Noise,
		"note":             c.Note,
		"roam_count":       c.RoamCount,
		"rssi":             c.Rssi,
		"rx_bytes":         c.RxBytes,
		"rx_bytes_r":       c.RxBytesR,
		"rx_packets":       c.RxPackets,
		"rx_rate":          c.RxRate,
		"signal":           c.Signal,
		"tx_bytes":         c.TxBytes,
		"tx_bytes_r":       c.TxBytesR,
		"tx_packets":       c.TxPackets,
		"tx_retries":       c.TxRetries,
		"tx_power":         c.TxPower,
		"tx_rate":          c.TxRate,
		"uptime":           c.Uptime,
		"wifi_tx_attempts": c.WifiTxAttempts,
		"wired-rx_bytes":   c.WiredRxBytes,
		"wired-rx_bytes-r": c.WiredRxBytesR,
		"wired-rx_packets": c.WiredRxPackets,
		"wired-tx_bytes":   c.WiredTxBytes,
		"wired-tx_bytes-r": c.WiredTxBytesR,
		"wired-tx_packets": c.WiredTxPackets,
		"dpi_app":          c.DpiStats.App.Val,
		"dpi_cat":          c.DpiStats.Cat.Val,
		"dpi_rx_bytes":     c.DpiStats.RxBytes.Val,
		"dpi_rx_packets":   c.DpiStats.RxPackets.Val,
		"dpi_tx_bytes":     c.DpiStats.TxBytes.Val,
		"dpi_tx_packets":   c.DpiStats.TxPackets.Val,
	}
	pt, err := influx.NewPoint("clients", tags, fields, now)
	if err != nil {
		return nil, err
	}
	return []*influx.Point{pt}, nil
}