deleted: app.py
deleted: requirements.txt
deleted: run.bat
deleted: run.sh
deleted: src/__init__.py
deleted: src/mainprocess.py
This commit is contained in:
97
app.py
97
app.py
@ -1,97 +0,0 @@
|
|||||||
import logging
|
|
||||||
from flask import Flask, request, jsonify
|
|
||||||
from logging.handlers import RotatingFileHandler
|
|
||||||
from functools import wraps
|
|
||||||
from datetime import datetime
|
|
||||||
from src import mainprocess as src
|
|
||||||
|
|
||||||
|
|
||||||
app = Flask(__name__)
|
|
||||||
|
|
||||||
#===rebot===#
|
|
||||||
# 处理私聊消息
|
|
||||||
# 处理群聊消息
|
|
||||||
@app.route('/', methods=["POST"])
|
|
||||||
def handle_event():
|
|
||||||
try:
|
|
||||||
event = request.get_json()
|
|
||||||
event_type = event.get('post_type')
|
|
||||||
|
|
||||||
# 1. 处理私聊消息
|
|
||||||
if event_type == 'message' and event.get('message_type') == 'private':
|
|
||||||
# 注意：私聊消息在顶层有 user_id
|
|
||||||
uid = event.get('user_id')
|
|
||||||
message = event.get('raw_message')
|
|
||||||
src.process_message(uid, None, message)
|
|
||||||
|
|
||||||
# 2. 处理群消息
|
|
||||||
elif event_type == 'message' and event.get('message_type') == 'group':
|
|
||||||
gid = event.get('group_id')
|
|
||||||
# 注意：群消息发送者在 sender 内
|
|
||||||
sender = event.get('sender', {})
|
|
||||||
uid = sender.get('user_id')
|
|
||||||
message = event.get('raw_message')
|
|
||||||
src.process_message(uid, gid, message)
|
|
||||||
|
|
||||||
# 3. 处理通知事件（如输入状态）
|
|
||||||
elif event_type == 'notice':
|
|
||||||
notice_type = event.get('notice_type')
|
|
||||||
|
|
||||||
if notice_type == 'notify' and event.get('sub_type') == 'input_status':
|
|
||||||
# 仅记录，不处理
|
|
||||||
logging.info(f"用户 {event.get('user_id')} 输入状态变化")
|
|
||||||
|
|
||||||
elif notice_type == 'group_recall':
|
|
||||||
# 示例：处理群消息撤回
|
|
||||||
logging.info(f"群 {event.get('group_id')} 撤回消息")
|
|
||||||
|
|
||||||
else:
|
|
||||||
# 其他通知类型
|
|
||||||
logging.info(f"Ignored notice: {event}")
|
|
||||||
|
|
||||||
# 通知事件直接返回成功
|
|
||||||
return jsonify({
|
|
||||||
"status": "ok",
|
|
||||||
"retcode": 0,
|
|
||||||
"data": None
|
|
||||||
})
|
|
||||||
|
|
||||||
# 4. 处理元事件（如心跳）
|
|
||||||
elif event_type == 'meta_event':
|
|
||||||
# 心跳等元事件直接返回成功
|
|
||||||
return jsonify({
|
|
||||||
"status": "ok",
|
|
||||||
"retcode": 0,
|
|
||||||
"data": None
|
|
||||||
})
|
|
||||||
|
|
||||||
# 5. 一切正常的消息事件返回成功
|
|
||||||
return jsonify({
|
|
||||||
"status": "ok",
|
|
||||||
"retcode": 0,
|
|
||||||
"data": "Processed successfully"
|
|
||||||
})
|
|
||||||
|
|
||||||
except KeyError:
|
|
||||||
logging.warning(f"Missing required field in event: {event}")
|
|
||||||
return jsonify({
|
|
||||||
"status": "failed",
|
|
||||||
"retcode": 10001,
|
|
||||||
"message": "Missing required field"
|
|
||||||
}), 400
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
logging.exception(f"Error processing event: {str(e)}")
|
|
||||||
return jsonify({
|
|
||||||
"status": "failed",
|
|
||||||
"retcode": 20001,
|
|
||||||
"message": "Internal server error"
|
|
||||||
}), 500
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
try:
|
|
||||||
port = 25580
|
|
||||||
app.run(debug=True, host='0.0.0.0', port=port)
|
|
||||||
except Exception as e:
|
|
||||||
print(f"启动失败: {e}")
|
|
||||||
@ -1,16 +0,0 @@
|
|||||||
blinker==1.9.0
|
|
||||||
certifi==2025.8.3
|
|
||||||
charset-normalizer==3.4.2
|
|
||||||
click==8.2.1
|
|
||||||
colorama==0.4.6
|
|
||||||
Flask==3.1.1
|
|
||||||
idna==3.10
|
|
||||||
itsdangerous==2.2.0
|
|
||||||
Jinja2==3.1.6
|
|
||||||
MarkupSafe==3.0.2
|
|
||||||
packaging==25.0
|
|
||||||
pkg==0.2
|
|
||||||
requests==2.32.4
|
|
||||||
toml==0.10.2
|
|
||||||
urllib3==2.5.0
|
|
||||||
Werkzeug==3.1.3
|
|
||||||
74
run.bat
74
run.bat
@ -1,74 +0,0 @@
|
|||||||
@echo off
|
|
||||||
|
|
||||||
|
|
||||||
set PROJECT_DIR=%~dp0
|
|
||||||
set VENV_DIR=%PROJECT_DIR%.venv
|
|
||||||
|
|
||||||
|
|
||||||
if exist "%VENV_DIR%\Scripts\activate.bat" (
|
|
||||||
|
|
||||||
call "%VENV_DIR%\Scripts\activate.bat"
|
|
||||||
) else (
|
|
||||||
|
|
||||||
python -m venv "%VENV_DIR%"
|
|
||||||
call "%VENV_DIR%\Scripts\activate.bat"
|
|
||||||
|
|
||||||
if errorlevel 1 (
|
|
||||||
echo error: fail to create env
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
if "%VIRTUAL_ENV%" == "" (
|
|
||||||
echo error: fail to activate env
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
echo installing dependence...
|
|
||||||
|
|
||||||
pip install -r requirements.txt
|
|
||||||
if errorlevel 1 (
|
|
||||||
echo error: fail to install dependence
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
pip install waitress
|
|
||||||
if errorlevel 1 (
|
|
||||||
echo error: fail to install waitress
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
echo reading port from config...
|
|
||||||
for /f "usebackq tokens=*" %%P in (`python -c "from src.file_store_api import ConfigManager; config=ConfigManager().load_config(); print(config.get('app', {}).get('list_port', 25580))"`) do (
|
|
||||||
set PORT=%%P
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
if "%PORT%"=="" (
|
|
||||||
set PORT=25580
|
|
||||||
echo can't read port,use custom port:25580
|
|
||||||
) else (
|
|
||||||
echo success read port: %PORT%
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
echo starting rebot_server...
|
|
||||||
echo listening at: %PORT%
|
|
||||||
|
|
||||||
|
|
||||||
waitress-serve --host=0.0.0.0 --port=%PORT% app:app
|
|
||||||
|
|
||||||
|
|
||||||
if errorlevel 1 (
|
|
||||||
echo error,fail to start
|
|
||||||
pause
|
|
||||||
exit /b 1
|
|
||||||
)
|
|
||||||
|
|
||||||
pause
|
|
||||||
51
run.sh
51
run.sh
@ -1,51 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
PROJECT_DIR=$(cd "$(dirname "$0")"; pwd)
|
|
||||||
VENV_DIR="$PROJECT_DIR/.venv"
|
|
||||||
FLASK_APP="app:app"
|
|
||||||
|
|
||||||
|
|
||||||
echo "Activating virtual environment..."
|
|
||||||
|
|
||||||
if [ -f "$VENV_DIR/bin/activate" ]; then
|
|
||||||
source "$VENV_DIR/bin/activate"
|
|
||||||
else
|
|
||||||
echo "Creating new virtual environment..."
|
|
||||||
python3 -m venv "$VENV_DIR"
|
|
||||||
source "$VENV_DIR/bin/activate"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -z "$VIRTUAL_ENV" ]; then
|
|
||||||
echo "Error: Failed to activate virtual environment"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "Installing dependencies..."
|
|
||||||
pip install --upgrade pip
|
|
||||||
pip install -r requirements.txt
|
|
||||||
pip install gunicorn
|
|
||||||
|
|
||||||
echo "Reading port from configuration..."
|
|
||||||
|
|
||||||
PORT=$(python3 -c \
|
|
||||||
"
|
|
||||||
from src.file_store_api import ConfigManager
|
|
||||||
try:
|
|
||||||
config = ConfigManager().load_config()
|
|
||||||
port = config.get('app', {}).get('list_port')
|
|
||||||
print(str(port) if port else '')
|
|
||||||
except Exception as e:
|
|
||||||
print('ERROR: ' + str(e))
|
|
||||||
exit(1)
|
|
||||||
")
|
|
||||||
|
|
||||||
if [[ "$PORT" == ERROR:* ]] || [ -z "$PORT" ]; then
|
|
||||||
echo "Failed to get port from config: $PORT"
|
|
||||||
echo "Using default port 25580"
|
|
||||||
PORT=25580
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "Starting rebot server..."
|
|
||||||
echo "Listening on port: $PORT"
|
|
||||||
|
|
||||||
gunicorn -w 4 -b 0.0.0.0:$PORT "$FLASK_APP" --access-logfile - --error-logfile -
|
|
||||||
@ -1,79 +0,0 @@
|
|||||||
import sys
|
|
||||||
import src.modules.user_modules as usermod
|
|
||||||
from src.modules.plugin_modules import BasePlugin, MessageContext
|
|
||||||
import src.file_store_api as file_M
|
|
||||||
import src.plugin_manager as plm
|
|
||||||
|
|
||||||
manager = plm.PluginManager()
|
|
||||||
config = file_M.ConfigManager()
|
|
||||||
rebot_id = config.load_config().get("rebot").get("id")
|
|
||||||
def process_message(uid: str, gid: str | None, message: str) -> str:
|
|
||||||
# 创建上下文
|
|
||||||
ctx = MessageContext(uid=uid, gid=gid, raw_message=message,id = rebot_id)
|
|
||||||
|
|
||||||
plugin_manager = manager
|
|
||||||
manager.scan_plugins()
|
|
||||||
# 阶段1: before_load 插件（加载数据前）
|
|
||||||
ctx.phase = "before_load"
|
|
||||||
early_plugins = []
|
|
||||||
for name, plugin_cls in plugin_manager._plugins.items():
|
|
||||||
plugin = plugin_cls(ctx)
|
|
||||||
if hasattr(plugin, 'before_load') and callable(plugin.before_load):
|
|
||||||
early_plugins.append(plugin)
|
|
||||||
|
|
||||||
for plugin in early_plugins:
|
|
||||||
try:
|
|
||||||
result = plugin.before_load()
|
|
||||||
if result is not None: # 拦截逻辑
|
|
||||||
return result
|
|
||||||
except Exception as e:
|
|
||||||
print(f"error:Plugin {plugin.__class__.__name__} before_load error: {str(e)}")
|
|
||||||
|
|
||||||
# 消息加载逻辑
|
|
||||||
if gid is not None:
|
|
||||||
ctx.group.messages = ctx.chat_manager.load_group_messages(ctx.group)
|
|
||||||
ctx.user.messages = ctx.chat_manager.load_user_group_messages(user=ctx.user, group=ctx.group)
|
|
||||||
else:
|
|
||||||
ctx.user.messages = ctx.chat_manager.load_private_messages(ctx.user)
|
|
||||||
|
|
||||||
# 阶段2: after_load 插件（加载数据后）
|
|
||||||
ctx.phase = "after_load"
|
|
||||||
loaded_plugins = []
|
|
||||||
for name, plugin_cls in plugin_manager._plugins.items():
|
|
||||||
plugin = plugin_cls(ctx)
|
|
||||||
if hasattr(plugin, 'after_load') and callable(plugin.after_load):
|
|
||||||
loaded_plugins.append(plugin)
|
|
||||||
|
|
||||||
for plugin in loaded_plugins:
|
|
||||||
try:
|
|
||||||
result = plugin.after_load()
|
|
||||||
if result is not None:
|
|
||||||
ctx.response = result
|
|
||||||
break
|
|
||||||
except Exception as e:
|
|
||||||
print(f"error:Plugin {plugin.__class__.__name__} after_load error: {str(e)}")
|
|
||||||
|
|
||||||
# 消息保存逻辑
|
|
||||||
if gid is not None:
|
|
||||||
ctx.chat_manager.save_group_message(ctx.group, role="user", content=ctx.raw_message, sender_id=ctx.user.user_id)
|
|
||||||
else:
|
|
||||||
ctx.chat_manager.save_private_message(ctx.user, role="user", content=ctx.raw_message)
|
|
||||||
|
|
||||||
# 阶段3: after_save 插件（保存数据后）
|
|
||||||
ctx.phase = "after_save"
|
|
||||||
saved_plugins = []
|
|
||||||
for name, plugin_cls in plugin_manager._plugins.items():
|
|
||||||
plugin = plugin_cls(ctx)
|
|
||||||
if hasattr(plugin, 'after_save') and callable(plugin.after_save):
|
|
||||||
saved_plugins.append(plugin)
|
|
||||||
|
|
||||||
for plugin in saved_plugins:
|
|
||||||
try:
|
|
||||||
result = plugin.after_save()
|
|
||||||
if result is not None and ctx.response is None:
|
|
||||||
ctx.response = result
|
|
||||||
except Exception as e:
|
|
||||||
print(f"error:Plugin {plugin.__class__.__name__} after_save error: {str(e)}")
|
|
||||||
plugin_manager.cleanup()
|
|
||||||
|
|
||||||
return ctx.response if ctx.response is not None else "ok"
|
|
||||||
Reference in New Issue
Block a user