From 5466287783c1740c5e9f7440f5c27cd419efa752 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Sun, 9 Jul 2023 17:38:38 +0800
Subject: [PATCH] fix indexer sites

---
 app/api/endpoints/system.py | 12 +++++++-----
 app/chain/search.py         |  6 ++++--
 app/chain/subscribe.py      |  9 ++++++---
 app/core/config.py          |  2 --
 app/schemas/types.py        |  2 ++
 5 files changed, 19 insertions(+), 12 deletions(-)

diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py
index e48dae210..18c84ca38 100644
--- a/app/api/endpoints/system.py
+++ b/app/api/endpoints/system.py
@@ -1,7 +1,7 @@
 import json
 import json
 import time
-from typing import Any
+from typing import Any, List
 
 from fastapi import APIRouter, HTTPException, Depends
 from fastapi.responses import StreamingResponse
@@ -37,16 +37,18 @@ def event_generator():
     return StreamingResponse(event_generator(), media_type="text/event-stream")
 
 
-@router.get("/setting", summary="查询系统设置")
+@router.get("/setting/{key}", summary="查询系统设置")
 def get_setting(key: str, _: schemas.TokenPayload = Depends(verify_token)):
     """
     查询系统设置
     """
-    return schemas.Response(success=True, data=SystemConfigOper().get(key))
+    return schemas.Response(success=True, data={
+        "value": SystemConfigOper().get(key)
+    })
 
 
-@router.put("/setting", summary="更新系统设置")
-def set_setting(key: str, value: Any, _: schemas.TokenPayload = Depends(verify_token)):
+@router.post("/setting/{key}", summary="更新系统设置")
+def set_setting(key: str, value: List[int], _: schemas.TokenPayload = Depends(verify_token)):
     """
     更新系统设置
     """
diff --git a/app/chain/search.py b/app/chain/search.py
index dc831b667..d449686cf 100644
--- a/app/chain/search.py
+++ b/app/chain/search.py
@@ -202,9 +202,11 @@ def __search_all_sites(self, mediainfo: Optional[MediaInfo] = None,
         """
         # 未开启的站点不搜索
         indexer_sites = []
+        # 配置的索引站点
+        config_indexers = self.systemconfig.get(SystemConfigKey.IndexerSites) or []
         for indexer in self.siteshelper.get_indexers():
-            if not settings.INDEXER_SITES \
-                    or any([s in indexer.get("domain") for s in settings.INDEXER_SITES.split(',')]):
+            # 检查站点索引开关
+            if not config_indexers or indexer.get("id") in config_indexers:
                 # 站点流控
                 state, msg = self.siteshelper.check(indexer.get("domain"))
                 if not state:
diff --git a/app/chain/subscribe.py b/app/chain/subscribe.py
index 0012015cd..272aebbc3 100644
--- a/app/chain/subscribe.py
+++ b/app/chain/subscribe.py
@@ -11,12 +11,13 @@
 from app.core.config import settings
 from app.db.models.subscribe import Subscribe
 from app.db.subscribe_oper import SubscribeOper
+from app.db.systemconfig_oper import SystemConfigOper
 from app.helper.message import MessageHelper
 from app.helper.sites import SitesHelper
 from app.log import logger
 from app.schemas import NotExistMediaInfo
 from app.utils.string import StringUtils
-from app.schemas.types import MediaType
+from app.schemas.types import MediaType, SystemConfigKey
 
 
 class SubscribeChain(ChainBase):
@@ -34,6 +35,7 @@ def __init__(self):
         self.subscribehelper = SubscribeOper()
         self.siteshelper = SitesHelper()
         self.message = MessageHelper()
+        self.systemconfig = SystemConfigOper()
 
     def add(self, title: str, year: str,
             mtype: MediaType = None,
@@ -269,11 +271,12 @@ def refresh(self):
         """
         # 所有站点索引
         indexers = self.siteshelper.get_indexers()
+        # 配置的索引站点
+        config_indexers = self.systemconfig.get(SystemConfigKey.IndexerSites) or []
         # 遍历站点缓存资源
         for indexer in indexers:
             # 未开启的站点不搜索
-            if settings.INDEXER_SITES \
-                    and not any([s in indexer.get("domain") for s in settings.INDEXER_SITES.split(',')]):
+            if config_indexers and indexer.get("id") not in config_indexers:
                 continue
             logger.info(f'开始刷新站点资源，站点：{indexer.get("name")} ...')
             domain = StringUtils.get_url_domain(indexer.get("domain"))
diff --git a/app/core/config.py b/app/core/config.py
index a5958a013..418466c53 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -59,8 +59,6 @@ class Settings(BaseSettings):
     RMT_AUDIO_TRACK_EXT: list = ['.mka']
     # 索引器
     INDEXER: str = "builtin"
-    # 索引站点，站点域名关键字使用,分隔
-    INDEXER_SITES: str = ""
     # 用户认证站点 hhclub/audiences/hddolby/zmpt/freefarm/hdfans/wintersakura/leaves/1ptba/icc2022/iyuu
     AUTH_SITE: str = ""
     # 消息通知渠道 telegram/wechat/slack
diff --git a/app/schemas/types.py b/app/schemas/types.py
index 6e87b0d35..99fec6c39 100644
--- a/app/schemas/types.py
+++ b/app/schemas/types.py
@@ -36,6 +36,8 @@ class SystemConfigKey(Enum):
     UserInstalledPlugins = "UserInstalledPlugins"
     # 搜索结果
     SearchResults = "SearchResults"
+    # 索引站点范围
+    IndexerSites = "IndexerSites"
 
 
 # 站点框架