From d82d59cbe4714a28ac3a3a9b4e0a775b21eabeea Mon Sep 17 00:00:00 2001
From: zs <you@example.com>
Date: Tue, 3 Mar 2026 15:32:33 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=89=A9=E5=B1=95=E5=BC=B9=E6=A1=86?=
 =?UTF-8?q?=E9=85=8D=E7=BD=AE=E9=87=8D=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 extension/src/content/Sidebar.vue        | 60 +++++++++--------------
 extension/src/content/sidebar-mount.ts   | 61 ++++++++++++++++++------
 extension/src/options/HotTweets.vue      |  7 ++-
 server/internal/handler/ai_handler.go    |  4 ++
 server/internal/handler/tweet_handler.go |  9 +++-
 server/internal/middleware/ratelimit.go  |  3 ++
 6 files changed, 89 insertions(+), 55 deletions(-)

diff --git a/extension/src/content/Sidebar.vue b/extension/src/content/Sidebar.vue
index 18701a8..dada3eb 100644
--- a/extension/src/content/Sidebar.vue
+++ b/extension/src/content/Sidebar.vue
@@ -1,20 +1,13 @@
 <script setup lang="ts">
-import { ref, onMounted } from 'vue'
+import { ref, inject, computed, type Ref } from 'vue'
 
-const props = defineProps<{
-  tweetData?: {
-    id: string;
-    author: string;
-    content: string;
-    stats: {
-      replies: string;
-      retweets: string;
-      likes: string;
-    }
-  }
-}>()
+// Injected reactive refs from sidebar-mount.ts
+const currentTweetData = inject<Ref<any>>('currentTweetData', ref(null))
+const isVisible = inject<Ref<boolean>>('sidebarVisible', ref(false))
+
+// Computed getter for template convenience
+const tweetData = computed(() => currentTweetData.value)
 
-const isVisible = ref(false) // Start hidden, wait for trigger
 const selectedStrategy = ref('Insightful')
 const generatedReplies = ref<Array<{strategy: string, content: string}>>([])
 const isGenerating = ref(false)
@@ -29,30 +22,21 @@ const defaultStrategies = [
 
 const strategies = ref([...defaultStrategies])
 
-onMounted(() => {
-  // Listen for toggle messages directly in the component
-  chrome.runtime.onMessage.addListener((message) => {
-    if (message.type === 'TOGGLE_SIDEBAR') {
-      isVisible.value = !isVisible.value
-    } else if (message.type === 'SHOW_INSIGHT') {
-      isVisible.value = true
-    }
-  })
-
-  chrome.runtime.sendMessage({ type: 'FETCH_CUSTOM_STRATEGIES' }, (response) => {
-    if (response && response.success && response.data) {
-      const customStrategies = response.data.map((s: any) => ({
-        id: s.strategy_key,
-        label: s.label,
-        icon: '<path d="m12 3-1.912 5.813a2 2 0 0 1-1.275 1.275L3 12l5.813 1.912a2 2 0 0 1 1.275 1.275L12 21l1.912-5.813a2 2 0 0 1 1.275-1.275L21 12l-5.813-1.912a2 2 0 0 1-1.275-1.275L12 3Z"/>'
-      }))
-      strategies.value = [...defaultStrategies, ...customStrategies]
-    }
-  })
+// Fetch custom strategies on mount
+chrome.runtime.sendMessage({ type: 'FETCH_CUSTOM_STRATEGIES' }, (response) => {
+  if (response && response.success && response.data) {
+    const customStrategies = response.data.map((s: any) => ({
+      id: s.strategy_key,
+      label: s.label,
+      icon: '<path d="m12 3-1.912 5.813a2 2 0 0 1-1.275 1.275L3 12l5.813 1.912a2 2 0 0 1 1.275 1.275L12 21l1.912-5.813a2 2 0 0 1 1.275-1.275L21 12l-5.813-1.912a2 2 0 0 1-1.275-1.275L12 3Z"/>'
+    }))
+    strategies.value = [...defaultStrategies, ...customStrategies]
+  }
 })
 
+
 const generate = () => {
-  if (!props.tweetData) return
+  if (!tweetData.value) return
   
   isGenerating.value = true
   generatedReplies.value = []
@@ -68,7 +52,7 @@ const generate = () => {
     chrome.runtime.sendMessage({
       type: 'GENERATE_REPLY',
       payload: {
-        tweetContent: props.tweetData!.content,
+        tweetContent: tweetData.value!.content,
         strategy: selectedStrategy.value,
         identity: finalIdentity
       }
@@ -93,7 +77,7 @@ const copyToClipboard = async (reply: any) => {
   showProfileTip.value = true
   setTimeout(() => { showProfileTip.value = false }, 7000)
 
-  if (!props.tweetData || !props.tweetData.id) return;
+  if (!tweetData.value || !tweetData.value.id) return;
 
   chrome.storage.local.get(['jwt_token'], async (result) => {
     if (result.jwt_token) {
@@ -107,7 +91,7 @@ const copyToClipboard = async (reply: any) => {
             'Authorization': `Bearer ${token}`
           },
           body: JSON.stringify({
-            tweet_id: props.tweetData?.id,
+            tweet_id: tweetData.value?.id,
             strategy_type: reply.strategy || 'General',
             content: reply.content,
             language: 'en'
diff --git a/extension/src/content/sidebar-mount.ts b/extension/src/content/sidebar-mount.ts
index d434a41..13d9ea6 100644
--- a/extension/src/content/sidebar-mount.ts
+++ b/extension/src/content/sidebar-mount.ts
@@ -1,13 +1,19 @@
-import { createApp } from 'vue'
+import { createApp, ref } from 'vue'
 import Sidebar from './Sidebar.vue'
-import '../assets/tailwind.css' // We might need to handle this specially for Shadow DOM
 
 const MOUNT_ID = 'insight-reply-sidebar-root'
 
-function initSidebar(tweetData?: any) {
-    let host = document.getElementById(MOUNT_ID)
+// Shared reactive tweet data that the Sidebar component will receive
+const currentTweetData = ref<any>(null)
+const sidebarVisible = ref(false)
 
-    if (host) return;
+let isMounted = false
+
+function initSidebar() {
+    if (isMounted) return
+
+    let host = document.getElementById(MOUNT_ID)
+    if (host) return
 
     // 1. Create Host Element
     host = document.createElement('div')
@@ -48,17 +54,42 @@ function initSidebar(tweetData?: any) {
     }
     injectStyles()
 
-    // 5. Create Vue App
-    const app = createApp(Sidebar, { tweetData })
+    // 5. Create Vue App with reactive provide
+    const app = createApp(Sidebar)
+    app.provide('currentTweetData', currentTweetData)
+    app.provide('sidebarVisible', sidebarVisible)
     app.mount(container)
+
+    isMounted = true
+    console.log('[InsightReply] Sidebar mounted successfully')
 }
 
-// Ensure it mounts on load if needed, but primarily triggered by messages
-chrome.runtime.onMessage.addListener((message, _sender, sendResponse) => {
-    if (message.type === 'SHOW_INSIGHT' || message.type === 'TOGGLE_SIDEBAR') {
-        initSidebar(message.payload);
-        // The component itself listens for TOGGLE_SIDEBAR to show/hide
-        sendResponse({ received: true });
+function showSidebar(tweetData?: any) {
+    initSidebar()
+
+    if (tweetData) {
+        currentTweetData.value = tweetData
+        console.log('[InsightReply] Tweet data updated:', tweetData.content?.substring(0, 60))
     }
-    return true;
-});
+
+    sidebarVisible.value = true
+}
+
+function toggleSidebar() {
+    initSidebar()
+    sidebarVisible.value = !sidebarVisible.value
+}
+
+// Listen for messages from background script
+chrome.runtime.onMessage.addListener((message, _sender, sendResponse) => {
+    console.log('[InsightReply SidebarMount] Received message:', message.type)
+
+    if (message.type === 'SHOW_INSIGHT') {
+        showSidebar(message.payload)
+        sendResponse({ received: true })
+    } else if (message.type === 'TOGGLE_SIDEBAR') {
+        toggleSidebar()
+        sendResponse({ received: true })
+    }
+    return true
+})
diff --git a/extension/src/options/HotTweets.vue b/extension/src/options/HotTweets.vue
index 63077fe..cd6c895 100644
--- a/extension/src/options/HotTweets.vue
+++ b/extension/src/options/HotTweets.vue
@@ -39,6 +39,11 @@ const loadHotTweets = async () => {
       }
     })
     
+    if (response.status === 429) {
+      error.value = '请求频率达到上限，请稍后再试。Free 用户每日限额 10 次 API 调用。'
+      return
+    }
+
     if (!response.ok) {
         throw new Error(`Opportunity discovery failed (${response.status})`)
     }
@@ -46,7 +51,7 @@ const loadHotTweets = async () => {
     const data = await response.json()
     tweets.value = data || []
   } catch (err: any) {
-    error.value = err.message || 'Quantum network sync error.'
+    error.value = err.message || 'Network connection error.'
   } finally {
     loading.value = false
   }
diff --git a/server/internal/handler/ai_handler.go b/server/internal/handler/ai_handler.go
index 8ff4019..4086622 100644
--- a/server/internal/handler/ai_handler.go
+++ b/server/internal/handler/ai_handler.go
@@ -2,6 +2,7 @@ package handler
 
 import (
 	"encoding/json"
+	"log"
 	"net/http"
 	"strings"
 
@@ -53,6 +54,7 @@ func (h *AIHandler) Generate(w http.ResponseWriter, r *http.Request) {
 
 	ctx := r.Context()
 	userID := ctx.Value("userID").(string)
+	log.Printf("[AIHandler] Generate request from userID=%s strategy=%q provider=%q model=%q", userID, body.Strategy, body.Provider, body.Model)
 
 	// Fetch Product Profile Context
 	var productContext string
@@ -79,9 +81,11 @@ func (h *AIHandler) Generate(w http.ResponseWriter, r *http.Request) {
 
 	replyString, err := h.svc.GenerateReply(ctx, body.TweetContent, productContext, body.Identity, body.Provider, body.Model)
 	if err != nil {
+		log.Printf("[AIHandler] ERROR GenerateReply for userID=%s: %v", userID, err)
 		SendError(w, http.StatusBadGateway, 5002, "Failed to generate AI reply: "+err.Error())
 		return
 	}
+	log.Printf("[AIHandler] GenerateReply success for userID=%s, reply length=%d", userID, len(replyString))
 
 	// Clean up potential markdown wrappers from LLM output
 	cleanReply := strings.TrimSpace(replyString)
diff --git a/server/internal/handler/tweet_handler.go b/server/internal/handler/tweet_handler.go
index 97cdd74..dd69382 100644
--- a/server/internal/handler/tweet_handler.go
+++ b/server/internal/handler/tweet_handler.go
@@ -2,6 +2,7 @@ package handler
 
 import (
 	"encoding/json"
+	"log"
 	"net/http"
 
 	"github.com/zs/InsightReply/internal/repository"
@@ -17,15 +18,18 @@ func NewTweetHandler(repo *repository.TweetRepository) *TweetHandler {
 
 // GetHotTweets returns the top heating tweets spanning across all tracking targets
 func (h *TweetHandler) GetHotTweets(w http.ResponseWriter, r *http.Request) {
-	// Standardize to take the top 50 hottest tweets that haven't been manually marked as processed
+	log.Printf("[TweetHandler] GetHotTweets called from %s", r.RemoteAddr)
+
 	tweets, err := h.repo.GetTopHeatingTweets(50)
 	if err != nil {
+		log.Printf("[TweetHandler] ERROR GetTopHeatingTweets: %v", err)
 		w.Header().Set("Content-Type", "application/json")
 		w.WriteHeader(http.StatusInternalServerError)
 		json.NewEncoder(w).Encode(map[string]interface{}{"error": "failed to retrieve hot tweets"})
 		return
 	}
 
+	log.Printf("[TweetHandler] GetHotTweets returning %d tweets", len(tweets))
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(tweets)
 }
@@ -34,15 +38,18 @@ func (h *TweetHandler) GetHotTweets(w http.ResponseWriter, r *http.Request) {
 func (h *TweetHandler) GetSearchTweets(w http.ResponseWriter, r *http.Request) {
 	keyword := r.URL.Query().Get("keyword")
 	handle := r.URL.Query().Get("handle")
+	log.Printf("[TweetHandler] SearchTweets called: keyword=%q handle=%q", keyword, handle)
 
 	tweets, err := h.repo.SearchTweets(keyword, handle, 50)
 	if err != nil {
+		log.Printf("[TweetHandler] ERROR SearchTweets: %v", err)
 		w.Header().Set("Content-Type", "application/json")
 		w.WriteHeader(http.StatusInternalServerError)
 		json.NewEncoder(w).Encode(map[string]interface{}{"error": "failed to search tweets"})
 		return
 	}
 
+	log.Printf("[TweetHandler] SearchTweets returning %d tweets", len(tweets))
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(tweets)
 }
diff --git a/server/internal/middleware/ratelimit.go b/server/internal/middleware/ratelimit.go
index 033241e..c54c34e 100644
--- a/server/internal/middleware/ratelimit.go
+++ b/server/internal/middleware/ratelimit.go
@@ -1,6 +1,7 @@
 package middleware
 
 import (
+	"log"
 	"net/http"
 	"sync"
 	"time"
@@ -49,6 +50,7 @@ func RateLimit(db *gorm.DB) func(http.Handler) http.Handler {
 				// For now, fallback to generic tight limit for anonymous usage
 				ipLimiter := getLimiter(r.RemoteAddr, "Free")
 				if !ipLimiter.Allow() {
+					log.Printf("[RateLimit] 429 for anonymous IP=%s path=%s", r.RemoteAddr, r.URL.Path)
 					http.Error(w, `{"code":429, "message":"Too Many Requests: Rate limit exceeded"}`, http.StatusTooManyRequests)
 					return
 				}
@@ -72,6 +74,7 @@ func RateLimit(db *gorm.DB) func(http.Handler) http.Handler {
 
 			limiter := getLimiter(userID, tier)
 			if !limiter.Allow() {
+				log.Printf("[RateLimit] 429 for userID=%s tier=%s path=%s", userID, tier, r.URL.Path)
 				http.Error(w, `{"code":429, "message":"Too Many Requests: Daily quota or rate limit exceeded"}`, http.StatusTooManyRequests)
 				return
 			}