feat(gemini-web): Implement proactive PSIDTS cookie rotation

internal/client/gemini-web/client.go

@@ -772,7 +772,18 @@ func (cs *ChatSession) RCID() string {
 	}
 	return ""
 }
-
+func (cs *ChatSession) setCID(v string) {
+	if len(cs.metadata) < 1 {
+		cs.metadata = normalizeMeta(cs.metadata)
+	}
+	cs.metadata[0] = v
+}
+func (cs *ChatSession) setRID(v string) {
+	if len(cs.metadata) < 2 {
+		cs.metadata = normalizeMeta(cs.metadata)
+	}
+	cs.metadata[1] = v
+}
 func (cs *ChatSession) setRCID(v string) {
 	if len(cs.metadata) < 3 {
 		cs.metadata = normalizeMeta(cs.metadata)

internal/logging/request_logger.go

@@ -13,7 +13,6 @@ import (
 	"path/filepath"
 	"regexp"
 	"strings"
-	"sync"
 	"time"
 
 	"github.com/router-for-me/CLIProxyAPI/v6/internal/interfaces"
@@ -494,12 +493,6 @@ type FileStreamingLogWriter struct {
 
 	// statusWritten indicates whether the response status has been written.
 	statusWritten bool
-
-	// mu protects concurrent access to the writer state.
-	mu sync.RWMutex
-
-	// closed indicates whether the streaming writer has been closed.
-	closed bool
 }
 
 // WriteChunkAsync writes a response chunk asynchronously (non-blocking).
@@ -507,10 +500,7 @@ type FileStreamingLogWriter struct {
 // Parameters:
 //   - chunk: The response chunk to write
 func (w *FileStreamingLogWriter) WriteChunkAsync(chunk []byte) {
-	w.mu.RLock()
-	defer w.mu.RUnlock()
-
-	if w.chunkChan == nil || w.closed {
+	if w.chunkChan == nil {
 		return
 	}
 
@@ -535,9 +525,6 @@ func (w *FileStreamingLogWriter) WriteChunkAsync(chunk []byte) {
 // Returns:
 //   - error: An error if writing fails, nil otherwise
 func (w *FileStreamingLogWriter) WriteStatus(status int, headers map[string][]string) error {
-	w.mu.Lock()
-	defer w.mu.Unlock()
-
 	if w.file == nil || w.statusWritten {
 		return nil
 	}
@@ -566,38 +553,21 @@ func (w *FileStreamingLogWriter) WriteStatus(status int, headers map[string][]st
 // Returns:
 //   - error: An error if closing fails, nil otherwise
 func (w *FileStreamingLogWriter) Close() error {
-	w.mu.Lock()
-	if w.closed {
-		w.mu.Unlock()
-		return nil
-	}
-	w.closed = true
-	chunkChan := w.chunkChan
-	closeChan := w.closeChan
-	file := w.file
-	w.mu.Unlock()
-
-	if chunkChan != nil {
-		close(chunkChan)
+	if w.chunkChan != nil {
+		close(w.chunkChan)
 	}
 
 	// Wait for async writer to finish
-	if closeChan != nil {
-		<-closeChan
+	if w.closeChan != nil {
+		<-w.closeChan
+		w.chunkChan = nil
 	}
 
-	var err error
-	if file != nil {
-		err = file.Close()
+	if w.file != nil {
+		return w.file.Close()
 	}
 
-	w.mu.Lock()
-	w.chunkChan = nil
-	w.closeChan = nil
-	w.file = nil
-	w.mu.Unlock()
-
-	return err
+	return nil
 }
 
 // asyncWriter runs in a goroutine to handle async chunk writing.

internal/runtime/executor/gemini_web_state.go

@@ -5,6 +5,8 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"net/http"
+	"net/url"
 	"path/filepath"
 	"strings"
 	"sync"
@@ -141,6 +143,7 @@ func (s *geminiWebState) ensureClient() error {
 	if s.cfg != nil && s.cfg.GeminiWeb.TokenRefreshSeconds > 0 {
 		refresh = s.cfg.GeminiWeb.TokenRefreshSeconds
 	}
+	// Use explicit refresh; background auto-refresh disabled here
 	if err := s.client.Init(float64(timeout), false, 300, false, float64(refresh), false); err != nil {
 		s.client = nil
 		return err
@@ -166,9 +169,12 @@ func (s *geminiWebState) refresh(ctx context.Context) error {
 	if s.cfg != nil && s.cfg.GeminiWeb.TokenRefreshSeconds > 0 {
 		refresh = s.cfg.GeminiWeb.TokenRefreshSeconds
 	}
+	// Use explicit refresh; background auto-refresh disabled here
 	if err := s.client.Init(float64(timeout), false, 300, false, float64(refresh), false); err != nil {
 		return err
 	}
+	// Attempt rotation proactively to persist new TS sooner
+	_ = s.tryRotatePSIDTS(proxyURL)
 	s.lastRefresh = time.Now()
 	return nil
 }
@@ -195,6 +201,59 @@ func (s *geminiWebState) tokenSnapshot() *gemini.GeminiWebTokenStorage {
 	return &c
 }
 
+// tryRotatePSIDTS performs a best-effort rotation of __Secure-1PSIDTS using
+// the public RotateCookies endpoint. On success it updates both the in-memory
+// token and the live client's cookie jar so that subsequent requests adopt the
+// new value. Any error is ignored by the caller to avoid disrupting refresh.
+func (s *geminiWebState) tryRotatePSIDTS(proxy string) error {
+	cookies := map[string]string{
+		"__Secure-1PSID":   s.token.Secure1PSID,
+		"__Secure-1PSIDTS": s.token.Secure1PSIDTS,
+	}
+
+	tr := &http.Transport{}
+	if proxy != "" {
+		if pu, err := url.Parse(proxy); err == nil {
+			tr.Proxy = http.ProxyURL(pu)
+		}
+	}
+	client := &http.Client{Transport: tr, Timeout: 60 * time.Second}
+
+	req, _ := http.NewRequest(http.MethodPost, geminiwebapi.EndpointRotateCookies, bytes.NewReader([]byte("[000,\"-0000000000000000000\"]")))
+	for k, vs := range geminiwebapi.HeadersRotateCookies {
+		for _, v := range vs {
+			req.Header.Add(k, v)
+		}
+	}
+	for k, v := range cookies {
+		req.AddCookie(&http.Cookie{Name: k, Value: v})
+	}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return err
+	}
+	defer func() { _ = resp.Body.Close() }()
+	if resp.StatusCode >= 200 && resp.StatusCode < 300 {
+		for _, c := range resp.Cookies() {
+			if c == nil {
+				continue
+			}
+			if c.Name == "__Secure-1PSIDTS" && c.Value != "" && c.Value != s.token.Secure1PSIDTS {
+				s.tokenMu.Lock()
+				s.token.Secure1PSIDTS = c.Value
+				s.tokenDirty = true
+				if s.client != nil && s.client.Cookies != nil {
+					s.client.Cookies["__Secure-1PSIDTS"] = c.Value
+				}
+				s.tokenMu.Unlock()
+				break
+			}
+		}
+	}
+	return nil
+}
+
 func (s *geminiWebState) ShouldRefresh(now time.Time, _ *cliproxyauth.Auth) bool {
 	interval := s.refreshInterval
 	if interval <= 0 {