Spaces:
Paused
Paused
File size: 3,624 Bytes
f16d50c | 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 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 | package platform
import (
"encoding/json"
"errors"
"io"
"net/url"
"strings"
http "github.com/bogdanfinn/fhttp"
"github.com/linweiyuan/go-chatgpt-api/api"
)
func (userLogin *UserLogin) GetAuthorizedUrl(csrfToken string) (string, int, error) {
urlParams := url.Values{
"client_id": {platformAuthClientID},
"audience": {platformAuthAudience},
"redirect_uri": {platformAuthRedirectURL},
"scope": {platformAuthScope},
"response_type": {platformAuthResponseType},
}
req, _ := http.NewRequest(http.MethodGet, platformAuth0Url+urlParams.Encode(), nil)
req.Header.Set("Content-Type", api.ContentType)
req.Header.Set("User-Agent", api.UserAgent)
resp, err := userLogin.client.Do(req)
if err != nil {
return "", http.StatusInternalServerError, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return "", resp.StatusCode, errors.New(api.GetAuthorizedUrlErrorMessage)
}
return resp.Request.URL.String(), http.StatusOK, nil
}
func (userLogin *UserLogin) GetState(authorizedUrl string) (string, int, error) {
split := strings.Split(authorizedUrl, "=")
return split[1], http.StatusOK, nil
}
func (userLogin *UserLogin) CheckUsername(state string, username string) (int, error) {
formParams := url.Values{
"state": {state},
"username": {username},
"js-available": {"true"},
"webauthn-available": {"true"},
"is-brave": {"false"},
"webauthn-platform-available": {"false"},
"action": {"default"},
}
req, _ := http.NewRequest(http.MethodPost, api.LoginUsernameUrl+state, strings.NewReader(formParams.Encode()))
req.Header.Set("Content-Type", api.ContentType)
req.Header.Set("User-Agent", api.UserAgent)
resp, err := userLogin.client.Do(req)
if err != nil {
return http.StatusInternalServerError, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return resp.StatusCode, errors.New(api.EmailInvalidErrorMessage)
}
return http.StatusOK, nil
}
func (userLogin *UserLogin) CheckPassword(state string, username string, password string) (string, int, error) {
formParams := url.Values{
"state": {state},
"username": {username},
"password": {password},
"action": {"default"},
}
req, _ := http.NewRequest(http.MethodPost, api.LoginPasswordUrl+state, strings.NewReader(formParams.Encode()))
req.Header.Set("Content-Type", api.ContentType)
req.Header.Set("User-Agent", api.UserAgent)
resp, err := userLogin.client.Do(req)
if err != nil {
return "", http.StatusInternalServerError, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return "", resp.StatusCode, errors.New(api.EmailOrPasswordInvalidErrorMessage)
}
return resp.Request.URL.Query().Get("code"), http.StatusOK, nil
}
func (userLogin *UserLogin) GetAccessToken(code string) (string, int, error) {
jsonBytes, _ := json.Marshal(GetAccessTokenRequest{
ClientID: platformAuthClientID,
Code: code,
GrantType: platformAuthGrantType,
RedirectURI: platformAuthRedirectURL,
})
req, _ := http.NewRequest(http.MethodPost, getTokenUrl, strings.NewReader(string(jsonBytes)))
req.Header.Set("Content-Type", "application/json")
req.Header.Set("User-Agent", api.UserAgent)
resp, err := userLogin.client.Do(req)
if err != nil {
return "", http.StatusInternalServerError, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return "", resp.StatusCode, errors.New(api.GetAccessTokenErrorMessage)
}
data, _ := io.ReadAll(resp.Body)
return string(data), http.StatusOK, nil
}
|