File size: 1,531 Bytes
93d826e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 | package api
import (
"encoding/json"
"log"
"net/http"
"plandex-cli/auth"
"plandex-cli/term"
"strings"
shared "plandex-shared"
)
func HandleApiError(r *http.Response, errBody []byte) *shared.ApiError {
// Check if the response is JSON
if r.Header.Get("Content-Type") != "application/json" {
return &shared.ApiError{
Type: shared.ApiErrorTypeOther,
Status: r.StatusCode,
Msg: strings.TrimSpace(string(errBody)),
}
}
var apiError shared.ApiError
if err := json.Unmarshal(errBody, &apiError); err != nil {
log.Printf("Error unmarshalling JSON: %v\n", err)
return &shared.ApiError{
Type: shared.ApiErrorTypeOther,
Status: r.StatusCode,
Msg: strings.TrimSpace(string(errBody)),
}
}
// return error if token/auth refresh is needed
if apiError.Type == shared.ApiErrorTypeInvalidToken || apiError.Type == shared.ApiErrorTypeAuthOutdated {
return &apiError
}
term.HandleApiError(&apiError)
return &apiError
}
func refreshAuthIfNeeded(apiErr *shared.ApiError) (bool, *shared.ApiError) {
if apiErr.Type == shared.ApiErrorTypeInvalidToken {
err := auth.RefreshInvalidToken()
if err != nil {
return false, &shared.ApiError{Type: shared.ApiErrorTypeOther, Msg: "error refreshing invalid token"}
}
return true, nil
} else if apiErr.Type == shared.ApiErrorTypeAuthOutdated {
err := auth.RefreshAuth()
if err != nil {
return false, &shared.ApiError{Type: shared.ApiErrorTypeOther, Msg: "error refreshing auth"}
}
return true, nil
}
return false, apiErr
}
|