From 5fd846bfc8267c42eac90bef8fa23a493389d4bc Mon Sep 17 00:00:00 2001
From: aits2026 <wxdaochu@gmail.com>
Date: Fri, 6 Mar 2026 15:39:13 +0800
Subject: [PATCH] feat(sns): show daily counts in jump calendar

---
 src/components/JumpToDatePopover.tsx  | 10 ++-
 src/components/Sns/SnsFilterPanel.tsx | 92 ++++++++++++++++++++++++++-
 2 files changed, 98 insertions(+), 4 deletions(-)

diff --git a/src/components/JumpToDatePopover.tsx b/src/components/JumpToDatePopover.tsx
index 0a21929..ef3c807 100644
--- a/src/components/JumpToDatePopover.tsx
+++ b/src/components/JumpToDatePopover.tsx
@@ -6,6 +6,7 @@ interface JumpToDatePopoverProps {
   isOpen: boolean
   onClose: () => void
   onSelect: (date: Date) => void
+  onMonthChange?: (date: Date) => void
   className?: string
   style?: React.CSSProperties
   currentDate?: Date
@@ -20,6 +21,7 @@ const JumpToDatePopover: React.FC<JumpToDatePopoverProps> = ({
   isOpen,
   onClose,
   onSelect,
+  onMonthChange,
   className,
   style,
   currentDate = new Date(),
@@ -112,13 +114,17 @@ const JumpToDatePopover: React.FC<JumpToDatePopoverProps> = ({
   const weekdays = ['日', '一', '二', '三', '四', '五', '六']
   const days = generateCalendar()
   const mergedClassName = ['jump-date-popover', className || ''].join(' ').trim()
+  const updateCalendarDate = (nextDate: Date) => {
+    setCalendarDate(nextDate)
+    onMonthChange?.(nextDate)
+  }
 
   return (
     <div className={mergedClassName} style={style} role="dialog" aria-label="跳转日期">
       <div className="calendar-nav">
         <button
           className="nav-btn"
-          onClick={() => setCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() - 1, 1))}
+          onClick={() => updateCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() - 1, 1))}
           aria-label="上一月"
         >
           <ChevronLeft size={16} />
@@ -126,7 +132,7 @@ const JumpToDatePopover: React.FC<JumpToDatePopoverProps> = ({
         <span className="current-month">{calendarDate.getFullYear()}年{calendarDate.getMonth() + 1}月</span>
         <button
           className="nav-btn"
-          onClick={() => setCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() + 1, 1))}
+          onClick={() => updateCalendarDate(new Date(calendarDate.getFullYear(), calendarDate.getMonth() + 1, 1))}
           aria-label="下一月"
         >
           <ChevronRight size={16} />
diff --git a/src/components/Sns/SnsFilterPanel.tsx b/src/components/Sns/SnsFilterPanel.tsx
index f6f8e93..6814e82 100644
--- a/src/components/Sns/SnsFilterPanel.tsx
+++ b/src/components/Sns/SnsFilterPanel.tsx
@@ -1,4 +1,4 @@
-import React, { useEffect, useRef, useState } from 'react'
+import React, { useEffect, useRef, useState, useCallback } from 'react'
 import { Search, Calendar, User, X, Loader2 } from 'lucide-react'
 import { Avatar } from '../Avatar'
 import JumpToDatePopover from '../JumpToDatePopover'
@@ -46,7 +46,13 @@ export const SnsFilterPanel: React.FC<SnsFilterPanelProps> = ({
 }) => {
     const [showJumpPopover, setShowJumpPopover] = useState(false)
     const [jumpPopoverDate, setJumpPopoverDate] = useState<Date>(jumpTargetDate || new Date())
+    const [jumpDateCounts, setJumpDateCounts] = useState<Record<string, number>>({})
+    const [jumpDateMessageDates, setJumpDateMessageDates] = useState<Set<string>>(new Set())
+    const [hasLoadedJumpDateCounts, setHasLoadedJumpDateCounts] = useState(false)
+    const [loadingJumpDateCounts, setLoadingJumpDateCounts] = useState(false)
     const jumpCalendarWrapRef = useRef<HTMLDivElement | null>(null)
+    const jumpDateCountsCacheRef = useRef<Map<string, Record<string, number>>>(new Map())
+    const jumpDateRequestSeqRef = useRef(0)
 
     useEffect(() => {
         if (!showJumpPopover) return
@@ -64,6 +70,78 @@ export const SnsFilterPanel: React.FC<SnsFilterPanelProps> = ({
         setJumpPopoverDate(jumpTargetDate || new Date())
     }, [jumpTargetDate, showJumpPopover])
 
+    const toMonthKey = useCallback((date: Date) => {
+        return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, '0')}`
+    }, [])
+
+    const toDateKey = useCallback((timestampSeconds: number) => {
+        const date = new Date(timestampSeconds * 1000)
+        const year = date.getFullYear()
+        const month = String(date.getMonth() + 1).padStart(2, '0')
+        const day = String(date.getDate()).padStart(2, '0')
+        return `${year}-${month}-${day}`
+    }, [])
+
+    const applyJumpDateCounts = useCallback((counts: Record<string, number>) => {
+        setJumpDateCounts(counts)
+        setJumpDateMessageDates(new Set(Object.keys(counts)))
+        setHasLoadedJumpDateCounts(true)
+    }, [])
+
+    const loadJumpDateCounts = useCallback(async (monthDate: Date) => {
+        const monthKey = toMonthKey(monthDate)
+        const cached = jumpDateCountsCacheRef.current.get(monthKey)
+        if (cached) {
+            applyJumpDateCounts(cached)
+            setLoadingJumpDateCounts(false)
+            return
+        }
+
+        const requestSeq = ++jumpDateRequestSeqRef.current
+        setLoadingJumpDateCounts(true)
+        setHasLoadedJumpDateCounts(false)
+
+        const year = monthDate.getFullYear()
+        const month = monthDate.getMonth()
+        const monthStart = new Date(year, month, 1, 0, 0, 0, 0)
+        const monthEnd = new Date(year, month + 1, 0, 23, 59, 59, 999)
+        const startTime = Math.floor(monthStart.getTime() / 1000)
+        const endTime = Math.floor(monthEnd.getTime() / 1000)
+        const pageSize = 200
+        let offset = 0
+        const counts: Record<string, number> = {}
+
+        try {
+            while (true) {
+                const result = await window.electronAPI.sns.getTimeline(pageSize, offset, [], '', startTime, endTime)
+                if (!result?.success || !Array.isArray(result.timeline) || result.timeline.length === 0) {
+                    break
+                }
+                result.timeline.forEach((post) => {
+                    const key = toDateKey(Number(post.createTime || 0))
+                    if (!key) return
+                    counts[key] = (counts[key] || 0) + 1
+                })
+                if (result.timeline.length < pageSize) break
+                offset += pageSize
+            }
+
+            if (requestSeq !== jumpDateRequestSeqRef.current) return
+            jumpDateCountsCacheRef.current.set(monthKey, counts)
+            applyJumpDateCounts(counts)
+        } catch (error) {
+            console.error('加载朋友圈按日条数失败:', error)
+            if (requestSeq !== jumpDateRequestSeqRef.current) return
+            setJumpDateCounts({})
+            setJumpDateMessageDates(new Set())
+            setHasLoadedJumpDateCounts(true)
+        } finally {
+            if (requestSeq === jumpDateRequestSeqRef.current) {
+                setLoadingJumpDateCounts(false)
+            }
+        }
+    }, [applyJumpDateCounts, toDateKey, toMonthKey])
+
     const filteredContacts = contacts.filter(c =>
         (c.displayName || '').toLowerCase().includes(contactSearch.toLowerCase()) ||
         c.username.toLowerCase().includes(contactSearch.toLowerCase())
@@ -139,7 +217,9 @@ export const SnsFilterPanel: React.FC<SnsFilterPanelProps> = ({
                                 className={`date-picker-trigger ${jumpTargetDate ? 'active' : ''}`}
                                 onClick={() => {
                                     if (!showJumpPopover) {
-                                        setJumpPopoverDate(jumpTargetDate || new Date())
+                                        const nextDate = jumpTargetDate || new Date()
+                                        setJumpPopoverDate(nextDate)
+                                        void loadJumpDateCounts(nextDate)
                                     }
                                     setShowJumpPopover(prev => !prev)
                                 }}
@@ -165,10 +245,18 @@ export const SnsFilterPanel: React.FC<SnsFilterPanelProps> = ({
                                 isOpen={showJumpPopover}
                                 currentDate={jumpPopoverDate}
                                 onClose={() => setShowJumpPopover(false)}
+                                onMonthChange={(date) => {
+                                    setJumpPopoverDate(date)
+                                    void loadJumpDateCounts(date)
+                                }}
                                 onSelect={(date) => {
                                     setJumpPopoverDate(date)
                                     setJumpTargetDate(date)
                                 }}
+                                messageDates={jumpDateMessageDates}
+                                hasLoadedMessageDates={hasLoadedJumpDateCounts}
+                                messageDateCounts={jumpDateCounts}
+                                loadingDateCounts={loadingJumpDateCounts}
                             />
                         </div>
                     </div>