From 2dc58c5c8ebdafcd46ddf7dee7b8ff811b64c273 Mon Sep 17 00:00:00 2001
From: Aidan Holm <aidanholm@gmail.com>
Date: Tue, 29 Oct 2024 22:34:24 +0800
Subject: [PATCH] Fix user agent using hardcoded 1.0 version.

---
 cmd/yarr/main.go     | 2 ++
 src/worker/client.go | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/cmd/yarr/main.go b/cmd/yarr/main.go
index dbb4680..9f91752 100644
--- a/cmd/yarr/main.go
+++ b/cmd/yarr/main.go
@@ -13,6 +13,7 @@ import (
 	"github.com/nkanaev/yarr/src/platform"
 	"github.com/nkanaev/yarr/src/server"
 	"github.com/nkanaev/yarr/src/storage"
+	"github.com/nkanaev/yarr/src/worker"
 )
 
 var Version string = "0.0"
@@ -131,6 +132,7 @@ func main() {
 		log.Fatal("Failed to initialise database: ", err)
 	}
 
+	worker.SetVersion(Version)
 	srv := server.NewServer(store, addr)
 
 	if basepath != "" {
diff --git a/src/worker/client.go b/src/worker/client.go
index d7cac5d..eee2edd 100644
--- a/src/worker/client.go
+++ b/src/worker/client.go
@@ -32,6 +32,10 @@ func (c *Client) getConditional(url, lastModified, etag string) (*http.Response,
 
 var client *Client
 
+func SetVersion(num string) {
+	client.userAgent = "Yarr/" + num
+}
+
 func init() {
 	transport := &http.Transport{
 		Proxy: http.ProxyFromEnvironment,