1 files changed, 109 insertions, 0 deletions
diff --git a/ATRIbot/plugins/chat.py b/ATRIbot/plugins/chat.py
new file mode 100644
index 0000000..93526e3
--- /dev/null
+++ b/ATRIbot/plugins/chat.py
@@ -0,0 +1,109 @@
+# -*- coding:utf-8 -*-
+import time
+import httpx
+import string
+import nonebot
+import hashlib
+from AyaBot.plugins import *
+from random import randint
+from urllib.parse import urlencode
+from nonebot.helpers import render_expression
+from nonebot import on_command, CommandSession, on_natural_language, NLPSession, IntentCommand
+
+bot = nonebot.get_bot()
+mastwr = bot.config.SUPERUSERS
+
+ERROR_REPLY = (
+    '吾辈不是很能理解主人的意思...',
+    '(。´・ω・)ん?',
+    'ん？',
+    'あつ...脑子坏了..理解不能',
+    '吾辈想听主人再说一次...'
+)
+
+class Chat(object):
+    URL = 'https://api.ai.qq.com/fcgi-bin/nlp/nlp_textchat'
+    app_id = bot.config.TX_APP_ID
+    app_key = bot.config.TX_APPKEY
+    print(app_id, app_key)
+    nonce_str_example = 'fa577ce340859f9fe'
+    ct = lambda: time.time()
+
+    @classmethod
+    def get_nonce_str(self):
+        nonce_str = ''
+        len_str = string.digits + string.ascii_letters
+        for i in range(len(self.nonce_str_example)):
+            nonce_str += len_str[randint(0, len(len_str) - 1)]
+        return nonce_str
+
+    @classmethod
+    def sign(self, req_data):
+        new_list = sorted(req_data.items())
+        encode_list = urlencode(new_list)
+        req_data = encode_list + "&" + "app_key" + "=" + self.app_key
+        md5 = hashlib.md5()
+        md5.update(req_data.encode('utf-8'))
+        data = md5.hexdigest()
+        return data.upper()
+
+    @classmethod
+    async def request(self, text):
+        req_data = {
+            'app_id': self.app_id,
+            'time_stamp': int(self.ct()),
+            'nonce_str': self.get_nonce_str(),
+            'session': 10000,
+            'question': text,
+        }
+        print(req_data)
+        req_data['sign'] = self.sign(req_data)
+        req_data = sorted(req_data.items())
+        requests = httpx.AsyncClient()
+        result = await requests.get(self.URL, params=req_data)
+        await requests.aclose()
+        result = result.json()
+        print(result)
+        if result['ret'] == 0:
+            return result['data']['answer']
+        return render_expression(ERROR_REPLY)
+
+
+@on_command('chat')
+async def chat(session: CommandSession):
+    if chat_switch:
+        msg = session.state.get('msg')
+        reply = await Chat.request(msg)
+        await session.finish(reply)
+        return
+    else:
+        await session.send('えつ，アトリ被主人告知不能与陌生人说话')
+
+@on_natural_language
+async def _(session: NLPSession):
+    return IntentCommand(60.0, ('chat'), {'msg': session.msg_text})
+
+chat_switch = True
+@on_command('chat_switch', aliases=['开启', '关闭'], only_to_me=False)
+async def _(session: CommandSession):
+    if session.event.user_id in master:
+        command = session.event.raw_message.split(' ', 1)
+        switch = command[0]
+        com = command[1]
+        global chat_switch
+        if switch == '开启':
+            if com == '闲聊':
+                chat_switch = True
+            else:
+                pass
+
+        elif switch == '关闭':
+            if com == '闲聊':
+                chat_switch = False
+            else:
+                pass
+        
+        await session.send('完成')
+    
+    else:
+        await session.send('恁哪位?')