From f9a954fbd3cbca686cd6f7786065bb0a7f4da75d Mon Sep 17 00:00:00 2001
From: Medzik <nivua1fn@duck.com>
Date: Fri, 15 Oct 2021 19:15:29 +0000
Subject: [PATCH] feat: add UserAgent

---
 config/vars.go | 8 +++++++-
 main.go        | 1 +
 ping/ping.go   | 3 +++
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/config/vars.go b/config/vars.go
index 4eadead..4297b23 100644
--- a/config/vars.go
+++ b/config/vars.go
@@ -1,6 +1,11 @@
 package config
 
-import "time"
+import (
+	"fmt"
+	"runtime"
+	"strings"
+	"time"
+)
 
 const GH_Repo = "MedzikUser/go-pingbot"
 
@@ -8,4 +13,5 @@ var (
 	Version   = "dev"
 	Build     = ""
 	StartTime time.Time
+	UserAgent = fmt.Sprintf("PingBot/%s go/%s", Version, strings.Replace(runtime.Version(), "go", "", 1))
 )
diff --git a/main.go b/main.go
index 94cb502..5636172 100644
--- a/main.go
+++ b/main.go
@@ -24,6 +24,7 @@ type cmdOpts struct {
 
 func main() {
 	log.Info("You're using verion: ", config.Version)
+	log.Info(config.UserAgent)
 
 	var wg sync.WaitGroup
 
diff --git a/ping/ping.go b/ping/ping.go
index da6ea05..e699d40 100644
--- a/ping/ping.go
+++ b/ping/ping.go
@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/MedzikUser/go-utils/common"
+	"gitlab.com/gaming0skar123/go/pingbot/config"
 )
 
 func ping() int {
@@ -30,6 +31,8 @@ func loop(url string) {
 		return
 	}
 
+	req.Header.Set("User-Agent", config.UserAgent)
+
 	client := http.DefaultClient
 	r, err := client.Do(req)
 	if err != nil {