1 files changed, 53 insertions, 0 deletions
diff --git a/ATRI/plugins/genshin/__init__.py b/ATRI/plugins/genshin/__init__.py
new file mode 100644
index 0000000..4705e0e
--- /dev/null
+++ b/ATRI/plugins/genshin/__init__.py
@@ -0,0 +1,53 @@
+import re
+from nonebot.plugin import on_command
+from nonebot.typing import Bot, Event
+
+from ATRI.config import GENSHIN_CONFIG
+from ATRI.exceptions import InvalidRequestError
+from ATRI.rule import isInBanList, isInDormant, isInService
+
+from .data_source import Genshin
+
+__plugin_name__ = 'genshin'
+
+genshin = on_command(GENSHIN_CONFIG['genshin']['command'][0],
+                     aliases=set(GENSHIN_CONFIG['genshin']['command']),
+                     rule=isInBanList() & isInDormant()
+                     & isInService(__plugin_name__))
+
+
+@genshin.handle()
+async def _(bot: Bot, event: Event, state: dict) -> None:
+    args = str(event.message)
+    
+    if args:
+        state['args'] = args
+
+@genshin.got('args', prompt='请告诉咱id以及是官服还是b服嗷~w\n用空格隔开！！')
+async def _(bot: Bot, event: Event, state: dict) -> None:
+    args = state['args'].split(' ')
+    uid_info = ''
+    server_id = ''
+    
+    if len(args[0]) != 9:
+        await genshin.finish('抱歉暂时只支持国服呢...')
+    else:
+        pass
+    
+    if re.findall('[Bb官]服', args[1]):
+        if re.findall('[Bb]服', args[1]):
+            server_id = 'cn_qd01'
+        else:
+            server_id = 'cn_gf01'
+        
+        await bot.send(event, '别急，在找！')
+        try:
+            uid_info = Genshin().jsonAnalysis(Genshin().getInfo(server_id=server_id, uid=args[0]))
+        except InvalidRequestError:
+            await genshin.finish('找不到呢...咱搜索区域或许出错了...')
+        
+        msg0 = (f'{args[0]} Genshin INFO:\n'
+                f'{uid_info}')
+        await genshin.finish(msg0)
+    else:
+        await genshin.finish('抱歉暂时只支持官服和b服...呐.')