mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-28 05:16:48 +08:00
95 lines
3.4 KiB
Python
95 lines
3.4 KiB
Python
import re
|
|
import sys
|
|
import importlib
|
|
|
|
import interactive
|
|
from config import config
|
|
from filter import add_filter
|
|
from command import CommandNotExistsError, CommandScopeError, CommandPermissionError
|
|
from little_shit import *
|
|
from commands import core
|
|
from apiclient import client as api
|
|
from command import hub as cmdhub
|
|
|
|
_fallback_command = config.get('fallback_command')
|
|
_command_start_flags = get_command_start_flags()
|
|
_command_args_start_flags = get_command_args_start_flags()
|
|
|
|
|
|
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)
|
|
|
|
|
|
def _dispatch_command(ctx_msg):
|
|
try:
|
|
content = ctx_msg.get('content', '')
|
|
source = get_source(ctx_msg)
|
|
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:
|
|
core.echo('暂时还没有这个命令哦～', ctx_msg)
|
|
except CommandPermissionError:
|
|
core.echo('你没有权限使用这个命令哦～', ctx_msg)
|
|
except CommandScopeError as se:
|
|
core.echo('这个命令不支持' + se.msg_type + '哦～', ctx_msg)
|
|
|
|
|
|
_load_commands()
|
|
add_filter(_dispatch_command, 0)
|