Luis Pater
e50cabac4b
- Updated all references from v6 to v7 for `github.com/router-for-me/CLIProxyAPI`. - Ensured consistency in imports within core libraries, tests, and integration tests. - Added missing tests for new features in Redis Protocol integration.
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v7/internal/auth/codex"
|
|
"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
|
|
sdkAuth "github.com/router-for-me/CLIProxyAPI/v7/sdk/auth"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
codexLoginModeMetadataKey = "codex_login_mode"
|
|
codexLoginModeDevice = "device"
|
|
)
|
|
|
|
// DoCodexDeviceLogin triggers the Codex device-code flow while keeping the
|
|
// existing codex-login OAuth callback flow intact.
|
|
func DoCodexDeviceLogin(cfg *config.Config, options *LoginOptions) {
|
|
if options == nil {
|
|
options = &LoginOptions{}
|
|
}
|
|
|
|
promptFn := options.Prompt
|
|
if promptFn == nil {
|
|
promptFn = defaultProjectPrompt()
|
|
}
|
|
|
|
manager := newAuthManager()
|
|
|
|
authOpts := &sdkAuth.LoginOptions{
|
|
NoBrowser: options.NoBrowser,
|
|
CallbackPort: options.CallbackPort,
|
|
Metadata: map[string]string{
|
|
codexLoginModeMetadataKey: codexLoginModeDevice,
|
|
},
|
|
Prompt: promptFn,
|
|
}
|
|
|
|
_, savedPath, err := manager.Login(context.Background(), "codex", cfg, authOpts)
|
|
if err != nil {
|
|
if authErr, ok := errors.AsType[*codex.AuthenticationError](err); ok {
|
|
log.Error(codex.GetUserFriendlyMessage(authErr))
|
|
if authErr.Type == codex.ErrPortInUse.Type {
|
|
os.Exit(codex.ErrPortInUse.Code)
|
|
}
|
|
return
|
|
}
|
|
fmt.Printf("Codex device authentication failed: %v\n", err)
|
|
return
|
|
}
|
|
|
|
if savedPath != "" {
|
|
fmt.Printf("Authentication saved to %s\n", savedPath)
|
|
}
|
|
fmt.Println("Codex device authentication successful!")
|
|
}
|