auto decrypt

.gitignore

@@ -27,6 +27,5 @@ go.work.sum
 # syncthing files
 .stfolder
 
-chatlog
 chatlog.exe# Added by goreleaser init:
 dist/

README.md

@@ -58,7 +58,7 @@ go install github.com/sjzar/chatlog@latest
 
 ### macOS 版本提示
 
-1. macOS 用户在获取密钥前，需要确认已经关闭 SIP 并安装 Xcode。由于 macOS 的安全机制，在正常情况在无法读取微信进程的内存数据，所以需要临时关闭 SIP。关闭 SIP 的方法：
+1. macOS 用户在获取密钥前，需要确认已经关闭 SIP 并安装 Xcode Command Line Tools。由于 macOS 的安全机制，在正常情况在无法读取微信进程的内存数据，所以需要临时关闭 SIP。关闭 SIP 的方法：
 
 ```shell
 # 1. 进入恢复模式
@@ -72,10 +72,17 @@ go install github.com/sjzar/chatlog@latest
 # 4. 重启系统
 ```
 
-2. 目前的 macOS 版本方案依赖 `lldb` 工具，所以需要安装 Xcode，可以从 App Store 进行下载。
+2. 目前的 macOS 版本方案依赖 `lldb` 工具，所以需要安装 Xcode Command Line Tools。
+
+```shell
+# 在 terminal 执行以下命令安装 Xcode Command Line Tools：
+xcode-select --install
+```
 
 3. 仅获取数据密钥步骤需要关闭 SIP；获取数据密钥后即可重新打开 SIP，不影响解密数据和 HTTP 服务的运行。
 
+4. 如果是 Apple Silicon 芯片的 mac 用户，请检查 微信、chatlog、terminal 均不要运行在 Rosetta 模式下运行，否则可能无法获取密钥。
+
 ### Terminal UI 模式
 
 1. 启动程序：

cmd/chatlog/cmd_dumpmemory.go

@@ -0,0 +1,146 @@
+package chatlog
+
+import (
+	"archive/zip"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"runtime"
+	"time"
+
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+
+	"github.com/sjzar/chatlog/internal/wechat"
+	"github.com/sjzar/chatlog/internal/wechat/key/darwin/glance"
+)
+
+func init() {
+	rootCmd.AddCommand(dumpmemoryCmd)
+}
+
+var dumpmemoryCmd = &cobra.Command{
+	Use:   "dumpmemory",
+	Short: "dump memory",
+	Run: func(cmd *cobra.Command, args []string) {
+		if runtime.GOOS != "darwin" {
+			log.Info().Msg("dump memory only support macOS")
+		}
+
+		session := time.Now().Format("20060102150405")
+
+		dir, err := os.Getwd()
+		if err != nil {
+			log.Fatal().Err(err).Msg("get current directory failed")
+			return
+		}
+		log.Info().Msgf("current directory: %s", dir)
+
+		// step 1. check pid
+		if err = wechat.Load(); err != nil {
+			log.Fatal().Err(err).Msg("load wechat failed")
+			return
+		}
+		accounts := wechat.GetAccounts()
+		if len(accounts) == 0 {
+			log.Fatal().Msg("no wechat account found")
+			return
+		}
+
+		log.Info().Msgf("found %d wechat account", len(accounts))
+		for i, a := range accounts {
+			log.Info().Msgf("%d. %s %d %s", i, a.FullVersion, a.PID, a.DataDir)
+		}
+
+		// step 2. dump memory
+		account := accounts[0]
+		file := fmt.Sprintf("wechat_%s_%d_%s.bin", account.FullVersion, account.PID, session)
+		path := filepath.Join(dir, file)
+		log.Info().Msgf("dumping memory to %s", path)
+
+		g := glance.NewGlance(account.PID)
+		b, err := g.Read()
+		if err != nil {
+			log.Fatal().Err(err).Msg("read memory failed")
+			return
+		}
+
+		if err = os.WriteFile(path, b, 0644); err != nil {
+			log.Fatal().Err(err).Msg("write memory failed")
+			return
+		}
+
+		log.Info().Msg("dump memory success")
+
+		// step 3. copy encrypted database file
+		dbFile := "db_storage/session/session.db"
+		if account.Version == 3 {
+			dbFile = "Session/session_new.db"
+		}
+		from := filepath.Join(account.DataDir, dbFile)
+		to := filepath.Join(dir, fmt.Sprintf("wechat_%s_%d_session.db", account.FullVersion, account.PID))
+
+		log.Info().Msgf("copying %s to %s", from, to)
+		b, err = os.ReadFile(from)
+		if err != nil {
+			log.Fatal().Err(err).Msg("read session.db failed")
+			return
+		}
+		if err = os.WriteFile(to, b, 0644); err != nil {
+			log.Fatal().Err(err).Msg("write session.db failed")
+			return
+		}
+		log.Info().Msg("copy session.db success")
+
+		// step 4. package
+		zipFile := fmt.Sprintf("wechat_%s_%d_%s.zip", account.FullVersion, account.PID, session)
+		zipPath := filepath.Join(dir, zipFile)
+		log.Info().Msgf("packaging to %s", zipPath)
+
+		zf, err := os.Create(zipPath)
+		if err != nil {
+			log.Fatal().Err(err).Msg("create zip file failed")
+			return
+		}
+		defer zf.Close()
+
+		zw := zip.NewWriter(zf)
+
+		for _, file := range []string{file, to} {
+			f, err := os.Open(file)
+			if err != nil {
+				log.Fatal().Err(err).Msg("open file failed")
+				return
+			}
+			defer f.Close()
+			info, err := f.Stat()
+			if err != nil {
+				log.Fatal().Err(err).Msg("get file info failed")
+				return
+			}
+			header, err := zip.FileInfoHeader(info)
+			if err != nil {
+				log.Fatal().Err(err).Msg("create zip file info header failed")
+				return
+			}
+			header.Name = filepath.Base(file)
+			header.Method = zip.Deflate
+			writer, err := zw.CreateHeader(header)
+			if err != nil {
+				log.Fatal().Err(err).Msg("create zip file header failed")
+				return
+			}
+			if _, err = io.Copy(writer, f); err != nil {
+				log.Fatal().Err(err).Msg("copy file to zip failed")
+				return
+			}
+		}
+		if err = zw.Close(); err != nil {
+			log.Fatal().Err(err).Msg("close zip writer failed")
+			return
+		}
+
+		log.Info().Msgf("package success, please send %s to developer", zipPath)
+	},
+}

cmd/chatlog/log.go

@@ -22,6 +22,8 @@ func initLog(cmd *cobra.Command, args []string) {
 	if Debug {
 		zerolog.SetGlobalLevel(zerolog.DebugLevel)
 	}
+
+	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339})
 }
 
 func initTuiLog(cmd *cobra.Command, args []string) {

go.mod

@@ -7,16 +7,18 @@ require (
 	github.com/gin-gonic/gin v1.10.0
 	github.com/google/uuid v1.6.0
 	github.com/klauspost/compress v1.18.0
-	github.com/mattn/go-sqlite3 v1.14.24
+	github.com/mattn/go-sqlite3 v1.14.27
 	github.com/pierrec/lz4/v4 v4.1.22
-	github.com/rivo/tview v0.0.0-20250325173046-7b72abf45814
+	github.com/rivo/tview v0.0.0-20250330220935-949945f8d922
 	github.com/rs/zerolog v1.34.0
-	github.com/shirou/gopsutil/v4 v4.25.2
+	github.com/shirou/gopsutil/v4 v4.25.3
 	github.com/sirupsen/logrus v1.9.3
+	github.com/sjzar/go-lame v0.0.8
+	github.com/sjzar/go-silk v0.0.1
 	github.com/spf13/cobra v1.9.1
 	github.com/spf13/viper v1.20.1
-	golang.org/x/crypto v0.36.0
+	golang.org/x/crypto v0.37.0
-	golang.org/x/sys v0.31.0
+	golang.org/x/sys v0.32.0
 	google.golang.org/protobuf v1.36.6
 	howett.net/plist v1.0.1
 )
@@ -26,14 +28,14 @@ require (
 	github.com/bytedance/sonic/loader v0.2.4 // indirect
 	github.com/cloudwego/base64x v0.1.5 // indirect
 	github.com/ebitengine/purego v0.8.2 // indirect
-	github.com/fsnotify/fsnotify v1.8.0 // indirect
+	github.com/fsnotify/fsnotify v1.9.0 // indirect
 	github.com/gabriel-vasile/mimetype v1.4.8 // indirect
 	github.com/gdamore/encoding v1.0.1 // indirect
-	github.com/gin-contrib/sse v1.0.0 // indirect
+	github.com/gin-contrib/sse v1.1.0 // indirect
 	github.com/go-ole/go-ole v1.3.0 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.25.0 // indirect
+	github.com/go-playground/validator/v10 v10.26.0 // indirect
 	github.com/go-viper/mapstructure/v2 v2.2.1 // indirect
 	github.com/goccy/go-json v0.10.5 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
@@ -42,12 +44,12 @@ require (
 	github.com/leodido/go-urn v1.4.0 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/lufia/plan9stats v0.0.0-20250317134145-8bc96cf8fc35 // indirect
-	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-colorable v0.1.14 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/mattn/go-runewidth v0.0.16 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.4 // indirect
 	github.com/power-devops/perfstat v0.0.0-20240221224432-82ca36839d55 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/sagikazarmark/locafero v0.9.0 // indirect
@@ -62,9 +64,9 @@ require (
 	github.com/ugorji/go/codec v1.2.12 // indirect
 	github.com/yusufpapurcu/wmi v1.2.4 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
-	golang.org/x/arch v0.15.0 // indirect
+	golang.org/x/arch v0.16.0 // indirect
-	golang.org/x/net v0.38.0 // indirect
+	golang.org/x/net v0.39.0 // indirect
-	golang.org/x/term v0.30.0 // indirect
+	golang.org/x/term v0.31.0 // indirect
-	golang.org/x/text v0.23.0 // indirect
+	golang.org/x/text v0.24.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

go.sum

@@ -15,16 +15,16 @@ github.com/ebitengine/purego v0.8.2 h1:jPPGWs2sZ1UgOSgD2bClL0MJIqu58nOmIcBuXr62z
 github.com/ebitengine/purego v0.8.2/go.mod h1:iIjxzd6CiRiOG0UyXP+V1+jWqUXVjPKLAI0mRfJZTmQ=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
-github.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M=
+github.com/fsnotify/fsnotify v1.9.0 h1:2Ml+OJNzbYCTzsxtv8vKSFD9PbJjmhYF14k/jKC7S9k=
-github.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
+github.com/fsnotify/fsnotify v1.9.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
 github.com/gabriel-vasile/mimetype v1.4.8 h1:FfZ3gj38NjllZIeJAmMhr+qKL8Wu+nOoI3GqacKw1NM=
 github.com/gabriel-vasile/mimetype v1.4.8/go.mod h1:ByKUIKGjh1ODkGM1asKUbQZOLGrPjydw3hYPU2YU9t8=
 github.com/gdamore/encoding v1.0.1 h1:YzKZckdBL6jVt2Gc+5p82qhrGiqMdG/eNs6Wy0u3Uhw=
 github.com/gdamore/encoding v1.0.1/go.mod h1:0Z0cMFinngz9kS1QfMjCP8TY7em3bZYeeklsSDPivEo=
 github.com/gdamore/tcell/v2 v2.8.1 h1:KPNxyqclpWpWQlPLx6Xui1pMk8S+7+R37h3g07997NU=
 github.com/gdamore/tcell/v2 v2.8.1/go.mod h1:bj8ori1BG3OYMjmb3IklZVWfZUJ1UBQt9JXrOCOhGWw=
-github.com/gin-contrib/sse v1.0.0 h1:y3bT1mUWUxDpW4JLQg/HnTqV4rozuW4tC9eFKTxYI9E=
+github.com/gin-contrib/sse v1.1.0 h1:n0w2GMuUpWDVp7qSpvze6fAu9iRxJY4Hmj6AmBOU05w=
-github.com/gin-contrib/sse v1.0.0/go.mod h1:zNuFdwarAygJBht0NTKiSi3jRf6RbqeILZ9Sp6Slhe0=
+github.com/gin-contrib/sse v1.1.0/go.mod h1:hxRZ5gVpWMT7Z0B0gSNYqqsSCNIJMjzvm6fqCz9vjwM=
 github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
 github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
 github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
@@ -36,8 +36,8 @@ github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/o
 github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
 github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
-github.com/go-playground/validator/v10 v10.25.0 h1:5Dh7cjvzR7BRZadnsVOzPhWsrwUr0nmsZJxEAnFLNO8=
+github.com/go-playground/validator/v10 v10.26.0 h1:SP05Nqhjcvz81uJaRfEV0YBSSSGMc/iMaVtFbr3Sw2k=
-github.com/go-playground/validator/v10 v10.25.0/go.mod h1:GGzBIJMuE98Ic/kJsBXbz1x/7cByt++cQ+YOuDM5wus=
+github.com/go-playground/validator/v10 v10.26.0/go.mod h1:I5QpIEbmr8On7W0TktmJAumgzX4CA1XNl4ZmDuVHKKo=
 github.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIxtHqx8aGss=
 github.com/go-viper/mapstructure/v2 v2.2.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=
 github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4=
@@ -70,23 +70,24 @@ github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69
 github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
 github.com/lufia/plan9stats v0.0.0-20250317134145-8bc96cf8fc35 h1:PpXWgLPs+Fqr325bN2FD2ISlRRztXibcX6e8f5FR5Dc=
 github.com/lufia/plan9stats v0.0.0-20250317134145-8bc96cf8fc35/go.mod h1:autxFIvghDt3jPTLoqZ9OZ7s9qTGNAWmYCjVFWPX/zg=
-github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
 github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
+github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
+github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
 github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
 github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
 github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
-github.com/mattn/go-sqlite3 v1.14.24 h1:tpSp2G2KyMnnQu99ngJ47EIkWVmliIizyZBfPrBWDRM=
+github.com/mattn/go-sqlite3 v1.14.27 h1:drZCnuvf37yPfs95E5jd9s3XhdVWLal+6BOK6qrv6IU=
-github.com/mattn/go-sqlite3 v1.14.24/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/mattn/go-sqlite3 v1.14.27/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
 github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
-github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
+github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0t5Ec4=
-github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
+github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
 github.com/pierrec/lz4/v4 v4.1.22 h1:cKFw6uJDK+/gfw5BcDL0JL5aBsAFdsIT18eRtLj7VIU=
 github.com/pierrec/lz4/v4 v4.1.22/go.mod h1:gZWDp/Ze/IJXGXf23ltt2EXimqmTUXEy0GFuRQyBid4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
@@ -94,8 +95,8 @@ github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZb
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/power-devops/perfstat v0.0.0-20240221224432-82ca36839d55 h1:o4JXh1EVt9k/+g42oCprj/FisM4qX9L3sZB3upGN2ZU=
 github.com/power-devops/perfstat v0.0.0-20240221224432-82ca36839d55/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=
-github.com/rivo/tview v0.0.0-20250325173046-7b72abf45814 h1:pJIO3sp+rkDbJTeqqpe2Oihq3hegiM5ASvsd6S0pvjg=
+github.com/rivo/tview v0.0.0-20250330220935-949945f8d922 h1:SMyqkaRfpE8ZQUSRTZKO3uN84xov++OGa+e3NCksaQw=
-github.com/rivo/tview v0.0.0-20250325173046-7b72abf45814/go.mod h1:02iFIz7K/A9jGCvrizLPvoqr4cEIx7q54RH5Qudkrss=
+github.com/rivo/tview v0.0.0-20250330220935-949945f8d922/go.mod h1:02iFIz7K/A9jGCvrizLPvoqr4cEIx7q54RH5Qudkrss=
 github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 github.com/rivo/uniseg v0.4.3/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
 github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
@@ -108,10 +109,14 @@ github.com/rs/zerolog v1.34.0/go.mod h1:bJsvje4Z08ROH4Nhs5iH600c3IkWhwp44iRc54W6
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/sagikazarmark/locafero v0.9.0 h1:GbgQGNtTrEmddYDSAH9QLRyfAHY12md+8YFTqyMTC9k=
 github.com/sagikazarmark/locafero v0.9.0/go.mod h1:UBUyz37V+EdMS3hDF3QWIiVr/2dPrx49OMO0Bn0hJqk=
-github.com/shirou/gopsutil/v4 v4.25.2 h1:NMscG3l2CqtWFS86kj3vP7soOczqrQYIEhO/pMvvQkk=
+github.com/shirou/gopsutil/v4 v4.25.3 h1:SeA68lsu8gLggyMbmCn8cmp97V1TI9ld9sVzAUcKcKE=
-github.com/shirou/gopsutil/v4 v4.25.2/go.mod h1:34gBYJzyqCDT11b6bMHP0XCvWeU3J61XRT7a2EmCRTA=
+github.com/shirou/gopsutil/v4 v4.25.3/go.mod h1:xbuxyoZj+UsgnZrENu3lQivsngRR5BdjbJwf2fv4szA=
 github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ=
 github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
+github.com/sjzar/go-lame v0.0.8 h1:AS9l32R6foMiMEXWfUY8i79WIMfDoBC2QqQ9s5yziIk=
+github.com/sjzar/go-lame v0.0.8/go.mod h1:8RmqWcAKSbBAk6bTRV9d8mdDxqK3hY9vFyoJ4DoQE6Y=
+github.com/sjzar/go-silk v0.0.1 h1:cXD9dsIZti3n+g0Fd3IUvLH9A7tyL4jvUsHEyhff21s=
+github.com/sjzar/go-silk v0.0.1/go.mod h1:IXVcHEXKiU9j3ZtHEiGS37OFKkex9pdAhZVcFzAIOlM=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
 github.com/spf13/afero v1.14.0 h1:9tH6MapGnn/j0eb0yIXiLjERO8RB6xIVZRDCX7PtqWA=
@@ -127,13 +132,11 @@ github.com/spf13/viper v1.20.1/go.mod h1:P9Mdzt1zoHIG8m2eZQinpiBjo6kCmZSKBClNNqj
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
 github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
-github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
-github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
@@ -151,15 +154,15 @@ github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo
 github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
 go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
-golang.org/x/arch v0.15.0 h1:QtOrQd0bTUnhNVNndMpLHNWrDmYzZ2KDqSrEymqInZw=
+golang.org/x/arch v0.16.0 h1:foMtLTdyOmIniqWCHjY6+JxuC54XP1fDwx4N0ASyW+U=
-golang.org/x/arch v0.15.0/go.mod h1:JmwW7aLIoRUKgaTzhkiEFxvcEiQGyOg9BMonBJUS7EE=
+golang.org/x/arch v0.16.0/go.mod h1:JmwW7aLIoRUKgaTzhkiEFxvcEiQGyOg9BMonBJUS7EE=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
 golang.org/x/crypto v0.13.0/go.mod h1:y6Z2r+Rw4iayiXXAIxJIDAJ1zMW4yaTpebo8fPOliYc=
 golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
 golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
-golang.org/x/crypto v0.36.0 h1:AnAEvhDddvBdpY+uR+MyHmuZzzNqXSe/GvuDeob5L34=
+golang.org/x/crypto v0.37.0 h1:kJNSjF/Xp7kU0iB2Z+9viTPMW4EqqsrywMXLJOOsXSE=
-golang.org/x/crypto v0.36.0/go.mod h1:Y4J0ReaxCR1IMaabaSMugxJES1EpwhBHhv2bDHklZvc=
+golang.org/x/crypto v0.37.0/go.mod h1:vg+k43peMZ0pUMhYmVAWysMK35e6ioLh3wB8ZCAfbVc=
 golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
 golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
 golang.org/x/mod v0.12.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
@@ -173,8 +176,8 @@ golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
 golang.org/x/net v0.15.0/go.mod h1:idbUs1IY1+zTqbi8yxTbhexhEEk5ur9LInksu6HrEpk=
 golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44=
 golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
-golang.org/x/net v0.38.0 h1:vRMAPTMaeGqVhG5QyLJHqNDwecKTomGeqbnfZyKlBI8=
+golang.org/x/net v0.39.0 h1:ZCu7HMWDxpXpaiKdhzIfaltL9Lp31x/3fCP11bc6/fY=
-golang.org/x/net v0.38.0/go.mod h1:ivrbrMbzFq5J41QOQh0siUuly180yBYtLp+CKbEaFx8=
+golang.org/x/net v0.39.0/go.mod h1:X7NRbYVEA+ewNkCNyJ513WmMdQ3BineSwVtN2zD/d+E=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -199,8 +202,8 @@ golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
-golang.org/x/sys v0.31.0 h1:ioabZlmFYtWhL+TRYpcnNlLwhyxaM9kWTDEmfnprqik=
+golang.org/x/sys v0.32.0 h1:s77OFDvIQeibCmezSnk/q6iAfkdiQaJi4VzroCFrN20=
-golang.org/x/sys v0.31.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
+golang.org/x/sys v0.32.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
 golang.org/x/telemetry v0.0.0-20240228155512-f48c80bd79b2/go.mod h1:TeRTkGYfJXctD9OcfyVLyj2J3IxLnKwHJR8f4D8a3YE=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
@@ -210,8 +213,8 @@ golang.org/x/term v0.12.0/go.mod h1:owVbMEjm3cBLCHdkQu9b1opXd4ETQWc3BhuQGKgXgvU=
 golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
 golang.org/x/term v0.20.0/go.mod h1:8UkIAJTvZgivsXaD6/pH6U9ecQzZ45awqEOzuCvwpFY=
 golang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek=
-golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y=
+golang.org/x/term v0.31.0 h1:erwDkOK1Msy6offm1mOgvspSkslFnIGsFnxOKoufg3o=
-golang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g=
+golang.org/x/term v0.31.0/go.mod h1:R4BeIy7D95HzImkxGkTW1UQTtP54tio2RyHz7PwK0aw=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
@@ -221,8 +224,8 @@ golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
 golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
-golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY=
+golang.org/x/text v0.24.0 h1:dd5Bzh4yt5KYA8f9CJHCP4FB4D51c2c6JvN37xJJkJ0=
-golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4=
+golang.org/x/text v0.24.0/go.mod h1:L8rBsPeo2pSS+xqN0d5u2ikmjtmoJbDBT1b7nHvFCdU=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
 golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=

internal/chatlog/app.go

@@ -2,14 +2,17 @@ package chatlog
 
 import (
 	"fmt"
+	"path/filepath"
 	"runtime"
 	"time"
 
 	"github.com/sjzar/chatlog/internal/chatlog/ctx"
 	"github.com/sjzar/chatlog/internal/ui/footer"
+	"github.com/sjzar/chatlog/internal/ui/form"
 	"github.com/sjzar/chatlog/internal/ui/help"
 	"github.com/sjzar/chatlog/internal/ui/infobar"
 	"github.com/sjzar/chatlog/internal/ui/menu"
+	"github.com/sjzar/chatlog/internal/wechat"
 
 	"github.com/gdamore/tcell/v2"
 	"github.com/rivo/tview"
@@ -54,6 +57,8 @@ func NewApp(ctx *ctx.Context, m *Manager) *App {
 
 	app.initMenu()
 
+	app.updateMenuItemsState()
+
 	return app
 }
 
@@ -91,6 +96,33 @@ func (a *App) Stop() {
 	a.Application.Stop()
 }
 
+func (a *App) updateMenuItemsState() {
+	// 查找并更新自动解密菜单项
+	for _, item := range a.menu.GetItems() {
+		// 更新自动解密菜单项
+		if item.Index == 5 {
+			if a.ctx.AutoDecrypt {
+				item.Name = "停止自动解密"
+				item.Description = "停止监控数据目录更新，不再自动解密新增数据"
+			} else {
+				item.Name = "开启自动解密"
+				item.Description = "监控数据目录更新，自动解密新增数据"
+			}
+		}
+
+		// 更新HTTP服务菜单项
+		if item.Index == 4 {
+			if a.ctx.HTTPEnabled {
+				item.Name = "停止 HTTP 服务"
+				item.Description = "停止本地 HTTP & MCP 服务器"
+			} else {
+				item.Name = "启动 HTTP 服务"
+				item.Description = "启动本地 HTTP & MCP 服务器"
+			}
+		}
+	}
+}
+
 func (a *App) switchTab(step int) {
 	index := (a.activeTab + step) % a.tabCount
 	if index < 0 {
@@ -109,17 +141,29 @@ func (a *App) refresh() {
 		case <-a.stopRefresh:
 			return
 		case <-tick.C:
+			if a.ctx.AutoDecrypt || a.ctx.HTTPEnabled {
+				a.m.RefreshSession()
+			}
 			a.infoBar.UpdateAccount(a.ctx.Account)
 			a.infoBar.UpdateBasicInfo(a.ctx.PID, a.ctx.FullVersion, a.ctx.ExePath)
 			a.infoBar.UpdateStatus(a.ctx.Status)
 			a.infoBar.UpdateDataKey(a.ctx.DataKey)
+			a.infoBar.UpdatePlatform(a.ctx.Platform)
 			a.infoBar.UpdateDataUsageDir(a.ctx.DataUsage, a.ctx.DataDir)
 			a.infoBar.UpdateWorkUsageDir(a.ctx.WorkUsage, a.ctx.WorkDir)
+			if a.ctx.LastSession.Unix() > 1000000000 {
+				a.infoBar.UpdateSession(a.ctx.LastSession.Format("2006-01-02 15:04:05"))
+			}
 			if a.ctx.HTTPEnabled {
 				a.infoBar.UpdateHTTPServer(fmt.Sprintf("[green][已启动][white] [%s]", a.ctx.HTTPAddr))
 			} else {
 				a.infoBar.UpdateHTTPServer("[未启动]")
 			}
+			if a.ctx.AutoDecrypt {
+				a.infoBar.UpdateAutoDecrypt("[green][已开启][white]")
+			} else {
+				a.infoBar.UpdateAutoDecrypt("[未开启]")
+			}
 
 			a.Draw()
 		}
@@ -257,11 +301,11 @@ func (a *App) initMenu() {
 						} else {
 							// 启动成功
 							modal.SetText("已启动 HTTP 服务")
-							// 更改菜单项名称
-							i.Name = "停止 HTTP 服务"
-							i.Description = "停止本地 HTTP & MCP 服务器"
 						}
 
+						// 更改菜单项名称
+						a.updateMenuItemsState()
+
 						// 添加确认按钮
 						modal.AddButtons([]string{"OK"})
 						modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
@@ -288,11 +332,89 @@ func (a *App) initMenu() {
 						} else {
 							// 停止成功
 							modal.SetText("已停止 HTTP 服务")
-							// 更改菜单项名称
-							i.Name = "启动 HTTP 服务"
-							i.Description = "启动本地 HTTP & MCP 服务器"
 						}
 
+						// 更改菜单项名称
+						a.updateMenuItemsState()
+
+						// 添加确认按钮
+						modal.AddButtons([]string{"OK"})
+						modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
+							a.mainPages.RemovePage("modal")
+						})
+						a.SetFocus(modal)
+					})
+				}()
+			}
+		},
+	}
+
+	autoDecrypt := &menu.Item{
+		Index:       5,
+		Name:        "开启自动解密",
+		Description: "自动解密新增的数据文件",
+		Selected: func(i *menu.Item) {
+			modal := tview.NewModal()
+
+			// 根据当前自动解密状态执行不同操作
+			if !a.ctx.AutoDecrypt {
+				// 自动解密未开启，开启自动解密
+				modal.SetText("正在开启自动解密...")
+				a.mainPages.AddPage("modal", modal, true, true)
+				a.SetFocus(modal)
+
+				// 在后台开启自动解密
+				go func() {
+					err := a.m.StartAutoDecrypt()
+
+					// 在主线程中更新UI
+					a.QueueUpdateDraw(func() {
+						if err != nil {
+							// 开启失败
+							modal.SetText("开启自动解密失败: " + err.Error())
+						} else {
+							// 开启成功
+							if a.ctx.Version == 3 {
+								modal.SetText("已开启自动解密\n3.x版本数据文件更新不及时，有低延迟需求请使用4.0版本")
+							} else {
+								modal.SetText("已开启自动解密")
+							}
+						}
+
+						// 更改菜单项名称
+						a.updateMenuItemsState()
+
+						// 添加确认按钮
+						modal.AddButtons([]string{"OK"})
+						modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
+							a.mainPages.RemovePage("modal")
+						})
+						a.SetFocus(modal)
+					})
+				}()
+			} else {
+				// 自动解密已开启，停止自动解密
+				modal.SetText("正在停止自动解密...")
+				a.mainPages.AddPage("modal", modal, true, true)
+				a.SetFocus(modal)
+
+				// 在后台停止自动解密
+				go func() {
+					err := a.m.StopAutoDecrypt()
+
+					// 在主线程中更新UI
+					a.QueueUpdateDraw(func() {
+						if err != nil {
+							// 停止失败
+							modal.SetText("停止自动解密失败: " + err.Error())
+						} else {
+							// 停止成功
+							modal.SetText("已停止自动解密")
+						}
+
+						// 更改菜单项名称
+						a.updateMenuItemsState()
+
 						// 添加确认按钮
 						modal.AddButtons([]string{"OK"})
 						modal.SetDoneFunc(func(buttonIndex int, buttonLabel string) {
@@ -306,19 +428,28 @@ func (a *App) initMenu() {
 	}
 
 	setting := &menu.Item{
-		Index:       5,
+		Index:       6,
 		Name:        "设置",
 		Description: "设置应用程序选项",
 		Selected:    a.settingSelected,
 	}
 
-	a.menu.AddItem(setting)
+	selectAccount := &menu.Item{
+		Index:       7,
+		Name:        "切换账号",
+		Description: "切换当前操作的账号，可以选择进程或历史账号",
+		Selected:    a.selectAccountSelected,
+	}
+
 	a.menu.AddItem(getDataKey)
 	a.menu.AddItem(decryptData)
 	a.menu.AddItem(httpServer)
+	a.menu.AddItem(autoDecrypt)
+	a.menu.AddItem(setting)
+	a.menu.AddItem(selectAccount)
 
 	a.menu.AddItem(&menu.Item{
-		Index:       6,
+		Index:       8,
 		Name:        "退出",
 		Description: "退出程序",
 		Selected: func(i *menu.Item) {
@@ -347,6 +478,16 @@ func (a *App) settingSelected(i *menu.Item) {
 			description: "配置数据解密后的存储目录",
 			action:      a.settingWorkDir,
 		},
+		{
+			name:        "设置数据密钥",
+			description: "配置数据解密密钥",
+			action:      a.settingDataKey,
+		},
+		{
+			name:        "设置数据目录",
+			description: "配置微信数据文件所在目录",
+			action:      a.settingDataDir,
+		},
 	}
 
 	subMenu := menu.NewSubMenu("设置")
@@ -370,43 +511,279 @@ func (a *App) settingSelected(i *menu.Item) {
 
 // settingHTTPPort 设置 HTTP 端口
 func (a *App) settingHTTPPort() {
-	// 实现端口设置逻辑
+	// 使用我们的自定义表单组件
-	// 这里可以使用 tview.InputField 让用户输入端口
+	formView := form.NewForm("设置 HTTP 地址")
-	form := tview.NewForm().
+
-		AddInputField("地址", a.ctx.HTTPAddr, 20, nil, func(text string) {
+	// 临时存储用户输入的值
-			a.m.SetHTTPAddr(text)
+	tempHTTPAddr := a.ctx.HTTPAddr
-		}).
+
-		AddButton("保存", func() {
+	// 添加输入字段 - 不再直接设置HTTP地址，而是更新临时变量
+	formView.AddInputField("地址", tempHTTPAddr, 0, nil, func(text string) {
+		tempHTTPAddr = text // 只更新临时变量
+	})
+
+	// 添加按钮 - 点击保存时才设置HTTP地址
+	formView.AddButton("保存", func() {
+		a.m.SetHTTPAddr(tempHTTPAddr) // 在这里设置HTTP地址
 		a.mainPages.RemovePage("submenu2")
 		a.showInfo("HTTP 地址已设置为 " + a.ctx.HTTPAddr)
-		}).
+	})
-		AddButton("取消", func() {
+
+	formView.AddButton("取消", func() {
 		a.mainPages.RemovePage("submenu2")
 	})
-	form.SetBorder(true).SetTitle("设置 HTTP 地址")
 
-	a.mainPages.AddPage("submenu2", form, true, true)
+	a.mainPages.AddPage("submenu2", formView, true, true)
-	a.SetFocus(form)
+	a.SetFocus(formView)
 }
 
 // settingWorkDir 设置工作目录
 func (a *App) settingWorkDir() {
-	// 实现工作目录设置逻辑
+	// 使用我们的自定义表单组件
-	form := tview.NewForm().
+	formView := form.NewForm("设置工作目录")
-		AddInputField("工作目录", a.ctx.WorkDir, 40, nil, func(text string) {
+
-			a.ctx.SetWorkDir(text)
+	// 临时存储用户输入的值
-		}).
+	tempWorkDir := a.ctx.WorkDir
-		AddButton("保存", func() {
+
+	// 添加输入字段 - 不再直接设置工作目录，而是更新临时变量
+	formView.AddInputField("工作目录", tempWorkDir, 0, nil, func(text string) {
+		tempWorkDir = text // 只更新临时变量
+	})
+
+	// 添加按钮 - 点击保存时才设置工作目录
+	formView.AddButton("保存", func() {
+		a.ctx.SetWorkDir(tempWorkDir) // 在这里设置工作目录
 		a.mainPages.RemovePage("submenu2")
 		a.showInfo("工作目录已设置为 " + a.ctx.WorkDir)
-		}).
+	})
-		AddButton("取消", func() {
+
+	formView.AddButton("取消", func() {
 		a.mainPages.RemovePage("submenu2")
 	})
-	form.SetBorder(true).SetTitle("设置工作目录")
 
-	a.mainPages.AddPage("submenu2", form, true, true)
+	a.mainPages.AddPage("submenu2", formView, true, true)
-	a.SetFocus(form)
+	a.SetFocus(formView)
+}
+
+// settingDataKey 设置数据密钥
+func (a *App) settingDataKey() {
+	// 使用我们的自定义表单组件
+	formView := form.NewForm("设置数据密钥")
+
+	// 临时存储用户输入的值
+	tempDataKey := a.ctx.DataKey
+
+	// 添加输入字段 - 不直接设置数据密钥，而是更新临时变量
+	formView.AddInputField("数据密钥", tempDataKey, 0, nil, func(text string) {
+		tempDataKey = text // 只更新临时变量
+	})
+
+	// 添加按钮 - 点击保存时才设置数据密钥
+	formView.AddButton("保存", func() {
+		a.ctx.DataKey = tempDataKey // 设置数据密钥
+		a.mainPages.RemovePage("submenu2")
+		a.showInfo("数据密钥已设置")
+	})
+
+	formView.AddButton("取消", func() {
+		a.mainPages.RemovePage("submenu2")
+	})
+
+	a.mainPages.AddPage("submenu2", formView, true, true)
+	a.SetFocus(formView)
+}
+
+// settingDataDir 设置数据目录
+func (a *App) settingDataDir() {
+	// 使用我们的自定义表单组件
+	formView := form.NewForm("设置数据目录")
+
+	// 临时存储用户输入的值
+	tempDataDir := a.ctx.DataDir
+
+	// 添加输入字段 - 不直接设置数据目录，而是更新临时变量
+	formView.AddInputField("数据目录", tempDataDir, 0, nil, func(text string) {
+		tempDataDir = text // 只更新临时变量
+	})
+
+	// 添加按钮 - 点击保存时才设置数据目录
+	formView.AddButton("保存", func() {
+		a.ctx.DataDir = tempDataDir // 设置数据目录
+		a.mainPages.RemovePage("submenu2")
+		a.showInfo("数据目录已设置为 " + a.ctx.DataDir)
+	})
+
+	formView.AddButton("取消", func() {
+		a.mainPages.RemovePage("submenu2")
+	})
+
+	a.mainPages.AddPage("submenu2", formView, true, true)
+	a.SetFocus(formView)
+}
+
+// selectAccountSelected 处理切换账号菜单项的选择事件
+func (a *App) selectAccountSelected(i *menu.Item) {
+	// 创建子菜单
+	subMenu := menu.NewSubMenu("切换账号")
+
+	// 添加微信进程
+	instances := a.m.wechat.GetWeChatInstances()
+	if len(instances) > 0 {
+		// 添加实例标题
+		subMenu.AddItem(&menu.Item{
+			Index:       0,
+			Name:        "--- 微信进程 ---",
+			Description: "",
+			Hidden:      false,
+			Selected:    nil,
+		})
+
+		// 添加实例列表
+		for idx, instance := range instances {
+			// 创建一个实例描述
+			description := fmt.Sprintf("版本: %s 目录: %s", instance.FullVersion, instance.DataDir)
+
+			// 标记当前选中的实例
+			name := fmt.Sprintf("%s [%d]", instance.Name, instance.PID)
+			if a.ctx.Current != nil && a.ctx.Current.PID == instance.PID {
+				name = name + " [当前]"
+			}
+
+			// 创建菜单项
+			instanceItem := &menu.Item{
+				Index:       idx + 1,
+				Name:        name,
+				Description: description,
+				Hidden:      false,
+				Selected: func(instance *wechat.Account) func(*menu.Item) {
+					return func(*menu.Item) {
+						// 如果是当前账号，则无需切换
+						if a.ctx.Current != nil && a.ctx.Current.PID == instance.PID {
+							a.mainPages.RemovePage("submenu")
+							a.showInfo("已经是当前账号")
+							return
+						}
+
+						// 显示切换中的模态框
+						modal := tview.NewModal().SetText("正在切换账号...")
+						a.mainPages.AddPage("modal", modal, true, true)
+						a.SetFocus(modal)
+
+						// 在后台执行切换操作
+						go func() {
+							err := a.m.Switch(instance, "")
+
+							// 在主线程中更新UI
+							a.QueueUpdateDraw(func() {
+								a.mainPages.RemovePage("modal")
+								a.mainPages.RemovePage("submenu")
+
+								if err != nil {
+									// 切换失败
+									a.showError(fmt.Errorf("切换账号失败: %v", err))
+								} else {
+									// 切换成功
+									a.showInfo("切换账号成功")
+									// 更新菜单状态
+									a.updateMenuItemsState()
+								}
+							})
+						}()
+					}
+				}(instance),
+			}
+			subMenu.AddItem(instanceItem)
+		}
+	}
+
+	// 添加历史账号
+	if len(a.ctx.History) > 0 {
+		// 添加历史账号标题
+		subMenu.AddItem(&menu.Item{
+			Index:       100,
+			Name:        "--- 历史账号 ---",
+			Description: "",
+			Hidden:      false,
+			Selected:    nil,
+		})
+
+		// 添加历史账号列表
+		idx := 101
+		for account, hist := range a.ctx.History {
+			// 创建一个账号描述
+			description := fmt.Sprintf("版本: %s 目录: %s", hist.FullVersion, hist.DataDir)
+
+			// 标记当前选中的账号
+			name := account
+			if name == "" {
+				name = filepath.Base(hist.DataDir)
+			}
+			if a.ctx.DataDir == hist.DataDir {
+				name = name + " [当前]"
+			}
+
+			// 创建菜单项
+			histItem := &menu.Item{
+				Index:       idx,
+				Name:        name,
+				Description: description,
+				Hidden:      false,
+				Selected: func(account string) func(*menu.Item) {
+					return func(*menu.Item) {
+						// 如果是当前账号，则无需切换
+						if a.ctx.Current != nil && a.ctx.DataDir == a.ctx.History[account].DataDir {
+							a.mainPages.RemovePage("submenu")
+							a.showInfo("已经是当前账号")
+							return
+						}
+
+						// 显示切换中的模态框
+						modal := tview.NewModal().SetText("正在切换账号...")
+						a.mainPages.AddPage("modal", modal, true, true)
+						a.SetFocus(modal)
+
+						// 在后台执行切换操作
+						go func() {
+							err := a.m.Switch(nil, account)
+
+							// 在主线程中更新UI
+							a.QueueUpdateDraw(func() {
+								a.mainPages.RemovePage("modal")
+								a.mainPages.RemovePage("submenu")
+
+								if err != nil {
+									// 切换失败
+									a.showError(fmt.Errorf("切换账号失败: %v", err))
+								} else {
+									// 切换成功
+									a.showInfo("切换账号成功")
+									// 更新菜单状态
+									a.updateMenuItemsState()
+								}
+							})
+						}()
+					}
+				}(account),
+			}
+			idx++
+			subMenu.AddItem(histItem)
+		}
+	}
+
+	// 如果没有账号可选择
+	if len(a.ctx.History) == 0 && len(instances) == 0 {
+		subMenu.AddItem(&menu.Item{
+			Index:       1,
+			Name:        "无可用账号",
+			Description: "未检测到微信进程或历史账号",
+			Hidden:      false,
+			Selected:    nil,
+		})
+	}
+
+	// 显示子菜单
+	a.mainPages.AddPage("submenu", subMenu, true, true)
+	a.SetFocus(subMenu)
 }
 
 // showModal 显示一个模态对话框

internal/chatlog/ctx/context.go

@@ -2,6 +2,7 @@ package ctx
 
 import (
 	"sync"
+	"time"
 
 	"github.com/sjzar/chatlog/internal/chatlog/conf"
 	"github.com/sjzar/chatlog/internal/wechat"
@@ -33,6 +34,10 @@ type Context struct {
 	HTTPEnabled bool
 	HTTPAddr    string
 
+	// 自动解密
+	AutoDecrypt bool
+	LastSession time.Time
+
 	// 当前选中的微信实例
 	Current *wechat.Account
 	PID     int
@@ -63,6 +68,10 @@ func (c *Context) loadConfig() {
 func (c *Context) SwitchHistory(account string) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
+	c.Current = nil
+	c.PID = 0
+	c.ExePath = ""
+	c.Status = ""
 	history, ok := c.History[account]
 	if ok {
 		c.Account = history.Account
@@ -74,6 +83,16 @@ func (c *Context) SwitchHistory(account string) {
 		c.WorkDir = history.WorkDir
 		c.HTTPEnabled = history.HTTPEnabled
 		c.HTTPAddr = history.HTTPAddr
+	} else {
+		c.Account = ""
+		c.Platform = ""
+		c.Version = 0
+		c.FullVersion = ""
+		c.DataKey = ""
+		c.DataDir = ""
+		c.WorkDir = ""
+		c.HTTPEnabled = false
+		c.HTTPAddr = ""
 	}
 }
 
@@ -143,6 +162,13 @@ func (c *Context) SetDataDir(dir string) {
 	c.Refresh()
 }
 
+func (c *Context) SetAutoDecrypt(enabled bool) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	c.AutoDecrypt = enabled
+	c.UpdateConfig()
+}
+
 // 更新配置
 func (c *Context) UpdateConfig() {
 	pconf := conf.ProcessConfig{

internal/chatlog/http/route.go

@@ -12,6 +12,7 @@ import (
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/pkg/util"
 	"github.com/sjzar/chatlog/pkg/util/dat2img"
+	"github.com/sjzar/chatlog/pkg/util/silk"
 
 	"github.com/gin-gonic/gin"
 )
@@ -34,6 +35,7 @@ func (s *Service) initRouter() {
 	router.GET("/image/:key", s.GetImage)
 	router.GET("/video/:key", s.GetVideo)
 	router.GET("/file/:key", s.GetFile)
+	router.GET("/voice/:key", s.GetVoice)
 	router.GET("/data/*path", s.GetMediaData)
 
 	// MCP Server
@@ -272,6 +274,9 @@ func (s *Service) GetVideo(c *gin.Context) {
 func (s *Service) GetFile(c *gin.Context) {
 	s.GetMedia(c, "file")
 }
+func (s *Service) GetVoice(c *gin.Context) {
+	s.GetMedia(c, "voice")
+}
 
 func (s *Service) GetMedia(c *gin.Context, _type string) {
 	key := c.Param("key")
@@ -291,7 +296,13 @@ func (s *Service) GetMedia(c *gin.Context, _type string) {
 		return
 	}
 
+	switch media.Type {
+	case "voice":
+		s.HandleVoice(c, media.Data)
+	default:
 		c.Redirect(http.StatusFound, "/data/"+media.Path)
+	}
+
 }
 
 func (s *Service) GetMediaData(c *gin.Context) {
@@ -343,3 +354,12 @@ func (s *Service) HandleDatFile(c *gin.Context, path string) {
 		c.File(path)
 	}
 }
+
+func (s *Service) HandleVoice(c *gin.Context, data []byte) {
+	out, err := silk.Silk2MP3(data)
+	if err != nil {
+		c.Data(http.StatusOK, "audio/silk", data)
+		return
+	}
+	c.Data(http.StatusOK, "audio/mp3", out)
+}

internal/chatlog/http/service.go

@@ -84,11 +84,12 @@ func (s *Service) Stop() error {
 	}
 
 	// 使用超时上下文优雅关闭
-	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
 	defer cancel()
 
 	if err := s.server.Shutdown(ctx); err != nil {
-		return errors.HTTPShutDown(err)
+		log.Debug().Err(err).Msg("Failed to shutdown HTTP server")
+		return nil
 	}
 
 	log.Info().Msg("HTTP server stopped")

internal/chatlog/manager.go

@@ -6,12 +6,14 @@ import (
 	"path/filepath"
 	"strings"
 
+	"github.com/rs/zerolog/log"
 	"github.com/sjzar/chatlog/internal/chatlog/conf"
 	"github.com/sjzar/chatlog/internal/chatlog/ctx"
 	"github.com/sjzar/chatlog/internal/chatlog/database"
 	"github.com/sjzar/chatlog/internal/chatlog/http"
 	"github.com/sjzar/chatlog/internal/chatlog/mcp"
 	"github.com/sjzar/chatlog/internal/chatlog/wechat"
+	iwechat "github.com/sjzar/chatlog/internal/wechat"
 	"github.com/sjzar/chatlog/pkg/util"
 	"github.com/sjzar/chatlog/pkg/util/dat2img"
 )
@@ -79,6 +81,33 @@ func (m *Manager) Run() error {
 	return nil
 }
 
+func (m *Manager) Switch(info *iwechat.Account, history string) error {
+	if m.ctx.AutoDecrypt {
+		if err := m.StopAutoDecrypt(); err != nil {
+			return err
+		}
+	}
+	if m.ctx.HTTPEnabled {
+		if err := m.stopService(); err != nil {
+			return err
+		}
+	}
+	if info != nil {
+		m.ctx.SwitchCurrent(info)
+	} else {
+		m.ctx.SwitchHistory(history)
+	}
+
+	if m.ctx.HTTPEnabled {
+		// 启动HTTP服务
+		if err := m.StartService(); err != nil {
+			log.Info().Err(err).Msg("启动服务失败")
+			m.StopService()
+		}
+	}
+	return nil
+}
+
 func (m *Manager) StartService() error {
 
 	// 按依赖顺序启动服务
@@ -109,6 +138,17 @@ func (m *Manager) StartService() error {
 }
 
 func (m *Manager) StopService() error {
+	if err := m.stopService(); err != nil {
+		return err
+	}
+
+	// 更新状态
+	m.ctx.SetHTTPEnabled(false)
+
+	return nil
+}
+
+func (m *Manager) stopService() error {
 	// 按依赖的反序停止服务
 	var errs []error
 
@@ -124,9 +164,6 @@ func (m *Manager) StopService() error {
 		errs = append(errs, err)
 	}
 
-	// 更新状态
-	m.ctx.SetHTTPEnabled(false)
-
 	// 如果有错误，返回第一个错误
 	if len(errs) > 0 {
 		return errs[0]
@@ -138,7 +175,7 @@ func (m *Manager) StopService() error {
 func (m *Manager) SetHTTPAddr(text string) error {
 	var addr string
 	if util.IsNumeric(text) {
-		addr = fmt.Sprintf("0.0.0.0:%s", text)
+		addr = fmt.Sprintf("127.0.0.1:%s", text)
 	} else if strings.HasPrefix(text, "http://") {
 		addr = strings.TrimPrefix(text, "http://")
 	} else if strings.HasPrefix(text, "https://") {
@@ -175,7 +212,7 @@ func (m *Manager) DecryptDBFiles() error {
 		m.ctx.WorkDir = util.DefaultWorkDir(m.ctx.Account)
 	}
 
-	if err := m.wechat.DecryptDBFiles(m.ctx.DataDir, m.ctx.WorkDir, m.ctx.DataKey, m.ctx.Platform, m.ctx.Version); err != nil {
+	if err := m.wechat.DecryptDBFiles(); err != nil {
 		return err
 	}
 	m.ctx.Refresh()
@@ -183,6 +220,48 @@ func (m *Manager) DecryptDBFiles() error {
 	return nil
 }
 
+func (m *Manager) StartAutoDecrypt() error {
+	if m.ctx.DataKey == "" || m.ctx.DataDir == "" {
+		return fmt.Errorf("请先获取密钥")
+	}
+	if m.ctx.WorkDir == "" {
+		return fmt.Errorf("请先执行解密数据")
+	}
+
+	if err := m.wechat.StartAutoDecrypt(); err != nil {
+		return err
+	}
+
+	m.ctx.SetAutoDecrypt(true)
+	return nil
+}
+
+func (m *Manager) StopAutoDecrypt() error {
+	if err := m.wechat.StopAutoDecrypt(); err != nil {
+		return err
+	}
+
+	m.ctx.SetAutoDecrypt(false)
+	return nil
+}
+
+func (m *Manager) RefreshSession() error {
+	if m.db.GetDB() == nil {
+		if err := m.db.Start(); err != nil {
+			return err
+		}
+	}
+	resp, err := m.db.GetSessions("", 1, 0)
+	if err != nil {
+		return err
+	}
+	if len(resp.Items) == 0 {
+		return nil
+	}
+	m.ctx.LastSession = resp.Items[0].NTime
+	return nil
+}
+
 func (m *Manager) CommandKey(pid int) (string, error) {
 	instances := m.wechat.GetWeChatInstances()
 	if len(instances) == 0 {
@@ -216,8 +295,12 @@ func (m *Manager) CommandDecrypt(dataDir string, workDir string, key string, pla
 	if workDir == "" {
 		workDir = util.DefaultWorkDir(filepath.Base(filepath.Dir(dataDir)))
 	}
-
+	m.ctx.DataDir = dataDir
-	if err := m.wechat.DecryptDBFiles(dataDir, workDir, key, platform, version); err != nil {
+	m.ctx.WorkDir = workDir
+	m.ctx.DataKey = key
+	m.ctx.Platform = platform
+	m.ctx.Version = version
+	if err := m.wechat.DecryptDBFiles(); err != nil {
 		return err
 	}
 

internal/chatlog/wechat/service.go

@@ -5,24 +5,38 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
-	"strings"
+	"sync"
+	"time"
+
+	"github.com/fsnotify/fsnotify"
+	"github.com/rs/zerolog/log"
 
 	"github.com/sjzar/chatlog/internal/chatlog/ctx"
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/wechat"
 	"github.com/sjzar/chatlog/internal/wechat/decrypt"
+	"github.com/sjzar/chatlog/pkg/filemonitor"
 	"github.com/sjzar/chatlog/pkg/util"
+)
 
-	"github.com/rs/zerolog/log"
+var (
+	DebounceTime = 1 * time.Second
+	MaxWaitTime  = 10 * time.Second
 )
 
 type Service struct {
 	ctx            *ctx.Context
+	lastEvents     map[string]time.Time
+	pendingActions map[string]bool
+	mutex          sync.Mutex
+	fm             *filemonitor.FileMonitor
 }
 
 func NewService(ctx *ctx.Context) *Service {
 	return &Service{
 		ctx:            ctx,
+		lastEvents:     make(map[string]time.Time),
+		pendingActions: make(map[string]bool),
 	}
 }
 
@@ -46,90 +60,128 @@ func (s *Service) GetDataKey(info *wechat.Account) (string, error) {
 	return key, nil
 }
 
-// FindDBFiles finds all .db files in the specified directory
+func (s *Service) StartAutoDecrypt() error {
-func (s *Service) FindDBFiles(rootDir string, recursive bool) ([]string, error) {
+	dbGroup, err := filemonitor.NewFileGroup("wechat", s.ctx.DataDir, `.*\.db$`, []string{"fts"})
-	// Check if directory exists
-	info, err := os.Stat(rootDir)
 	if err != nil {
-		return nil, fmt.Errorf("cannot access directory %s: %w", rootDir, err)
+		return err
 	}
+	dbGroup.AddCallback(s.DecryptFileCallback)
 
-	if !info.IsDir() {
+	s.fm = filemonitor.NewFileMonitor()
-		return nil, fmt.Errorf("%s is not a directory", rootDir)
+	s.fm.AddGroup(dbGroup)
+	if err := s.fm.Start(); err != nil {
+		log.Debug().Err(err).Msg("failed to start file monitor")
+		return err
 	}
-
-	var dbFiles []string
-
-	// Define walk function
-	walkFunc := func(path string, info os.FileInfo, err error) error {
-		if err != nil {
-			// If a file or directory can't be accessed, log the error but continue
-			log.Err(err).Msgf("Warning: Cannot access %s", path)
 	return nil
-		}
-
-		// If it's a directory and not the root directory, and we're not recursively searching, skip it
-		if info.IsDir() && path != rootDir && !recursive {
-			return filepath.SkipDir
-		}
-
-		// Check if file extension is .db
-		if !info.IsDir() && strings.ToLower(filepath.Ext(path)) == ".db" {
-			dbFiles = append(dbFiles, path)
-		}
-
-		return nil
-	}
-
-	// Start traversal
-	err = filepath.Walk(rootDir, walkFunc)
-	if err != nil {
-		return nil, fmt.Errorf("error traversing directory: %w", err)
-	}
-
-	if len(dbFiles) == 0 {
-		return nil, fmt.Errorf("no .db files found")
-	}
-
-	return dbFiles, nil
 }
 
-func (s *Service) DecryptDBFiles(dataDir string, workDir string, key string, platform string, version int) error {
+func (s *Service) StopAutoDecrypt() error {
+	if s.fm != nil {
+		if err := s.fm.Stop(); err != nil {
+			return err
+		}
+	}
+	s.fm = nil
+	return nil
+}
 
-	ctx := context.Background()
+func (s *Service) DecryptFileCallback(event fsnotify.Event) error {
+	if event.Op.Has(fsnotify.Chmod) || !event.Op.Has(fsnotify.Write) {
+		return nil
+	}
 
-	dbfiles, err := s.FindDBFiles(dataDir, true)
+	s.mutex.Lock()
+	s.lastEvents[event.Name] = time.Now()
+
+	if !s.pendingActions[event.Name] {
+		s.pendingActions[event.Name] = true
+		s.mutex.Unlock()
+		go s.waitAndProcess(event.Name)
+	} else {
+		s.mutex.Unlock()
+	}
+
+	return nil
+}
+
+func (s *Service) waitAndProcess(dbFile string) {
+	start := time.Now()
+	for {
+		time.Sleep(DebounceTime)
+
+		s.mutex.Lock()
+		lastEventTime := s.lastEvents[dbFile]
+		elapsed := time.Since(lastEventTime)
+		totalElapsed := time.Since(start)
+
+		if elapsed >= DebounceTime || totalElapsed >= MaxWaitTime {
+			s.pendingActions[dbFile] = false
+			s.mutex.Unlock()
+
+			log.Debug().Msgf("Processing file: %s", dbFile)
+			s.DecryptDBFile(dbFile)
+			return
+		}
+		s.mutex.Unlock()
+	}
+}
+
+func (s *Service) DecryptDBFile(dbFile string) error {
+
+	decryptor, err := decrypt.NewDecryptor(s.ctx.Platform, s.ctx.Version)
 	if err != nil {
 		return err
 	}
 
-	decryptor, err := decrypt.NewDecryptor(platform, version)
+	output := filepath.Join(s.ctx.WorkDir, dbFile[len(s.ctx.DataDir):])
-	if err != nil {
-		return err
-	}
-
-	for _, dbfile := range dbfiles {
-		output := filepath.Join(workDir, dbfile[len(dataDir):])
 	if err := util.PrepareDir(filepath.Dir(output)); err != nil {
 		return err
 	}
 
-		outputFile, err := os.Create(output)
+	outputTemp := output + ".tmp"
+	outputFile, err := os.Create(outputTemp)
 	if err != nil {
 		return fmt.Errorf("failed to create output file: %v", err)
 	}
-		defer outputFile.Close()
+	defer func() {
+		outputFile.Close()
+		if err := os.Rename(outputTemp, output); err != nil {
+			log.Debug().Err(err).Msgf("failed to rename %s to %s", outputTemp, output)
+		}
+	}()
 
-		if err := decryptor.Decrypt(ctx, dbfile, key, outputFile); err != nil {
+	if err := decryptor.Decrypt(context.Background(), dbFile, s.ctx.DataKey, outputFile); err != nil {
-			log.Err(err).Msgf("failed to decrypt %s", dbfile)
 		if err == errors.ErrAlreadyDecrypted {
-				if data, err := os.ReadFile(dbfile); err == nil {
+			if data, err := os.ReadFile(dbFile); err == nil {
 				outputFile.Write(data)
 			}
-				continue
+			return nil
 		}
+		log.Err(err).Msgf("failed to decrypt %s", dbFile)
+		return err
+	}
+
+	log.Debug().Msgf("Decrypted %s to %s", dbFile, output)
+
+	return nil
+}
+
+func (s *Service) DecryptDBFiles() error {
+	dbGroup, err := filemonitor.NewFileGroup("wechat", s.ctx.DataDir, `.*\.db$`, []string{"fts"})
+	if err != nil {
+		return err
+	}
+
+	dbFiles, err := dbGroup.List()
+	if err != nil {
+		return err
+	}
+
+	for _, dbFile := range dbFiles {
+		if err := s.DecryptDBFile(dbFile); err != nil {
+			log.Debug().Msgf("DecryptDBFile %s failed: %v", dbFile, err)
 			continue
-			// return err
 		}
 	}
 

internal/errors/errors.go

@@ -112,7 +112,7 @@ func RootCause(err error) error {
 
 func Err(c *gin.Context, err error) {
 	if appErr, ok := err.(*Error); ok {
-		c.JSON(appErr.Code, appErr)
+		c.JSON(appErr.Code, appErr.Error())
 		return
 	}
 

internal/errors/wechatdb_errors.go

@@ -60,3 +60,7 @@ func ContactNotFound(key string) *Error {
 func InitCacheFailed(cause error) *Error {
 	return New(cause, http.StatusInternalServerError, "init cache failed").WithStack()
 }
+
+func FileGroupNotFound(name string) *Error {
+	return Newf(nil, http.StatusNotFound, "file group not found: %s", name).WithStack()
+}

internal/model/chatroom_v4.go

@@ -20,9 +20,17 @@ func (c *ChatRoomV4) Wrap() *ChatRoom {
 		users = ParseRoomData(c.ExtBuffer)
 	}
 
+	user2DisplayName := make(map[string]string, len(users))
+	for _, user := range users {
+		if user.DisplayName != "" {
+			user2DisplayName[user.UserName] = user.DisplayName
+		}
+	}
+
 	return &ChatRoom{
 		Name:             c.UserName,
 		Owner:            c.Owner,
 		Users:            users,
+		User2DisplayName: user2DisplayName,
 	}
 }

internal/model/contact.go

@@ -47,33 +47,6 @@ type ContactV3 struct {
 	Remark    string `json:"Remark"`
 	NickName  string `json:"NickName"`
 	Reserved1 int    `json:"Reserved1"` // 1 自己好友或自己加入的群聊; 0 群聊成员(非好友)
-
-	// EncryptUserName string `json:"EncryptUserName"`
-	// DelFlag         int    `json:"DelFlag"`
-	// Type            int    `json:"Type"`
-	// VerifyFlag      int    `json:"VerifyFlag"`
-	// Reserved2       int    `json:"Reserved2"`
-	// Reserved3       string `json:"Reserved3"`
-	// Reserved4       string `json:"Reserved4"`
-	// LabelIDList     string `json:"LabelIDList"`
-	// DomainList      string `json:"DomainList"`
-	// ChatRoomType    int    `json:"ChatRoomType"`
-	// PYInitial       string `json:"PYInitial"`
-	// QuanPin         string `json:"QuanPin"`
-	// RemarkPYInitial string `json:"RemarkPYInitial"`
-	// RemarkQuanPin   string `json:"RemarkQuanPin"`
-	// BigHeadImgUrl   string `json:"BigHeadImgUrl"`
-	// SmallHeadImgUrl string `json:"SmallHeadImgUrl"`
-	// HeadImgMd5      string `json:"HeadImgMd5"`
-	// ChatRoomNotify  int    `json:"ChatRoomNotify"`
-	// Reserved5       int    `json:"Reserved5"`
-	// Reserved6       string `json:"Reserved6"`
-	// Reserved7       string `json:"Reserved7"`
-	// ExtraBuf        []byte `json:"ExtraBuf"`
-	// Reserved8       int    `json:"Reserved8"`
-	// Reserved9       int    `json:"Reserved9"`
-	// Reserved10      string `json:"Reserved10"`
-	// Reserved11      string `json:"Reserved11"`
 }
 
 func (c *ContactV3) Wrap() *Contact {

internal/model/contact_darwinv3.go

@@ -40,33 +40,6 @@ type ContactDarwinV3 struct {
 	M_nsRemark    string `json:"m_nsRemark"`
 	M_uiSex       int    `json:"m_uiSex"`
 	M_nsAliasName string `json:"m_nsAliasName"`
-
-	// M_uiConType           int    `json:"m_uiConType"`
-	// M_nsShortPY           string `json:"m_nsShortPY"`
-	// M_nsRemarkPYFull      string `json:"m_nsRemarkPYFull"`
-	// M_nsRemarkPYShort     string `json:"m_nsRemarkPYShort"`
-	// M_uiCertificationFlag int    `json:"m_uiCertificationFlag"`
-	// M_uiType              int    `json:"m_uiType"` // 本来想拿这个字段来区分是否是好友，但是数据比较乱，好在 darwin v3 Contact 表中没有群聊成员
-	// M_nsImgStatus         string `json:"m_nsImgStatus"`
-	// M_uiImgKey            int    `json:"m_uiImgKey"`
-	// M_nsHeadImgUrl        string `json:"m_nsHeadImgUrl"`
-	// M_nsHeadHDImgUrl      string `json:"m_nsHeadHDImgUrl"`
-	// M_nsHeadHDMd5         string `json:"m_nsHeadHDMd5"`
-	// M_nsChatRoomMemList   string `json:"m_nsChatRoomMemList"`
-	// M_nsChatRoomAdminList string `json:"m_nsChatRoomAdminList"`
-	// M_uiChatRoomStatus    int    `json:"m_uiChatRoomStatus"`
-	// M_nsChatRoomDesc      string `json:"m_nsChatRoomDesc"`
-	// M_nsDraft             string `json:"m_nsDraft"`
-	// M_nsBrandIconUrl      string `json:"m_nsBrandIconUrl"`
-	// M_nsGoogleContactName string `json:"m_nsGoogleContactName"`
-	// M_nsEncodeUserName    string `json:"m_nsEncodeUserName"`
-	// M_uiChatRoomVersion   int    `json:"m_uiChatRoomVersion"`
-	// M_uiChatRoomMaxCount  int    `json:"m_uiChatRoomMaxCount"`
-	// M_uiChatRoomType      int    `json:"m_uiChatRoomType"`
-	// M_patSuffix           string `json:"m_patSuffix"`
-	// RichChatRoomDesc      string `json:"richChatRoomDesc"`
-	// Packed_WCContactData  string `json:"_packed_WCContactData"`
-	// OpenIMInfo            string `json:"openIMInfo"`
 }
 
 func (c *ContactDarwinV3) Wrap() *Contact {

internal/model/contact_v4.go

@@ -30,25 +30,6 @@ type ContactV4 struct {
 	Remark    string `json:"remark"`
 	NickName  string `json:"nick_name"`
 	LocalType int    `json:"local_type"` // 2 群聊; 3 群聊成员(非好友); 5,6 企业微信;
-
-	// ID                  int    `json:"id"`
-
-	// EncryptUserName     string `json:"encrypt_username"`
-	// Flag                int    `json:"flag"`
-	// DeleteFlag          int    `json:"delete_flag"`
-	// VerifyFlag          int    `json:"verify_flag"`
-	// RemarkQuanPin       string `json:"remark_quan_pin"`
-	// RemarkPinYinInitial string `json:"remark_pin_yin_initial"`
-	// PinYinInitial       string `json:"pin_yin_initial"`
-	// QuanPin             string `json:"quan_pin"`
-	// BigHeadUrl          string `json:"big_head_url"`
-	// SmallHeadUrl        string `json:"small_head_url"`
-	// HeadImgMd5          string `json:"head_img_md5"`
-	// ChatRoomNotify      int    `json:"chat_room_notify"`
-	// IsInChatRoom        int    `json:"is_in_chat_room"`
-	// Description         string `json:"description"`
-	// ExtraBuffer         []byte `json:"extra_buffer"`
-	// ChatRoomType        int    `json:"chat_room_type"`
 }
 
 func (c *ContactV4) Wrap() *Contact {

internal/model/media.go

@@ -10,6 +10,7 @@ type Media struct {
 	Path       string `json:"path"`
 	Name       string `json:"name"`
 	Size       int64  `json:"size"`
+	Data       []byte `json:"data"` // for voice
 	ModifyTime int64  `json:"modifyTime"`
 }
 

internal/model/mediamessage.go

@@ -283,9 +283,34 @@ type FinderMegaVideo struct {
 }
 
 type SysMsg struct {
-	SysMsgTemplate SysMsgTemplate `xml:"sysmsgtemplate"`
+	Type              string             `xml:"type,attr"`
+	DelChatRoomMember *DelChatRoomMember `xml:"delchatroommember,omitempty"`
+	SysMsgTemplate    *SysMsgTemplate    `xml:"sysmsgtemplate,omitempty"`
 }
 
+// 第一种消息类型：删除群成员/二维码邀请
+type DelChatRoomMember struct {
+	Plain string `xml:"plain"`
+	Text  string `xml:"text"`
+	Link  QRLink `xml:"link"`
+}
+
+type QRLink struct {
+	Scene      string       `xml:"scene"`
+	Text       string       `xml:"text"`
+	MemberList QRMemberList `xml:"memberlist"`
+	QRCode     string       `xml:"qrcode"`
+}
+
+type QRMemberList struct {
+	Usernames []UsernameItem `xml:"username"`
+}
+
+type UsernameItem struct {
+	Value string `xml:",chardata"`
+}
+
+// 第二种消息类型：系统消息模板
 type SysMsgTemplate struct {
 	ContentTemplate ContentTemplate `xml:"content_template"`
 }
@@ -305,7 +330,8 @@ type Link struct {
 	Name       string     `xml:"name,attr"`
 	Type       string     `xml:"type,attr"`
 	MemberList MemberList `xml:"memberlist"`
-	Separator  string     `xml:"separator"`
+	Separator  string     `xml:"separator,omitempty"`
+	Title      string     `xml:"title,omitempty"`
 }
 
 type MemberList struct {
@@ -318,6 +344,24 @@ type Member struct {
 }
 
 func (s *SysMsg) String() string {
+	if s.Type == "delchatroommember" {
+		return s.DelChatRoomMemberString()
+	}
+	return s.SysMsgTemplateString()
+}
+
+func (s *SysMsg) DelChatRoomMemberString() string {
+	if s.DelChatRoomMember == nil {
+		return ""
+	}
+	return s.DelChatRoomMember.Plain
+}
+
+func (s *SysMsg) SysMsgTemplateString() string {
+	if s.SysMsgTemplate == nil {
+		return ""
+	}
+
 	template := s.SysMsgTemplate.ContentTemplate.Template
 	links := s.SysMsgTemplate.ContentTemplate.LinkList.Links
 
@@ -354,8 +398,12 @@ func (s *SysMsg) String() string {
 
 		// 可以根据需要添加其他链接类型的处理逻辑
 		default:
+			if link.Title != "" {
+				replacement = link.Title
+			} else {
 				replacement = ""
 			}
+		}
 
 		// 将占位符名称和替换内容存入映射
 		replacements["$"+link.Name+"$"] = replacement

internal/model/message.go

@@ -55,6 +55,8 @@ func (m *Message) ParseMediaInfo(data string) error {
 		if Debug {
 			m.SysMsg = &sysMsg
 		}
+		m.Sender = "系统消息"
+		m.SenderName = ""
 		m.Content = sysMsg.String()
 		return nil
 	}
@@ -188,13 +190,8 @@ func (m *Message) PlainText(showChatRoom bool, host string) string {
 	buf := strings.Builder{}
 
 	sender := m.Sender
-	switch {
+	if m.IsSelf {
-	case m.Type == 10000:
-		sender = "系统消息"
-	case m.IsSelf:
 		sender = "我"
-	default:
-		sender = m.Sender
 	}
 	if m.SenderName != "" {
 		buf.WriteString(m.SenderName)
@@ -235,6 +232,9 @@ func (m *Message) PlainTextContent() string {
 	case 3:
 		return fmt.Sprintf("![图片](http://%s/image/%s)", m.Contents["host"], m.Contents["md5"])
 	case 34:
+		if voice, ok := m.Contents["voice"]; ok {
+			return fmt.Sprintf("[语音](http://%s/voice/%s)", m.Contents["host"], voice)
+		}
 		return "[语音]"
 	case 42:
 		return "[名片]"

internal/model/message_v3.go

@@ -1,6 +1,7 @@
 package model
 
 import (
+	"fmt"
 	"path/filepath"
 	"strings"
 	"time"
@@ -39,6 +40,7 @@ import (
 // BytesTrans BLOB
 // )
 type MessageV3 struct {
+	MsgSvrID        int64  `json:"MsgSvrID"`        // 消息 ID
 	Sequence        int64  `json:"Sequence"`        // 消息序号，10位时间戳 + 3位序号
 	CreateTime      int64  `json:"CreateTime"`      // 消息创建时间，10位时间戳
 	StrTalker       string `json:"StrTalker"`       // 聊天对象，微信 ID or 群 ID
@@ -77,6 +79,11 @@ func (m *MessageV3) Wrap() *Message {
 
 	_m.ParseMediaInfo(_m.Content)
 
+	// 语音消息
+	if _m.Type == 34 {
+		_m.Contents["voice"] = fmt.Sprint(m.MsgSvrID)
+	}
+
 	if len(m.BytesExtra) != 0 {
 		if bytesExtra := ParseBytesExtra(m.BytesExtra); bytesExtra != nil {
 			if _m.IsChatRoom {

internal/model/message_v4.go

@@ -2,6 +2,7 @@ package model
 
 import (
 	"bytes"
+	"fmt"
 	"strings"
 	"time"
 
@@ -31,6 +32,7 @@ import (
 // )
 type MessageV4 struct {
 	SortSeq        int64  `json:"sort_seq"`         // 消息序号，10位时间戳 + 3位序号
+	ServerID       int64  `json:"server_id"`        // 消息 ID，用于关联 voice
 	LocalType      int64  `json:"local_type"`       // 消息类型
 	UserName       string `json:"user_name"`        // 发送人，通过 Join Name2Id 表获得
 	CreateTime     int64  `json:"create_time"`      // 消息创建时间，10位时间戳
@@ -74,6 +76,11 @@ func (m *MessageV4) Wrap(talker string) *Message {
 
 	_m.ParseMediaInfo(content)
 
+	// 语音消息
+	if _m.Type == 34 {
+		_m.Contents["voice"] = fmt.Sprint(m.ServerID)
+	}
+
 	if len(m.PackedInfoData) != 0 {
 		if packedInfo := ParsePackedInfo(m.PackedInfoData); packedInfo != nil {
 			// FIXME 尝试解决 v4 版本 xml 数据无法匹配到 hardlink 记录的问题

internal/ui/form/form.go

@@ -0,0 +1,258 @@
+package form
+
+import (
+	"fmt"
+
+	"github.com/gdamore/tcell/v2"
+	"github.com/rivo/tview"
+	"github.com/sjzar/chatlog/internal/ui/style"
+)
+
+const (
+	// DialogPadding dialog inner padding.
+	DialogPadding = 3
+
+	// DialogHelpHeight dialog help text height.
+	DialogHelpHeight = 1
+
+	// DialogMinWidth dialog min width.
+	DialogMinWidth = 40
+
+	// FormHeightOffset form height offset for border.
+	FormHeightOffset = 3
+
+	// 额外的宽度补偿，类似于 submenu 的 cmdWidthOffset
+	formWidthOffset = 10
+)
+
+// Form is a modal form component with a title, form fields, and help text.
+type Form struct {
+	*tview.Box
+	title         string
+	layout        *tview.Flex
+	form          *tview.Form
+	helpText      *tview.TextView
+	width         int
+	height        int
+	cancelHandler func()
+	fields        []formField // 存储字段信息以便重新计算宽度
+}
+
+// formField 存储表单字段的信息
+type formField struct {
+	label      string
+	value      string
+	fieldWidth int
+}
+
+// NewForm creates a new form with the given title.
+func NewForm(title string) *Form {
+	f := &Form{
+		Box:    tview.NewBox(),
+		title:  title,
+		layout: tview.NewFlex().SetDirection(tview.FlexRow),
+		form:   tview.NewForm(),
+		fields: make([]formField, 0),
+	}
+
+	// 设置表单样式
+	f.form.SetBorderPadding(1, 1, 1, 1)
+	f.form.SetBackgroundColor(style.DialogBgColor)
+	f.form.SetFieldBackgroundColor(style.BgColor)
+	f.form.SetFieldTextColor(style.FgColor)
+	f.form.SetButtonBackgroundColor(style.ButtonBgColor)
+	f.form.SetButtonTextColor(style.FgColor)
+	f.form.SetLabelColor(style.DialogFgColor)
+	f.form.SetButtonsAlign(tview.AlignCenter)
+
+	// 创建帮助文本
+	f.helpText = tview.NewTextView()
+	f.helpText.SetDynamicColors(true)
+	f.helpText.SetTextAlign(tview.AlignCenter)
+	f.helpText.SetTextColor(style.DialogFgColor)
+	f.helpText.SetBackgroundColor(style.DialogBgColor)
+	fmt.Fprintf(f.helpText,
+		"[%s::b]Tab[%s::b]: 导航  [%s::b]Enter[%s::b]: 选择  [%s::b]ESC[%s::b]: 返回",
+		style.GetColorHex(style.MenuBgColor), style.GetColorHex(style.PageHeaderFgColor),
+		style.GetColorHex(style.MenuBgColor), style.GetColorHex(style.PageHeaderFgColor),
+		style.GetColorHex(style.MenuBgColor), style.GetColorHex(style.PageHeaderFgColor),
+	)
+
+	// 创建布局
+	formLayout := tview.NewFlex().SetDirection(tview.FlexColumn)
+	formLayout.AddItem(EmptyBoxSpace(style.DialogBgColor), 1, 0, false)
+	formLayout.AddItem(f.form, 0, 1, true)
+	formLayout.AddItem(EmptyBoxSpace(style.DialogBgColor), 1, 0, false)
+
+	// 设置主布局
+	f.layout.SetTitle(fmt.Sprintf("[::b]%s", f.title))
+	f.layout.SetTitleColor(style.DialogFgColor)
+	f.layout.SetTitleAlign(tview.AlignCenter)
+	f.layout.SetBorder(true)
+	f.layout.SetBorderColor(style.DialogBorderColor)
+	f.layout.SetBackgroundColor(style.DialogBgColor)
+
+	// 添加表单区域
+	f.layout.AddItem(formLayout, 0, 1, true)
+
+	// 添加帮助文本区域
+	f.layout.AddItem(f.helpText, DialogHelpHeight, 0, false)
+
+	return f
+}
+
+// AddInputField adds an input field to the form.
+func (f *Form) AddInputField(label, value string, fieldWidth int, accept func(textToCheck string, lastChar rune) bool, changed func(text string)) *Form {
+	// 存储字段信息
+	f.fields = append(f.fields, formField{
+		label:      label,
+		value:      value,
+		fieldWidth: fieldWidth,
+	})
+
+	// 添加输入字段到表单
+	f.form.AddInputField(label, value, fieldWidth, accept, changed)
+
+	// 更新表单尺寸
+	f.recalculateSize()
+
+	return f
+}
+
+// AddButton adds a button to the form.
+func (f *Form) AddButton(label string, selected func()) *Form {
+	f.form.AddButton(label, selected)
+	// 更新表单尺寸
+	f.recalculateSize()
+	return f
+}
+
+// AddCheckbox adds a checkbox to the form.
+func (f *Form) AddCheckbox(label string, checked bool, changed func(checked bool)) *Form {
+	f.form.AddCheckbox(label, checked, changed)
+	// 更新表单尺寸
+	f.recalculateSize()
+	return f
+}
+
+// SetCancelFunc sets the function to be called when the form is cancelled.
+func (f *Form) SetCancelFunc(handler func()) *Form {
+	f.cancelHandler = handler
+	return f
+}
+
+// recalculateSize 重新计算表单尺寸
+func (f *Form) recalculateSize() {
+	// 计算表单项数量
+	itemCount := f.form.GetFormItemCount()
+
+	// 计算高度 - 每个表单项占2行，按钮区域至少占2行，再加上边框和帮助文本
+	f.height = (itemCount * 2) + 2 + FormHeightOffset + DialogHelpHeight
+
+	// 计算宽度 - 类似于 submenu 的实现
+	maxLabelWidth := 0
+	maxValueWidth := 0
+
+	// 遍历所有字段，找出最长的标签和值
+	for _, field := range f.fields {
+		if len(field.label) > maxLabelWidth {
+			maxLabelWidth = len(field.label)
+		}
+
+		// 对于值，使用字段宽度和实际值长度中的较大者
+		valueWidth := field.fieldWidth
+		if len(field.value) > valueWidth {
+			valueWidth = len(field.value)
+		}
+
+		if valueWidth > maxValueWidth {
+			maxValueWidth = valueWidth
+		}
+	}
+
+	// 计算总宽度，类似于 submenu 的计算方式
+	f.width = maxLabelWidth + maxValueWidth + formWidthOffset
+
+	// 确保宽度不小于最小值
+	if f.width < DialogMinWidth {
+		f.width = DialogMinWidth
+	}
+}
+
+// Draw draws the form on the screen.
+func (f *Form) Draw(screen tcell.Screen) {
+	// 在绘制前重新计算尺寸，确保尺寸是最新的
+	f.recalculateSize()
+
+	// 绘制
+	f.Box.DrawForSubclass(screen, f)
+	f.layout.Draw(screen)
+}
+
+// SetRect sets the position and size of the form.
+func (f *Form) SetRect(x, y, width, height int) {
+	// 确保尺寸是最新的
+	f.recalculateSize()
+
+	// 类似于 submenu 的实现
+	ws := (width - f.width) / 2
+	hs := (height - f.height) / 2
+
+	// 确保不会超出屏幕
+	if f.width > width {
+		ws = 0
+		f.width = width - 1
+	}
+
+	if f.height > height {
+		hs = 0
+		f.height = height - 1
+	}
+
+	// 设置表单位置
+	f.Box.SetRect(x+ws, y+hs, f.width, f.height)
+
+	// 获取内部矩形并设置布局
+	x, y, width, height = f.Box.GetInnerRect()
+	f.layout.SetRect(x, y, width, height)
+}
+
+// Focus is called when this primitive receives focus.
+func (f *Form) Focus(delegate func(p tview.Primitive)) {
+	// 确保表单获得焦点
+	if f.form != nil {
+		delegate(f.form)
+	} else {
+		// 如果表单为空，则让Box获得焦点
+		delegate(f.Box)
+	}
+}
+
+// HasFocus returns whether or not this primitive has focus.
+func (f *Form) HasFocus() bool {
+	return f.form.HasFocus()
+}
+
+// InputHandler returns the handler for this primitive.
+func (f *Form) InputHandler() func(event *tcell.EventKey, setFocus func(p tview.Primitive)) {
+	return f.WrapInputHandler(func(event *tcell.EventKey, setFocus func(p tview.Primitive)) {
+		// ESC键处理
+		if event.Key() == tcell.KeyEscape && f.cancelHandler != nil {
+			f.cancelHandler()
+			return
+		}
+
+		// 将事件传递给表单
+		if handler := f.form.InputHandler(); handler != nil {
+			handler(event, setFocus)
+		}
+	})
+}
+
+// EmptyBoxSpace creates an empty box with the specified background color.
+func EmptyBoxSpace(bgColor tcell.Color) *tview.Box {
+	box := tview.NewBox()
+	box.SetBackgroundColor(bgColor)
+	box.SetBorder(false)
+	return box
+}

internal/ui/infobar/infobar.go

@@ -15,13 +15,14 @@ const (
 
 // InfoBarViewHeight info bar height.
 const (
-	InfoBarViewHeight = 6
+	InfoBarViewHeight = 7
 	accountRow        = 0
-	pidRow            = 1
+	statusRow         = 1
-	statusRow         = 2
+	platformRow       = 2
-	dataUsageRow      = 3
+	sessionRow        = 3
-	workUsageRow      = 4
+	dataUsageRow      = 4
-	httpServerRow     = 5
+	workUsageRow      = 5
+	httpServerRow     = 6
 
 	// 列索引
 	labelCol1 = 0 // 第一列标签
@@ -43,7 +44,7 @@ func New() *InfoBar {
 	table := tview.NewTable()
 	headerColor := style.InfoBarItemFgColor
 
-	// Account 和 Version 行
+	// Account 和 PID 行
 	table.SetCell(
 		accountRow,
 		labelCol1,
@@ -54,26 +55,11 @@ func New() *InfoBar {
 	table.SetCell(
 		accountRow,
 		labelCol2,
-		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Version:")),
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "PID:")),
 	)
 	table.SetCell(accountRow, valueCol2, tview.NewTableCell(""))
 
-	// PID 和 ExePath 行
+	// Status 和 ExePath 行
-	table.SetCell(
-		pidRow,
-		labelCol1,
-		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "PID:")),
-	)
-	table.SetCell(pidRow, valueCol1, tview.NewTableCell(""))
-
-	table.SetCell(
-		pidRow,
-		labelCol2,
-		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "ExePath:")),
-	)
-	table.SetCell(pidRow, valueCol2, tview.NewTableCell(""))
-
-	// Status 和 Key 行
 	table.SetCell(
 		statusRow,
 		labelCol1,
@@ -84,10 +70,40 @@ func New() *InfoBar {
 	table.SetCell(
 		statusRow,
 		labelCol2,
-		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Data Key:")),
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "ExePath:")),
 	)
 	table.SetCell(statusRow, valueCol2, tview.NewTableCell(""))
 
+	// Platform 和 Version 行
+	table.SetCell(
+		platformRow,
+		labelCol1,
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Platform:")),
+	)
+	table.SetCell(platformRow, valueCol1, tview.NewTableCell(""))
+
+	table.SetCell(
+		platformRow,
+		labelCol2,
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Version:")),
+	)
+	table.SetCell(platformRow, valueCol2, tview.NewTableCell(""))
+
+	// Session 和 Data Key 行
+	table.SetCell(
+		sessionRow,
+		labelCol1,
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Session:")),
+	)
+	table.SetCell(sessionRow, valueCol1, tview.NewTableCell(""))
+
+	table.SetCell(
+		sessionRow,
+		labelCol2,
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Data Key:")),
+	)
+	table.SetCell(sessionRow, valueCol2, tview.NewTableCell(""))
+
 	// Data Usage 和 Data Dir 行
 	table.SetCell(
 		dataUsageRow,
@@ -126,6 +142,13 @@ func New() *InfoBar {
 	)
 	table.SetCell(httpServerRow, valueCol1, tview.NewTableCell(""))
 
+	table.SetCell(
+		httpServerRow,
+		labelCol2,
+		tview.NewTableCell(fmt.Sprintf(" [%s::]%s", headerColor, "Auto Decrypt:")),
+	)
+	table.SetCell(httpServerRow, valueCol2, tview.NewTableCell(""))
+
 	// infobar
 	infoBar := &InfoBar{
 		Box:   tview.NewBox(),
@@ -141,17 +164,25 @@ func (info *InfoBar) UpdateAccount(account string) {
 }
 
 func (info *InfoBar) UpdateBasicInfo(pid int, version string, exePath string) {
-	info.table.GetCell(pidRow, valueCol1).SetText(fmt.Sprintf("%d", pid))
+	info.table.GetCell(accountRow, valueCol2).SetText(fmt.Sprintf("%d", pid))
-	info.table.GetCell(pidRow, valueCol2).SetText(exePath)
+	info.table.GetCell(statusRow, valueCol2).SetText(exePath)
-	info.table.GetCell(accountRow, valueCol2).SetText(version)
+	info.table.GetCell(platformRow, valueCol2).SetText(version)
 }
 
 func (info *InfoBar) UpdateStatus(status string) {
 	info.table.GetCell(statusRow, valueCol1).SetText(status)
 }
 
+func (info *InfoBar) UpdatePlatform(text string) {
+	info.table.GetCell(platformRow, valueCol1).SetText(text)
+}
+
+func (info *InfoBar) UpdateSession(text string) {
+	info.table.GetCell(sessionRow, valueCol1).SetText(text)
+}
+
 func (info *InfoBar) UpdateDataKey(key string) {
-	info.table.GetCell(statusRow, valueCol2).SetText(key)
+	info.table.GetCell(sessionRow, valueCol2).SetText(key)
 }
 
 func (info *InfoBar) UpdateDataUsageDir(dataUsage string, dataDir string) {
@@ -169,6 +200,11 @@ func (info *InfoBar) UpdateHTTPServer(server string) {
 	info.table.GetCell(httpServerRow, valueCol1).SetText(server)
 }
 
+// UpdateAutoDecrypt updates Auto Decrypt value.
+func (info *InfoBar) UpdateAutoDecrypt(text string) {
+	info.table.GetCell(httpServerRow, valueCol2).SetText(text)
+}
+
 // Draw draws this primitive onto the screen.
 func (info *InfoBar) Draw(screen tcell.Screen) {
 	info.Box.DrawForSubclass(screen, info)

internal/wechat/decrypt/validator.go

@@ -15,13 +15,17 @@ type Validator struct {
 }
 
 // NewValidator 创建一个仅用于验证的验证器
-func NewValidator(dataDir string, platform string, version int) (*Validator, error) {
+func NewValidator(platform string, version int, dataDir string) (*Validator, error) {
+	dbFile := GetSimpleDBFile(platform, version)
+	dbPath := filepath.Join(dataDir + "/" + dbFile)
+	return NewValidatorWithFile(platform, version, dbPath)
+}
+
+func NewValidatorWithFile(platform string, version int, dbPath string) (*Validator, error) {
 	decryptor, err := NewDecryptor(platform, version)
 	if err != nil {
 		return nil, err
 	}
-	dbFile := GetSimpleDBFile(platform, version)
-	dbPath := filepath.Join(dataDir + "/" + dbFile)
 	d, err := common.OpenDBFile(dbPath, decryptor.GetPageSize())
 	if err != nil {
 		return nil, err

internal/wechat/key/darwin/v3.go

@@ -19,12 +19,22 @@ const (
 	MaxWorkersV3 = 8
 )
 
+var V3KeyPatterns = []KeyPatternInfo{
+	{
+		Pattern: []byte{0x72, 0x74, 0x72, 0x65, 0x65, 0x5f, 0x69, 0x33, 0x32},
+		Offset:  24,
+	},
+}
+
 type V3Extractor struct {
 	validator   *decrypt.Validator
+	keyPatterns []KeyPatternInfo
 }
 
 func NewV3Extractor() *V3Extractor {
-	return &V3Extractor{}
+	return &V3Extractor{
+		keyPatterns: V3KeyPatterns,
+	}
 }
 
 func (e *V3Extractor) Extract(ctx context.Context, proc *model.Process) (string, error) {
@@ -127,7 +137,6 @@ func (e *V3Extractor) findMemory(ctx context.Context, pid uint32, memoryChannel
 
 // worker processes memory regions to find V3 version key
 func (e *V3Extractor) worker(ctx context.Context, memoryChannel <-chan []byte, resultChannel chan<- string) {
-	keyPattern := []byte{0x72, 0x74, 0x72, 0x65, 0x65, 0x5f, 0x69, 0x33, 0x32}
 	for {
 		select {
 		case <-ctx.Done():
@@ -137,49 +146,58 @@ func (e *V3Extractor) worker(ctx context.Context, memoryChannel <-chan []byte, r
 				return
 			}
 
+			if key, ok := e.SearchKey(ctx, memory); ok {
+				select {
+				case resultChannel <- key:
+				default:
+				}
+			}
+		}
+	}
+}
+
+func (e *V3Extractor) SearchKey(ctx context.Context, memory []byte) (string, bool) {
+	for _, keyPattern := range e.keyPatterns {
 		index := len(memory)
 
 		for {
 			select {
 			case <-ctx.Done():
-					return // Exit if context cancelled
+				return "", false
 			default:
 			}
 
-				log.Debug().Msgf("Searching for V3 key in memory region, size: %d bytes", len(memory))
-
 			// Find pattern from end to beginning
-				index = bytes.LastIndex(memory[:index], keyPattern)
+			index = bytes.LastIndex(memory[:index], keyPattern.Pattern)
 			if index == -1 {
 				break // No more matches found
 			}
 
-				log.Debug().Msgf("Found potential V3 key pattern in memory region, index: %d", index)
+			// Check if we have enough space for the key
-
+			keyOffset := index + keyPattern.Offset
-				// For V3, the key is 32 bytes and starts right after the pattern
+			if keyOffset < 0 || keyOffset+32 > len(memory) {
-				if index+24+32 > len(memory) {
 				index -= 1
 				continue
 			}
 
-				// Extract the key data, which is right after the pattern and 32 bytes long
+			// Extract the key data, which is 32 bytes long
-				keyOffset := index + 24
 			keyData := memory[keyOffset : keyOffset+32]
 
 			// Validate key against database header
 			if e.validator.Validate(keyData) {
-					select {
+				log.Debug().
-					case resultChannel <- hex.EncodeToString(keyData):
+					Str("pattern", hex.EncodeToString(keyPattern.Pattern)).
-						log.Debug().Msg("Key found: " + hex.EncodeToString(keyData))
+					Int("offset", keyPattern.Offset).
-						return
+					Str("key", hex.EncodeToString(keyData)).
-					default:
+					Msg("Key found")
-					}
+				return hex.EncodeToString(keyData), true
 			}
 
 			index -= 1
 		}
 	}
-	}
+
+	return "", false
 }
 
 func (e *V3Extractor) SetValidate(validator *decrypt.Validator) {

internal/wechat/key/darwin/v4.go

@@ -19,12 +19,26 @@ const (
 	MaxWorkers = 8
 )
 
+var V4KeyPatterns = []KeyPatternInfo{
+	{
+		Pattern: []byte{0x20, 0x66, 0x74, 0x73, 0x35, 0x28, 0x25, 0x00},
+		Offset:  16,
+	},
+	{
+		Pattern: []byte{0x20, 0x66, 0x74, 0x73, 0x35, 0x28, 0x25, 0x00},
+		Offset:  -80,
+	},
+}
+
 type V4Extractor struct {
 	validator   *decrypt.Validator
+	keyPatterns []KeyPatternInfo
 }
 
 func NewV4Extractor() *V4Extractor {
-	return &V4Extractor{}
+	return &V4Extractor{
+		keyPatterns: V4KeyPatterns,
+	}
 }
 
 func (e *V4Extractor) Extract(ctx context.Context, proc *model.Process) (string, error) {
@@ -127,8 +141,6 @@ func (e *V4Extractor) findMemory(ctx context.Context, pid uint32, memoryChannel
 
 // worker processes memory regions to find V4 version key
 func (e *V4Extractor) worker(ctx context.Context, memoryChannel <-chan []byte, resultChannel chan<- string) {
-	keyPattern := []byte{0x20, 0x66, 0x74, 0x73, 0x35, 0x28, 0x25, 0x00}
-
 	for {
 		select {
 		case <-ctx.Done():
@@ -138,47 +150,65 @@ func (e *V4Extractor) worker(ctx context.Context, memoryChannel <-chan []byte, r
 				return
 			}
 
+			if key, ok := e.SearchKey(ctx, memory); ok {
+				select {
+				case resultChannel <- key:
+				default:
+				}
+			}
+		}
+	}
+}
+
+func (e *V4Extractor) SearchKey(ctx context.Context, memory []byte) (string, bool) {
+	for _, keyPattern := range e.keyPatterns {
 		index := len(memory)
 
 		for {
 			select {
 			case <-ctx.Done():
-					return // Exit if context cancelled
+				return "", false
 			default:
 			}
 
 			// Find pattern from end to beginning
-				index = bytes.LastIndex(memory[:index], keyPattern)
+			index = bytes.LastIndex(memory[:index], keyPattern.Pattern)
 			if index == -1 {
 				break // No more matches found
 			}
 
 			// Check if we have enough space for the key
-				if index+16+32 > len(memory) {
+			keyOffset := index + keyPattern.Offset
+			if keyOffset < 0 || keyOffset+32 > len(memory) {
 				index -= 1
 				continue
 			}
 
 			// Extract the key data, which is 16 bytes after the pattern and 32 bytes long
-				keyOffset := index + 16
 			keyData := memory[keyOffset : keyOffset+32]
 
 			// Validate key against database header
 			if e.validator.Validate(keyData) {
-					select {
+				log.Debug().
-					case resultChannel <- hex.EncodeToString(keyData):
+					Str("pattern", hex.EncodeToString(keyPattern.Pattern)).
-						log.Debug().Msg("Key found: " + hex.EncodeToString(keyData))
+					Int("offset", keyPattern.Offset).
-						return
+					Str("key", hex.EncodeToString(keyData)).
-					default:
+					Msg("Key found")
-					}
+				return hex.EncodeToString(keyData), true
 			}
 
 			index -= 1
 		}
 	}
-	}
+
+	return "", false
 }
 
 func (e *V4Extractor) SetValidate(validator *decrypt.Validator) {
 	e.validator = validator
 }
+
+type KeyPatternInfo struct {
+	Pattern []byte
+	Offset  int
+}

internal/wechat/key/extractor.go

@@ -15,6 +15,9 @@ type Extractor interface {
 	// Extract 从进程中提取密钥
 	Extract(ctx context.Context, proc *model.Process) (string, error)
 
+	// SearchKey 在内存中搜索密钥
+	SearchKey(ctx context.Context, memory []byte) (string, bool)
+
 	SetValidate(validator *decrypt.Validator)
 }
 

internal/wechat/key/windows/v3.go

@@ -1,6 +1,8 @@
 package windows
 
 import (
+	"context"
+
 	"github.com/sjzar/chatlog/internal/wechat/decrypt"
 )
 
@@ -12,6 +14,11 @@ func NewV3Extractor() *V3Extractor {
 	return &V3Extractor{}
 }
 
+func (e *V3Extractor) SearchKey(ctx context.Context, memory []byte) (string, bool) {
+	// TODO : Implement the key search logic for V3
+	return "", false
+}
+
 func (e *V3Extractor) SetValidate(validator *decrypt.Validator) {
 	e.validator = validator
 }

internal/wechat/key/windows/v4.go

@@ -1,6 +1,8 @@
 package windows
 
 import (
+	"context"
+
 	"github.com/sjzar/chatlog/internal/wechat/decrypt"
 )
 
@@ -12,6 +14,11 @@ func NewV4Extractor() *V4Extractor {
 	return &V4Extractor{}
 }
 
+func (e *V4Extractor) SearchKey(ctx context.Context, memory []byte) (string, bool) {
+	// TODO : Implement the key search logic for V4
+	return "", false
+}
+
 func (e *V4Extractor) SetValidate(validator *decrypt.Validator) {
 	e.validator = validator
 }

internal/wechat/process/darwin/detector.go

@@ -15,10 +15,10 @@ import (
 )
 
 const (
-	V3ProcessName = "WeChat"
+	ProcessNameOfficial = "WeChat"
-	V4ProcessName = "Weixin"
+	ProcessNameBeta     = "Weixin"
 	V3DBFile            = "Message/msg_0.db"
-	V4DBFile      = "db_storage/message/message_0.db"
+	V4DBFile            = "db_storage/session/session.db"
 )
 
 // Detector 实现 macOS 平台的进程检测器
@@ -40,7 +40,7 @@ func (d *Detector) FindProcesses() ([]*model.Process, error) {
 	var result []*model.Process
 	for _, p := range processes {
 		name, err := p.Name()
-		if err != nil || (name != V3ProcessName && name != V4ProcessName) {
+		if err != nil || (name != ProcessNameOfficial && name != ProcessNameBeta) {
 			continue
 		}
 

internal/wechat/wechat.go

@@ -90,7 +90,7 @@ func (a *Account) GetKey(ctx context.Context) (string, error) {
 		return "", err
 	}
 
-	validator, err := decrypt.NewValidator(process.DataDir, process.Platform, process.Version)
+	validator, err := decrypt.NewValidator(process.Platform, process.Version, process.DataDir)
 	if err != nil {
 		return "", err
 	}

internal/wechatdb/datasource/darwinv3/datasource.go

@@ -3,87 +3,127 @@ package darwinv3
 import (
 	"context"
 	"crypto/md5"
-	"database/sql"
 	"encoding/hex"
 	"fmt"
 	"strings"
 	"time"
 
+	"github.com/fsnotify/fsnotify"
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/rs/zerolog/log"
 
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/model"
-	"github.com/sjzar/chatlog/pkg/util"
+	"github.com/sjzar/chatlog/internal/wechatdb/datasource/dbm"
 )
 
 const (
-	MessageFilePattern  = "^msg_([0-9]?[0-9])?\\.db$"
+	Message  = "message"
-	ContactFilePattern  = "^wccontact_new2\\.db$"
+	Contact  = "contact"
-	ChatRoomFilePattern = "^group_new\\.db$"
+	ChatRoom = "chatroom"
-	SessionFilePattern  = "^session_new\\.db$"
+	Session  = "session"
-	MediaFilePattern    = "^hldata\\.db$"
+	Media    = "media"
 )
 
+var Groups = []dbm.Group{
+	{
+		Name:      Message,
+		Pattern:   `^msg_([0-9]?[0-9])?\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Contact,
+		Pattern:   `^wccontact_new2\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      ChatRoom,
+		Pattern:   `group_new\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Session,
+		Pattern:   `^session_new\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Media,
+		Pattern:   `^hldata\.db$`,
+		BlackList: []string{},
+	},
+}
+
 type DataSource struct {
 	path string
-	messageDbs []*sql.DB
+	dbm  *dbm.DBManager
-	contactDb  *sql.DB
-	chatRoomDb *sql.DB
-	sessionDb  *sql.DB
-	mediaDb    *sql.DB
 
-	talkerDBMap      map[string]*sql.DB
+	talkerDBMap      map[string]string
 	user2DisplayName map[string]string
 }
 
 func New(path string) (*DataSource, error) {
 	ds := &DataSource{
 		path:             path,
-		messageDbs:       make([]*sql.DB, 0),
+		dbm:              dbm.NewDBManager(path),
-		talkerDBMap:      make(map[string]*sql.DB),
+		talkerDBMap:      make(map[string]string),
 		user2DisplayName: make(map[string]string),
 	}
 
-	if err := ds.initMessageDbs(path); err != nil {
+	for _, g := range Groups {
+		ds.dbm.AddGroup(g)
+	}
+
+	if err := ds.dbm.Start(); err != nil {
+		return nil, err
+	}
+
+	if err := ds.initMessageDbs(); err != nil {
 		return nil, errors.DBInitFailed(err)
 	}
-	if err := ds.initContactDb(path); err != nil {
+	if err := ds.initChatRoomDb(); err != nil {
 		return nil, errors.DBInitFailed(err)
 	}
-	if err := ds.initChatRoomDb(path); err != nil {
+
-		return nil, errors.DBInitFailed(err)
+	ds.dbm.AddCallback(Message, func(event fsnotify.Event) error {
+		if !event.Op.Has(fsnotify.Create) {
+			return nil
 		}
-	if err := ds.initSessionDb(path); err != nil {
+		if err := ds.initMessageDbs(); err != nil {
-		return nil, errors.DBInitFailed(err)
+			log.Err(err).Msgf("Failed to reinitialize message DBs: %s", event.Name)
 		}
-	if err := ds.initMediaDb(path); err != nil {
+		return nil
-		return nil, errors.DBInitFailed(err)
+	})
+	ds.dbm.AddCallback(ChatRoom, func(event fsnotify.Event) error {
+		if !event.Op.Has(fsnotify.Create) {
+			return nil
 		}
+		if err := ds.initChatRoomDb(); err != nil {
+			log.Err(err).Msgf("Failed to reinitialize chatroom DB: %s", event.Name)
+		}
+		return nil
+	})
 
 	return ds, nil
 }
 
-func (ds *DataSource) initMessageDbs(path string) error {
+func (ds *DataSource) SetCallback(name string, callback func(event fsnotify.Event) error) error {
+	return ds.dbm.AddCallback(name, callback)
+}
 
-	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
+func (ds *DataSource) initMessageDbs() error {
+
+	dbPaths, err := ds.dbm.GetDBPath(Message)
 	if err != nil {
-		return errors.DBFileNotFound(path, MessageFilePattern, err)
+		return err
 	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, MessageFilePattern, nil)
-	}
-
 	// 处理每个数据库文件
-	for _, filePath := range files {
+	talkerDBMap := make(map[string]string)
-		// 连接数据库
+	for _, filePath := range dbPaths {
-		db, err := sql.Open("sqlite3", filePath)
+		db, err := ds.dbm.OpenDB(filePath)
 		if err != nil {
-			log.Err(err).Msgf("连接数据库 %s 失败", filePath)
+			log.Err(err).Msgf("获取数据库 %s 失败", filePath)
 			continue
 		}
-		ds.messageDbs = append(ds.messageDbs, db)
 
 		// 获取所有表名
 		rows, err := db.Query("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'Chat_%'")
@@ -104,96 +144,42 @@ func (ds *DataSource) initMessageDbs(path string) error {
 			if talkerMd5 == "" {
 				continue
 			}
-			ds.talkerDBMap[talkerMd5] = db
+			talkerDBMap[talkerMd5] = filePath
 		}
 		rows.Close()
-
 	}
+	ds.talkerDBMap = talkerDBMap
 	return nil
 }
 
-func (ds *DataSource) initContactDb(path string) error {
+func (ds *DataSource) initChatRoomDb() error {
-
+	db, err := ds.dbm.GetDB(ChatRoom)
-	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
 	if err != nil {
-		return errors.DBFileNotFound(path, ContactFilePattern, err)
+		return err
 	}
 
-	if len(files) == 0 {
+	rows, err := db.Query("SELECT m_nsUsrName, IFNULL(nickname,\"\") FROM GroupMember")
-		return errors.DBFileNotFound(path, ContactFilePattern, nil)
-	}
-
-	ds.contactDb, err = sql.Open("sqlite3", files[0])
 	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
+		log.Err(err).Msg("获取群聊成员失败")
-	}
-
-	return nil
-}
-
-func (ds *DataSource) initChatRoomDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, ChatRoomFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, ChatRoomFilePattern, err)
-	}
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, ChatRoomFilePattern, nil)
-	}
-	ds.chatRoomDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-
-	rows, err := ds.chatRoomDb.Query("SELECT m_nsUsrName, IFNULL(nickname,\"\") FROM GroupMember")
-	if err != nil {
-		log.Err(err).Msgf("数据库 %s 获取群聊成员失败", files[0])
 		return nil
 	}
 
+	user2DisplayName := make(map[string]string)
 	for rows.Next() {
 		var user string
 		var nickName string
 		if err := rows.Scan(&user, &nickName); err != nil {
-			log.Err(err).Msgf("数据库 %s 扫描表名失败", files[0])
+			log.Err(err).Msg("扫描表名失败")
 			continue
 		}
-		ds.user2DisplayName[user] = nickName
+		user2DisplayName[user] = nickName
 	}
 	rows.Close()
+	ds.user2DisplayName = user2DisplayName
 
 	return nil
 }
 
-func (ds *DataSource) initSessionDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, SessionFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, SessionFilePattern, err)
-	}
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, SessionFilePattern, nil)
-	}
-	ds.sessionDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-	return nil
-}
-
-func (ds *DataSource) initMediaDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, MediaFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, MediaFilePattern, err)
-	}
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, MediaFilePattern, nil)
-	}
-	ds.mediaDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-	return nil
-}
-
 // GetMessages 实现获取消息的方法
 func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
 	// 在 darwinv3 中，每个联系人/群聊的消息存储在单独的表中，表名为 Chat_md5(talker)
@@ -204,10 +190,14 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 
 	_talkerMd5Bytes := md5.Sum([]byte(talker))
 	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
-	db, ok := ds.talkerDBMap[talkerMd5]
+	dbPath, ok := ds.talkerDBMap[talkerMd5]
 	if !ok {
 		return nil, errors.TalkerNotFound(talker)
 	}
+	db, err := ds.dbm.OpenDB(dbPath)
+	if err != nil {
+		return nil, err
+	}
 	tableName := fmt.Sprintf("Chat_%s", talkerMd5)
 
 	// 构建查询条件
@@ -297,7 +287,11 @@ func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Contact)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -351,7 +345,11 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 	}
 
 	// 执行查询
-	rows, err := ds.chatRoomDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(ChatRoom)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -380,7 +378,7 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 		contacts, err := ds.GetContacts(ctx, key, 1, 0)
 		if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
 			// 再次尝试通过用户名查找群聊
-			rows, err := ds.chatRoomDb.QueryContext(ctx,
+			rows, err := db.QueryContext(ctx,
 				`SELECT IFNULL(m_nsUsrName,""), IFNULL(nickname,""), IFNULL(m_nsRemark,""), IFNULL(m_nsChatRoomMemList,""), IFNULL(m_nsChatRoomAdminList,"") 
 				FROM GroupContact 
 				WHERE m_nsUsrName = ?`,
@@ -448,7 +446,11 @@ func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.sessionDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Session)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -506,7 +508,11 @@ WHERE
     r.mediaMd5 = ?`
 	args := []interface{}{key}
 	// 执行查询
-	rows, err := ds.mediaDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Media)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -541,46 +547,5 @@ WHERE
 
 // Close 实现关闭数据库连接的方法
 func (ds *DataSource) Close() error {
-	var errs []error
+	return ds.dbm.Close()
-
-	// 关闭消息数据库连接
-	for _, db := range ds.messageDbs {
-		if err := db.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	// 关闭联系人数据库连接
-	if ds.contactDb != nil {
-		if err := ds.contactDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	// 关闭群聊数据库连接
-	if ds.chatRoomDb != nil {
-		if err := ds.chatRoomDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	// 关闭会话数据库连接
-	if ds.sessionDb != nil {
-		if err := ds.sessionDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	// 关闭媒体数据库连接
-	if ds.mediaDb != nil {
-		if err := ds.mediaDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	if len(errs) > 0 {
-		return errors.DBCloseFailed(errs[0])
-	}
-
-	return nil
 }

internal/wechatdb/datasource/datasource.go

@@ -4,6 +4,8 @@ import (
 	"context"
 	"time"
 
+	"github.com/fsnotify/fsnotify"
+
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/model"
 	"github.com/sjzar/chatlog/internal/wechatdb/datasource/darwinv3"
@@ -28,6 +30,9 @@ type DataSource interface {
 	// 媒体
 	GetMedia(ctx context.Context, _type string, key string) (*model.Media, error)
 
+	// 设置回调函数
+	SetCallback(name string, callback func(event fsnotify.Event) error) error
+
 	Close() error
 }
 

internal/wechatdb/datasource/dbm/dbm.go

@@ -0,0 +1,170 @@
+package dbm
+
+import (
+	"database/sql"
+	"runtime"
+	"sync"
+	"time"
+
+	"github.com/fsnotify/fsnotify"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/rs/zerolog/log"
+
+	"github.com/sjzar/chatlog/internal/errors"
+	"github.com/sjzar/chatlog/pkg/filecopy"
+	"github.com/sjzar/chatlog/pkg/filemonitor"
+)
+
+type DBManager struct {
+	path    string
+	fm      *filemonitor.FileMonitor
+	fgs     map[string]*filemonitor.FileGroup
+	dbs     map[string]*sql.DB
+	dbPaths map[string][]string
+	mutex   sync.RWMutex
+}
+
+func NewDBManager(path string) *DBManager {
+	return &DBManager{
+		path:    path,
+		fm:      filemonitor.NewFileMonitor(),
+		fgs:     make(map[string]*filemonitor.FileGroup),
+		dbs:     make(map[string]*sql.DB),
+		dbPaths: make(map[string][]string),
+	}
+}
+
+func (d *DBManager) AddGroup(g Group) error {
+	fg, err := filemonitor.NewFileGroup(g.Name, d.path, g.Pattern, g.BlackList)
+	if err != nil {
+		return err
+	}
+	fg.AddCallback(d.Callback)
+	d.fm.AddGroup(fg)
+	d.mutex.Lock()
+	d.fgs[g.Name] = fg
+	d.mutex.Unlock()
+	return nil
+}
+
+func (d *DBManager) AddCallback(name string, callback func(event fsnotify.Event) error) error {
+	d.mutex.RLock()
+	fg, ok := d.fgs[name]
+	d.mutex.RUnlock()
+	if !ok {
+		return errors.FileGroupNotFound(name)
+	}
+	fg.AddCallback(callback)
+	return nil
+}
+
+func (d *DBManager) GetDB(name string) (*sql.DB, error) {
+	dbPaths, err := d.GetDBPath(name)
+	if err != nil {
+		return nil, err
+	}
+	return d.OpenDB(dbPaths[0])
+}
+
+func (d *DBManager) GetDBs(name string) ([]*sql.DB, error) {
+	dbPaths, err := d.GetDBPath(name)
+	if err != nil {
+		return nil, err
+	}
+	dbs := make([]*sql.DB, 0)
+	for _, file := range dbPaths {
+		db, err := d.OpenDB(file)
+		if err != nil {
+			return nil, err
+		}
+		dbs = append(dbs, db)
+	}
+	return dbs, nil
+}
+
+func (d *DBManager) GetDBPath(name string) ([]string, error) {
+	d.mutex.RLock()
+	dbPaths, ok := d.dbPaths[name]
+	d.mutex.RUnlock()
+	if !ok {
+		d.mutex.RLock()
+		fg, ok := d.fgs[name]
+		d.mutex.RUnlock()
+		if !ok {
+			return nil, errors.FileGroupNotFound(name)
+		}
+		list, err := fg.List()
+		if err != nil {
+			return nil, errors.DBFileNotFound(d.path, fg.PatternStr, err)
+		}
+		if len(list) == 0 {
+			return nil, errors.DBFileNotFound(d.path, fg.PatternStr, nil)
+		}
+		dbPaths = list
+		d.mutex.Lock()
+		d.dbPaths[name] = dbPaths
+		d.mutex.Unlock()
+	}
+	return dbPaths, nil
+}
+
+func (d *DBManager) OpenDB(path string) (*sql.DB, error) {
+	d.mutex.RLock()
+	db, ok := d.dbs[path]
+	d.mutex.RUnlock()
+	if ok {
+		return db, nil
+	}
+	var err error
+	tempPath := path
+	if runtime.GOOS == "windows" {
+		tempPath, err = filecopy.GetTempCopy(path)
+		if err != nil {
+			log.Err(err).Msgf("获取临时拷贝文件 %s 失败", path)
+			return nil, err
+		}
+	}
+	db, err = sql.Open("sqlite3", tempPath)
+	if err != nil {
+		log.Err(err).Msgf("连接数据库 %s 失败", path)
+		return nil, err
+	}
+	d.mutex.Lock()
+	d.dbs[path] = db
+	d.mutex.Unlock()
+	return db, nil
+}
+
+func (d *DBManager) Callback(event fsnotify.Event) error {
+	if !event.Op.Has(fsnotify.Create) {
+		return nil
+	}
+
+	d.mutex.Lock()
+	db, ok := d.dbs[event.Name]
+	if ok {
+		delete(d.dbs, event.Name)
+		go func(db *sql.DB) {
+			time.Sleep(time.Second * 5)
+			db.Close()
+		}(db)
+	}
+	d.mutex.Unlock()
+
+	return nil
+}
+
+func (d *DBManager) Start() error {
+	return d.fm.Start()
+}
+
+func (d *DBManager) Stop() error {
+	return d.fm.Stop()
+}
+
+func (d *DBManager) Close() error {
+	for _, db := range d.dbs {
+		db.Close()
+	}
+	return d.fm.Stop()
+}

internal/wechatdb/datasource/dbm/dbm_test.go

@@ -0,0 +1,42 @@
+package dbm
+
+import (
+	"fmt"
+	"testing"
+	"time"
+)
+
+func TestXxx(t *testing.T) {
+	path := "/Users/sarv/Documents/chatlog/bigjun_9e7a"
+
+	g := Group{
+		Name:      "session",
+		Pattern:   `session\.db$`,
+		BlackList: []string{},
+	}
+
+	d := NewDBManager(path)
+	d.AddGroup(g)
+	d.Start()
+
+	i := 0
+	for {
+		db, err := d.GetDB("session")
+		if err != nil {
+			fmt.Println(err)
+			break
+		}
+
+		var username string
+		row := db.QueryRow(`SELECT username FROM SessionTable LIMIT 1`)
+		if err := row.Scan(&username); err != nil {
+			fmt.Printf("Error scanning row: %v\n", err)
+			time.Sleep(100 * time.Millisecond)
+			continue
+		}
+		fmt.Printf("%d: Username: %s\n", i, username)
+		i++
+		time.Sleep(1000 * time.Millisecond)
+	}
+
+}

internal/wechatdb/datasource/dbm/group.go

@@ -0,0 +1,7 @@
+package dbm
+
+type Group struct {
+	Name      string
+	Pattern   string
+	BlackList []string
+}

internal/wechatdb/datasource/v4/datasource.go

@@ -10,21 +10,51 @@ import (
 	"strings"
 	"time"
 
+	"github.com/fsnotify/fsnotify"
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/rs/zerolog/log"
 
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/model"
-	"github.com/sjzar/chatlog/pkg/util"
+	"github.com/sjzar/chatlog/internal/wechatdb/datasource/dbm"
 )
 
 const (
-	MessageFilePattern = "^message_([0-9]?[0-9])?\\.db$"
+	Message = "message"
-	ContactFilePattern = "^contact\\.db$"
+	Contact = "contact"
-	SessionFilePattern = "^session\\.db$"
+	Session = "session"
-	MediaFilePattern   = "^hardlink\\.db$"
+	Media   = "media"
+	Voice   = "voice"
 )
 
+var Groups = []dbm.Group{
+	{
+		Name:      Message,
+		Pattern:   `^message_([0-9]?[0-9])?\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Contact,
+		Pattern:   `^contact\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Session,
+		Pattern:   `session\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Media,
+		Pattern:   `^hardlink\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Voice,
+		Pattern:   `^media_([0-9]?[0-9])?\.db$`,
+		BlackList: []string{},
+	},
+}
+
 // MessageDBInfo 存储消息数据库的信息
 type MessageDBInfo struct {
 	FilePath  string
@@ -34,55 +64,64 @@ type MessageDBInfo struct {
 
 type DataSource struct {
 	path string
-	messageDbs map[string]*sql.DB
+	dbm  *dbm.DBManager
-	contactDb  *sql.DB
-	sessionDb  *sql.DB
-	mediaDb    *sql.DB
 
 	// 消息数据库信息
-	messageFiles []MessageDBInfo
+	messageInfos []MessageDBInfo
 }
 
 func New(path string) (*DataSource, error) {
+
 	ds := &DataSource{
 		path:         path,
-		messageDbs:   make(map[string]*sql.DB),
+		dbm:          dbm.NewDBManager(path),
-		messageFiles: make([]MessageDBInfo, 0),
+		messageInfos: make([]MessageDBInfo, 0),
 	}
 
-	if err := ds.initMessageDbs(path); err != nil {
+	for _, g := range Groups {
+		ds.dbm.AddGroup(g)
+	}
+
+	if err := ds.dbm.Start(); err != nil {
+		return nil, err
+	}
+
+	if err := ds.initMessageDbs(); err != nil {
 		return nil, errors.DBInitFailed(err)
 	}
-	if err := ds.initContactDb(path); err != nil {
+
-		return nil, errors.DBInitFailed(err)
+	ds.dbm.AddCallback(Message, func(event fsnotify.Event) error {
+		if !event.Op.Has(fsnotify.Create) {
+			return nil
 		}
-	if err := ds.initSessionDb(path); err != nil {
+		if err := ds.initMessageDbs(); err != nil {
-		return nil, errors.DBInitFailed(err)
+			log.Err(err).Msgf("Failed to reinitialize message DBs: %s", event.Name)
-	}
-	if err := ds.initMediaDb(path); err != nil {
-		return nil, errors.DBInitFailed(err)
 		}
+		return nil
+	})
 
 	return ds, nil
 }
 
-func (ds *DataSource) initMessageDbs(path string) error {
+func (ds *DataSource) SetCallback(name string, callback func(event fsnotify.Event) error) error {
-	// 查找所有消息数据库文件
+	if name == "chatroom" {
-	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
+		name = Contact
-	if err != nil {
-		return errors.DBFileNotFound(path, MessageFilePattern, err)
 	}
+	return ds.dbm.AddCallback(name, callback)
+}
 
-	if len(files) == 0 {
+func (ds *DataSource) initMessageDbs() error {
-		return errors.DBFileNotFound(path, MessageFilePattern, nil)
+	dbPaths, err := ds.dbm.GetDBPath(Message)
+	if err != nil {
+		return err
 	}
 
 	// 处理每个数据库文件
-	for _, filePath := range files {
+	infos := make([]MessageDBInfo, 0)
-		// 连接数据库
+	for _, filePath := range dbPaths {
-		db, err := sql.Open("sqlite3", filePath)
+		db, err := ds.dbm.OpenDB(filePath)
 		if err != nil {
-			log.Err(err).Msgf("连接数据库 %s 失败", filePath)
+			log.Err(err).Msgf("获取数据库 %s 失败", filePath)
 			continue
 		}
 
@@ -93,90 +132,38 @@ func (ds *DataSource) initMessageDbs(path string) error {
 		row := db.QueryRow("SELECT timestamp FROM Timestamp LIMIT 1")
 		if err := row.Scan(&timestamp); err != nil {
 			log.Err(err).Msgf("获取数据库 %s 的时间戳失败", filePath)
-			db.Close()
 			continue
 		}
 		startTime = time.Unix(timestamp, 0)
 
 		// 保存数据库信息
-		ds.messageFiles = append(ds.messageFiles, MessageDBInfo{
+		infos = append(infos, MessageDBInfo{
 			FilePath:  filePath,
 			StartTime: startTime,
 		})
-
-		// 保存数据库连接
-		ds.messageDbs[filePath] = db
 	}
 
 	// 按照 StartTime 排序数据库文件
-	sort.Slice(ds.messageFiles, func(i, j int) bool {
+	sort.Slice(infos, func(i, j int) bool {
-		return ds.messageFiles[i].StartTime.Before(ds.messageFiles[j].StartTime)
+		return infos[i].StartTime.Before(infos[j].StartTime)
 	})
 
 	// 设置结束时间
-	for i := range ds.messageFiles {
+	for i := range infos {
-		if i == len(ds.messageFiles)-1 {
+		if i == len(infos)-1 {
-			ds.messageFiles[i].EndTime = time.Now()
+			infos[i].EndTime = time.Now()
 		} else {
-			ds.messageFiles[i].EndTime = ds.messageFiles[i+1].StartTime
+			infos[i].EndTime = infos[i+1].StartTime
 		}
 	}
-
+	ds.messageInfos = infos
-	return nil
-}
-
-func (ds *DataSource) initContactDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, ContactFilePattern, err)
-	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, ContactFilePattern, nil)
-	}
-
-	ds.contactDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-
-	return nil
-}
-
-func (ds *DataSource) initSessionDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, SessionFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, SessionFilePattern, err)
-	}
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, SessionFilePattern, nil)
-	}
-	ds.sessionDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-	return nil
-}
-
-func (ds *DataSource) initMediaDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, MediaFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, MediaFilePattern, err)
-	}
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, MediaFilePattern, nil)
-	}
-	ds.mediaDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
 	return nil
 }
 
 // getDBInfosForTimeRange 获取时间范围内的数据库信息
 func (ds *DataSource) getDBInfosForTimeRange(startTime, endTime time.Time) []MessageDBInfo {
 	var dbs []MessageDBInfo
-	for _, info := range ds.messageFiles {
+	for _, info := range ds.messageInfos {
 		if info.StartTime.Before(endTime) && info.EndTime.After(startTime) {
 			dbs = append(dbs, info)
 		}
@@ -188,6 +175,7 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 	if talker == "" {
 		return nil, errors.ErrTalkerEmpty
 	}
+	log.Debug().Msg(talker)
 
 	// 找到时间范围内的数据库文件
 	dbInfos := ds.getDBInfosForTimeRange(startTime, endTime)
@@ -209,13 +197,13 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 			return nil, err
 		}
 
-		db, ok := ds.messageDbs[dbInfo.FilePath]
+		db, err := ds.dbm.OpenDB(dbInfo.FilePath)
-		if !ok {
+		if err != nil {
 			log.Error().Msgf("数据库 %s 未打开", dbInfo.FilePath)
 			continue
 		}
 
-		messages, err := ds.getMessagesFromDB(ctx, db, dbInfo, startTime, endTime, talker)
+		messages, err := ds.getMessagesFromDB(ctx, db, startTime, endTime, talker)
 		if err != nil {
 			log.Err(err).Msgf("从数据库 %s 获取消息失败", dbInfo.FilePath)
 			continue
@@ -250,8 +238,8 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 
 // getMessagesSingleFile 从单个数据库文件获取消息
 func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
-	db, ok := ds.messageDbs[dbInfo.FilePath]
+	db, err := ds.dbm.OpenDB(dbInfo.FilePath)
-	if !ok {
+	if err != nil {
 		return nil, errors.DBConnectFailed(dbInfo.FilePath, nil)
 	}
 
@@ -260,12 +248,26 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
 	tableName := "Msg_" + talkerMd5
 
+	// 检查表是否存在
+	var exists bool
+	err = db.QueryRowContext(ctx,
+		"SELECT 1 FROM sqlite_master WHERE type='table' AND name=?",
+		tableName).Scan(&exists)
+
+	if err != nil {
+		if err == sql.ErrNoRows {
+			// 表不存在，返回空结果
+			return []*model.Message{}, nil
+		}
+		return nil, errors.QueryFailed("", err)
+	}
+
 	// 构建查询条件
 	conditions := []string{"create_time >= ? AND create_time <= ?"}
 	args := []interface{}{startTime.Unix(), endTime.Unix()}
 
 	query := fmt.Sprintf(`
-		SELECT m.sort_seq, m.local_type, n.user_name, m.create_time, m.message_content, m.packed_info_data, m.status
+		SELECT m.sort_seq, m.server_id, m.local_type, n.user_name, m.create_time, m.message_content, m.packed_info_data, m.status
 		FROM %s m
 		LEFT JOIN Name2Id n ON m.real_sender_id = n.rowid
 		WHERE %s 
@@ -293,6 +295,7 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 		var msg model.MessageV4
 		err := rows.Scan(
 			&msg.SortSeq,
+			&msg.ServerID,
 			&msg.LocalType,
 			&msg.UserName,
 			&msg.CreateTime,
@@ -311,7 +314,7 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 }
 
 // getMessagesFromDB 从数据库获取消息
-func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string) ([]*model.Message, error) {
+func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, startTime, endTime time.Time, talker string) ([]*model.Message, error) {
 	// 构建表名
 	_talkerMd5Bytes := md5.Sum([]byte(talker))
 	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
@@ -336,7 +339,7 @@ func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, dbInfo
 	args := []interface{}{startTime.Unix(), endTime.Unix()}
 
 	query := fmt.Sprintf(`
-		SELECT m.sort_seq, m.local_type, n.user_name, m.create_time, m.message_content, m.packed_info_data, m.status
+		SELECT m.sort_seq, m.server_id, m.local_type, n.user_name, m.create_time, m.message_content, m.packed_info_data, m.status
 		FROM %s m
 		LEFT JOIN Name2Id n ON m.real_sender_id = n.rowid
 		WHERE %s 
@@ -361,6 +364,7 @@ func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, dbInfo
 		var msg model.MessageV4
 		err := rows.Scan(
 			&msg.SortSeq,
+			&msg.ServerID,
 			&msg.LocalType,
 			&msg.UserName,
 			&msg.CreateTime,
@@ -404,7 +408,11 @@ func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Contact)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -436,13 +444,18 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 	var query string
 	var args []interface{}
 
+	// 执行查询
+	db, err := ds.dbm.GetDB(Contact)
+	if err != nil {
+		return nil, err
+	}
+
 	if key != "" {
 		// 按照关键字查询
 		query = `SELECT username, owner, ext_buffer FROM chat_room WHERE username = ?`
 		args = []interface{}{key}
 
-		// 执行查询
+		rows, err := db.QueryContext(ctx, query, args...)
-		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
 		if err != nil {
 			return nil, errors.QueryFailed(query, err)
 		}
@@ -469,7 +482,7 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 			contacts, err := ds.GetContacts(ctx, key, 1, 0)
 			if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
 				// 再次尝试通过用户名查找群聊
-				rows, err := ds.contactDb.QueryContext(ctx,
+				rows, err := db.QueryContext(ctx,
 					`SELECT username, owner, ext_buffer FROM chat_room WHERE username = ?`,
 					contacts[0].UserName)
 
@@ -519,7 +532,7 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 		}
 
 		// 执行查询
-		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		rows, err := db.QueryContext(ctx, query, args...)
 		if err != nil {
 			return nil, errors.QueryFailed(query, err)
 		}
@@ -573,7 +586,11 @@ func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.sessionDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Session)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -605,10 +622,6 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 		return nil, errors.ErrKeyEmpty
 	}
 
-	if len(key) != 32 {
-		return nil, errors.ErrKeyLengthMust32
-	}
-
 	var table string
 	switch _type {
 	case "image":
@@ -617,6 +630,8 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 		table = "video_hardlink_info_v3"
 	case "file":
 		table = "file_hardlink_info_v3"
+	case "voice":
+		return ds.GetVoice(ctx, key)
 	default:
 		return nil, errors.MediaTypeUnsupported(_type)
 	}
@@ -639,7 +654,12 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 	query += " WHERE f.md5 = ? OR f.file_name LIKE ? || '%'"
 	args := []interface{}{key, key}
 
-	rows, err := ds.mediaDb.QueryContext(ctx, query, args...)
+	// 执行查询
+	db, err := ds.dbm.GetDB(Media)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -675,39 +695,51 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 	return media, nil
 }
 
-func (ds *DataSource) Close() error {
+func (ds *DataSource) GetVoice(ctx context.Context, key string) (*model.Media, error) {
-	var errs []error
+	if key == "" {
+		return nil, errors.ErrKeyEmpty
+	}
 
-	// 关闭消息数据库连接
+	query := `
-	for _, db := range ds.messageDbs {
+	SELECT voice_data
-		if err := db.Close(); err != nil {
+	FROM VoiceInfo
-			errs = append(errs, err)
+	WHERE svr_id = ? 
+	`
+	args := []interface{}{key}
+
+	dbs, err := ds.dbm.GetDBs(Voice)
+	if err != nil {
+		return nil, errors.DBConnectFailed("", err)
+	}
+
+	for _, db := range dbs {
+		rows, err := db.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, errors.QueryFailed(query, err)
+		}
+		defer rows.Close()
+
+		for rows.Next() {
+			var voiceData []byte
+			err := rows.Scan(
+				&voiceData,
+			)
+			if err != nil {
+				return nil, errors.ScanRowFailed(err)
+			}
+			if len(voiceData) > 0 {
+				return &model.Media{
+					Type: "voice",
+					Key:  key,
+					Data: voiceData,
+				}, nil
+			}
 		}
 	}
 
-	// 关闭联系人数据库连接
+	return nil, errors.ErrMediaNotFound
-	if ds.contactDb != nil {
+}
-		if err := ds.contactDb.Close(); err != nil {
+
-			errs = append(errs, err)
+func (ds *DataSource) Close() error {
-		}
+	return ds.dbm.Close()
-	}
-
-	// 关闭会话数据库连接
-	if ds.sessionDb != nil {
-		if err := ds.sessionDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	if ds.mediaDb != nil {
-		if err := ds.mediaDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	if len(errs) > 0 {
-		return errors.DBCloseFailed(errs[0])
-	}
-
-	return nil
 }

internal/wechatdb/datasource/windowsv3/datasource.go

@@ -9,22 +9,57 @@ import (
 	"strings"
 	"time"
 
+	"github.com/fsnotify/fsnotify"
 	_ "github.com/mattn/go-sqlite3"
 	"github.com/rs/zerolog/log"
 
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/model"
-	"github.com/sjzar/chatlog/pkg/util"
+	"github.com/sjzar/chatlog/internal/wechatdb/datasource/dbm"
 )
 
 const (
-	MessageFilePattern = "^MSG([0-9]?[0-9])?\\.db$"
+	Message = "message"
-	ContactFilePattern = "^MicroMsg.db$"
+	Contact = "contact"
-	ImageFilePattern   = "^HardLinkImage\\.db$"
+	Image   = "image"
-	VideoFilePattern   = "^HardLinkVideo\\.db$"
+	Video   = "video"
-	FileFilePattern    = "^HardLinkFile\\.db$"
+	File    = "file"
+	Voice   = "voice"
 )
 
+var Groups = []dbm.Group{
+	{
+		Name:      Message,
+		Pattern:   `^MSG([0-9]?[0-9])?\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Contact,
+		Pattern:   `^MicroMsg.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Image,
+		Pattern:   `^HardLinkImage\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Video,
+		Pattern:   `^HardLinkVideo\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      File,
+		Pattern:   `^HardLinkFile\.db$`,
+		BlackList: []string{},
+	},
+	{
+		Name:      Voice,
+		Pattern:   `^MediaMSG([0-9])?\.db$`,
+		BlackList: []string{},
+	},
+}
+
 // MessageDBInfo 保存消息数据库的信息
 type MessageDBInfo struct {
 	FilePath  string
@@ -35,61 +70,67 @@ type MessageDBInfo struct {
 
 // DataSource 实现了 DataSource 接口
 type DataSource struct {
-	// 消息数据库
+	path string
-	messageFiles []MessageDBInfo
+	dbm  *dbm.DBManager
-	messageDbs   map[string]*sql.DB
 
-	// 联系人数据库
+	// 消息数据库信息
-	contactDbFile string
+	messageInfos []MessageDBInfo
-	contactDb     *sql.DB
-
-	imageDb *sql.DB
-	videoDb *sql.DB
-	fileDb  *sql.DB
 }
 
 // New 创建一个新的 WindowsV3DataSource
 func New(path string) (*DataSource, error) {
 	ds := &DataSource{
-		messageFiles: make([]MessageDBInfo, 0),
+		path:         path,
-		messageDbs:   make(map[string]*sql.DB),
+		dbm:          dbm.NewDBManager(path),
+		messageInfos: make([]MessageDBInfo, 0),
 	}
 
-	// 初始化消息数据库
+	for _, g := range Groups {
-	if err := ds.initMessageDbs(path); err != nil {
+		ds.dbm.AddGroup(g)
+	}
+
+	if err := ds.dbm.Start(); err != nil {
+		return nil, err
+	}
+
+	if err := ds.initMessageDbs(); err != nil {
 		return nil, errors.DBInitFailed(err)
 	}
 
-	// 初始化联系人数据库
+	ds.dbm.AddCallback(Message, func(event fsnotify.Event) error {
-	if err := ds.initContactDb(path); err != nil {
+		if !event.Op.Has(fsnotify.Create) {
-		return nil, errors.DBInitFailed(err)
+			return nil
 		}
-
+		if err := ds.initMessageDbs(); err != nil {
-	if err := ds.initMediaDb(path); err != nil {
+			log.Err(err).Msgf("Failed to reinitialize message DBs: %s", event.Name)
-		return nil, errors.DBInitFailed(err)
 		}
+		return nil
+	})
 
 	return ds, nil
 }
 
-// initMessageDbs 初始化消息数据库
+func (ds *DataSource) SetCallback(name string, callback func(event fsnotify.Event) error) error {
-func (ds *DataSource) initMessageDbs(path string) error {
+	if name == "chatroom" {
-	// 查找所有消息数据库文件
+		name = Contact
-	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, MessageFilePattern, err)
 	}
+	return ds.dbm.AddCallback(name, callback)
+}
 
-	if len(files) == 0 {
+// initMessageDbs 初始化消息数据库
-		return errors.DBFileNotFound(path, MessageFilePattern, nil)
+func (ds *DataSource) initMessageDbs() error {
+	// 获取所有消息数据库文件路径
+	dbPaths, err := ds.dbm.GetDBPath(Message)
+	if err != nil {
+		return err
 	}
 
 	// 处理每个数据库文件
-	for _, filePath := range files {
+	infos := make([]MessageDBInfo, 0)
-		// 连接数据库
+	for _, filePath := range dbPaths {
-		db, err := sql.Open("sqlite3", filePath)
+		db, err := ds.dbm.OpenDB(filePath)
 		if err != nil {
-			log.Err(err).Msgf("连接数据库 %s 失败", filePath)
+			log.Err(err).Msgf("获取数据库 %s 失败", filePath)
 			continue
 		}
 
@@ -99,7 +140,6 @@ func (ds *DataSource) initMessageDbs(path string) error {
 		rows, err := db.Query("SELECT tableIndex, tableVersion, tableDesc FROM DBInfo")
 		if err != nil {
 			log.Err(err).Msgf("查询数据库 %s 的 DBInfo 表失败", filePath)
-			db.Close()
 			continue
 		}
 
@@ -126,7 +166,6 @@ func (ds *DataSource) initMessageDbs(path string) error {
 		rows, err = db.Query("SELECT UsrName FROM Name2ID")
 		if err != nil {
 			log.Err(err).Msgf("查询数据库 %s 的 Name2ID 表失败", filePath)
-			db.Close()
 			continue
 		}
 
@@ -143,105 +182,34 @@ func (ds *DataSource) initMessageDbs(path string) error {
 		rows.Close()
 
 		// 保存数据库信息
-		ds.messageFiles = append(ds.messageFiles, MessageDBInfo{
+		infos = append(infos, MessageDBInfo{
 			FilePath:  filePath,
 			StartTime: startTime,
 			TalkerMap: talkerMap,
 		})
-
-		// 保存数据库连接
-		ds.messageDbs[filePath] = db
 	}
 
 	// 按照 StartTime 排序数据库文件
-	sort.Slice(ds.messageFiles, func(i, j int) bool {
+	sort.Slice(infos, func(i, j int) bool {
-		return ds.messageFiles[i].StartTime.Before(ds.messageFiles[j].StartTime)
+		return infos[i].StartTime.Before(infos[j].StartTime)
 	})
 
 	// 设置结束时间
-	for i := range ds.messageFiles {
+	for i := range infos {
-		if i == len(ds.messageFiles)-1 {
+		if i == len(infos)-1 {
-			ds.messageFiles[i].EndTime = time.Now()
+			infos[i].EndTime = time.Now()
 		} else {
-			ds.messageFiles[i].EndTime = ds.messageFiles[i+1].StartTime
+			infos[i].EndTime = infos[i+1].StartTime
 		}
 	}
-
+	ds.messageInfos = infos
-	return nil
-}
-
-// initContactDb 初始化联系人数据库
-func (ds *DataSource) initContactDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, ContactFilePattern, err)
-	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, ContactFilePattern, nil)
-	}
-
-	ds.contactDbFile = files[0]
-
-	ds.contactDb, err = sql.Open("sqlite3", ds.contactDbFile)
-	if err != nil {
-		return errors.DBConnectFailed(ds.contactDbFile, err)
-	}
-
-	return nil
-}
-
-// initContactDb 初始化联系人数据库
-func (ds *DataSource) initMediaDb(path string) error {
-	files, err := util.FindFilesWithPatterns(path, ImageFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, ImageFilePattern, err)
-	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, ImageFilePattern, nil)
-	}
-
-	ds.imageDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-
-	files, err = util.FindFilesWithPatterns(path, VideoFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, VideoFilePattern, err)
-	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, VideoFilePattern, nil)
-	}
-
-	ds.videoDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-
-	files, err = util.FindFilesWithPatterns(path, FileFilePattern, true)
-	if err != nil {
-		return errors.DBFileNotFound(path, FileFilePattern, err)
-	}
-
-	if len(files) == 0 {
-		return errors.DBFileNotFound(path, FileFilePattern, nil)
-	}
-
-	ds.fileDb, err = sql.Open("sqlite3", files[0])
-	if err != nil {
-		return errors.DBConnectFailed(files[0], err)
-	}
-
 	return nil
 }
 
 // getDBInfosForTimeRange 获取时间范围内的数据库信息
 func (ds *DataSource) getDBInfosForTimeRange(startTime, endTime time.Time) []MessageDBInfo {
 	var dbs []MessageDBInfo
-	for _, info := range ds.messageFiles {
+	for _, info := range ds.messageInfos {
 		if info.StartTime.Before(endTime) && info.EndTime.After(startTime) {
 			dbs = append(dbs, info)
 		}
@@ -271,69 +239,19 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 			return nil, err
 		}
 
-		db, ok := ds.messageDbs[dbInfo.FilePath]
+		db, err := ds.dbm.OpenDB(dbInfo.FilePath)
-		if !ok {
+		if err != nil {
 			log.Error().Msgf("数据库 %s 未打开", dbInfo.FilePath)
 			continue
 		}
 
-		// 构建查询条件
+		messages, err := ds.getMessagesFromDB(ctx, db, dbInfo, startTime, endTime, talker)
-		conditions := []string{"Sequence >= ? AND Sequence <= ?"}
-		args := []interface{}{startTime.Unix() * 1000, endTime.Unix() * 1000}
-
-		if len(talker) > 0 {
-			talkerID, ok := dbInfo.TalkerMap[talker]
-			if ok {
-				conditions = append(conditions, "TalkerId = ?")
-				args = append(args, talkerID)
-			} else {
-				conditions = append(conditions, "StrTalker = ?")
-				args = append(args, talker)
-			}
-		}
-
-		query := fmt.Sprintf(`
-            SELECT Sequence, CreateTime, StrTalker, IsSender, 
-                Type, SubType, StrContent, CompressContent, BytesExtra
-            FROM MSG 
-            WHERE %s 
-            ORDER BY Sequence ASC
-        `, strings.Join(conditions, " AND "))
-
-		// 执行查询
-		rows, err := db.QueryContext(ctx, query, args...)
 		if err != nil {
-			log.Err(err).Msgf("查询数据库 %s 失败", dbInfo.FilePath)
+			log.Err(err).Msgf("从数据库 %s 获取消息失败", dbInfo.FilePath)
 			continue
 		}
 
-		// 处理查询结果
+		totalMessages = append(totalMessages, messages...)
-		for rows.Next() {
-			var msg model.MessageV3
-			var compressContent []byte
-			var bytesExtra []byte
-
-			err := rows.Scan(
-				&msg.Sequence,
-				&msg.CreateTime,
-				&msg.StrTalker,
-				&msg.IsSender,
-				&msg.Type,
-				&msg.SubType,
-				&msg.StrContent,
-				&compressContent,
-				&bytesExtra,
-			)
-			if err != nil {
-				log.Err(err).Msg("扫描消息行失败")
-				continue
-			}
-			msg.CompressContent = compressContent
-			msg.BytesExtra = bytesExtra
-
-			totalMessages = append(totalMessages, msg.Wrap())
-		}
-		rows.Close()
 
 		if limit+offset > 0 && len(totalMessages) >= limit+offset {
 			break
@@ -362,6 +280,11 @@ func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.T
 
 // getMessagesSingleFile 从单个数据库文件获取消息
 func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	db, err := ds.dbm.OpenDB(dbInfo.FilePath)
+	if err != nil {
+		return nil, errors.DBConnectFailed(dbInfo.FilePath, nil)
+	}
+
 	// 构建查询条件
 	conditions := []string{"Sequence >= ? AND Sequence <= ?"}
 	args := []interface{}{startTime.Unix() * 1000, endTime.Unix() * 1000}
@@ -377,7 +300,7 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 		}
 	}
 	query := fmt.Sprintf(`
-        SELECT Sequence, CreateTime, StrTalker, IsSender, 
+        SELECT MsgSvrID, Sequence, CreateTime, StrTalker, IsSender, 
             Type, SubType, StrContent, CompressContent, BytesExtra
         FROM MSG 
         WHERE %s 
@@ -393,7 +316,7 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 	}
 
 	// 执行查询
-	rows, err := ds.messageDbs[dbInfo.FilePath].QueryContext(ctx, query, args...)
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -406,6 +329,7 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 		var compressContent []byte
 		var bytesExtra []byte
 		err := rows.Scan(
+			&msg.MsgSvrID,
 			&msg.Sequence,
 			&msg.CreateTime,
 			&msg.StrTalker,
@@ -426,6 +350,69 @@ func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageD
 	return totalMessages, nil
 }
 
+// getMessagesFromDB 从数据库获取消息
+func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string) ([]*model.Message, error) {
+	// 构建查询条件
+	conditions := []string{"Sequence >= ? AND Sequence <= ?"}
+	args := []interface{}{startTime.Unix() * 1000, endTime.Unix() * 1000}
+
+	if len(talker) > 0 {
+		talkerID, ok := dbInfo.TalkerMap[talker]
+		if ok {
+			conditions = append(conditions, "TalkerId = ?")
+			args = append(args, talkerID)
+		} else {
+			conditions = append(conditions, "StrTalker = ?")
+			args = append(args, talker)
+		}
+	}
+
+	query := fmt.Sprintf(`
+		SELECT MsgSvrID, Sequence, CreateTime, StrTalker, IsSender, 
+			Type, SubType, StrContent, CompressContent, BytesExtra
+		FROM MSG 
+		WHERE %s 
+		ORDER BY Sequence ASC
+	`, strings.Join(conditions, " AND "))
+
+	// 执行查询
+	rows, err := db.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, errors.QueryFailed(query, err)
+	}
+	defer rows.Close()
+
+	// 处理查询结果
+	messages := []*model.Message{}
+	for rows.Next() {
+		var msg model.MessageV3
+		var compressContent []byte
+		var bytesExtra []byte
+
+		err := rows.Scan(
+			&msg.MsgSvrID,
+			&msg.Sequence,
+			&msg.CreateTime,
+			&msg.StrTalker,
+			&msg.IsSender,
+			&msg.Type,
+			&msg.SubType,
+			&msg.StrContent,
+			&compressContent,
+			&bytesExtra,
+		)
+		if err != nil {
+			return nil, errors.ScanRowFailed(err)
+		}
+		msg.CompressContent = compressContent
+		msg.BytesExtra = bytesExtra
+
+		messages = append(messages, msg.Wrap())
+	}
+
+	return messages, nil
+}
+
 // GetContacts 实现获取联系人信息的方法
 func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset int) ([]*model.Contact, error) {
 	var query string
@@ -451,7 +438,11 @@ func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Contact)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -489,7 +480,11 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 		args = []interface{}{key}
 
 		// 执行查询
-		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		db, err := ds.dbm.GetDB(Contact)
+		if err != nil {
+			return nil, err
+		}
+		rows, err := db.QueryContext(ctx, query, args...)
 		if err != nil {
 			return nil, errors.QueryFailed(query, err)
 		}
@@ -516,7 +511,7 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 			contacts, err := ds.GetContacts(ctx, key, 1, 0)
 			if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
 				// 再次尝试通过用户名查找群聊
-				rows, err := ds.contactDb.QueryContext(ctx,
+				rows, err := db.QueryContext(ctx,
 					`SELECT ChatRoomName, Reserved2, RoomData FROM ChatRoom WHERE ChatRoomName = ?`,
 					contacts[0].UserName)
 
@@ -566,7 +561,11 @@ func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offse
 		}
 
 		// 执行查询
-		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		db, err := ds.dbm.GetDB(Contact)
+		if err != nil {
+			return nil, err
+		}
+		rows, err := db.QueryContext(ctx, query, args...)
 		if err != nil {
 			return nil, errors.QueryFailed(query, err)
 		}
@@ -620,7 +619,11 @@ func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset
 	}
 
 	// 执行查询
-	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	db, err := ds.dbm.GetDB(Contact)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := db.QueryContext(ctx, query, args...)
 	if err != nil {
 		return nil, errors.QueryFailed(query, err)
 	}
@@ -652,30 +655,38 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 		return nil, errors.ErrKeyEmpty
 	}
 
+	if _type == "voice" {
+		return ds.GetVoice(ctx, key)
+	}
+
 	md5key, err := hex.DecodeString(key)
 	if err != nil {
 		return nil, errors.DecodeKeyFailed(err)
 	}
 
-	var db *sql.DB
+	var dbType string
 	var table1, table2 string
 
 	switch _type {
 	case "image":
-		db = ds.imageDb
+		dbType = Image
 		table1 = "HardLinkImageAttribute"
 		table2 = "HardLinkImageID"
 	case "video":
-		db = ds.videoDb
+		dbType = Video
 		table1 = "HardLinkVideoAttribute"
 		table2 = "HardLinkVideoID"
 	case "file":
-		db = ds.fileDb
+		dbType = File
 		table1 = "HardLinkFileAttribute"
 		table2 = "HardLinkFileID"
 	default:
 		return nil, errors.MediaTypeUnsupported(_type)
+	}
 
+	db, err := ds.dbm.GetDB(dbType)
+	if err != nil {
+		return nil, err
 	}
 
 	query := fmt.Sprintf(`
@@ -725,43 +736,52 @@ func (ds *DataSource) GetMedia(ctx context.Context, _type string, key string) (*
 	return media, nil
 }
 
+func (ds *DataSource) GetVoice(ctx context.Context, key string) (*model.Media, error) {
+	if key == "" {
+		return nil, errors.ErrKeyEmpty
+	}
+
+	query := `
+	SELECT Buf
+	FROM Media
+	WHERE Reserved0 = ? 
+	`
+	args := []interface{}{key}
+
+	dbs, err := ds.dbm.GetDBs(Voice)
+	if err != nil {
+		return nil, errors.DBConnectFailed("", err)
+	}
+
+	for _, db := range dbs {
+		rows, err := db.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, errors.QueryFailed(query, err)
+		}
+		defer rows.Close()
+
+		for rows.Next() {
+			var voiceData []byte
+			err := rows.Scan(
+				&voiceData,
+			)
+			if err != nil {
+				return nil, errors.ScanRowFailed(err)
+			}
+			if len(voiceData) > 0 {
+				return &model.Media{
+					Type: "voice",
+					Key:  key,
+					Data: voiceData,
+				}, nil
+			}
+		}
+	}
+
+	return nil, errors.ErrMediaNotFound
+}
+
 // Close 实现 DataSource 接口的 Close 方法
 func (ds *DataSource) Close() error {
-	var errs []error
+	return ds.dbm.Close()
-
-	// 关闭消息数据库连接
-	for _, db := range ds.messageDbs {
-		if err := db.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	// 关闭联系人数据库连接
-	if ds.contactDb != nil {
-		if err := ds.contactDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	if ds.imageDb != nil {
-		if err := ds.imageDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-	if ds.videoDb != nil {
-		if err := ds.videoDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-	if ds.fileDb != nil {
-		if err := ds.fileDb.Close(); err != nil {
-			errs = append(errs, err)
-		}
-	}
-
-	if len(errs) > 0 {
-		return errors.DBCloseFailed(errs[0])
-	}
-
-	return nil
 }

internal/wechatdb/repository/repository.go

@@ -3,6 +3,9 @@ package repository
 import (
 	"context"
 
+	"github.com/fsnotify/fsnotify"
+	"github.com/rs/zerolog/log"
+
 	"github.com/sjzar/chatlog/internal/errors"
 	"github.com/sjzar/chatlog/internal/model"
 	"github.com/sjzar/chatlog/internal/wechatdb/datasource"
@@ -61,6 +64,9 @@ func New(ds datasource.DataSource) (*Repository, error) {
 		return nil, errors.InitCacheFailed(err)
 	}
 
+	ds.SetCallback("contact", r.contactCallback)
+	ds.SetCallback("chatroom", r.chatroomCallback)
+
 	return r, nil
 }
 
@@ -79,6 +85,26 @@ func (r *Repository) initCache(ctx context.Context) error {
 	return nil
 }
 
+func (r *Repository) contactCallback(event fsnotify.Event) error {
+	if !event.Op.Has(fsnotify.Create) {
+		return nil
+	}
+	if err := r.initContactCache(context.Background()); err != nil {
+		log.Err(err).Msgf("Failed to reinitialize contact cache: %s", event.Name)
+	}
+	return nil
+}
+
+func (r *Repository) chatroomCallback(event fsnotify.Event) error {
+	if !event.Op.Has(fsnotify.Create) {
+		return nil
+	}
+	if err := r.initChatRoomCache(context.Background()); err != nil {
+		log.Err(err).Msgf("Failed to reinitialize contact cache: %s", event.Name)
+	}
+	return nil
+}
+
 // Close 实现 Repository 接口的 Close 方法
 func (r *Repository) Close() error {
 	return r.ds.Close()

pkg/filecopy/filecopy.go

@@ -0,0 +1,628 @@
+package filecopy
+
+import (
+	"encoding/json"
+	"fmt"
+	"hash/fnv"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+	"time"
+)
+
+var (
+	// Singleton locks to ensure only one thread processes the same file at a time
+	fileOperationLocks = make(map[string]*sync.Mutex)
+	locksMutex         = sync.RWMutex{}
+
+	// Mapping from original file paths to temporary file paths
+	pathToTempFile = make(map[string]string)
+	// Metadata information for original files
+	fileMetadata = make(map[string]fileMetaInfo)
+	// Track old versions of temporary files for each original file
+	oldVersions = make(map[string]string)
+	mapMutex    = sync.RWMutex{}
+
+	// Temporary directory
+	tempDir string
+	// Path to the mapping file
+	mappingFilePath string
+
+	// Channel for delayed file deletion
+	fileDeletionChan = make(chan FileDeletion, 1000)
+
+	// Default deletion delay time (30 seconds)
+	DefaultDeletionDelay = 30 * time.Second
+)
+
+type FileDeletion struct {
+	Path string
+	Time time.Time
+}
+
+// File metadata information
+type fileMetaInfo struct {
+	ModTime time.Time `json:"mod_time"`
+	Size    int64     `json:"size"`
+}
+
+// Persistent mapping information
+type persistentMapping struct {
+	OriginalPath string       `json:"original_path"`
+	TempPath     string       `json:"temp_path"`
+	Metadata     fileMetaInfo `json:"metadata"`
+}
+
+// Initialize temporary directory
+func initTempDir() {
+	// Get process name to create a unique temporary directory
+	procName := getProcessName()
+	tempDir = filepath.Join(os.TempDir(), "filecopy_"+procName)
+
+	if err := os.MkdirAll(tempDir, 0755); err != nil {
+		tempDir = filepath.Join(os.TempDir(), "filecopy")
+		if err := os.MkdirAll(tempDir, 0755); err != nil {
+			tempDir = os.TempDir()
+		}
+	}
+
+	// Set mapping file path
+	mappingFilePath = filepath.Join(tempDir, "file_mappings.json")
+
+	// Load existing mappings if available
+	loadMappings()
+
+	// Scan and clean existing temporary files
+	cleanupExistingTempFiles()
+}
+
+// Get process name
+func getProcessName() string {
+	executable, err := os.Executable()
+	if err != nil {
+		return "unknown"
+	}
+
+	// Extract base name (without extension)
+	baseName := filepath.Base(executable)
+	ext := filepath.Ext(baseName)
+	if ext != "" {
+		baseName = baseName[:len(baseName)-len(ext)]
+	}
+
+	// Clean name, keep only letters, numbers, underscores and hyphens
+	baseName = strings.Map(func(r rune) rune {
+		if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9') || r == '-' || r == '_' {
+			return r
+		}
+		return '_'
+	}, baseName)
+
+	return baseName
+}
+
+// Load file mappings from persistent storage
+func loadMappings() {
+	file, err := os.Open(mappingFilePath)
+	if err != nil {
+		// It's okay if the file doesn't exist yet
+		return
+	}
+	defer file.Close()
+
+	var mappings []persistentMapping
+	decoder := json.NewDecoder(file)
+	if err := decoder.Decode(&mappings); err != nil {
+		// If the file is corrupted, we'll just start fresh
+		return
+	}
+
+	// Restore mappings
+	mapMutex.Lock()
+	defer mapMutex.Unlock()
+
+	for _, mapping := range mappings {
+		// Verify that both the original file and temp file still exist
+		origStat, origErr := os.Stat(mapping.OriginalPath)
+		_, tempErr := os.Stat(mapping.TempPath)
+
+		if origErr == nil && tempErr == nil {
+			// Check if the original file has changed since the mapping was saved
+			if origStat.ModTime() == mapping.Metadata.ModTime && origStat.Size() == mapping.Metadata.Size {
+				// The mapping is still valid
+				pathToTempFile[mapping.OriginalPath] = mapping.TempPath
+				fileMetadata[mapping.OriginalPath] = mapping.Metadata
+			}
+		}
+	}
+}
+
+// Save file mappings to persistent storage
+func saveMappings() {
+	mapMutex.RLock()
+	defer mapMutex.RUnlock()
+
+	var mappings []persistentMapping
+	for origPath, tempPath := range pathToTempFile {
+		if meta, exists := fileMetadata[origPath]; exists {
+			mappings = append(mappings, persistentMapping{
+				OriginalPath: origPath,
+				TempPath:     tempPath,
+				Metadata:     meta,
+			})
+		}
+	}
+
+	// Create the file
+	file, err := os.Create(mappingFilePath)
+	if err != nil {
+		return
+	}
+	defer file.Close()
+
+	// Write the mappings
+	encoder := json.NewEncoder(file)
+	encoder.SetIndent("", "  ")
+	if err := encoder.Encode(mappings); err != nil {
+		// If we can't save, just continue - it's not critical
+		return
+	}
+}
+
+// Clean up existing temporary files
+func cleanupExistingTempFiles() {
+	files, err := os.ReadDir(tempDir)
+	if err != nil {
+		return
+	}
+
+	// Skip the mapping file
+	mappingFileName := filepath.Base(mappingFilePath)
+
+	// First, collect all files that are already in our mapping
+	knownFiles := make(map[string]bool)
+	mapMutex.RLock()
+	for _, tempPath := range pathToTempFile {
+		knownFiles[tempPath] = true
+	}
+	mapMutex.RUnlock()
+
+	// Group files by prefix (baseName_hashPrefix)
+	fileGroups := make(map[string][]tempFileInfo)
+
+	for _, file := range files {
+		if file.IsDir() {
+			continue
+		}
+
+		fileName := file.Name()
+
+		// Skip the mapping file
+		if fileName == mappingFileName {
+			continue
+		}
+
+		filePath := filepath.Join(tempDir, fileName)
+		parts := strings.Split(fileName, "_")
+
+		// Skip files that don't match our naming convention
+		if len(parts) < 3 {
+			removeFileImmediately(filePath)
+			continue
+		}
+
+		// Extract base name and hash part as key
+		baseName := parts[0]
+		hashPart := parts[1]
+		groupKey := baseName + "_" + hashPart
+
+		// Extract timestamp
+		timeStr := strings.Split(parts[2], ".")[0] // Remove extension part
+		var timestamp int64
+		if _, err := fmt.Sscanf(timeStr, "%d", &timestamp); err != nil {
+			removeFileImmediately(filePath)
+			continue
+		}
+
+		// Add file info to corresponding group
+		fileGroups[groupKey] = append(fileGroups[groupKey], tempFileInfo{
+			path:      filePath,
+			timestamp: timestamp,
+		})
+	}
+
+	// Process each group of files, keep only the newest one
+	for _, fileInfos := range fileGroups {
+		if len(fileInfos) == 0 {
+			continue
+		}
+
+		// Find the newest file
+		var newestFile tempFileInfo
+		for _, fileInfo := range fileInfos {
+			if fileInfo.timestamp > newestFile.timestamp {
+				newestFile = fileInfo
+			}
+		}
+
+		// Delete all files except the newest one
+		for _, fileInfo := range fileInfos {
+			if fileInfo.path != newestFile.path {
+				// If this file is already in our mapping, keep it
+				if knownFiles[fileInfo.path] {
+					continue
+				}
+				removeFileImmediately(fileInfo.path)
+			}
+		}
+	}
+}
+
+// Temporary file information
+type tempFileInfo struct {
+	path      string
+	timestamp int64
+}
+
+// Get file lock
+func getFileLock(path string) *sync.Mutex {
+	locksMutex.RLock()
+	lock, exists := fileOperationLocks[path]
+	locksMutex.RUnlock()
+
+	if exists {
+		return lock
+	}
+
+	locksMutex.Lock()
+	defer locksMutex.Unlock()
+
+	// Check again, might have been created while we were acquiring the write lock
+	lock, exists = fileOperationLocks[path]
+	if !exists {
+		lock = &sync.Mutex{}
+		fileOperationLocks[path] = lock
+	}
+
+	return lock
+}
+
+// GetTempCopy returns a temporary copy path of the original file
+// If the file hasn't changed since the last copy, returns the existing copy
+func GetTempCopy(originalPath string) (string, error) {
+	// Get the operation lock for this file to ensure thread safety
+	fileLock := getFileLock(originalPath)
+	fileLock.Lock()
+	defer fileLock.Unlock()
+
+	// Check if original file exists
+	stat, err := os.Stat(originalPath)
+	if err != nil {
+		return "", fmt.Errorf("original file does not exist: %w", err)
+	}
+
+	// Current file info
+	currentInfo := fileMetaInfo{
+		ModTime: stat.ModTime(),
+		Size:    stat.Size(),
+	}
+
+	// Check existing mapping
+	mapMutex.RLock()
+	tempPath, pathExists := pathToTempFile[originalPath]
+	cachedInfo, infoExists := fileMetadata[originalPath]
+	mapMutex.RUnlock()
+
+	// If we have an existing temp file and original file hasn't changed, return it
+	if pathExists && infoExists {
+		fileChanged := currentInfo.ModTime.After(cachedInfo.ModTime) ||
+			currentInfo.Size != cachedInfo.Size
+
+		if !fileChanged {
+			// Verify temp file still exists
+			if _, err := os.Stat(tempPath); err == nil {
+				// Try to open file to verify accessibility
+				if file, err := os.Open(tempPath); err == nil {
+					file.Close()
+					return tempPath, nil
+				}
+			}
+		}
+	}
+
+	// Generate new temp file path
+	fileName := filepath.Base(originalPath)
+	fileExt := filepath.Ext(fileName)
+	baseName := fileName[:len(fileName)-len(fileExt)]
+	if baseName == "" {
+		baseName = "file" // Use default name if empty
+	}
+
+	// Generate hash for original path
+	pathHash := hashString(originalPath)
+	hashPrefix := getHashPrefix(pathHash, 8)
+
+	// Format: basename_pathhash_timestamp.ext
+	timestamp := time.Now().UnixNano()
+	tempPath = filepath.Join(tempDir,
+		fmt.Sprintf("%s_%s_%d%s",
+			baseName,
+			hashPrefix,
+			timestamp,
+			fileExt))
+
+	// Copy file (with retry mechanism)
+	if err := copyFileWithRetry(originalPath, tempPath, 3); err != nil {
+		return "", err
+	}
+
+	// Update mappings
+	mapMutex.Lock()
+	oldPath := pathToTempFile[originalPath]
+
+	// If there's an old path and it's different, move it to old versions and schedule for deletion
+	if oldPath != "" && oldPath != tempPath {
+		// First clean up previous old version (if any)
+		if oldVersionPath, hasOldVersion := oldVersions[originalPath]; hasOldVersion && oldVersionPath != oldPath {
+			removeFileImmediately(oldVersionPath)
+		}
+
+		// Set current version as old version
+		oldVersions[originalPath] = oldPath
+		scheduleForDeletion(oldPath)
+	}
+
+	// Update to new temp file
+	pathToTempFile[originalPath] = tempPath
+	fileMetadata[originalPath] = currentInfo
+	mapMutex.Unlock()
+
+	// Save mappings to persistent storage
+	go saveMappings()
+
+	// Immediately clean up any other related temp files
+	go cleanupRelatedTempFiles(originalPath, tempPath, oldPath)
+
+	return tempPath, nil
+}
+
+// Immediately clean up other temp files related to the specified original file
+func cleanupRelatedTempFiles(originalPath, currentTempPath, knownOldPath string) {
+	// Extract hash prefix of original file to match related files
+	fileName := filepath.Base(originalPath)
+	fileExt := filepath.Ext(fileName)
+	baseName := fileName[:len(fileName)-len(fileExt)]
+	if baseName == "" {
+		baseName = "file"
+	}
+
+	pathHash := hashString(originalPath)
+	hashPrefix := getHashPrefix(pathHash, 8)
+
+	// File name prefix pattern
+	filePrefix := baseName + "_" + hashPrefix
+
+	currentTempPathNoExt := strings.TrimSuffix(currentTempPath, filepath.Ext(currentTempPath))
+	knownOldPathNoExt := strings.TrimSuffix(knownOldPath, filepath.Ext(knownOldPath))
+
+	files, err := os.ReadDir(tempDir)
+	if err != nil {
+		return
+	}
+
+	for _, file := range files {
+		if file.IsDir() {
+			continue
+		}
+
+		fileName := file.Name()
+
+		// Skip the mapping file
+		if fileName == filepath.Base(mappingFilePath) {
+			continue
+		}
+
+		filePath := filepath.Join(tempDir, fileName)
+		filePathNoExt := strings.TrimSuffix(filePath, filepath.Ext(filePath))
+
+		// Skip current file and known old version
+		if filePathNoExt == currentTempPathNoExt || filePathNoExt == knownOldPathNoExt {
+			continue
+		}
+
+		// If file name matches our pattern, delete it immediately
+		if strings.HasPrefix(fileName, filePrefix) {
+			removeFileImmediately(filePath)
+		}
+	}
+}
+
+// Immediately delete file without waiting for delay
+func removeFileImmediately(path string) {
+	if path == "" {
+		return
+	}
+
+	// Try to delete file
+	if err := os.Remove(path); err != nil && !os.IsNotExist(err) {
+		// Silently fail if we can't delete
+	}
+}
+
+// Schedule file for delayed deletion
+func scheduleForDeletion(path string) {
+	if path == "" {
+		return
+	}
+
+	// Check if file exists
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return
+	}
+
+	// Put file in deletion channel
+	select {
+	case fileDeletionChan <- FileDeletion{Path: path, Time: time.Now().Add(DefaultDeletionDelay)}:
+		// Successfully scheduled
+	default:
+		// If channel is full, delete file immediately
+		removeFileImmediately(path)
+	}
+}
+
+// File deletion handler
+func fileDeletionHandler() {
+	for {
+		// Get file to delete from channel
+		file := <-fileDeletionChan
+
+		if !time.Now().After(file.Time) {
+			time.Sleep(time.Until(file.Time))
+		}
+
+		// Ensure file is not in active mappings
+		isActive := false
+		mapMutex.RLock()
+		for _, activePath := range pathToTempFile {
+			if activePath == file.Path {
+				isActive = true
+				break
+			}
+		}
+
+		mapMutex.RUnlock()
+
+		if isActive {
+			continue
+		}
+
+		// Delete file
+		removeFileImmediately(file.Path)
+	}
+}
+
+// CleanupTempFiles cleans up unused temporary files
+func CleanupTempFiles() {
+	files, err := os.ReadDir(tempDir)
+	if err != nil {
+		return
+	}
+
+	// Skip the mapping file
+	mappingFileName := filepath.Base(mappingFilePath)
+
+	// Get current active temp file paths and old version paths
+	mapMutex.RLock()
+	activeTempFiles := make(map[string]bool)
+	for _, tempFilePath := range pathToTempFile {
+		tempFilePath = strings.TrimSuffix(tempFilePath, filepath.Ext(tempFilePath))
+		activeTempFiles[tempFilePath] = true
+	}
+	for _, oldVersionPath := range oldVersions {
+		oldVersionPath = strings.TrimSuffix(oldVersionPath, filepath.Ext(oldVersionPath))
+		activeTempFiles[oldVersionPath] = true
+	}
+	mapMutex.RUnlock()
+
+	// Schedule deletion of inactive temp files
+	for _, file := range files {
+		if file.IsDir() {
+			continue
+		}
+
+		fileName := file.Name()
+
+		// Skip the mapping file
+		if fileName == mappingFileName {
+			continue
+		}
+
+		tempFilePath := filepath.Join(tempDir, fileName)
+		tempFilePath = strings.TrimSuffix(tempFilePath, filepath.Ext(tempFilePath))
+		if !activeTempFiles[tempFilePath] {
+			scheduleForDeletion(tempFilePath)
+		}
+	}
+}
+
+// Copy file with retry mechanism
+func copyFileWithRetry(src, dst string, maxRetries int) error {
+	var err error
+	for i := 0; i < maxRetries; i++ {
+		err = copyFile(src, dst)
+		if err == nil {
+			return nil
+		}
+
+		// Wait before retrying
+		time.Sleep(time.Duration(100*(i+1)) * time.Millisecond)
+	}
+	return fmt.Errorf("failed to copy file after %d attempts: %w", maxRetries, err)
+}
+
+// Copy file
+func copyFile(src, dst string) error {
+	in, err := os.Open(src)
+	if err != nil {
+		return fmt.Errorf("failed to open source file: %w", err)
+	}
+	defer in.Close()
+
+	out, err := os.Create(dst)
+	if err != nil {
+		return fmt.Errorf("failed to create destination file: %w", err)
+	}
+	defer func() {
+		cerr := out.Close()
+		if err == nil && cerr != nil {
+			err = fmt.Errorf("failed to close destination file: %w", cerr)
+		}
+	}()
+
+	// Use buffered copy for better performance
+	buf := make([]byte, 256*1024) // 256KB buffer
+	if _, err = io.CopyBuffer(out, in, buf); err != nil {
+		return fmt.Errorf("failed to copy file contents: %w", err)
+	}
+
+	return out.Sync()
+}
+
+// Generate hash for string
+func hashString(s string) string {
+	h := fnv.New32a()
+	h.Write([]byte(s))
+	return fmt.Sprintf("%x", h.Sum32())
+}
+
+// Safely get hash prefix, avoid index out of bounds
+func getHashPrefix(hash string, length int) string {
+	if len(hash) <= length {
+		return hash
+	}
+	return hash[:length]
+}
+
+// Initialize temp directory and start background cleanup
+func init() {
+	// Initialize temp directory and scan existing files
+	initTempDir()
+
+	// Start multiple file deletion handlers
+	for i := 0; i < 2; i++ {
+		go fileDeletionHandler()
+	}
+
+	// Start periodic cleanup routine
+	go func() {
+		for {
+			time.Sleep(30 * time.Second)
+			CleanupTempFiles()
+
+			// Also periodically save mappings
+			saveMappings()
+		}
+	}()
+}

pkg/filemonitor/filegroup.go

@@ -0,0 +1,182 @@
+package filemonitor
+
+import (
+	"errors"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"regexp"
+	"strings"
+	"sync"
+
+	"github.com/fsnotify/fsnotify"
+	"github.com/rs/zerolog/log"
+)
+
+// FileChangeCallback defines the callback function signature for file change events
+type FileChangeCallback func(event fsnotify.Event) error
+
+// FileGroup represents a group of files with the same processing logic
+type FileGroup struct {
+	ID         string               // Unique identifier
+	RootDir    string               // Root directory
+	Pattern    *regexp.Regexp       // File matching pattern
+	PatternStr string               // Original pattern string for rebuilding
+	Blacklist  []string             // Blacklist patterns
+	Callbacks  []FileChangeCallback // File change callbacks
+	mutex      sync.RWMutex         // Concurrency control
+}
+
+// NewFileGroup creates a new file group
+func NewFileGroup(id, rootDir, pattern string, blacklist []string) (*FileGroup, error) {
+	// Compile the regular expression
+	re, err := regexp.Compile(pattern)
+	if err != nil {
+		return nil, fmt.Errorf("invalid pattern '%s': %w", pattern, err)
+	}
+
+	// Normalize root directory path
+	rootDir = filepath.Clean(rootDir)
+
+	return &FileGroup{
+		ID:         id,
+		RootDir:    rootDir,
+		Pattern:    re,
+		PatternStr: pattern,
+		Blacklist:  blacklist,
+		Callbacks:  []FileChangeCallback{},
+	}, nil
+}
+
+// AddCallback adds a callback function to the file group
+func (fg *FileGroup) AddCallback(callback FileChangeCallback) {
+	fg.mutex.Lock()
+	defer fg.mutex.Unlock()
+
+	fg.Callbacks = append(fg.Callbacks, callback)
+}
+
+// RemoveCallback removes a callback function from the file group
+func (fg *FileGroup) RemoveCallback(callbackToRemove FileChangeCallback) bool {
+	fg.mutex.Lock()
+	defer fg.mutex.Unlock()
+
+	for i, callback := range fg.Callbacks {
+		// Compare function addresses
+		if fmt.Sprintf("%p", callback) == fmt.Sprintf("%p", callbackToRemove) {
+			// Remove the callback
+			fg.Callbacks = append(fg.Callbacks[:i], fg.Callbacks[i+1:]...)
+			return true
+		}
+	}
+
+	return false
+}
+
+// Match checks if a file path matches this group's criteria
+func (fg *FileGroup) Match(path string) bool {
+	// Normalize paths for comparison
+	path = filepath.Clean(path)
+	rootDir := filepath.Clean(fg.RootDir)
+
+	// Check if path is under root directory
+	// Use filepath.Rel to handle path comparison safely across different OSes
+	relPath, err := filepath.Rel(rootDir, path)
+	if err != nil || strings.HasPrefix(relPath, "..") {
+		return false
+	}
+
+	// Check if file matches pattern
+	if !fg.Pattern.MatchString(filepath.Base(path)) {
+		return false
+	}
+
+	// Check blacklist
+	for _, blackItem := range fg.Blacklist {
+		if strings.Contains(relPath, blackItem) {
+			return false
+		}
+	}
+
+	return true
+}
+
+// List returns a list of files in the group (real-time scan)
+func (fg *FileGroup) List() ([]string, error) {
+	files := []string{}
+
+	// Scan directory for matching files using fs.WalkDir
+	err := fs.WalkDir(os.DirFS(fg.RootDir), ".", func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return fs.SkipDir
+		}
+
+		// Skip directories
+		if d.IsDir() {
+			return nil
+		}
+
+		// Convert relative path to absolute
+		absPath := filepath.Join(fg.RootDir, path)
+
+		// Use Match function to check if file belongs to this group
+		if fg.Match(absPath) {
+			files = append(files, absPath)
+		}
+
+		return nil
+	})
+
+	if err != nil && !errors.Is(err, fs.ErrNotExist) {
+		return nil, fmt.Errorf("error listing files: %w", err)
+	}
+
+	return files, nil
+}
+
+// ListMatchingDirectories returns directories containing matching files
+func (fg *FileGroup) ListMatchingDirectories() (map[string]bool, error) {
+	directories := make(map[string]bool)
+
+	// Get matching files
+	files, err := fg.List()
+	if err != nil {
+		return nil, err
+	}
+
+	// Extract directories from matching files
+	for _, file := range files {
+		dir := filepath.Dir(file)
+		directories[dir] = true
+	}
+
+	return directories, nil
+}
+
+// HandleEvent processes a file event and triggers callbacks if the file matches
+func (fg *FileGroup) HandleEvent(event fsnotify.Event) {
+	// Check if this event is relevant for this group
+	if !fg.Match(event.Name) {
+		return
+	}
+
+	// Get callbacks under read lock
+	fg.mutex.RLock()
+	callbacks := make([]FileChangeCallback, len(fg.Callbacks))
+	copy(callbacks, fg.Callbacks)
+	fg.mutex.RUnlock()
+
+	// Asynchronously call callbacks
+	for _, callback := range callbacks {
+		go func(cb FileChangeCallback) {
+			if err := cb(event); err != nil {
+				log.Error().
+					Str("file", event.Name).
+					Str("op", event.Op.String()).
+					Err(err).
+					Msg("Callback error")
+			}
+		}(callback)
+	}
+}

pkg/filemonitor/filemonitor.go

@@ -0,0 +1,430 @@
+package filemonitor
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+
+	"github.com/fsnotify/fsnotify"
+	"github.com/rs/zerolog/log"
+)
+
+// FileMonitor manages multiple file groups
+type FileMonitor struct {
+	groups     map[string]*FileGroup // Map of file groups
+	watcher    *fsnotify.Watcher     // File system watcher
+	watchDirs  map[string]bool       // Monitored directories
+	blacklist  []string              // Global blacklist patterns
+	mutex      sync.RWMutex          // Concurrency control for groups and watchDirs
+	stopCh     chan struct{}         // Stop signal
+	wg         sync.WaitGroup        // Wait group
+	isRunning  bool                  // Running state flag
+	stateMutex sync.RWMutex          // State mutex
+}
+
+func (fm *FileMonitor) Watcher() *fsnotify.Watcher {
+	return fm.watcher
+}
+
+// NewFileMonitor creates a new file monitor
+func NewFileMonitor() *FileMonitor {
+	return &FileMonitor{
+		groups:    make(map[string]*FileGroup),
+		watchDirs: make(map[string]bool),
+		blacklist: []string{},
+		isRunning: false,
+	}
+}
+
+// SetBlacklist sets the global directory blacklist
+func (fm *FileMonitor) SetBlacklist(blacklist []string) {
+	fm.mutex.Lock()
+	defer fm.mutex.Unlock()
+
+	fm.blacklist = make([]string, len(blacklist))
+	copy(fm.blacklist, blacklist)
+}
+
+// AddGroup adds a new file group
+func (fm *FileMonitor) AddGroup(group *FileGroup) error {
+	if group == nil {
+		return errors.New("group cannot be nil")
+	}
+
+	// First check if monitor is running
+	isRunning := fm.IsRunning()
+
+	// Add group to monitor
+	fm.mutex.Lock()
+	// Check if ID already exists
+	if _, exists := fm.groups[group.ID]; exists {
+		fm.mutex.Unlock()
+		return fmt.Errorf("group with ID '%s' already exists", group.ID)
+	}
+	// Add to monitor
+	fm.groups[group.ID] = group
+	fm.mutex.Unlock()
+
+	// If monitor is running, set up watching
+	if isRunning {
+		if err := fm.setupWatchForGroup(group); err != nil {
+			// Remove group on failure
+			fm.mutex.Lock()
+			delete(fm.groups, group.ID)
+			fm.mutex.Unlock()
+			return err
+		}
+	}
+
+	return nil
+}
+
+// CreateGroup creates and adds a new file group (convenience method)
+func (fm *FileMonitor) CreateGroup(id, rootDir, pattern string, blacklist []string) (*FileGroup, error) {
+	// Create file group
+	group, err := NewFileGroup(id, rootDir, pattern, blacklist)
+	if err != nil {
+		return nil, err
+	}
+
+	// Add to monitor
+	if err := fm.AddGroup(group); err != nil {
+		return nil, err
+	}
+
+	return group, nil
+}
+
+// RemoveGroup removes a file group
+func (fm *FileMonitor) RemoveGroup(id string) error {
+	fm.mutex.Lock()
+	defer fm.mutex.Unlock()
+
+	// Check if group exists
+	_, exists := fm.groups[id]
+	if !exists {
+		return fmt.Errorf("group with ID '%s' does not exist", id)
+	}
+
+	// Remove group
+	delete(fm.groups, id)
+	// log.Info().Str("groupID", id).Msg("Removed file group")
+
+	return nil
+}
+
+// GetGroups returns a list of all file group IDs
+func (fm *FileMonitor) GetGroups() []*FileGroup {
+	fm.mutex.RLock()
+	defer fm.mutex.RUnlock()
+
+	groups := make([]*FileGroup, 0, len(fm.groups))
+	for _, group := range fm.groups {
+		groups = append(groups, group)
+	}
+
+	return groups
+}
+
+// GetGroup returns the specified file group
+func (fm *FileMonitor) GetGroup(id string) (*FileGroup, bool) {
+	fm.mutex.RLock()
+	defer fm.mutex.RUnlock()
+
+	group, exists := fm.groups[id]
+	return group, exists
+}
+
+// Start starts the file monitor
+func (fm *FileMonitor) Start() error {
+	// Check if already running
+	fm.stateMutex.Lock()
+	if fm.isRunning {
+		fm.stateMutex.Unlock()
+		return errors.New("file monitor is already running")
+	}
+
+	// Create new watcher
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		fm.stateMutex.Unlock()
+		return fmt.Errorf("failed to create watcher: %w", err)
+	}
+	fm.watcher = watcher
+
+	// Reset stop channel
+	fm.stopCh = make(chan struct{})
+
+	// Get groups to monitor (without holding the state lock)
+	fm.mutex.RLock()
+	groups := make([]*FileGroup, 0, len(fm.groups))
+	for _, group := range fm.groups {
+		groups = append(groups, group)
+	}
+	fm.mutex.RUnlock()
+
+	// Reset monitored directories
+	fm.mutex.Lock()
+	fm.watchDirs = make(map[string]bool)
+	fm.mutex.Unlock()
+
+	// Mark as running before setting up watches
+	fm.isRunning = true
+	fm.stateMutex.Unlock()
+
+	// Set up monitoring for all groups (without holding any locks)
+	for _, group := range groups {
+		if err := fm.setupWatchForGroup(group); err != nil {
+			// Clean up resources on failure
+			_ = fm.watcher.Close()
+
+			// Reset running state
+			fm.stateMutex.Lock()
+			fm.watcher = nil
+			fm.isRunning = false
+			fm.stateMutex.Unlock()
+
+			return fmt.Errorf("failed to setup watch for group '%s': %w", group.ID, err)
+		}
+	}
+
+	// Start watch loop
+	fm.wg.Add(1)
+	go fm.watchLoop()
+
+	// log.Info().Msg("File monitor started")
+	return nil
+}
+
+// Stop stops the file monitor
+func (fm *FileMonitor) Stop() error {
+	// Check if already stopped
+	fm.stateMutex.Lock()
+	if !fm.isRunning {
+		fm.stateMutex.Unlock()
+		return errors.New("file monitor is not running")
+	}
+
+	// Get watcher reference before changing state
+	watcher := fm.watcher
+
+	// Send stop signal
+	close(fm.stopCh)
+
+	// Mark as not running
+	fm.isRunning = false
+	fm.stateMutex.Unlock()
+
+	// Wait for all goroutines to exit
+	fm.wg.Wait()
+
+	// Close watcher
+	if watcher != nil {
+		if err := watcher.Close(); err != nil {
+			return fmt.Errorf("failed to close watcher: %w", err)
+		}
+
+		fm.stateMutex.Lock()
+		fm.watcher = nil
+		fm.stateMutex.Unlock()
+	}
+
+	// log.Info().Msg("File monitor stopped")
+	return nil
+}
+
+// IsRunning returns whether the file monitor is running
+func (fm *FileMonitor) IsRunning() bool {
+	fm.stateMutex.RLock()
+	defer fm.stateMutex.RUnlock()
+	return fm.isRunning
+}
+
+// addWatchDir adds a directory to monitoring
+func (fm *FileMonitor) addWatchDir(dirPath string) error {
+	// Check global blacklist first
+	fm.mutex.RLock()
+	for _, pattern := range fm.blacklist {
+		if strings.Contains(dirPath, pattern) {
+			fm.mutex.RUnlock()
+			log.Debug().Str("dir", dirPath).Msg("Skipping blacklisted directory")
+			return nil
+		}
+	}
+	fm.mutex.RUnlock()
+
+	fm.mutex.Lock()
+	defer fm.mutex.Unlock()
+
+	// Check if directory is already being monitored
+	if _, watched := fm.watchDirs[dirPath]; watched {
+		return nil // Already monitored, no need to add again
+	}
+
+	// Add to monitoring
+	if err := fm.watcher.Add(dirPath); err != nil {
+		return fmt.Errorf("failed to watch directory '%s': %w", dirPath, err)
+	}
+
+	fm.watchDirs[dirPath] = true
+	// log.Debug().Str("dir", dirPath).Msg("Added watch for directory")
+	return nil
+}
+
+// setupWatchForGroup sets up monitoring for a file group
+func (fm *FileMonitor) setupWatchForGroup(group *FileGroup) error {
+	// Check if file monitor is running
+	if !fm.IsRunning() {
+		return errors.New("file monitor is not running")
+	}
+
+	// Find directories containing matching files
+	matchingDirs, err := group.ListMatchingDirectories()
+	if err != nil {
+		return fmt.Errorf("failed to list matching directories: %w", err)
+	}
+
+	// Always watch the root directory to catch new files
+	rootDir := filepath.Clean(group.RootDir)
+	if err := fm.addWatchDir(rootDir); err != nil {
+		return err
+	}
+
+	// Watch directories containing matching files
+	for dir := range matchingDirs {
+		if err := fm.addWatchDir(dir); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+// RefreshWatches updates the watched directories based on current matching files
+func (fm *FileMonitor) RefreshWatches() error {
+	// Check if file monitor is running
+	if !fm.IsRunning() {
+		return errors.New("file monitor is not running")
+	}
+
+	// Get groups to refresh
+	fm.mutex.RLock()
+	groups := make([]*FileGroup, 0, len(fm.groups))
+	for _, group := range fm.groups {
+		groups = append(groups, group)
+	}
+	fm.mutex.RUnlock()
+
+	// Reset watched directories
+	fm.mutex.Lock()
+	oldWatchDirs := fm.watchDirs
+	fm.watchDirs = make(map[string]bool)
+	fm.mutex.Unlock()
+
+	// Setup watches for each group
+	for _, group := range groups {
+		if err := fm.setupWatchForGroup(group); err != nil {
+			return fmt.Errorf("failed to refresh watches for group '%s': %w", group.ID, err)
+		}
+	}
+
+	// Remove watches for directories no longer needed
+	for dir := range oldWatchDirs {
+		fm.mutex.RLock()
+		_, stillWatched := fm.watchDirs[dir]
+		fm.mutex.RUnlock()
+
+		if !stillWatched && fm.watcher != nil {
+			_ = fm.watcher.Remove(dir)
+			log.Debug().Str("dir", dir).Msg("Removed watch for directory")
+		}
+	}
+
+	return nil
+}
+
+// watchLoop monitors for file system events
+func (fm *FileMonitor) watchLoop() {
+	defer fm.wg.Done()
+
+	for {
+		select {
+		case <-fm.stopCh:
+			return
+
+		case event, ok := <-fm.watcher.Events:
+			if !ok {
+				// Channel closed, exit loop
+				return
+			}
+
+			// Handle directory creation events to add new watches
+			info, err := os.Stat(event.Name)
+			if err == nil && info.IsDir() && event.Op&(fsnotify.Create|fsnotify.Rename) != 0 {
+				// Add new directory to monitoring
+				if err := fm.addWatchDir(event.Name); err != nil {
+					log.Error().
+						Str("dir", event.Name).
+						Err(err).
+						Msg("Error watching new directory")
+				}
+				continue
+			}
+
+			// For file creation/modification, check if we need to watch its directory
+			if event.Op&(fsnotify.Create|fsnotify.Write) != 0 {
+				// Check if this file matches any group
+				shouldWatch := false
+
+				fm.mutex.RLock()
+				for _, group := range fm.groups {
+					if group.Match(event.Name) {
+						shouldWatch = true
+						break
+					}
+				}
+				fm.mutex.RUnlock()
+
+				// If file matches, ensure its directory is watched
+				if shouldWatch {
+					dir := filepath.Dir(event.Name)
+					if err := fm.addWatchDir(dir); err != nil {
+						log.Error().
+							Str("dir", dir).
+							Err(err).
+							Msg("Error watching directory of matching file")
+					}
+				}
+			}
+
+			// Forward event to all groups
+			fm.forwardEventToGroups(event)
+
+		case err, ok := <-fm.watcher.Errors:
+			if !ok {
+				// Channel closed, exit loop
+				return
+			}
+			log.Error().Err(err).Msg("Watcher error")
+		}
+	}
+}
+
+// forwardEventToGroups forwards file events to matching groups
+func (fm *FileMonitor) forwardEventToGroups(event fsnotify.Event) {
+	// Get a copy of groups to avoid holding lock during processing
+	fm.mutex.RLock()
+	groupsCopy := make([]*FileGroup, 0, len(fm.groups))
+	for _, group := range fm.groups {
+		groupsCopy = append(groupsCopy, group)
+	}
+	fm.mutex.RUnlock()
+
+	// Forward to all groups - each group will check if the event is relevant
+	for _, group := range groupsCopy {
+		group.HandleEvent(event)
+	}
+}

pkg/util/silk/silk.go

@@ -0,0 +1,36 @@
+package silk
+
+import (
+	"fmt"
+
+	"github.com/sjzar/go-lame"
+	"github.com/sjzar/go-silk"
+)
+
+func Silk2MP3(data []byte) ([]byte, error) {
+
+	sd := silk.SilkInit()
+	defer sd.Close()
+
+	pcmdata := sd.Decode(data)
+	if len(pcmdata) == 0 {
+		return nil, fmt.Errorf("silk decode failed")
+	}
+
+	le := lame.Init()
+	defer le.Close()
+
+	le.SetInSamplerate(24000)
+	le.SetOutSamplerate(24000)
+	le.SetNumChannels(1)
+	le.SetBitrate(16)
+	// IMPORTANT!
+	le.InitParams()
+
+	mp3data := le.Encode(pcmdata)
+	if len(mp3data) == 0 {
+		return nil, fmt.Errorf("mp3 encode failed")
+	}
+
+	return mp3data, nil
+}