1 files changed, 152 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/wireguard/charts.go b/src/go/collectors/go.d.plugin/modules/wireguard/charts.go
new file mode 100644
index 000000000..fe7f89e9b
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/wireguard/charts.go
@@ -0,0 +1,152 @@
+// 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()
+		}
+	}
+}