1 files changed, 90 insertions, 5 deletions
diff --git a/ATRI/service/__init__.py b/ATRI/service/__init__.py
index 5dddb97..6b18ba1 100644
--- a/ATRI/service/__init__.py
+++ b/ATRI/service/__init__.py
@@ -1,10 +1,95 @@
 import os
+import json
 from pathlib import Path
+from pydantic import BaseModel
+from typing import Optional
 
+from ATRI.log import logger
+from ATRI.exceptions import WriteError
 
-SERVICE_PATH = Path('.') / 'ATRI' / 'data' / 'service'
-ERROR_PATH = Path('.') / 'ATRI' / 'data' / 'error'
-os.makedirs(SERVICE_PATH, exist_ok=True)
-os.makedirs(ERROR_PATH, exist_ok=True)
 
-state = 0
+SERVICE_DIR = Path('.') / 'ATRI' / 'data' / 'service'
+PLUGIN_INFO_DIR = Path('.') / 'ATRI' / 'data' / 'service' / 'plugins'
+os.makedirs(SERVICE_DIR, exist_ok=True)
+os.makedirs(PLUGIN_INFO_DIR, exist_ok=True)
+
+sleep = False
+
+class Service:
+    class ServiceInfo(BaseModel):
+        name: str
+        _type: str
+        docs: Optional[str] = None
+        command: Optional[list] = None
+    
+    @classmethod
+    def register(
+        cls,
+        service_name: str,
+        service_type: str,
+        docs: Optional[str] = None,
+        command: Optional[list] = None
+    ) -> None:
+        """
+        启动时保存各个服务信息，便于后续网页控制台调用
+        增强管理
+        """
+        file_name = f"{service_name}.function.json"
+        path = SERVICE_DIR / file_name
+        path.parent.mkdir(exist_ok=True, parents=True)
+        try:
+            data = json.loads(path.read_bytes())
+        except:
+            data = {}
+        
+        data = cls.ServiceInfo(
+            name=service_name,
+            _type=service_type,
+            docs=docs,
+            command=command
+        )
+        try:
+            with open(path, 'w', encoding='utf-8') as target:
+                target.write(
+                    json.dumps(
+                        data.dict(), indent=4
+                    )
+                )
+        except WriteError:
+            raise WriteError("Writing file failed!")
+        else:
+            pass
+        
+        # TODO: shitcode-style
+        docs_judge = "N" if not docs else "Y"
+        a = " "
+        log_center = ""
+        log_head = f"Success register service: [{service_name}]."
+        log_suffix = f"Docs [{docs_judge}]. Type [{service_type}]"
+        log_head_lenght = len(log_head)
+        log_suffix_lenght = len(log_suffix)
+        log_center_lenght = 120 - (
+            log_head_lenght + log_suffix_lenght
+        )
+        for _ in range(log_center_lenght): log_center = log_center + a
+        log_print = log_head + log_center + log_suffix
+        logger.info(log_print)
+    
+    @staticmethod
+    def is_dormant() -> bool:
+        """
+        为bot提供休眠，期间不会响应除了 superusers 以外的用户的信息
+        触发在于 Rule
+        """
+        return False if sleep else True
+    
+    @staticmethod
+    def control_dormant(_type: bool) -> None:
+        """
+        更改休眠状态
+        """
+        global sleep
+        if _type:
+            sleep = True
+        else:
+            sleep = False