mirror of
https://github.com/kuaifan/dootask.git
synced 2026-03-02 06:54:19 +00:00
365 lines
13 KiB
PHP
365 lines
13 KiB
PHP
<?php
|
||
|
||
namespace App\Http\Controllers\Api;
|
||
|
||
use Request;
|
||
use App\Models\File;
|
||
use App\Models\User;
|
||
use App\Models\WebSocketDialogMsg;
|
||
use App\Module\Base;
|
||
use App\Module\Apps;
|
||
use App\Module\Manticore\ManticoreFile;
|
||
use App\Module\Manticore\ManticoreUser;
|
||
use App\Module\Manticore\ManticoreProject;
|
||
use App\Module\Manticore\ManticoreTask;
|
||
use App\Module\Manticore\ManticoreMsg;
|
||
|
||
/**
|
||
* @apiDefine search
|
||
*
|
||
* 智能搜索
|
||
*/
|
||
class SearchController extends AbstractController
|
||
{
|
||
/**
|
||
* @api {get} api/search/contact AI 搜索联系人
|
||
*
|
||
* @apiDescription 需要token身份，需要安装 Manticore Search 应用
|
||
* @apiVersion 1.0.0
|
||
* @apiGroup search
|
||
* @apiName contact
|
||
*
|
||
* @apiParam {String} key 搜索关键词
|
||
* @apiParam {String} [search_type] 搜索类型（text/vector/hybrid，默认：hybrid）
|
||
* @apiParam {Number} [take] 获取数量（默认：20，最大：50）
|
||
*
|
||
* @apiSuccess {Number} ret 返回状态码（1正确、0错误）
|
||
* @apiSuccess {String} msg 返回信息（错误描述）
|
||
* @apiSuccess {Object} data 返回数据
|
||
*/
|
||
public function contact()
|
||
{
|
||
User::auth();
|
||
|
||
if (!Apps::isInstalled('manticore')) {
|
||
return Base::retError('Manticore Search 应用未安装');
|
||
}
|
||
|
||
$key = trim(Request::input('key'));
|
||
$searchType = Request::input('search_type', 'hybrid');
|
||
$take = Base::getPaginate(50, 20, 'take');
|
||
|
||
if (empty($key)) {
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
$results = ManticoreUser::search($key, $searchType, $take);
|
||
|
||
// 补充用户完整信息
|
||
$userids = array_column($results, 'userid');
|
||
if (!empty($userids)) {
|
||
$users = User::whereIn('userid', $userids)
|
||
->select(User::$basicField)
|
||
->get()
|
||
->keyBy('userid');
|
||
|
||
foreach ($results as &$item) {
|
||
$userData = $users->get($item['userid']);
|
||
if ($userData) {
|
||
$item = array_merge($userData->toArray(), [
|
||
'relevance' => $item['relevance'] ?? 0,
|
||
'introduction_preview' => $item['introduction_preview'] ?? null,
|
||
]);
|
||
}
|
||
}
|
||
}
|
||
|
||
return Base::retSuccess('success', $results);
|
||
}
|
||
|
||
/**
|
||
* @api {get} api/search/project AI 搜索项目
|
||
*
|
||
* @apiDescription 需要token身份，需要安装 Manticore Search 应用
|
||
* @apiVersion 1.0.0
|
||
* @apiGroup search
|
||
* @apiName project
|
||
*
|
||
* @apiParam {String} key 搜索关键词
|
||
* @apiParam {String} [search_type] 搜索类型（text/vector/hybrid，默认：hybrid）
|
||
* @apiParam {Number} [take] 获取数量（默认：20，最大：50）
|
||
*
|
||
* @apiSuccess {Number} ret 返回状态码（1正确、0错误）
|
||
* @apiSuccess {String} msg 返回信息（错误描述）
|
||
* @apiSuccess {Object} data 返回数据
|
||
*/
|
||
public function project()
|
||
{
|
||
$user = User::auth();
|
||
|
||
if (!Apps::isInstalled('manticore')) {
|
||
return Base::retError('Manticore Search 应用未安装');
|
||
}
|
||
|
||
$key = trim(Request::input('key'));
|
||
$searchType = Request::input('search_type', 'hybrid');
|
||
$take = Base::getPaginate(50, 20, 'take');
|
||
|
||
if (empty($key)) {
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
$results = ManticoreProject::search($user->userid, $key, $searchType, $take);
|
||
|
||
// 补充项目完整信息
|
||
$projectIds = array_column($results, 'project_id');
|
||
if (!empty($projectIds)) {
|
||
$projects = \App\Models\Project::whereIn('id', $projectIds)
|
||
->get()
|
||
->keyBy('id');
|
||
|
||
foreach ($results as &$item) {
|
||
$projectData = $projects->get($item['project_id']);
|
||
if ($projectData) {
|
||
$item = array_merge($projectData->toArray(), [
|
||
'relevance' => $item['relevance'] ?? 0,
|
||
'desc_preview' => $item['desc_preview'] ?? null,
|
||
]);
|
||
}
|
||
}
|
||
}
|
||
|
||
return Base::retSuccess('success', $results);
|
||
}
|
||
|
||
/**
|
||
* @api {get} api/search/task AI 搜索任务
|
||
*
|
||
* @apiDescription 需要token身份，需要安装 Manticore Search 应用
|
||
* @apiVersion 1.0.0
|
||
* @apiGroup search
|
||
* @apiName task
|
||
*
|
||
* @apiParam {String} key 搜索关键词
|
||
* @apiParam {String} [search_type] 搜索类型（text/vector/hybrid，默认：hybrid）
|
||
* @apiParam {Number} [take] 获取数量（默认：20，最大：50）
|
||
*
|
||
* @apiSuccess {Number} ret 返回状态码（1正确、0错误）
|
||
* @apiSuccess {String} msg 返回信息（错误描述）
|
||
* @apiSuccess {Object} data 返回数据
|
||
*/
|
||
public function task()
|
||
{
|
||
$user = User::auth();
|
||
|
||
if (!Apps::isInstalled('manticore')) {
|
||
return Base::retError('Manticore Search 应用未安装');
|
||
}
|
||
|
||
$key = trim(Request::input('key'));
|
||
$searchType = Request::input('search_type', 'hybrid');
|
||
$take = Base::getPaginate(50, 20, 'take');
|
||
|
||
if (empty($key)) {
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
$results = ManticoreTask::search($user->userid, $key, $searchType, $take);
|
||
|
||
// 补充任务完整信息
|
||
$taskIds = array_column($results, 'task_id');
|
||
if (!empty($taskIds)) {
|
||
$tasks = \App\Models\ProjectTask::whereIn('id', $taskIds)
|
||
->get()
|
||
->keyBy('id');
|
||
|
||
foreach ($results as &$item) {
|
||
$taskData = $tasks->get($item['task_id']);
|
||
if ($taskData) {
|
||
$item = array_merge($taskData->toArray(), [
|
||
'relevance' => $item['relevance'] ?? 0,
|
||
'desc_preview' => $item['desc_preview'] ?? null,
|
||
'content_preview' => $item['content_preview'] ?? null,
|
||
]);
|
||
}
|
||
}
|
||
}
|
||
|
||
return Base::retSuccess('success', $results);
|
||
}
|
||
|
||
/**
|
||
* @api {get} api/search/file AI 搜索文件
|
||
*
|
||
* @apiDescription 需要token身份，需要安装 Manticore Search 应用
|
||
* @apiVersion 1.0.0
|
||
* @apiGroup search
|
||
* @apiName file
|
||
*
|
||
* @apiParam {String} key 搜索关键词
|
||
* @apiParam {String} [search_type] 搜索类型（text/vector/hybrid，默认：hybrid）
|
||
* @apiParam {Number} [take] 获取数量（默认：20，最大：50）
|
||
*
|
||
* @apiSuccess {Number} ret 返回状态码（1正确、0错误）
|
||
* @apiSuccess {String} msg 返回信息（错误描述）
|
||
* @apiSuccess {Object} data 返回数据
|
||
*/
|
||
public function file()
|
||
{
|
||
$user = User::auth();
|
||
|
||
if (!Apps::isInstalled('manticore')) {
|
||
return Base::retError('Manticore Search 应用未安装');
|
||
}
|
||
|
||
$key = trim(Request::input('key'));
|
||
$searchType = Request::input('search_type', 'hybrid');
|
||
$take = Base::getPaginate(50, 20, 'take');
|
||
|
||
if (empty($key)) {
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
$results = ManticoreFile::search($user->userid, $key, $searchType, 0, $take);
|
||
|
||
// 补充文件完整信息
|
||
$fileIds = array_column($results, 'file_id');
|
||
if (!empty($fileIds)) {
|
||
$files = File::whereIn('id', $fileIds)
|
||
->get()
|
||
->keyBy('id');
|
||
|
||
$formattedResults = [];
|
||
foreach ($results as $item) {
|
||
$fileData = $files->get($item['file_id']);
|
||
if ($fileData) {
|
||
$formattedResults[] = array_merge($fileData->toArray(), [
|
||
'relevance' => $item['relevance'] ?? 0,
|
||
'content_preview' => $item['content_preview'] ?? null,
|
||
]);
|
||
}
|
||
}
|
||
return Base::retSuccess('success', $formattedResults);
|
||
}
|
||
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
/**
|
||
* @api {get} api/search/message AI 搜索消息
|
||
*
|
||
* @apiDescription 需要token身份，需要安装 Manticore Search 应用
|
||
* @apiVersion 1.0.0
|
||
* @apiGroup search
|
||
* @apiName message
|
||
*
|
||
* @apiParam {String} key 搜索关键词
|
||
* @apiParam {String} [search_type] 搜索类型（text/vector/hybrid，默认：hybrid）
|
||
* @apiParam {Number} [take] 获取数量（默认：20，最大：50）
|
||
* @apiParam {String} [mode] 返回模式（message/position/dialog，默认：message）
|
||
* - message: 返回消息详细信息
|
||
* - position: 只返回消息ID
|
||
* - dialog: 返回对话级数据
|
||
* @apiParam {Number} [dialog_id] 对话ID（筛选指定对话内的消息）
|
||
*
|
||
* @apiSuccess {Number} ret 返回状态码（1正确、0错误）
|
||
* @apiSuccess {String} msg 返回信息（错误描述）
|
||
* @apiSuccess {Object} data 返回数据
|
||
*/
|
||
public function message()
|
||
{
|
||
$user = User::auth();
|
||
|
||
if (!Apps::isInstalled('manticore')) {
|
||
return Base::retError('Manticore Search 应用未安装');
|
||
}
|
||
|
||
$key = trim(Request::input('key'));
|
||
$searchType = Request::input('search_type', 'hybrid');
|
||
$take = Base::getPaginate(50, 20, 'take');
|
||
$mode = Request::input('mode', 'message');
|
||
$dialogId = intval(Request::input('dialog_id', 0));
|
||
|
||
// 验证 mode 参数
|
||
if (!in_array($mode, ['message', 'position', 'dialog'])) {
|
||
$mode = 'message';
|
||
}
|
||
|
||
if (empty($key)) {
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
|
||
// 如果指定了 dialog_id，需要验证用户有权限访问该对话
|
||
if ($dialogId > 0) {
|
||
\App\Models\WebSocketDialog::checkDialog($dialogId);
|
||
}
|
||
|
||
$results = ManticoreMsg::search($user->userid, $key, $searchType, 0, $take, $dialogId);
|
||
|
||
// 根据 mode 返回不同格式的数据
|
||
switch ($mode) {
|
||
case 'position':
|
||
// 只返回消息ID
|
||
$data = array_column($results, 'msg_id');
|
||
return Base::retSuccess('success', compact('data'));
|
||
|
||
case 'dialog':
|
||
// 返回对话级数据
|
||
$list = [];
|
||
$seenDialogs = [];
|
||
foreach ($results as $item) {
|
||
$dialogIdFromResult = $item['dialog_id'];
|
||
// 每个对话只返回一次
|
||
if (isset($seenDialogs[$dialogIdFromResult])) {
|
||
continue;
|
||
}
|
||
$seenDialogs[$dialogIdFromResult] = true;
|
||
|
||
if ($dialog = \App\Models\WebSocketDialog::find($dialogIdFromResult)) {
|
||
$dialogData = array_merge($dialog->toArray(), [
|
||
'search_msg_id' => $item['msg_id'],
|
||
]);
|
||
$list[] = \App\Models\WebSocketDialog::synthesizeData($dialogData, $user->userid);
|
||
}
|
||
}
|
||
return Base::retSuccess('success', ['data' => $list]);
|
||
|
||
case 'message':
|
||
default:
|
||
// 返回消息详细信息（默认行为）
|
||
$msgIds = array_column($results, 'msg_id');
|
||
if (!empty($msgIds)) {
|
||
$msgs = WebSocketDialogMsg::whereIn('id', $msgIds)
|
||
->with(['user' => function ($query) {
|
||
$query->select(User::$basicField);
|
||
}])
|
||
->get()
|
||
->keyBy('id');
|
||
|
||
$formattedResults = [];
|
||
foreach ($results as $item) {
|
||
$msgData = $msgs->get($item['msg_id']);
|
||
if ($msgData) {
|
||
$formattedResults[] = [
|
||
'id' => $msgData->id,
|
||
'msg_id' => $msgData->id,
|
||
'dialog_id' => $msgData->dialog_id,
|
||
'userid' => $msgData->userid,
|
||
'type' => $msgData->type,
|
||
'msg' => $msgData->msg,
|
||
'created_at' => $msgData->created_at,
|
||
'user' => $msgData->user,
|
||
'relevance' => $item['relevance'] ?? 0,
|
||
'content_preview' => $item['content_preview'] ?? null,
|
||
];
|
||
}
|
||
}
|
||
return Base::retSuccess('success', $formattedResults);
|
||
}
|
||
|
||
return Base::retSuccess('success', []);
|
||
}
|
||
}
|
||
}
|
||
|