117 lines
4.2 KiB
PHP
117 lines
4.2 KiB
PHP
<?php
|
||
|
||
namespace app\cozeai\controller;
|
||
|
||
use app\cozeai\model\Conversation as ConversationModel;
|
||
use app\cozeai\model\Message as MessageModel;
|
||
use think\facade\Env;
|
||
|
||
/**
|
||
* Coze AI 消息控制器
|
||
*/
|
||
class MessageController extends BaseController
|
||
{
|
||
/**
|
||
* 获取用户与AI的对话记录
|
||
*/
|
||
public function getMessages()
|
||
{
|
||
try {
|
||
// 获取用户信息
|
||
$userInfo = request()->userInfo;
|
||
$uid = $userInfo['id'];
|
||
$companyId = $userInfo['companyId'];
|
||
|
||
// 获取会话ID
|
||
$conversation_id = input('conversation_id', '');
|
||
|
||
// 如果没有传入会话ID，则查询用户最新的会话
|
||
if (empty($conversation_id)) {
|
||
// 查询用户是否有会话记录
|
||
$conversation = ConversationModel::where([
|
||
['userId', '=', $uid],
|
||
['companyId', '=', $companyId]
|
||
])->order('create_time', 'desc')->find();
|
||
|
||
// 如果没有会话记录，创建新会话
|
||
if (empty($conversation)) {
|
||
$conversationController = new ConversationController();
|
||
$result = $conversationController->create(true);
|
||
$result = json_decode($result, true);
|
||
if ($result['code'] != 200) {
|
||
return errorJson('创建会话失败：' . $result['msg']);
|
||
}
|
||
|
||
$conversation_id = $result['data']['id'];
|
||
} else {
|
||
$conversation_id = $conversation['conversation_id'];
|
||
}
|
||
} else {
|
||
// 验证会话是否属于当前用户
|
||
$conversation = ConversationModel::where([
|
||
['conversation_id', '=', $conversation_id],
|
||
['userId', '=', $uid],
|
||
['companyId', '=', $companyId]
|
||
])->find();
|
||
|
||
if (empty($conversation)) {
|
||
return errorJson('会话不存在或无权访问');
|
||
}
|
||
}
|
||
|
||
// 分页参数
|
||
$page = input('page', 1);
|
||
$limit = input('limit', 20);
|
||
|
||
// 查询消息记录
|
||
$messages = MessageModel::where('conversation_id', $conversation_id)
|
||
->order('id', 'DESC')
|
||
->page($page, $limit)
|
||
->select()
|
||
->each(function($item) {
|
||
// 格式化时间显示
|
||
$timestamp = $item['created_at'];
|
||
$today = strtotime(date('Y-m-d'));
|
||
$yesterday = strtotime('-1 day', $today);
|
||
$thisYear = strtotime(date('Y-01-01'));
|
||
|
||
if($timestamp >= $today) {
|
||
$item['show_time'] = date('H:i', $timestamp);
|
||
} elseif($timestamp >= $yesterday) {
|
||
$item['show_time'] = '昨天 ' . date('H:i', $timestamp);
|
||
} elseif($timestamp >= $thisYear) {
|
||
$item['show_time'] = date('m-d H:i', $timestamp);
|
||
} else {
|
||
$item['show_time'] = date('Y-m-d H:i', $timestamp);
|
||
}
|
||
|
||
// 根据role设置type
|
||
if ($item['role'] == 'assistant') {
|
||
$item['type'] = 'assistant';
|
||
} else {
|
||
$item['type'] = 'user';
|
||
}
|
||
unset($item['role']);
|
||
|
||
return $item;
|
||
});
|
||
|
||
// 对消息进行倒序处理
|
||
$messages = array_reverse($messages->toArray());
|
||
|
||
// 获取总记录数
|
||
$total = MessageModel::where('conversation_id', $conversation_id)->count();
|
||
|
||
$data = [
|
||
'list' => $messages,
|
||
'total' => $total,
|
||
'conversation_id' => $conversation_id
|
||
];
|
||
|
||
return successJson($data, '获取成功');
|
||
|
||
} catch (\Exception $e) {
|
||
return errorJson('获取对话记录失败：' . $e->getMessage());
|
||
}
|
||
}
|
||
}
|