v0.3.8

internal/cursor/state_db.go

@@ -0,0 +1,145 @@
+package cursor
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"sort"
+	"strings"
+
+	"cursor/internal/logger"
+
+	_ "modernc.org/sqlite"
+)
+
+const (
+	cursorStateMembershipType      = "ultra"
+	cursorStateSubscriptionStatus  = "active"
+	cursorStateDefaultSignUpType   = "Google"
+	cursorStateSQLiteBusyTimeoutMS = 2000
+	cursorStateDBRelativePath      = "Cursor/User/globalStorage/state.vscdb"
+	cursorStateDarwinRelativePath  = "Library/Application Support/Cursor/User/globalStorage/state.vscdb"
+	cursorStateLinuxRelativePath   = ".config/Cursor/User/globalStorage/state.vscdb"
+)
+
+// InjectCursorUserInfo synchronizes the Cursor user-level auth cache used by the
+// Settings page. It does not modify the installed Cursor app bundle.
+func InjectCursorUserInfo(email, token string) error {
+	stateDBPath, err := resolveCursorStateDBPath()
+	if err != nil {
+		return err
+	}
+	if err := os.MkdirAll(filepath.Dir(stateDBPath), 0o755); err != nil {
+		return fmt.Errorf("创建 Cursor 状态目录失败: %w", err)
+	}
+
+	values := buildCursorAuthStateValues(email, token)
+	if err := syncCursorAuthStateDB(stateDBPath, values); err != nil {
+		return fmt.Errorf("同步 Cursor 状态库失败 path=%s: %w", stateDBPath, err)
+	}
+
+	logger.Infof(
+		"injectCursorUserInfo synced path=%s email=%s membership=%s subscription=%s",
+		stateDBPath,
+		values["cursorAuth/cachedEmail"],
+		values["cursorAuth/stripeMembershipType"],
+		values["cursorAuth/stripeSubscriptionStatus"],
+	)
+	return nil
+}
+
+func buildCursorAuthStateValues(email, token string) map[string]string {
+	email = strings.TrimSpace(email)
+	token = strings.TrimSpace(token)
+
+	return map[string]string{
+		"cursorAuth/accessToken":              token,
+		"cursorAuth/cachedEmail":              email,
+		"cursorAuth/cachedSignUpType":         cursorStateDefaultSignUpType,
+		"cursorAuth/refreshToken":             token,
+		"cursorAuth/stripeMembershipType":     cursorStateMembershipType,
+		"cursorAuth/stripeSubscriptionStatus": cursorStateSubscriptionStatus,
+	}
+}
+
+func syncCursorAuthStateDB(path string, values map[string]string) error {
+	db, err := sql.Open("sqlite", path)
+	if err != nil {
+		return err
+	}
+	defer db.Close()
+	db.SetMaxOpenConns(1)
+	db.SetMaxIdleConns(1)
+
+	ctx := context.Background()
+	if _, err := db.ExecContext(ctx, fmt.Sprintf("PRAGMA busy_timeout = %d", cursorStateSQLiteBusyTimeoutMS)); err != nil {
+		return err
+	}
+	if _, err := db.ExecContext(ctx, "CREATE TABLE IF NOT EXISTS ItemTable (key TEXT UNIQUE ON CONFLICT REPLACE, value BLOB)"); err != nil {
+		return err
+	}
+
+	tx, err := db.BeginTx(ctx, &sql.TxOptions{})
+	if err != nil {
+		return err
+	}
+	committed := false
+	defer func() {
+		if !committed {
+			_ = tx.Rollback()
+		}
+	}()
+
+	keys := make([]string, 0, len(values))
+	for key := range values {
+		keys = append(keys, key)
+	}
+	sort.Strings(keys)
+
+	stmt, err := tx.PrepareContext(ctx, "INSERT OR REPLACE INTO ItemTable(key, value) VALUES(?, ?)")
+	if err != nil {
+		return err
+	}
+	defer stmt.Close()
+
+	for _, key := range keys {
+		if _, err := stmt.ExecContext(ctx, key, values[key]); err != nil {
+			return err
+		}
+	}
+
+	if err := tx.Commit(); err != nil {
+		return err
+	}
+	committed = true
+	return nil
+}
+
+func resolveCursorStateDBPath() (string, error) {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return "", fmt.Errorf("获取用户目录失败: %w", err)
+	}
+
+	switch runtime.GOOS {
+	case "darwin":
+		return filepath.Join(homeDir, filepath.FromSlash(cursorStateDarwinRelativePath)), nil
+	case "windows":
+		appData := strings.TrimSpace(os.Getenv("APPDATA"))
+		if appData == "" {
+			appData = filepath.Join(homeDir, "AppData", "Roaming")
+		}
+		return filepath.Join(appData, "Cursor", "User", "globalStorage", "state.vscdb"), nil
+	case "linux":
+		configDir := strings.TrimSpace(os.Getenv("XDG_CONFIG_HOME"))
+		if configDir == "" {
+			return filepath.Join(homeDir, filepath.FromSlash(cursorStateLinuxRelativePath)), nil
+		}
+		return filepath.Join(configDir, filepath.FromSlash(cursorStateDBRelativePath)), nil
+	default:
+		return "", fmt.Errorf("不支持的系统: %s", runtime.GOOS)
+	}
+}