fix: use better url parsing

danielgtaylor · Dec 6, 2023 · 9a25266 · 9a25266
1 parent ac559b9
commit 9a25266
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/oauth/authcode.go b/oauth/authcode.go
@@ -244,8 +244,11 @@ func (ac *AuthorizationCodeTokenSource) Token() (*oauth2.Token, error) {
 	}
 
 	// strip protocol prefix from configured redirect url for local webserver
-	redirectServer := strings.TrimPrefix(ac.getRedirectUrl(), "https://")
-	redirectServer = strings.TrimPrefix(redirectServer, "http://")
+	u, err := url.Parse(ac.getRedirectUrl())
+	if err != nil {
+		panic(err)
+	}
+	redirectServer := fmt.Sprintf("%s:%s", u.Hostname(), u.Port())
 
 	s := &http.Server{
 		Addr:           redirectServer,