From 08a2d7f41d10974aa74971feea9663af023c2fdb Mon Sep 17 00:00:00 2001
From: Giovanni Harting <539@idlegandalf.com>
Date: Fri, 14 Apr 2023 20:48:05 +0200
Subject: [PATCH] added waybar support

---
 ping-package.go | 60 +++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 51 insertions(+), 9 deletions(-)

diff --git a/ping-package.go b/ping-package.go
index 8126d0e..40dfc7e 100644
--- a/ping-package.go
+++ b/ping-package.go
@@ -1,17 +1,32 @@
 package main
 
 import (
+	"encoding/json"
+	"flag"
 	"fmt"
 	"github.com/prometheus-community/pro-bing"
 	"math"
+	"strconv"
 	"time"
 )
 
-const (
-	PingCount    = 3
-	PingInterval = 5
+var (
+	Waybar              = flag.Bool("waybar", false, "output waybar json format")
+	PingCount           = flag.Int("count", 3, "how many pings to average")
+	PingInterval        = flag.Int("interval", 5, "ping interval")
+	PingWarningLimit    = flag.Int("warning", 50, "ping warn limit")
+	PingCritLimit       = flag.Int("crit", 100, "ping crit limit")
+	PacketLossWarnLimit = flag.Int("pwarn", 10, "package loss warn limit")
+	PacketLossCritLimit = flag.Int("pcrit", 25, "package loss crit limit")
+	Host                = flag.String("host", "google.com", "host to ping")
 )
 
+type WaybarOut struct {
+	Class       string `json:"class"`
+	PackageLoss string `json:"package_loss"`
+	Ping        string `json:"ping"`
+}
+
 func formatLine(stats *probing.Statistics) {
 	if stats.PacketLoss >= 100.0 {
 		// fontawesome/forkawesome doesn't have the fitting icon...
@@ -46,17 +61,44 @@ func formatLine(stats *probing.Statistics) {
 	fmt.Printf("%s%dms %%{F-}| %s%d%%\n", rttColor, stats.AvgRtt.Milliseconds(), packetColor, int(math.Round(stats.PacketLoss)))
 }
 
-func main() {
-	for {
-		time.Sleep(time.Duration(PingInterval) * time.Second)
+func formatLineWaybar(stats *probing.Statistics) {
+	res := new(WaybarOut)
 
-		pinger, err := probing.NewPinger("itsh.dev")
+	res.PackageLoss = strconv.Itoa(int(math.Round(stats.PacketLoss)))
+	res.Ping = strconv.Itoa(int(stats.AvgRtt.Milliseconds()))
+
+	switch {
+	case int(math.Round(stats.PacketLoss)) > *PacketLossWarnLimit || int(stats.AvgRtt.Milliseconds()) > *PingWarningLimit:
+		res.Class = "warning"
+	case int(math.Round(stats.PacketLoss)) > *PacketLossCritLimit || int(stats.AvgRtt.Milliseconds()) > *PingCritLimit:
+		res.Class = "critical"
+	}
+
+	jOut, err := json.Marshal(res)
+	if err != nil {
+		return
+	}
+
+	fmt.Println(string(jOut))
+}
+
+func main() {
+	flag.Parse()
+
+	for {
+		time.Sleep(time.Duration(*PingInterval) * time.Second)
+
+		pinger, err := probing.NewPinger(*Host)
 		if err != nil {
 			fmt.Println(err)
 			continue
 		}
-		pinger.Count = PingCount
-		pinger.OnFinish = formatLine
+		pinger.Count = *PingCount
+		if *Waybar {
+			pinger.OnFinish = formatLineWaybar
+		} else {
+			pinger.OnFinish = formatLine
+		}
 
 		if pinger.Run() != nil {
 			fmt.Println(err)