From 92905d4f04445d0cf30ad85bd6a58b4a0681079b Mon Sep 17 00:00:00 2001
From: yzhh <31803608+OasisAkari@users.noreply.github.com>
Date: Thu, 11 Jan 2024 21:24:07 +0800
Subject: [PATCH] update

---
 core/builtins/message/internal.py | 15 +++++++++++----
 modules/secret/__init__.py        |  2 +-
 modules/wiki/audit.py             | 32 ++++++++++++++++++++++---------
 modules/wiki/locales/zh_cn.json   |  1 +
 modules/wiki/utils/__init__.py    | 32 +++++++++++++++++++++----------
 modules/wiki/utils/ab.py          |  5 +++--
 modules/wiki/utils/ab_qq.py       |  2 +-
 modules/wiki/utils/rc.py          |  5 +++--
 modules/wiki/utils/rc_qq.py       |  3 ++-
 modules/wiki/utils/wikilib.py     | 13 +++++++------
 schedulers/wiki_bot.py            | 15 +++++++++++++++
 11 files changed, 89 insertions(+), 36 deletions(-)
 create mode 100644 schedulers/wiki_bot.py

diff --git a/core/builtins/message/internal.py b/core/builtins/message/internal.py
index 4b868b60..b71662e3 100644
--- a/core/builtins/message/internal.py
+++ b/core/builtins/message/internal.py
@@ -84,10 +84,10 @@ class FormattedTime:
                 else:
                     ftime_template.append(msg.locale.t("time.time.nosec.format"))
             if self.timezone:
-                if self._tz_offset == "+0":
+                if msg._tz_offset == "+0":
                     ftime_template.append("(UTC)")
                 else:
-                    ftime_template.append(f"(UTC{self._tz_offset})")
+                    ftime_template.append(f"(UTC{msg._tz_offset})")
         else:
             ftime_template.append('%Y-%m-%d %H:%M:%S')
         if not msg:
@@ -102,8 +102,15 @@ class FormattedTime:
         return f'FormattedTime(time={self.timestamp})'
 
     def to_dict(self):
-        return {'type': 'formatted_time', 'data': {'timestamp': self.timestamp, 'date': self.date, 'iso': self.iso,
-                                                   'time': self.time, 'seconds': self.seconds, 'timezone': self.timezone}}
+        return {
+            'type': 'formatted_time',
+            'data': {
+                'timestamp': self.timestamp,
+                'date': self.date,
+                'iso': self.iso,
+                'time': self.time,
+                'seconds': self.seconds,
+                'timezone': self.timezone}}
 
 
 class I18NContext:
diff --git a/modules/secret/__init__.py b/modules/secret/__init__.py
index 1bf6456d..019a7b0c 100644
--- a/modules/secret/__init__.py
+++ b/modules/secret/__init__.py
@@ -14,7 +14,7 @@ bot = Bot.FetchTarget
 ca = module('__check_abuse__', required_superuser=True, developers=['OasisAkari'])
 
 
-@ca.handle(DateTrigger(datetime.now() + timedelta(seconds=10)))
+@ca.handle(DateTrigger(datetime.now() + timedelta(seconds=20)))
 async def _():
     if bot.name not in ['QQ', 'TEST']:
         return
diff --git a/modules/wiki/audit.py b/modules/wiki/audit.py
index 045dda2b..ae599be6 100644
--- a/modules/wiki/audit.py
+++ b/modules/wiki/audit.py
@@ -6,6 +6,7 @@ from core.component import module
 from core.scheduler import DateTrigger
 from core.logger import Logger
 from core.utils.image_table import image_table_render, ImageTable
+from database import BotDBUtil
 from modules.wiki.utils.bot import BotAccount, LoginFailed
 from modules.wiki.utils.dbutils import Audit
 from modules.wiki.utils.wikilib import WikiLib
@@ -131,13 +132,13 @@ async def _(msg: Bot.MessageSession):
     check = await WikiLib(api_link).check_wiki_available()
     if check.available:
         try:
-            await BotAccount._login(api_link, account, password)
+            login = await BotAccount._login(check.value.api, account, password)
+            BotAccountDB.add(check.value.api, account, password)
+            BotAccount.cookies[check.value.api] = login
+            await msg.finish('Login success')
         except LoginFailed as e:
             Logger.error(f'Login failed: {e}')
             await msg.finish(f'Login failed: {e}')
-        else:
-            await msg.finish('Login success')
-            BotAccountDB.add(api_link, account, password)
     else:
         result = msg.locale.t('wiki.message.error.query') + \
             ('\n' + msg.locale.t('wiki.message.error.info') + check.message if check.message != '' else '')
@@ -148,10 +149,23 @@ async def _(msg: Bot.MessageSession):
 async def _(msg: Bot.MessageSession):
     api_link = msg.parsed_msg['<apiLink>']
     BotAccountDB.remove(api_link)
-    await msg.finish('Done')
+    await msg.finish(msg.locale.t("success"))
 
 
-@aud.handle(DateTrigger(datetime.now() + timedelta(seconds=10)))
-async def login_bots():
-    Logger.info('Start login wiki bot account...')
-    await BotAccount.login()
+@aud.handle('bot use')
+async def _(msg: Bot.MessageSession):
+    target_data = BotDBUtil.TargetInfo(msg)
+    target_data.edit_option('use_bot_account', True)
+    await msg.finish(msg.locale.t("success"))
+
+
+@aud.handle('bot unuse')
+async def _(msg: Bot.MessageSession):
+    target_data = BotDBUtil.TargetInfo(msg)
+    target_data.edit_option('use_bot_account', False)
+    await msg.finish(msg.locale.t("success"))
+
+
+@aud.hook('login_wiki_bots')
+async def mcbv_jira_rss(fetch: Bot.FetchTarget, ctx: Bot.ModuleHookContext):
+    BotAccount.cookies.update(ctx.args['cookies'])
diff --git a/modules/wiki/locales/zh_cn.json b/modules/wiki/locales/zh_cn.json
index 5a32e214..71975ca2 100644
--- a/modules/wiki/locales/zh_cn.json
+++ b/modules/wiki/locales/zh_cn.json
@@ -39,6 +39,7 @@
     "wiki.message.error.info": "详细信息：",
     "wiki.message.error.query": "发生错误：无法查询此 Wiki。",
     "wiki.message.error.set": "发生错误：无法设置此 Wiki。",
+    "wiki.message.error.fetch_log": "无法获取日志，可能是本wiki设置了日志的查看权限。",
     "wiki.message.error.unable_to_render_section": "（章节渲染失败，请联系开发者检查原因）",
     "wiki.message.fandom.disable": "已关闭 Fandom 全局 Interwiki 查询。",
     "wiki.message.fandom.enable": "已开启 Fandom 全局 Interwiki 查询。",
diff --git a/modules/wiki/utils/__init__.py b/modules/wiki/utils/__init__.py
index 1dc4af8c..cfa02ac3 100644
--- a/modules/wiki/utils/__init__.py
+++ b/modules/wiki/utils/__init__.py
@@ -14,8 +14,8 @@ rc_ = module('rc', developers=['OasisAkari'], recommend_modules='wiki')
 
 
 @rc_.command(['{{wiki.help.rc}}',
-             'legacy {{wiki.help.rc.legacy}}'],
-           available_for=['QQ', 'QQ|Group'])
+              'legacy {{wiki.help.rc.legacy}}'],
+             available_for=['QQ', 'QQ|Group'])
 async def rc_loader(msg: Bot.MessageSession):
     start_wiki = WikiTargetInfo(msg).get_start_wiki()
     if not start_wiki:
@@ -35,7 +35,7 @@ async def rc_loader(msg: Bot.MessageSession):
 
 
 @rc_.command('{{wiki.help.rc}}',
-           exclude_from=['QQ', 'QQ|Group'])
+             exclude_from=['QQ', 'QQ|Group'])
 async def rc_loader(msg: Bot.MessageSession):
     start_wiki = WikiTargetInfo(msg).get_start_wiki()
     if not start_wiki:
@@ -48,7 +48,7 @@ a = module('ab', developers=['OasisAkari'], recommend_modules='wiki')
 
 
 @a.command(['{{wiki.help.ab}}',
-           'legacy {{wiki.help.ab.legacy}}'],
+            'legacy {{wiki.help.ab.legacy}}'],
            available_for=['QQ', 'QQ|Group'])
 async def ab_loader(msg: Bot.MessageSession):
     start_wiki = WikiTargetInfo(msg).get_start_wiki()
@@ -64,8 +64,12 @@ async def ab_loader(msg: Bot.MessageSession):
             traceback.print_exc()
             await msg.send_message(msg.locale.t('wiki.message.rollback'))
     if legacy:
-        res = await ab(msg, start_wiki)
-        await msg.finish(res)
+        try:
+            res = await ab(msg, start_wiki)
+            await msg.finish(res)
+        except Exception:
+            traceback.print_exc()
+            await msg.send_message(msg.locale.t('wiki.message.error.fetch_log'))
 
 
 @a.command('{{wiki.help.ab}}',
@@ -74,8 +78,12 @@ async def ab_loader(msg: Bot.MessageSession):
     start_wiki = WikiTargetInfo(msg).get_start_wiki()
     if not start_wiki:
         await msg.finish(msg.locale.t('wiki.message.not_set'))
-    res = await ab(msg, start_wiki)
-    await msg.finish(res)
+    try:
+        res = await ab(msg, start_wiki)
+        await msg.finish(res)
+    except Exception:
+        traceback.print_exc()
+        await msg.send_message(msg.locale.t('wiki.message.error.fetch_log'))
 
 
 n = module('newbie', developers=['OasisAkari'], recommend_modules='wiki')
@@ -86,5 +94,9 @@ async def newbie_loader(msg: Bot.MessageSession):
     start_wiki = WikiTargetInfo(msg).get_start_wiki()
     if not start_wiki:
         await msg.finish(msg.locale.t('wiki.message.not_set'))
-    res = await newbie(msg, start_wiki)
-    await msg.finish(res)
+    try:
+        res = await newbie(msg, start_wiki)
+        await msg.finish(res)
+    except Exception:
+        traceback.print_exc()
+        await msg.send_message(msg.locale.t('wiki.message.error.fetch_log'))
diff --git a/modules/wiki/utils/ab.py b/modules/wiki/utils/ab.py
index fe808466..8fea00ad 100644
--- a/modules/wiki/utils/ab.py
+++ b/modules/wiki/utils/ab.py
@@ -6,7 +6,8 @@ from modules.wiki.utils.wikilib import WikiLib
 
 async def ab(msg: Bot.MessageSession, wiki_url):
     wiki = WikiLib(wiki_url)
-    query = await wiki.get_json(action='query', list='abuselog', aflprop='user|title|action|result|filter|timestamp')
+    query = await wiki.get_json(action='query', list='abuselog', aflprop='user|title|action|result|filter|timestamp',
+                                _no_login=not msg.options.get("use_bot_account", False))
     pageurl = wiki.wiki_info.articlepath.replace('$1', 'Special:AbuseLog')
     d = []
     for x in query['query']['abuselog'][:5]:
@@ -21,4 +22,4 @@ async def ab(msg: Bot.MessageSession, wiki_url):
         y = y.replace("<全部吃掉了>", msg.locale.t("check.redacted.all"))
         return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="5")}\n{msg.locale.t("wiki.message.utils.redacted")}'
     else:
-        return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="5")}'
\ No newline at end of file
+        return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="5")}'
diff --git a/modules/wiki/utils/ab_qq.py b/modules/wiki/utils/ab_qq.py
index c3bd4ff4..d135c1b7 100644
--- a/modules/wiki/utils/ab_qq.py
+++ b/modules/wiki/utils/ab_qq.py
@@ -10,7 +10,7 @@ async def ab_qq(msg: MessageSession, wiki_url):
     wiki = WikiLib(wiki_url)
     qq_account = Config("qq_account")
     query = await wiki.get_json(action='query', list='abuselog', aflprop='user|title|action|result|filter|timestamp',
-                                afllimit=99)
+                                afllimit=99, _no_login=not msg.options.get("use_bot_account", False))
     pageurl = wiki.wiki_info.articlepath.replace("$1", 'Special:AbuseLog')
     nodelist = [{
         "type": "node",
diff --git a/modules/wiki/utils/rc.py b/modules/wiki/utils/rc.py
index 56965f12..b2a4679b 100644
--- a/modules/wiki/utils/rc.py
+++ b/modules/wiki/utils/rc.py
@@ -6,7 +6,8 @@ from modules.wiki.utils.wikilib import WikiLib
 
 async def rc(msg: Bot.MessageSession, wiki_url):
     wiki = WikiLib(wiki_url)
-    query = await wiki.get_json(action='query', list='recentchanges', rcprop='title|user|timestamp', rctype='edit')
+    query = await wiki.get_json(action='query', list='recentchanges', rcprop='title|user|timestamp', rctype='edit',
+                                _no_login=not msg.options.get("use_bot_account", False))
     pageurl = wiki.wiki_info.articlepath.replace('$1', 'Special:RecentChanges')
     d = []
     for x in query['query']['recentchanges'][:10]:
@@ -20,4 +21,4 @@ async def rc(msg: Bot.MessageSession, wiki_url):
         y = y.replace("<全部吃掉了>", msg.locale.t("check.redacted.all"))
         return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="10")}\n{msg.locale.t("wiki.message.utils.redacted")}'
     else:
-        return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="10")}'
\ No newline at end of file
+        return f'{str(Url(pageurl))}\n{y}\n{msg.locale.t("message.collapse", amount="10")}'
diff --git a/modules/wiki/utils/rc_qq.py b/modules/wiki/utils/rc_qq.py
index 7b7db202..902f4a3c 100644
--- a/modules/wiki/utils/rc_qq.py
+++ b/modules/wiki/utils/rc_qq.py
@@ -15,7 +15,8 @@ async def rc_qq(msg: MessageSession, wiki_url):
     query = await wiki.get_json(action='query', list='recentchanges',
                                 rcprop='title|user|timestamp|loginfo|comment|redirect|flags|sizes|ids',
                                 rclimit=99,
-                                rctype='edit|new|log'
+                                rctype='edit|new|log',
+                                _no_login=not msg.options.get("use_bot_account", False)
                                 )
     pageurl = wiki.wiki_info.articlepath
 
diff --git a/modules/wiki/utils/wikilib.py b/modules/wiki/utils/wikilib.py
index b463db99..1f5facc1 100644
--- a/modules/wiki/utils/wikilib.py
+++ b/modules/wiki/utils/wikilib.py
@@ -155,7 +155,10 @@ class WikiLib:
         self.headers = headers
         self.locale = Locale(locale)
 
-    async def get_json_from_api(self, api, **kwargs) -> dict:
+    async def get_json_from_api(self, api, _no_login=False, **kwargs) -> dict:
+        cookies = None
+        if api in BotAccount.cookies and not _no_login:
+            cookies = BotAccount.cookies[api]
         if api in redirect_list:
             api = redirect_list[api]
         if kwargs:
@@ -171,9 +174,7 @@ class WikiLib:
                     api = (web_render_local if use_local else web_render) + 'source?url=' + urllib.parse.quote(api)
                 request_local = True
                 break
-        cookies = None
-        if api in BotAccount.cookies:
-            cookies = BotAccount.cookies[api]
+
         try:
             return await get_url(api, status_code=200, headers=self.headers, fmt="json", request_private_ip=request_local,
                                  cookies=cookies)
@@ -309,10 +310,10 @@ class WikiLib:
             else:
                 raise InvalidWikiError(wiki_info.message if wiki_info.message != '' else '')
 
-    async def get_json(self, **kwargs) -> dict:
+    async def get_json(self, _no_login=False, **kwargs) -> dict:
         await self.fixup_wiki_info()
         api = self.wiki_info.api
-        return await self.get_json_from_api(api, **kwargs)
+        return await self.get_json_from_api(api, _no_login=_no_login, **kwargs)
 
     @staticmethod
     def parse_text(text):
diff --git a/schedulers/wiki_bot.py b/schedulers/wiki_bot.py
new file mode 100644
index 00000000..907c35ce
--- /dev/null
+++ b/schedulers/wiki_bot.py
@@ -0,0 +1,15 @@
+from datetime import datetime, timedelta
+
+from core.builtins import Bot
+from core.logger import Logger
+from core.queue import JobQueue
+from core.scheduler import DateTrigger, Scheduler
+from modules.wiki import BotAccount
+
+
+@Scheduler.scheduled_job(DateTrigger(datetime.now() + timedelta(seconds=10)))
+async def login_bots():
+    Logger.info('Start login wiki bot account...')
+    await BotAccount.login()
+    await JobQueue.trigger_hook_all('login_wiki_bots', cookies=BotAccount.cookies)
+    Logger.info('Login wiki bot account done')