netdata/netdata

// SPDX-License-Identifier: GPL-3.0-or-later

package wireguard

import (
    "fmt"
    "strings"

    "github.com/netdata/netdata/go/go.d.plugin/agent/module"
)

const (
    prioDeviceNetworkIO = module.Priority + iota
    prioDevicePeers
    prioPeerNetworkIO
    prioPeerLatestHandShake
)

var (
    deviceChartsTmpl = module.Charts{
        deviceNetworkIOChartTmpl.Copy(),
        devicePeersChartTmpl.Copy(),
    }

    deviceNetworkIOChartTmpl = module.Chart{
        ID:       "device_%s_network_io",
        Title:    "Device traffic",
        Units:    "B/s",
        Fam:      "device traffic",
        Ctx:      "wireguard.device_network_io",
        Type:     module.Area,
        Priority: prioDeviceNetworkIO,
        Dims: module.Dims{
            {ID: "device_%s_receive", Name: "receive", Algo: module.Incremental},
            {ID: "device_%s_transmit", Name: "transmit", Algo: module.Incremental, Mul: -1},
        },
    }
    devicePeersChartTmpl = module.Chart{
        ID:       "device_%s_peers",
        Title:    "Device peers",
        Units:    "peers",
        Fam:      "device peers",
        Ctx:      "wireguard.device_peers",
        Priority: prioDevicePeers,
        Dims: module.Dims{
            {ID: "device_%s_peers", Name: "peers"},
        },
    }
)

var (
    peerChartsTmpl = module.Charts{
        peerNetworkIOChartTmpl.Copy(),
        peerLatestHandShakeChartTmpl.Copy(),
    }

    peerNetworkIOChartTmpl = module.Chart{
        ID:       "peer_%s_network_io",
        Title:    "Peer traffic",
        Units:    "B/s",
        Fam:      "peer traffic",
        Ctx:      "wireguard.peer_network_io",
        Type:     module.Area,
        Priority: prioPeerNetworkIO,
        Dims: module.Dims{
            {ID: "peer_%s_receive", Name: "receive", Algo: module.Incremental},
            {ID: "peer_%s_transmit", Name: "transmit", Algo: module.Incremental, Mul: -1},
        },
    }
    peerLatestHandShakeChartTmpl = module.Chart{
        ID:       "peer_%s_latest_handshake_ago",
        Title:    "Peer time elapsed since the latest handshake",
        Units:    "seconds",
        Fam:      "peer latest handshake",
        Ctx:      "wireguard.peer_latest_handshake_ago",
        Priority: prioPeerLatestHandShake,
        Dims: module.Dims{
            {ID: "peer_%s_latest_handshake_ago", Name: "time"},
        },
    }
)

func newDeviceCharts(device string) *module.Charts {
    charts := deviceChartsTmpl.Copy()

    for _, c := range *charts {
        c.ID = fmt.Sprintf(c.ID, device)
        c.Labels = []module.Label{
            {Key: "device", Value: device},
        }
        for _, d := range c.Dims {
            d.ID = fmt.Sprintf(d.ID, device)
        }
    }

    return charts
}

func (w *WireGuard) addNewDeviceCharts(device string) {
    charts := newDeviceCharts(device)

    if err := w.Charts().Add(*charts...); err != nil {
        w.Warning(err)
    }
}

func (w *WireGuard) removeDeviceCharts(device string) {
    prefix := fmt.Sprintf("device_%s", device)

    for _, c := range *w.Charts() {
        if strings.HasPrefix(c.ID, prefix) {
            c.MarkRemove()
            c.MarkNotCreated()
        }
    }
}

func newPeerCharts(id, device, pubKey string) *module.Charts {
    charts := peerChartsTmpl.Copy()

    for _, c := range *charts {
        c.ID = fmt.Sprintf(c.ID, id)
        c.Labels = []module.Label{
            {Key: "device", Value: device},
            {Key: "public_key", Value: pubKey},
        }
        for _, d := range c.Dims {
            d.ID = fmt.Sprintf(d.ID, id)
        }
    }

    return charts
}

func (w *WireGuard) addNewPeerCharts(id, device, pubKey string) {
    charts := newPeerCharts(id, device, pubKey)

    if err := w.Charts().Add(*charts...); err != nil {
        w.Warning(err)
    }
}

func (w *WireGuard) removePeerCharts(id string) {
    prefix := fmt.Sprintf("peer_%s", id)

    for _, c := range *w.Charts() {
        if strings.HasPrefix(c.ID, prefix) {
            c.MarkRemove()
            c.MarkNotCreated()
        }
    }
}