diff --git a/package.json b/package.json index bb22788..45039d3 100644 --- a/package.json +++ b/package.json @@ -342,5 +342,13 @@ "icon": "nfo.png", "author": "叮叮当", "level": 1 + }, + "RemoveLink": { + "name": "清理硬链接", + "description": "监控目录内文件被删除时，同步删除所有和它硬链接的文件", + "version": "1.0", + "icon": "delete.png", + "author": "DzAvril", + "level": 1 } } diff --git a/plugins/removelink/__init__.py b/plugins/removelink/__init__.py new file mode 100644 index 0000000..1146efd --- /dev/null +++ b/plugins/removelink/__init__.py @@ -0,0 +1,280 @@ +import traceback +from pathlib import Path +from typing import List, Tuple, Dict, Any +import os +import re + +from watchdog.events import FileSystemEventHandler +from watchdog.observers import Observer + +from app.log import logger +from app.plugins import _PluginBase +from app.schemas import Notification, NotificationType +import time + +class FileMonitorHandler(FileSystemEventHandler): + """ + 目录监控处理 + """ + def __init__(self, monpath: str, sync: Any, **kwargs): + super(FileMonitorHandler, self).__init__(**kwargs) + self._watch_path = monpath + self.sync = sync + + def on_created(self, event): + logger.info("新增文件：%s" % event.event_data) + self.sync._state_set.add((event.src_path, event.src_path.stat().st_ino)) + + def on_deleted(self, event): + logger.info("监测到删除：%s" % event.src_path) + self.sync.event_handler(event=event, event_path=event.src_path) + +def updateState(monitor_dirs: List[str]): + """ + 更新监控目录的文件列表 + """ + start_time = time.time() # 记录开始时间 + state_set = set() + for mon_path in monitor_dirs: + for root, dirs, files in os.walk(mon_path): + for file in files: + file = Path(root) / file + if not file.exists(): + continue + state_set.add((file, file.stat().st_ino)) + end_time = time.time() # 记录结束时间 + elapsed_time = end_time - start_time # 计算耗时 + logger.info(f"更新文件列表完成，共计{len(state_set)}个文件，耗时：{elapsed_time}秒") + + return state_set + +class RemoveLink(_PluginBase): + # 插件名称 + plugin_name = "清理硬链接" + # 插件描述 + plugin_desc = "监控目录内文件被删除时，同步删除所有和它硬链接的文件" + # 插件图标 + plugin_icon = "delete.png" + # 插件版本 + plugin_version = "1.0" + # 插件作者 + plugin_author = "DzAvril" + # 作者主页 + author_url = "https://github.com/DzAvril" + # 插件配置项ID前缀 + plugin_config_prefix = "linkdeleted_" + # 加载顺序 + plugin_order = 27 + # 可使用的用户级别 + auth_level = 1 + + # preivate property + _monitor_dirs = "" + _exclude_keywords = "" + _enabled = False + _observer = [] + _state_set = set() + + def init_plugin(self, config: dict = None): + logger.info(f"Hello, RemoveLink! config {config}") + if config: + self._enabled = config.get("enabled") + self._notify = config.get("notify") + self._monitor_dirs = config.get("monitor_dirs") + self._exclude_keywords = config.get("exclude_keywords") or "" + self.__update_config() + # 停止现有任务 + self.stop_service() + if self._enabled: + # 读取目录配置 + monitor_dirs = self._monitor_dirs.split("\n") + logger.info(f"监控目录：{monitor_dirs}") + if not monitor_dirs: + return + for mon_path in monitor_dirs: + # 格式源目录:目的目录 + if not mon_path: + continue + try: + observer = Observer(timeout=10) + self._observer.append(observer) + observer.schedule(FileMonitorHandler(mon_path, self), mon_path, recursive=True) + observer.daemon = True + observer.start() + logger.info(f"{mon_path} 的目录监控服务启动") + except Exception as e: + err_msg = str(e) + logger.error(f"{mon_path} 启动目录监控失败：{err_msg}") + self.systemmessage.put(f"{mon_path} 启动目录监控失败：{err_msg}") + self._state_set = updateState(monitor_dirs) + + def __update_config(self): + """ + 更新配置 + """ + self.update_config({ + "enabled": self._enabled, + "notify": self._notify, + "monitor_dirs": self._monitor_dirs, + "exclude_keywords": self._exclude_keywords, + }) + + def get_state(self) -> bool: + return self._enabled + + @staticmethod + def get_command() -> List[Dict[str, Any]]: + pass + + def get_api(self) -> List[Dict[str, Any]]: + pass + + def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: + return [ + { + 'component': 'VForm', + 'content': [ + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'enabled', + 'label': '启用插件', + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 4 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'notify', + 'label': '发送通知', + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12 + }, + 'content': [ + { + 'component': 'VTextarea', + 'props': { + 'model': 'monitor_dirs', + 'label': '监控目录', + 'rows': 5, + 'placeholder': '每一行一个目录' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VTextarea', + 'props': { + 'model': 'exclude_keywords', + 'label': '排除关键词', + 'rows': 2, + 'placeholder': '每一行一个关键词' + } + } + ] + } + ] + } + ] + } + ], { + "enabled": False, + "notify": False, + "onlyonce": False, + "monitor_dirs": "", + "exclude_keywords": "", + } + + def get_page(self) -> List[dict]: + pass + + def stop_service(self): + """ + 退出插件 + """ + if self._observer: + for observer in self._observer: + try: + observer.stop() + observer.join() + except Exception as e: + print(str(e)) + logger.error(f"停止目录监控失败：{str(e)}") + self._observer = [] + + def event_handler(self, event, event_path: str): + """ + 处理删除事件 + """ + deleted_files = [] + if not event.is_directory: + # 命中过滤关键字不处理 + if self._exclude_keywords: + for keyword in self._exclude_keywords.split("\n"): + if keyword and re.findall(keyword, event_path): + logger.info(f"{event_path} 命中过滤关键字 {keyword}，不处理") + return + + file_path = Path(event_path) + current_set = updateState(self._monitor_dirs.split("\n")) + deleted_set = self._state_set - current_set + deleted_inode = [x[1] for x in deleted_set] + try: + # 在current_set中查找与deleted_inode有相同inode的文件并删除 + for path, inode in current_set: + if inode in deleted_inode: + file = Path(path) + file.unlink() + deleted_files.append(path) + logger.info(f"删除硬链接文件：{path}") + except Exception as e: + logger.error("目录监控发生错误：%s - %s" % (str(e), traceback.format_exc())) + + if self._notify and (len(deleted_files) != 0): + self.chain.post_message(Notification( + mtype=NotificationType.SiteMessage, + title=f"[删除硬链接]", + text=f"根据源文件：{file_path}\n" + f"删除硬链接文件：{[str(x) for x in deleted_files]}", + )) + self._state_set = updateState(self._monitor_dirs.split("\n")) \ No newline at end of file