mirror of
https://github.com/larksuite/cli.git
synced 2026-07-07 00:55:53 +08:00
a6de8360f0
* feat(auth): add scope hint for missing authorization errors * fix(auth): handle existing hints in missing scope error * refactor(auth): centralize user authorization error detection * fix(auth): handle nil error case in IsNeedUserAuthorizationError
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package auth
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/larksuite/cli/internal/output"
|
|
)
|
|
|
|
const (
|
|
LarkErrBlockByPolicy = 21001 // access denied by access control policy
|
|
LarkErrBlockByPolicyTryAuth = 21000 // access denied by access control policy; challenge is required to be completed by user in order to gain access
|
|
needUserAuthorizationMarker = "need_user_authorization"
|
|
)
|
|
|
|
// RefreshTokenRetryable contains error codes that allow one immediate retry.
|
|
// All other refresh errors clear the token immediately.
|
|
var RefreshTokenRetryable = map[int]bool{
|
|
output.LarkErrRefreshServerError: true,
|
|
}
|
|
|
|
// TokenRetryCodes contains error codes that allow retry after token refresh.
|
|
var TokenRetryCodes = map[int]bool{
|
|
output.LarkErrTokenInvalid: true,
|
|
output.LarkErrTokenExpired: true,
|
|
}
|
|
|
|
// NeedAuthorizationError is thrown when no valid UAT exists.
|
|
type NeedAuthorizationError struct {
|
|
UserOpenId string
|
|
}
|
|
|
|
// Error returns the error message for NeedAuthorizationError.
|
|
func (e *NeedAuthorizationError) Error() string {
|
|
return fmt.Sprintf("%s (user: %s)", needUserAuthorizationMarker, e.UserOpenId)
|
|
}
|
|
|
|
// IsNeedUserAuthorizationError reports whether err represents a missing-UAT
|
|
// failure, either as the original auth error or as a wrapped ExitError.
|
|
func IsNeedUserAuthorizationError(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
|
|
var needAuthErr *NeedAuthorizationError
|
|
if errors.As(err, &needAuthErr) {
|
|
return true
|
|
}
|
|
|
|
var exitErr *output.ExitError
|
|
if errors.As(err, &exitErr) && exitErr.Detail != nil {
|
|
return strings.Contains(exitErr.Detail.Message, needUserAuthorizationMarker)
|
|
}
|
|
return strings.Contains(err.Error(), needUserAuthorizationMarker)
|
|
}
|
|
|
|
// SecurityPolicyError is returned when a request is blocked by access control policies.
|
|
type SecurityPolicyError struct {
|
|
Code int
|
|
Message string
|
|
ChallengeURL string
|
|
CLIHint string
|
|
Err error
|
|
}
|
|
|
|
// Error returns the error message for SecurityPolicyError.
|
|
func (e *SecurityPolicyError) Error() string {
|
|
if e.Err != nil {
|
|
return fmt.Sprintf("security policy error [%d]: %s: %v", e.Code, e.Message, e.Err)
|
|
}
|
|
return fmt.Sprintf("security policy error [%d]: %s", e.Code, e.Message)
|
|
}
|
|
|
|
// Unwrap returns the underlying error.
|
|
func (e *SecurityPolicyError) Unwrap() error {
|
|
return e.Err
|
|
}
|