perf: 机器人支持webhook

This commit is contained in:
kuaifan 2023-02-14 15:10:54 +08:00
parent 19f806e429
commit 9a8eae723f
4 changed files with 186 additions and 32 deletions

View File

@ -10,6 +10,8 @@ namespace App\Models;
* @property int|null $bot_id 机器人ID * @property int|null $bot_id 机器人ID
* @property int|null $clear_day 消息自动清理天数 * @property int|null $clear_day 消息自动清理天数
* @property string|null $clear_at 下一次清理时间 * @property string|null $clear_at 下一次清理时间
* @property string|null $webhook_url 消息webhook地址
* @property int|null $webhook_num 消息webhook请求次数
* @property \Illuminate\Support\Carbon|null $created_at * @property \Illuminate\Support\Carbon|null $created_at
* @property \Illuminate\Support\Carbon|null $updated_at * @property \Illuminate\Support\Carbon|null $updated_at
* @method static \Illuminate\Database\Eloquent\Builder|UserBot newModelQuery() * @method static \Illuminate\Database\Eloquent\Builder|UserBot newModelQuery()
@ -22,6 +24,8 @@ namespace App\Models;
* @method static \Illuminate\Database\Eloquent\Builder|UserBot whereId($value) * @method static \Illuminate\Database\Eloquent\Builder|UserBot whereId($value)
* @method static \Illuminate\Database\Eloquent\Builder|UserBot whereUpdatedAt($value) * @method static \Illuminate\Database\Eloquent\Builder|UserBot whereUpdatedAt($value)
* @method static \Illuminate\Database\Eloquent\Builder|UserBot whereUserid($value) * @method static \Illuminate\Database\Eloquent\Builder|UserBot whereUserid($value)
* @method static \Illuminate\Database\Eloquent\Builder|UserBot whereWebhookNum($value)
* @method static \Illuminate\Database\Eloquent\Builder|UserBot whereWebhookUrl($value)
* @mixin \Eloquent * @mixin \Eloquent
*/ */
class UserBot extends AbstractModel class UserBot extends AbstractModel

View File

@ -7,6 +7,7 @@ use App\Models\UserBot;
use App\Models\WebSocketDialog; use App\Models\WebSocketDialog;
use App\Models\WebSocketDialogMsg; use App\Models\WebSocketDialogMsg;
use App\Module\Base; use App\Module\Base;
use App\Module\Ihttp;
use Carbon\Carbon; use Carbon\Carbon;
@error_reporting(E_ALL & ~E_NOTICE & ~E_WARNING); @error_reporting(E_ALL & ~E_NOTICE & ~E_WARNING);
@ -48,9 +49,7 @@ class BotReceiveMsgTask extends AbstractTask
if ($dialog->type !== 'user') { if ($dialog->type !== 'user') {
return; return;
} }
if ($botUser->email === 'bot-manager@bot.system') { $this->botManagerReceive($msg, $botUser);
$this->botManagerReceive($msg);
}
} }
public function end() public function end()
@ -61,25 +60,46 @@ class BotReceiveMsgTask extends AbstractTask
/** /**
* 机器人管理处理消息 * 机器人管理处理消息
* @param WebSocketDialogMsg $msg * @param WebSocketDialogMsg $msg
* @param User $botUser
* @return void * @return void
*/ */
private function botManagerReceive(WebSocketDialogMsg $msg) private function botManagerReceive(WebSocketDialogMsg $msg, User $botUser)
{ {
if ($msg->type === 'text') { if ($msg->type !== 'text') {
$text = trim(strip_tags($msg->msg['text'])); return;
if (empty($text)) { }
$pureText = trim(strip_tags($msg->msg['text']));
if (str_starts_with($pureText, '/')) {
// 管理机器人
if ($botUser->email === 'bot-manager@bot.system') {
$isManager = true;
} elseif (UserBot::whereBotId($botUser->userid)->whereUserid($msg->userid)->exists()) {
$isManager = false;
} else {
$text = "非常抱歉,我不是你的机器人,无法完成你的指令。";
WebSocketDialogMsg::sendMsg(null, $msg->dialog_id, 'text', ['text' => $text], $botUser->userid, false, false, true); // todo 未能在任务end事件来发送任务
return; return;
} }
$array = Base::newTrim(explode(" ", "{$text} ")); //
$array = Base::newTrim(explode(" ", "{$pureText} "));
$type = $array[0]; $type = $array[0];
$data = []; $data = [];
$notice = ""; $notice = "";
if (!$isManager && in_array($type, ['/list', '/newbot'])) {
return; // 这些操作仅支持【机器人管理】机器人
}
switch ($type) { switch ($type) {
/** /**
* 列表 * 列表
*/ */
case '/list': case '/list':
$data = User::select(['users.*']) $data = User::select([
'users.*',
'user_bots.clear_day',
'user_bots.clear_at',
'user_bots.webhook_url',
'user_bots.webhook_num'
])
->join('user_bots', 'users.userid', '=', 'user_bots.bot_id') ->join('user_bots', 'users.userid', '=', 'user_bots.bot_id')
->where('users.bot', 1) ->where('users.bot', 1)
->where('user_bots.userid', $msg->userid) ->where('user_bots.userid', $msg->userid)
@ -92,6 +112,18 @@ class BotReceiveMsgTask extends AbstractTask
} }
break; break;
/**
* 详情
*/
case '/info':
$botId = $isManager ? $array[1] : $botUser->userid;
$data = $this->botManagerOne($botId, $msg->userid);
if (!$data) {
$type = "notice";
$notice = "机器人不存在。";
}
break;
/** /**
* 创建 * 创建
*/ */
@ -120,7 +152,7 @@ class BotReceiveMsgTask extends AbstractTask
} }
$dialog = WebSocketDialog::checkUserDialog($data->userid, $msg->userid); $dialog = WebSocketDialog::checkUserDialog($data->userid, $msg->userid);
if ($dialog) { if ($dialog) {
$text = "你好,我是你的机器人:{$data->nickname}, 我的机器人ID是{$data->userid}"; $text = "<p>您好,我是机器人:{$data->nickname}我的机器人ID是{$data->userid}</p><p>你可以发送 <u><b>/help</b></u> 查看我支持什么命令。</p>";
WebSocketDialogMsg::sendMsg(null, $dialog->id, 'text', ['text' => $text], $data->userid); // todo 未能在任务end事件来发送任务 WebSocketDialogMsg::sendMsg(null, $dialog->id, 'text', ['text' => $text], $data->userid); // todo 未能在任务end事件来发送任务
} }
break; break;
@ -129,16 +161,18 @@ class BotReceiveMsgTask extends AbstractTask
* 修改名字 * 修改名字
*/ */
case '/setname': case '/setname':
if (strlen($array[2]) < 2 || strlen($array[2]) > 20) { $botId = $isManager ? $array[1] : $botUser->userid;
$nameString = $isManager ? $array[2] : $array[1];
if (strlen($nameString) < 2 || strlen($nameString) > 20) {
$type = "notice"; $type = "notice";
$notice = "机器人名称由2-20个字符组成。"; $notice = "机器人名称由2-20个字符组成。";
break; break;
} }
$data = $this->botManagerOne($array[1], $msg->userid); $data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
$data->nickname = $array[2]; $data->nickname = $nameString;
$data->az = Base::getFirstCharter($array[2]); $data->az = Base::getFirstCharter($nameString);
$data->pinyin = Base::cn2pinyin($array[2]); $data->pinyin = Base::cn2pinyin($nameString);
$data->save(); $data->save();
} else { } else {
$type = "notice"; $type = "notice";
@ -151,7 +185,8 @@ class BotReceiveMsgTask extends AbstractTask
* 删除 * 删除
*/ */
case '/deletebot': case '/deletebot':
$data = $this->botManagerOne($array[1], $msg->userid); $botId = $isManager ? $array[1] : $botUser->userid;
$data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
$data->deleteUser('delete bot'); $data->deleteUser('delete bot');
} else { } else {
@ -164,7 +199,8 @@ class BotReceiveMsgTask extends AbstractTask
* 获取Token * 获取Token
*/ */
case '/token': case '/token':
$data = $this->botManagerOne($array[1], $msg->userid); $botId = $isManager ? $array[1] : $botUser->userid;
$data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
User::token($data); User::token($data);
} else { } else {
@ -177,7 +213,8 @@ class BotReceiveMsgTask extends AbstractTask
* 更新Token * 更新Token
*/ */
case '/revoke': case '/revoke':
$data = $this->botManagerOne($array[1], $msg->userid); $botId = $isManager ? $array[1] : $botUser->userid;
$data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
$data->encrypt = Base::generatePassword(6); $data->encrypt = Base::generatePassword(6);
$data->password = Base::md52(Base::generatePassword(32), $data->encrypt); $data->password = Base::md52(Base::generatePassword(32), $data->encrypt);
@ -192,11 +229,13 @@ class BotReceiveMsgTask extends AbstractTask
* 设置自动清理消息时间 * 设置自动清理消息时间
*/ */
case '/clearday': case '/clearday':
$data = $this->botManagerOne($array[1], $msg->userid); $botId = $isManager ? $array[1] : $botUser->userid;
$clearDay = $isManager ? $array[2] : $array[1];
$data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
$userBot = UserBot::whereBotId($array[1])->whereUserid($msg->userid)->first(); $userBot = UserBot::whereBotId($botId)->whereUserid($msg->userid)->first();
if ($userBot) { if ($userBot) {
$userBot->clear_day = min(intval($array[2]) ?: 30, 999); $userBot->clear_day = min(intval($clearDay) ?: 30, 999);
$userBot->clear_at = Carbon::now()->addDays($userBot->clear_day); $userBot->clear_at = Carbon::now()->addDays($userBot->clear_day);
$userBot->save(); $userBot->save();
} }
@ -208,15 +247,42 @@ class BotReceiveMsgTask extends AbstractTask
} }
break; break;
/**
* 设置webhook
*/
case '/webhook':
$botId = $isManager ? $array[1] : $botUser->userid;
$webhookUrl = $isManager ? $array[2] : $array[1];
$data = $this->botManagerOne($botId, $msg->userid);
if (strlen($webhookUrl) > 255) {
$type = "notice";
$notice = "webhook地址最长仅支持255个字符。";
} elseif ($data) {
$userBot = UserBot::whereBotId($botId)->whereUserid($msg->userid)->first();
if ($userBot) {
$userBot->webhook_url = $webhookUrl ?: "";
$userBot->webhook_num = 0;
$userBot->save();
}
$data->webhook_url = $userBot->webhook_url ?: '-';
$data->webhook_num = $userBot->webhook_num; // 这两个参数只是作为输出,所以不保存
} else {
$type = "notice";
$notice = "机器人不存在。";
}
break;
/** /**
* 会话搜索 * 会话搜索
*/ */
case '/dialog': case '/dialog':
$data = $this->botManagerOne($array[1], $msg->userid); $botId = $isManager ? $array[1] : $botUser->userid;
$nameKey = $isManager ? $array[2] : $array[1];
$data = $this->botManagerOne($botId, $msg->userid);
if ($data) { if ($data) {
$list = WebSocketDialog::select(['web_socket_dialogs.*', 'u.top_at', 'u.mark_unread', 'u.silence']) $list = WebSocketDialog::select(['web_socket_dialogs.*', 'u.top_at', 'u.mark_unread', 'u.silence'])
->join('web_socket_dialog_users as u', 'web_socket_dialogs.id', '=', 'u.dialog_id') ->join('web_socket_dialog_users as u', 'web_socket_dialogs.id', '=', 'u.dialog_id')
->where('web_socket_dialogs.name', 'LIKE', "%{$array[2]}%") ->where('web_socket_dialogs.name', 'LIKE', "%{$nameKey}%")
->where('u.userid', $data->userid) ->where('u.userid', $data->userid)
->orderByDesc('u.top_at') ->orderByDesc('u.top_at')
->orderByDesc('web_socket_dialogs.last_at') ->orderByDesc('web_socket_dialogs.last_at')
@ -229,7 +295,7 @@ class BotReceiveMsgTask extends AbstractTask
$list->transform(function (WebSocketDialog $item) use ($data) { $list->transform(function (WebSocketDialog $item) use ($data) {
return $item->formatData($data->userid); return $item->formatData($data->userid);
}); });
$data->list = $list; $data->list = $list; // 这个参数只是作为输出,所以不保存
} }
} else { } else {
$type = "notice"; $type = "notice";
@ -242,11 +308,26 @@ class BotReceiveMsgTask extends AbstractTask
'type' => $type, 'type' => $type,
'data' => $data, 'data' => $data,
'notice' => $notice, 'notice' => $notice,
'manager' => $isManager,
'version' => Base::getVersion() 'version' => Base::getVersion()
])->render(); ])->render();
if (!$isManager) {
$text = preg_replace("/\s*\{机器人ID\}/", "", $text);
}
$text = preg_replace("/^\x20+/", "", $text); $text = preg_replace("/^\x20+/", "", $text);
$text = preg_replace("/\n\x20+/", "\n", $text); $text = preg_replace("/\n\x20+/", "\n", $text);
WebSocketDialogMsg::sendMsg(null, $msg->dialog_id, 'text', ['text' => $text], $this->userid, false, false, true); // todo 未能在任务end事件来发送任务 WebSocketDialogMsg::sendMsg(null, $msg->dialog_id, 'text', ['text' => $text], $botUser->userid, false, false, true); // todo 未能在任务end事件来发送任务
} elseif ($pureText) {
// 推送Webhook
$userBot = UserBot::whereBotId($botUser->userid)->first();
if ($userBot && preg_match("/^https*:\/\//", $userBot->webhook_url)) {
Ihttp::ihttp_post($userBot->webhook_url, [
'text' => $pureText,
'token' => User::token($botUser),
'msg_id' => $msg->id,
'dialog_id' => $msg->dialog_id,
], 10);
}
} }
} }
@ -260,7 +341,13 @@ class BotReceiveMsgTask extends AbstractTask
$botId = intval($botId); $botId = intval($botId);
$userid = intval($userid); $userid = intval($userid);
if ($botId > 0) { if ($botId > 0) {
return User::select(['users.*']) return User::select([
'users.*',
'user_bots.clear_day',
'user_bots.clear_at',
'user_bots.webhook_url',
'user_bots.webhook_num'
])
->join('user_bots', 'users.userid', '=', 'user_bots.bot_id') ->join('user_bots', 'users.userid', '=', 'user_bots.bot_id')
->where('users.bot', 1) ->where('users.bot', 1)
->where('user_bots.bot_id', $botId) ->where('user_bots.bot_id', $botId)

View File

@ -0,0 +1,36 @@
<?php
use Illuminate\Database\Migrations\Migration;
use Illuminate\Database\Schema\Blueprint;
use Illuminate\Support\Facades\Schema;
class AddUserBotsWebhook extends Migration
{
/**
* Run the migrations.
*
* @return void
*/
public function up()
{
Schema::table('user_bots', function (Blueprint $table) {
if (!Schema::hasColumn('user_bots', 'webhook_url')) {
$table->string('webhook_url', 255)->nullable()->default('')->after('clear_at')->comment('消息webhook地址');
$table->integer('webhook_num')->nullable()->default(0)->after('webhook_url')->comment('消息webhook请求次数');
}
});
}
/**
* Reverse the migrations.
*
* @return void
*/
public function down()
{
Schema::table('user_bots', function (Blueprint $table) {
$table->dropColumn("webhook_url");
$table->dropColumn("webhook_num");
});
}
}

View File

@ -1,17 +1,22 @@
@if ($type === '/help') @if ($type === '/help')
您可以通过发送以下命令来控制我: 您可以通过发送以下命令来控制我:
<span style="color:#84c56a">/list</span> - 机器人列表 @if ($manager)
<span style="color:#84c56a">/newbot {机器人名称}</span> - 创建机器人 <span style="color:#84c56a">/list</span> - 机器人列表
<span style="color:#84c56a">/newbot {机器人名称}</span> - 创建机器人
@else
<span style="color:#84c56a">/info</span> - 查看机器人详情
@endif
<b>修改机器人</b> <b>修改机器人</b>
<span style="color:#84c56a">/setname {机器人ID} {机器人名称}</span> - 修改机器人名称 <span style="color:#84c56a">/setname {机器人ID} {机器人名称}</span> - 修改机器人名称
<span style="color:#84c56a">/deletebot {机器人ID}</span> - 删除机器人 <span style="color:#84c56a">/deletebot {机器人ID}</span> - 删除机器人
<span style="color:#84c56a">/clearday {机器人ID} {天数}</span> - 设置自动清理消息时间默认30天
<span style="color:#84c56a">/webhook {机器人ID} [url]</span> - 设置消息Webhook详细说明看 <u>/api</u>
<b>机器人设置</b> <b>机器人设置</b>
<span style="color:#84c56a">/token {机器人ID}</span> - 生成Token令牌 <span style="color:#84c56a">/token {机器人ID}</span> - 生成Token令牌
<span style="color:#84c56a">/revoke {机器人ID}</span> - 撤销机器人Token令牌 <span style="color:#84c56a">/revoke {机器人ID}</span> - 撤销机器人Token令牌
<span style="color:#84c56a">/clearday {机器人ID} {天数}</span> - 设置自动清理消息时间默认30天
<b>会话管理</b> <b>会话管理</b>
<span style="color:#84c56a">/dialog {机器人ID} [搜索关键词]</span> - 查看会话ID <span style="color:#84c56a">/dialog {机器人ID} [搜索关键词]</span> - 查看会话ID
@ -21,10 +26,19 @@
@elseif ($type === '/list') @elseif ($type === '/list')
<b>我的机器人。</b> <b>我的机器人。</b>
<b>机器人ID | 机器人名称</b> <b>ID | 名称 | 清理时间 | Webhook</b>
@foreach($data as $item) @foreach($data as $item)
{{$item->userid}} | {{$item->nickname}} {{$item->userid}} | {{$item->nickname}} | {{$item->clear_day}} | {{$item->webhook_url ?: '-'}}
@endforeach @endforeach
@elseif ($type === '/info')
<b>机器人详情。</b>
机器人ID<span style="color:#84c56a">{{$data->userid}}</span>
机器人名称:<span style="color:#84c56a">{{$data->nickname}}</span>
自动清理消息时间:<span style="color:#84c56a">{{$data->clear_day}}</span>
最后一次清理时间:<span style="color:#84c56a">{{$data->clear_at ?: '-'}}</span>
Webhook地址<span style="color:#84c56a">{{$data->webhook_url ?: '-'}}</span>
Webhook请求次数<span style="color:#84c56a">{{$data->webhook_num}}</span>
@elseif ($type === '/newbot') @elseif ($type === '/newbot')
<b>创建成功。</b> <b>创建成功。</b>
@ -51,6 +65,12 @@
机器人ID<span style="color:#84c56a">{{$data->userid}}</span> 机器人ID<span style="color:#84c56a">{{$data->userid}}</span>
机器人名称:<span style="color:#84c56a">{{$data->nickname}}</span> 机器人名称:<span style="color:#84c56a">{{$data->nickname}}</span>
@elseif ($type === '/webhook')
<b>设置Webhook地址。</b>
机器人ID<span style="color:#84c56a">{{$data->userid}}</span>
机器人名称:<span style="color:#84c56a">{{$data->nickname}}</span>
Webhook地址<span style="color:#84c56a">{{$data->webhook_url}}</span>
@elseif ($type === '/clearday') @elseif ($type === '/clearday')
<b>设置自动清理消息时间。</b> <b>设置自动清理消息时间。</b>
@ -68,13 +88,20 @@
@elseif ($type === '/api') @elseif ($type === '/api')
你可以通过执行以下命令来请求我: 你可以通过执行以下命令来请求我:
<b>发送文本消息</b> <b>发送文本消息</b>
curl --request POST '{{url('api/dialog/msg/sendtext')}}' \ curl --request POST '{{url('api/dialog/msg/sendtext')}}' \
--header 'version: {{ $version }}' \ --header 'version: {{ $version }}' \
--header 'token: <span style="color:#84c56a">{机器人Token}</span>' \ --header 'token: <span style="color:#84c56a">{机器人Token}</span>' \
--form 'dialog_id="<span style="color:#84c56a">{对话ID}</span>"' \ --form 'dialog_id="<span style="color:#84c56a">{对话ID}</span>"' \
--form 'text="<span style="color:#84c56a">{消息内容}</span>"' --form 'text="<span style="color:#84c56a">{消息内容}</span>"'
--form 'silence="<span style="color:#84c56a">[yes|no]</span>"' --form 'silence="<span style="color:#84c56a">[yes|no]</span>"'
<b>Webhook说明</b>
机器人收到个人对话消息后会将消息POST推送到Webhook地址请求超时为10秒请求参数如下
<span style="color:#84c56a">text</span>: 消息文本
<span style="color:#84c56a">token</span>: 机器人Token
<span style="color:#84c56a">msg_id</span>: 消息ID
<span style="color:#84c56a">dialog_id</span>: 对话ID
@elseif ($type === 'notice') @elseif ($type === 'notice')
{{$notice}} {{$notice}}
@else @else