working on botTokens

This commit is contained in:
bryce
2025-07-22 11:33:24 +12:00
parent b0be1f99c8
commit 963e829b7f
5 changed files with 89 additions and 9 deletions

View File

@@ -39,6 +39,7 @@ func StartOAuthServer(addr string) error {
http.HandleFunc("/", handleIndex)
http.HandleFunc("/callback", handleCallback)
http.HandleFunc("/refresh", handleRefresh)
log.Printf("OAuth server listening on %s", addr)
return http.ListenAndServe(addr, nil)
@@ -46,7 +47,7 @@ func StartOAuthServer(addr string) error {
func handleIndex(w http.ResponseWriter, r *http.Request) {
authURL := fmt.Sprintf(
"https://id.twitch.tv/oauth2/authorize?response_type=code&client_id=%s&redirect_uri=%s&scope=%s",
"https://id.twitch.tv/oauth2/authorize?response_type=code&client_id=%s&redirect_uri=%s&scope=%s&force_verify=true",
url.QueryEscape(clientID),
url.QueryEscape(redirectURI),
url.QueryEscape(strings.Join(scopes, " ")),
@@ -77,10 +78,26 @@ func handleCallback(w http.ResponseWriter, r *http.Request) {
}
jsonData, _ := json.MarshalIndent(token, "", " ")
w.Header().Set("Content-Type", "application/json")
w.Header().Set("Content-Type", "application/ijson")
w.Write(jsonData)
}
func handleRefresh(w http.ResponseWriter, r *http.Request) {
refresh := r.URL.Query().Get("refresh_token")
if refresh == "" {
http.Error(w, "Missing refresh_token parameter", http.StatusBadRequest)
return
}
tr, err := ExchangeRefreshToken(refresh)
if err != nil {
http.Error(w, "Refresh failed: "+err.Error(), http.StatusInternalServerError)
return
}
data, _ := json.MarshalIndent(tr, "", " ")
w.Header().Set("Content-Type", "application/json")
w.Write(data)
}
func exchangeCodeForToken(code string) (*TokenResponse, error) {
data := url.Values{}
data.Set("client_id", clientID)
@@ -110,3 +127,31 @@ func exchangeCodeForToken(code string) (*TokenResponse, error) {
return &tokenResp, nil
}
// refresh tokens programtically
func ExchangeRefreshToken(refreshToken string) (*TokenResponse, error) {
form := url.Values{}
form.Set("client_id", clientID)
form.Set("client_secret", clientSecret)
form.Set("grant_type", "refresh_token")
form.Set("refresh_token", refreshToken)
resp, err := http.Post(
"https://id.twitch.tv/oauth2/token",
"application/x-www-form-urlencoded",
strings.NewReader(form.Encode()),
)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("refresh failed: %s", resp.Status)
}
var tr TokenResponse
if err := json.NewDecoder(resp.Body).Decode(&tr); err != nil {
return nil, err
}
return &tr, nil
}