nonebot2/app.py

import re
import sys
import importlib

from flask import Flask, request

import interactive
from little_shit import *
from config import config
from command import hub as cmdhub
from command import CommandNotExistsError, CommandScopeError, CommandPermissionError
from apiclient import client as api

app = Flask(__name__)

_fallback_command = config.get('fallback_command')
_command_start_flags = get_command_start_flags()
_command_args_start_flags = get_command_args_start_flags()


def _send_text(text, ctx_msg):
    msg_type = ctx_msg.get('type')
    if msg_type == 'group_message':
        api.send_group_message(gnumber=ctx_msg.get('gnumber'), content=text)
    elif msg_type == 'message':
        api.send_message(qq=ctx_msg.get('sender_qq'), content=text)


@app.route('/', methods=['POST'])
def _index():
    ctx_msg = request.json
    source = get_source(ctx_msg)
    try:
        if ctx_msg.get('msg_class') != 'recv':
            raise SkipException
        content = ctx_msg.get('content', '')
        if content.startswith('@'):
            my_group_nick = ctx_msg.get('receiver')
            if not my_group_nick:
                raise SkipException
            at_me = '@' + my_group_nick
            if not content.startswith(at_me):
                my_nick = api.get_user_info().json().get('nick', my_group_nick)
                at_me = '@' + my_nick
                if not content.startswith(at_me):
                    raise SkipException
            content = content[len(at_me):]
        else:
            # Not starts with '@'
            if ctx_msg.get('type') == 'group_message':
                # And it's a group message, so we don't reply
                raise SkipException
        content = content.lstrip()
        start_flag = None
        for flag in _command_start_flags:
            # Match the command start flag
            if content.startswith(flag):
                start_flag = flag
                break
        if not start_flag or len(content) <= len(start_flag):
            # No command, check if a session exists
            if interactive.has_session(source):
                command = [interactive.get_session(source).cmd, content]
            else:
                # Use fallback
                if _fallback_command:
                    command = [_fallback_command, content]
                else:
                    # No fallback
                    raise SkipException
        else:
            # Split command and arguments
            command = re.split('|'.join(_command_args_start_flags),
                               content[len(start_flag):], 1)
            if len(command) == 1:
                # Add an empty argument
                command.append('')
            # Starting a new command, so remove any previous command session
            interactive.remove_session(source)

        cmdhub.call(command[0], command[1], ctx_msg)
    except SkipException:
        # Skip this message
        pass
    except CommandNotExistsError:
        _send_text('暂时还没有这个命令哦～', ctx_msg)
    except CommandPermissionError:
        _send_text('你没有权限使用这个命令哦～', ctx_msg)
    except CommandScopeError as se:
        _send_text('这个命令不支持' + se.msg_type + '哦～', ctx_msg)
    return '', 204


def load_commands():
    command_mod_files = filter(
        lambda filename: filename.endswith('.py') and not filename.startswith('_'),
        os.listdir(get_commands_dir())
    )
    command_mods = [os.path.splitext(file)[0] for file in command_mod_files]
    for mod_name in command_mods:
        cmd_mod = importlib.import_module('commands.' + mod_name)
        try:
            cmdhub.add_registry(mod_name, cmd_mod.__registry__)
        except AttributeError:
            print('Failed to load command module "' + mod_name + '.py".', file=sys.stderr)


if __name__ == '__main__':
    load_commands()
    app.run(host='127.0.0.1', port=8888)