Add projects

backend/sdk/auth/manager.go

@@ -0,0 +1,95 @@
+package auth
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
+	coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
+)
+
+// Manager aggregates authenticators and coordinates persistence via a token store.
+type Manager struct {
+	authenticators map[string]Authenticator
+	store          coreauth.Store
+}
+
+// NewManager constructs a manager with the provided token store and authenticators.
+// If store is nil, the caller must set it later using SetStore.
+func NewManager(store coreauth.Store, authenticators ...Authenticator) *Manager {
+	mgr := &Manager{
+		authenticators: make(map[string]Authenticator),
+		store:          store,
+	}
+	for i := range authenticators {
+		mgr.Register(authenticators[i])
+	}
+	return mgr
+}
+
+// Register adds or replaces an authenticator keyed by its provider identifier.
+func (m *Manager) Register(a Authenticator) {
+	if a == nil {
+		return
+	}
+	if m.authenticators == nil {
+		m.authenticators = make(map[string]Authenticator)
+	}
+	m.authenticators[a.Provider()] = a
+}
+
+// SetStore updates the token store used for persistence.
+func (m *Manager) SetStore(store coreauth.Store) {
+	m.store = store
+}
+
+// Login executes the provider login flow and persists the resulting auth record.
+func (m *Manager) Login(ctx context.Context, provider string, cfg *config.Config, opts *LoginOptions) (*coreauth.Auth, string, error) {
+	auth, ok := m.authenticators[provider]
+	if !ok {
+		return nil, "", fmt.Errorf("cliproxy auth: authenticator %s not registered", provider)
+	}
+
+	record, err := auth.Login(ctx, cfg, opts)
+	if err != nil {
+		return nil, "", err
+	}
+	if record == nil {
+		return nil, "", fmt.Errorf("cliproxy auth: authenticator %s returned nil record", provider)
+	}
+
+	if m.store == nil {
+		return record, "", nil
+	}
+
+	if cfg != nil {
+		if dirSetter, ok := m.store.(interface{ SetBaseDir(string) }); ok {
+			dirSetter.SetBaseDir(cfg.AuthDir)
+		}
+		if strings.TrimSpace(cfg.AuthDir) != "" {
+			targetFile := record.FileName
+			if targetFile == "" {
+				targetFile = record.ID
+			}
+			if targetFile != "" {
+				fullPath := filepath.Join(cfg.AuthDir, targetFile)
+				if raw, errRead := os.ReadFile(fullPath); errRead == nil && len(raw) > 0 {
+					var existingMap map[string]any
+					if errUnmarshal := json.Unmarshal(raw, &existingMap); errUnmarshal == nil && len(existingMap) > 0 {
+						coreauth.MergeExistingAuthMetadata(record, existingMap)
+					}
+				}
+			}
+		}
+	}
+
+	savedPath, err := m.store.Save(ctx, record)
+	if err != nil {
+		return record, "", err
+	}
+	return record, savedPath, nil
+}