🎨 Custom Emoji folders no longer require manual refresh when they cha…

…nge siyuan-note#11749

kernel/main.go

@@ -53,5 +53,6 @@ func main() {
 	go util.CheckFileSysStatus()
 
 	model.WatchAssets()
+	model.WatchEmojis()
 	model.HandleSignal()
 }

kernel/model/emojis_watcher.go

@@ -0,0 +1,111 @@
+// SiYuan - Refactor your thinking
+// Copyright (c) 2020-present, b3log.org
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+//go:build !darwin
+
+package model
+
+import (
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/88250/gulu"
+	"github.com/fsnotify/fsnotify"
+	"github.com/siyuan-note/logging"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+var emojisWatcher *fsnotify.Watcher
+
+func WatchEmojis() {
+	if util.ContainerAndroid == util.Container || util.ContainerIOS == util.Container {
+		return
+	}
+
+	go func() {
+		watchEmojis()
+	}()
+}
+
+func watchEmojis() {
+	emojisDir := filepath.Join(util.DataDir, "emojis")
+	if nil != emojisWatcher {
+		emojisWatcher.Close()
+	}
+
+	var err error
+	if emojisWatcher, err = fsnotify.NewWatcher(); nil != err {
+		logging.LogErrorf("add emojis watcher for folder [%s] failed: %s", emojisDir, err)
+		return
+	}
+
+	go func() {
+		defer logging.Recover()
+
+		var (
+			timer     *time.Timer
+			lastEvent fsnotify.Event
+		)
+		timer = time.NewTimer(100 * time.Millisecond)
+		<-timer.C // timer should be expired at first
+
+		for {
+			select {
+			case event, ok := <-emojisWatcher.Events:
+				if !ok {
+					return
+				}
+
+				lastEvent = event
+				timer.Reset(time.Millisecond * 100)
+
+				if lastEvent.Op&fsnotify.Rename == fsnotify.Rename || lastEvent.Op&fsnotify.Write == fsnotify.Write {
+					util.PushReloadEmojiConf()
+				} else if lastEvent.Op&fsnotify.Remove == fsnotify.Remove {
+					util.PushReloadEmojiConf()
+				}
+			case err, ok := <-emojisWatcher.Errors:
+				if !ok {
+					return
+				}
+				logging.LogErrorf("watch emojis failed: %s", err)
+			case <-timer.C:
+				//logging.LogInfof("emojis changed: %s", lastEvent)
+				if lastEvent.Op&fsnotify.Remove == fsnotify.Remove {
+					util.PushReloadEmojiConf()
+				} else {
+					util.PushReloadEmojiConf()
+				}
+			}
+		}
+	}()
+
+	if !gulu.File.IsDir(emojisDir) {
+		os.MkdirAll(emojisDir, 0755)
+	}
+
+	if err = emojisWatcher.Add(emojisDir); err != nil {
+		logging.LogErrorf("add emojis watcher for folder [%s] failed: %s", emojisDir, err)
+	}
+	//logging.LogInfof("added file watcher [%s]", emojisDir)
+}
+
+func CloseWatchEmojis() {
+	if nil != emojisWatcher {
+		emojisWatcher.Close()
+	}
+}

kernel/model/emojis_watcher_darwin.go

@@ -0,0 +1,91 @@
+// SiYuan - Refactor your thinking
+// Copyright (c) 2020-present, b3log.org
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+//go:build darwin
+
+package model
+
+import (
+	"path/filepath"
+	"time"
+
+	"github.com/radovskyb/watcher"
+	"github.com/siyuan-note/logging"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+var emojisWatcher *watcher.Watcher
+
+func WatchEmojis() {
+	go func() {
+		watchEmojis()
+	}()
+}
+
+func watchEmojis() {
+	if nil != emojisWatcher {
+		emojisWatcher.Close()
+	}
+	emojisWatcher = watcher.New()
+
+	emojisDir := filepath.Join(util.DataDir, "emojis")
+
+	go func() {
+		for {
+			select {
+			case event, ok := <-emojisWatcher.Event:
+				if !ok {
+					return
+				}
+
+				//logging.LogInfof("emojis changed: %s", event)
+				if watcher.Write == event.Op {
+					IncSync()
+				}
+
+				if watcher.Remove == event.Op {
+					util.PushReloadEmojiConf()
+				} else {
+					util.PushReloadEmojiConf()
+				}
+			case err, ok := <-emojisWatcher.Error:
+				if !ok {
+					return
+				}
+				logging.LogErrorf("watch emojis failed: %s", err)
+			case <-emojisWatcher.Closed:
+				return
+			}
+		}
+	}()
+
+	if err := emojisWatcher.Add(emojisDir); nil != err {
+		logging.LogErrorf("add emojis watcher for folder [%s] failed: %s", emojisDir, err)
+		return
+	}
+
+	//logging.LogInfof("added file watcher [%s]", emojisDir)
+	if err := emojisWatcher.Start(10 * time.Second); nil != err {
+		logging.LogErrorf("start emojis watcher for folder [%s] failed: %s", emojisDir, err)
+		return
+	}
+}
+
+func CloseWatchEmojis() {
+	if nil != emojisWatcher {
+		emojisWatcher.Close()
+	}
+}

kernel/model/repository.go

@@ -634,6 +634,8 @@ func checkoutRepo(id string) {
 	WaitForWritingFiles()
 	CloseWatchAssets()
 	defer WatchAssets()
+	CloseWatchEmojis()
+	defer WatchEmojis()
 
 	// 恢复快照时自动暂停同步，避免刚刚恢复后的数据又被同步覆盖
 	syncEnabled := Conf.Sync.Enabled

kernel/util/websocket.go

@@ -266,6 +266,10 @@ func PushProtyleLoading(rootID, msg string) {
 	BroadcastByType("protyle", "addLoading", 0, msg, rootID)
 }
 
+func PushReloadEmojiConf() {
+	BroadcastByType("main", "reloadEmojiConf", 0, "", nil)
+}
+
 func PushDownloadProgress(id string, percent float32) {
 	evt := NewCmdResult("downloadProgress", 0, PushModeBroadcast)
 	evt.Data = map[string]interface{}{