refactor

Author: llanc

autostart.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"golang.org/x/sys/windows/registry"
+	"os"
+)
+
+func isAutoStartEnabled() bool {
+	k, err := registry.OpenKey(registry.CURRENT_USER, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.QUERY_VALUE)
+	if err != nil {
+		return false
+	}
+	defer k.Close()
+
+	val, _, err := k.GetStringValue("HIDELL")
+	if err != nil {
+		return false
+	}
+
+	exePath, err := os.Executable()
+	if err != nil {
+		return false
+	}
+
+	return val == exePath
+}
+
+func enableAutoStart() {
+	exePath, err := os.Executable()
+	if err != nil {
+		return
+	}
+
+	k, err := registry.OpenKey(registry.CURRENT_USER, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.SET_VALUE)
+	if err != nil {
+		return
+	}
+	defer k.Close()
+
+	_ = k.SetStringValue("HIDELL", exePath)
+}
+
+func disableAutoStart() {
+	k, err := registry.OpenKey(registry.CURRENT_USER, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.SET_VALUE)
+	if err != nil {
+		return
+	}
+	defer k.Close()
+
+	_ = k.DeleteValue("HIDELL")
+}

config.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"encoding/json"
+	"os"
+	"path/filepath"
+)
+
+func LoadConfig() {
+	configPath := filepath.Join(getUserHome(), ".hidell", "config.json")
+	data, err := os.ReadFile(configPath)
+	if err != nil {
+		if os.IsNotExist(err) {
+			customDirs = []CustomDir{}
+			return
+		}
+		panic(err)
+	}
+
+	err = json.Unmarshal(data, &customDirs)
+	if err != nil {
+		panic(err)
+	}
+}
+
+func saveConfig() {
+	configPath := filepath.Join(getUserHome(), ".hidell", "config.json")
+	data, err := json.MarshalIndent(customDirs, "", "  ")
+	if err != nil {
+		panic(err)
+	}
+
+	err = os.MkdirAll(filepath.Dir(configPath), 0755)
+	if err != nil {
+		panic(err)
+	}
+
+	err = os.WriteFile(configPath, data, 0644)
+	if err != nil {
+		panic(err)
+	}
+}

custom_dir.go

@@ -0,0 +1,198 @@
+package main
+
+import (
+	"github.com/fsnotify/fsnotify"
+	"github.com/getlantern/systray"
+	"github.com/ncruces/zenity"
+	"path/filepath"
+	"strings"
+)
+
+type CustomDir struct {
+	Path    string `json:"path"`
+	Alias   string `json:"alias"`
+	Active  bool   `json:"active"`
+	Hidden  bool   `json:"hidden"`
+	watcher *fsnotify.Watcher
+	done    chan struct{}
+}
+
+var customDirs []CustomDir
+
+func addCustomDirMenu(dir *CustomDir, parentMenu *systray.MenuItem) {
+	menuName := dir.Alias
+	if menuName == "" {
+		menuName = dir.Path
+	}
+	dirMenu := parentMenu.AddSubMenuItem(menuName, "")
+	mActivate := dirMenu.AddSubMenuItem("激活", "")
+	mShow := dirMenu.AddSubMenuItem("显示", "")
+	mHide := dirMenu.AddSubMenuItem("隐藏", "")
+	mRemove := dirMenu.AddSubMenuItem("移除", "")
+
+	if dir.Active {
+		mActivate.Check()
+		dir.startWatching()
+	}
+	if dir.Hidden {
+		mHide.Check()
+	} else {
+		mShow.Check()
+	}
+
+	go func() {
+		for {
+			select {
+			case <-mActivate.ClickedCh:
+				toggleActivate(dir, mActivate)
+			case <-mShow.ClickedCh:
+				showCustomDir(dir, mShow, mHide)
+			case <-mHide.ClickedCh:
+				hideCustomDir(dir, mShow, mHide)
+			case <-mRemove.ClickedCh:
+				removeCustomDir(dir, dirMenu, parentMenu)
+				return
+			}
+		}
+	}()
+}
+
+func addNewCustomDir(parentMenu *systray.MenuItem) {
+	path, err := zenity.SelectFile(
+		zenity.Title("选择目录"),
+		zenity.Directory(),
+	)
+	if err != nil {
+		if err == zenity.ErrCanceled {
+			return
+		}
+		zenity.Error("选择目录时出错: " + err.Error())
+		return
+	}
+
+	alias, err := zenity.Entry(
+		"为这个目录输入一个别名（可选）",
+		zenity.Title("设置别名"),
+	)
+	if err != nil {
+		if err == zenity.ErrCanceled {
+			alias = ""
+		} else {
+			zenity.Error("输入别名时出错: " + err.Error())
+			return
+		}
+	}
+
+	newDir := CustomDir{
+		Path:    path,
+		Alias:   alias,
+		Active:  true,
+		Hidden:  true,
+		watcher: nil,
+		done:    make(chan struct{}),
+	}
+	customDirs = append(customDirs, newDir)
+	addCustomDirMenu(&customDirs[len(customDirs)-1], parentMenu)
+	hideDotFiles(path)
+	customDirs[len(customDirs)-1].startWatching()
+	saveConfig()
+}
+
+func toggleActivate(dir *CustomDir, menuItem *systray.MenuItem) {
+	dir.Active = !dir.Active
+	if dir.Active {
+		menuItem.Check()
+		dir.startWatching()
+	} else {
+		menuItem.Uncheck()
+		dir.stopWatching()
+	}
+	saveConfig()
+}
+
+func showCustomDir(dir *CustomDir, showItem, hideItem *systray.MenuItem) {
+	dir.Hidden = false
+	showItem.Check()
+	hideItem.Uncheck()
+	unhideDotFiles(dir.Path)
+	saveConfig()
+}
+
+func hideCustomDir(dir *CustomDir, showItem, hideItem *systray.MenuItem) {
+	dir.Hidden = true
+	hideItem.Check()
+	showItem.Uncheck()
+	hideDotFiles(dir.Path)
+	saveConfig()
+}
+
+func removeCustomDir(dir *CustomDir, dirMenu, parentMenu *systray.MenuItem) {
+	dir.stopWatching()
+	for i, d := range customDirs {
+		if d.Path == dir.Path {
+			customDirs = append(customDirs[:i], customDirs[i+1:]...)
+			break
+		}
+	}
+	dirMenu.Hide()
+	saveConfig()
+}
+
+func (dir *CustomDir) startWatching() {
+	if dir.watcher != nil {
+		return
+	}
+
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		zenity.Error("创建监视器时出错: " + err.Error())
+		return
+	}
+
+	dir.watcher = watcher
+	dir.done = make(chan struct{})
+
+	go func() {
+		defer watcher.Close()
+		dir.watchDir()
+	}()
+
+	err = watcher.Add(dir.Path)
+	if err != nil {
+		zenity.Error("添加监视目录时出错: " + err.Error())
+		dir.stopWatching()
+	}
+}
+
+func (dir *CustomDir) stopWatching() {
+	if dir.watcher == nil {
+		return
+	}
+
+	close(dir.done)
+	dir.watcher.Close()
+	dir.watcher = nil
+}
+
+func (dir *CustomDir) watchDir() {
+	for {
+		select {
+		case <-dir.done:
+			return
+		case event, ok := <-dir.watcher.Events:
+			if !ok {
+				return
+			}
+			if event.Op&fsnotify.Create == fsnotify.Create {
+				if strings.HasPrefix(filepath.Base(event.Name), ".") {
+					hideFile(event.Name)
+				}
+			}
+		case err, ok := <-dir.watcher.Errors:
+			if !ok {
+				return
+			}
+			zenity.Error("监视目录时出错: " + err.Error())
+		}
+	}
+}

file_operations.go

@@ -0,0 +1,91 @@
+package main
+
+import (
+	"github.com/fsnotify/fsnotify"
+	"os"
+	"path/filepath"
+	"strings"
+	"syscall"
+)
+
+func hideDotFiles(dir string) {
+	files, err := os.ReadDir(dir)
+	if err != nil {
+		panic(err)
+	}
+	for _, file := range files {
+		if strings.HasPrefix(file.Name(), ".") {
+			hideFile(filepath.Join(dir, file.Name()))
+		}
+	}
+}
+
+func unhideDotFiles(dir string) {
+	files, err := os.ReadDir(dir)
+	if err != nil {
+		panic(err)
+	}
+	for _, file := range files {
+		if strings.HasPrefix(file.Name(), ".") {
+			unhideFile(filepath.Join(dir, file.Name()))
+		}
+	}
+}
+
+func hideFile(path string) {
+	ptr, err := syscall.UTF16PtrFromString(path)
+	if err != nil {
+		panic(err)
+	}
+	attrs, err := syscall.GetFileAttributes(ptr)
+	if err != nil {
+		panic(err)
+	}
+	_ = syscall.SetFileAttributes(ptr, attrs|syscall.FILE_ATTRIBUTE_HIDDEN)
+}
+
+func unhideFile(path string) {
+	ptr, err := syscall.UTF16PtrFromString(path)
+	if err != nil {
+		panic(err)
+	}
+	attrs, err := syscall.GetFileAttributes(ptr)
+	if err != nil {
+		panic(err)
+	}
+	_ = syscall.SetFileAttributes(ptr, attrs&^syscall.FILE_ATTRIBUTE_HIDDEN)
+}
+
+func watchDir(dir string, done chan struct{}) {
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		panic(err)
+	}
+	defer watcher.Close()
+
+	err = watcher.Add(dir)
+	if err != nil {
+		panic(err)
+	}
+
+	for {
+		select {
+		case <-done:
+			return
+		case event, ok := <-watcher.Events:
+			if !ok {
+				return
+			}
+			if event.Op&fsnotify.Create == fsnotify.Create {
+				if strings.HasPrefix(filepath.Base(event.Name), ".") {
+					hideFile(event.Name)
+				}
+			}
+		case err, ok := <-watcher.Errors:
+			if !ok {
+				return
+			}
+			panic(err)
+		}
+	}
+}