From 136bd2c802f75717a8f1778d20ee98b87968cd28 Mon Sep 17 00:00:00 2001
From: Daniil Tsivinsky <daniil@tsivinsky.com>
Date: Sat, 7 Mar 2026 20:26:42 +0300
Subject: [PATCH] refresh token when needed and use oauth2.Client for requests

---
 main.go | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 2b9559e..203f5db 100644
--- a/main.go
+++ b/main.go
@@ -34,6 +34,8 @@ func main() {
 		RedirectURL: os.Getenv("REDIRECT_URI"),
 	}
 
+	oauthClient := oauthConfig.Client(context.Background(), token)
+
 	mux := http.NewServeMux()
 
 	mux.HandleFunc("GET /oauth", func(w http.ResponseWriter, r *http.Request) {
@@ -47,6 +49,7 @@ func main() {
 			return
 		}
 
+		oauthClient = oauthConfig.Client(context.Background(), token)
 		if err := writeConfig(authConfigPath, token); err != nil {
 			http.Error(w, err.Error(), 500)
 			return
@@ -57,6 +60,20 @@ func main() {
 	})
 
 	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		tokenSource := oauthConfig.TokenSource(context.Background(), token)
+		newToken, err := tokenSource.Token()
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to get new token: %v", err), 500)
+			return
+		}
+
+		if newToken.AccessToken != token.AccessToken {
+			if err := writeConfig(authConfigPath, newToken); err != nil {
+				http.Error(w, fmt.Sprintf("failed to save new token: %v", err), 500)
+				return
+			}
+		}
+
 		baseURL := os.Getenv("API_URL")
 		req, err := http.NewRequest(r.Method, baseURL+r.URL.Path, r.Body)
 		if err != nil {
@@ -65,12 +82,7 @@ func main() {
 		}
 		req.URL.RawQuery = r.URL.RawQuery
 
-		if token.AccessToken != "" {
-			authHeader := fmt.Sprintf("%s %s", token.TokenType, token.AccessToken)
-			req.Header.Add("Authorization", authHeader)
-		}
-
-		resp, err := http.DefaultClient.Do(req)
+		resp, err := oauthClient.Do(req)
 		if err != nil {
 			http.Error(w, err.Error(), 500)
 			return