X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/d9fddc0f506eb4ead7049f5638674b2b8773fcaa..HEAD:/chatgpt.go?ds=sidebyside

diff --git a/chatgpt.go b/chatgpt.go
index 7a5afc2..8849ada 100644
--- a/chatgpt.go
+++ b/chatgpt.go
@@ -131,6 +131,7 @@ func (a *ChatGPTAuth) Login(ctx context.Context, out io.Writer) error {
 	defer a.mutex.Unlock()
 
 	var code ChatGPTDeviceCode
+
 	err := a.postJSON(ctx, "/api/accounts/deviceauth/usercode", map[string]string{
 		"client_id": chatgptClientID,
 	}, &code)
@@ -183,6 +184,7 @@ func (a *ChatGPTAuth) read() (ChatGPTTokenFile, error) {
 	}
 
 	var tokens ChatGPTTokenFile
+
 	err = json.Unmarshal(data, &tokens)
 	if err == nil && (tokens.AccessToken == "" || tokens.RefreshToken == "" || tokens.AccountID == "") {
 		err = errors.New("a token or the account ID is missing")
@@ -199,6 +201,7 @@ func (a *ChatGPTAuth) read() (ChatGPTTokenFile, error) {
 // It keeps the old refresh token and ID token when the answer has none.
 func (a *ChatGPTAuth) refresh(ctx context.Context, tokens ChatGPTTokenFile) (ChatGPTTokenFile, error) {
 	var response ChatGPTTokenResponse
+
 	err := a.postJSON(ctx, "/oauth/token", map[string]string{
 		"client_id":     chatgptClientID,
 		"grant_type":    "refresh_token",
@@ -255,6 +258,7 @@ func (a *ChatGPTAuth) poll(ctx context.Context, deviceID, userCode string, inter
 
 	for {
 		var grant ChatGPTDeviceGrant
+
 		status, err := a.post(ctx, "/api/accounts/deviceauth/token", "application/json", body, &grant)
 		if err == nil && (grant.AuthorizationCode == "" || grant.CodeVerifier == "") {
 			err = errors.New("device grant has no authorization code or code verifier")
@@ -272,6 +276,7 @@ func (a *ChatGPTAuth) poll(ctx context.Context, deviceID, userCode string, inter
 		select {
 		case <-ctx.Done():
 			timer.Stop()
+
 			return ChatGPTDeviceGrant{}, fmt.Errorf("wait for approval: %w", ctx.Err())
 		case <-timer.C:
 		}
@@ -289,6 +294,7 @@ func (a *ChatGPTAuth) exchange(ctx context.Context, grant ChatGPTDeviceGrant) (C
 	}
 
 	var response ChatGPTTokenResponse
+
 	_, err := a.post(ctx, "/oauth/token", "application/x-www-form-urlencoded", []byte(form.Encode()), &response)
 	if err != nil {
 		return ChatGPTTokenFile{}, fmt.Errorf("exchange code for tokens: %w", err)
@@ -337,6 +343,7 @@ func (a *ChatGPTAuth) post(ctx context.Context, path, contentType string, body [
 	}
 
 	request.Header.Set("Content-Type", contentType)
+
 	response, err := a.http.Do(request)
 	if err != nil {
 		return 0, fmt.Errorf("send request: %w", err)
@@ -417,6 +424,7 @@ func chatgptPollInterval(raw json.RawMessage) time.Duration {
 // none.
 func chatgptAccountID(token string) string {
 	var claims map[string]json.RawMessage
+
 	if !chatgptDecodeClaims(token, &claims) {
 		return ""
 	}