90 lines
1.7 KiB
Go
90 lines
1.7 KiB
Go
package authclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/steipete/gogcli/internal/config"
|
|
)
|
|
|
|
type (
|
|
contextKey struct{}
|
|
accessTokenKey struct{}
|
|
)
|
|
|
|
func WithClient(ctx context.Context, client string) context.Context {
|
|
client = strings.TrimSpace(client)
|
|
if client == "" {
|
|
return ctx
|
|
}
|
|
|
|
return context.WithValue(ctx, contextKey{}, client)
|
|
}
|
|
|
|
func ClientOverrideFromContext(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
|
|
if v := ctx.Value(contextKey{}); v != nil {
|
|
if s, ok := v.(string); ok {
|
|
return s
|
|
}
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func WithAccessToken(ctx context.Context, token string) context.Context {
|
|
token = strings.TrimSpace(token)
|
|
if token == "" {
|
|
return ctx
|
|
}
|
|
|
|
return context.WithValue(ctx, accessTokenKey{}, token)
|
|
}
|
|
|
|
func AccessTokenFromContext(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
|
|
if v := ctx.Value(accessTokenKey{}); v != nil {
|
|
if s, ok := v.(string); ok {
|
|
return s
|
|
}
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func ResolveClient(ctx context.Context, email string) (string, error) {
|
|
cfg, err := config.ReadConfig()
|
|
if err != nil {
|
|
return "", fmt.Errorf("read config: %w", err)
|
|
}
|
|
override := ClientOverrideFromContext(ctx)
|
|
|
|
client, err := config.ResolveClientForAccount(cfg, email, override)
|
|
if err != nil {
|
|
return "", fmt.Errorf("resolve client: %w", err)
|
|
}
|
|
|
|
return client, nil
|
|
}
|
|
|
|
func ResolveClientWithOverride(email string, override string) (string, error) {
|
|
cfg, err := config.ReadConfig()
|
|
if err != nil {
|
|
return "", fmt.Errorf("read config: %w", err)
|
|
}
|
|
|
|
client, err := config.ResolveClientForAccount(cfg, email, override)
|
|
if err != nil {
|
|
return "", fmt.Errorf("resolve client: %w", err)
|
|
}
|
|
|
|
return client, nil
|
|
}
|