1 files changed, 89 insertions, 0 deletions
diff --git a/ATRIbot/plugins/welcome.py b/ATRIbot/plugins/welcome.py
new file mode 100644
index 0000000..da0a4c3
--- /dev/null
+++ b/ATRIbot/plugins/welcome.py
@@ -0,0 +1,89 @@
+import nonebot
+from nonebot import on_command, on_notice
+from nonebot import CommandSession
+from nonebot.notice_request import NoticeSession, RequestSession
+from nonebot.plugin import on_request
+from aiocqhttp.exceptions import ActionFailed
+
+bot = nonebot.get_bot()
+master = bot.config.SUPERUSERS
+
+
+@on_notice('group_increase')
+async def _(session: NoticeSession):
+    q = session.event.user_id
+    me = session.event.self_id
+
+    if q == me :
+        await session.send('在下アトリ，今后请多指教呐❤')
+    else:
+        await session.send(f'ねえ❤...是新人欸！\nここでは遠慮はいらないのだからね❤')
+
+@on_notice('group_decrease')
+async def _(session: NoticeSession):
+    q = session.event.user_id
+    qm = session.event.operator_id
+
+    if qm == q:
+        inf = await bot.get_stranger_info(user_id = q)
+
+        name = inf['nickname']
+
+        await session.send(f'ねえ...{name}({q}) 跑了...')
+
+@on_notice('firend_add')
+async def _(session: NoticeSession):
+    u = session.event.user_id
+    await bot.send_private_msg(self_id = session.self_id, user_id = u, message=f'有人来加我啦！\n{u}')
+
+
+@on_request('group')
+async def bot_request(session: RequestSession):
+    g = session.event.group_id
+    u = session.event.user_id
+    mt = master[0]
+    print(mt)
+
+    if session.event.sub_type == 'invite':
+        
+        if welcome_switch:
+
+            await session.bot.send_private_msg(self_id = session.self_id, user_id = u, message='嗯哼？想邀请吾辈入群嘛，既然这样，那吾辈来了呐。\n作者联系方式：\nhttps://lolihub.icu/#/about')
+
+            await session.bot.send_private_msg(self_id = session.self_id, user_id = mt, message=f'吾辈收到一个新邀请:\n裙: {g}\n邀请人: {u}')
+
+            try:
+                await session.approve()
+            except ActionFailed as e:
+                print(e.retcode)
+
+        else:
+            await session.bot.send_private_msg(self_id = session.self_id, user_id = u, message='主人告诉吾辈不能同意任何人的请求呢...\n作者联系方式：\nhttps://lolihub.icu/#/about')
+            
+            await session.bot.send_private_msg(self_id = session.self_id, user_id = mt, message=f'吾辈收到一个新的邀请请求，由于主人并未告知吾辈可以邀请，故做出拒绝的回应。\n裙: {g}\n邀请人: {u}')
+
+
+welcome_switch = True
+@on_command('welcome_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 welcome_switch
+        if switch == '开启':
+            if com == '邀请':
+                welcome_switch = True
+            else:
+                pass
+
+        elif switch == '关闭':
+            if com == '邀请':
+                welcome_switch = False
+            else:
+                pass
+        
+        await session.send('完成')
+    
+    else:
+        await session.send('恁哪位?')