xiaoai/php_server/app/api/logic/sv/SvBaseLogic.php

<?php

namespace app\api\logic\sv;

use app\api\logic\ApiLogic;
use app\common\model\sv\SvAccount;
use app\common\model\sv\SvAccountContact;
use app\common\model\sv\SvDevice;
use app\common\model\sv\SvRobot;
use app\common\model\sv\SvRobotKeyword;
use app\common\model\sv\SvReplyStrategy;
use app\common\service\FileService;
use app\api\logic\service\TokenLogService;
use app\common\enum\user\AccountLogEnum;
use app\common\model\ChatPrompt;
use think\facade\Queue;

/**
 * SvBaseLogic
 * @desc 基础逻辑
 * @author Qasim
 */
class SvBaseLogic extends ApiLogic
{


    /**
     * @desc 获取账号信息
     * @param string $account
     * @return bool|SvAccount
     */
    protected static function accountInfo(string $account, bool $check = true): bool|SvAccount
    {
        $account = SvAccount::where('account', $account)->when($check, function ($query) {
            $query->where('user_id', self::$uid);
        })->findOrEmpty();
        if ($account->isEmpty()) {
            self::setError('账号不存在');
            return false;
        }
        return $account;
    }


    /**
     * @desc 获取好友信息
     * @param string $account
     * @param string $friendId
     * @return bool|SvAccountContact
     */
    protected static function friendInfo(string $account, string $friendId, $accountType = 1): bool|SvAccountContact
    {
        $friend = SvAccountContact::where('account', $account)->where('friend_id', $friendId)->where('account_type', $accountType)->findOrEmpty();
        if ($friend->isEmpty()) {
            self::setError('好友不存在');
            return false;
        }
        return $friend;
    }

    /**
     * @desc 获取设备信息
     * @param string $deviceCode
     * @return bool|SvDevice
     */
    protected static function deviceInfo(string $deviceCode, bool $check = true): bool|SvDevice
    {
        $device = SvDevice::where('device_code', $deviceCode)->when($check, function ($query) {
            $query->where('user_id', self::$uid);
        })->findOrEmpty();

        if ($device->isEmpty()) {
            self::setError('设备不存在');
            return false;
        }
        return $device;
    }


    /**
     * @desc 正则匹配关键词
     * @param SvRobot $robot
     * @param array $request
     * @return bool
     */
    protected static function regularMatchKeyword(SvRobot $robot, array $request)
    {

        $match = false;
        // 获取机器人设置的正关键词
        SvRobotKeyword::where('robot_id', $robot->id)->select()->each(function ($item) use ($request, &$match) {

            // 模糊匹配
            if ($item->match_type == 0) {
                if (str_contains($request['message'], $item->keyword)) {

                    self::parseMessage($request, $item->reply);
                    $match = true;
                }
            } else {
                if ((string)$item->keyword === $request['message']) {

                    self::parseMessage($request, $item->reply);
                    $match = true;
                }
            }
        });

        return $match;
    }


    /**
     * @desc 解析AI提示词
     * @param array $request
     * @param array $content
     * @return void
     */
    protected static function parseAiPrompt(SvRobot $robot, array $request, array $logs): void
    {

        //检查扣费
        $unit = TokenLogService::checkToken(self::$uid, 'sv_chat');

        //获取提示词
        $keyword = ChatPrompt::where('id', 12)->value('prompt_text') ?? '';

        if (!$keyword) {

            message("提示词不存在");
        }
        $keyword = str_replace(
            ['企业背景', '角色设定', '用户备注', '用户标签', '咨询', '最近对话记录', '用户发送的内容'],
            [$robot->company_background, $robot->description, $request['friend_remark'], "", "", json_encode($logs, JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES), $request['message']],
            $keyword
        );
        $task_id = generate_unique_task_id();

        // 检查是否挂载知识库
        $bind = \app\common\model\knowledge\KnowledgeBind::where('data_id', $robot->id)->where('user_id', self::$uid)->where('type', 1)->limit(1)->find();
        $knowledge = [];
        if (!empty($bind)) {
            $knowledge = \app\common\model\knowledge\Knowledge::where('id', $bind['kid'])->limit(1)->find();
            if (empty($knowledge)) {
                message("挂载的知识库不存在");
            }
            $knowledge['task_id'] = $task_id;
        }

       /* $request = [
            'user_id' => self::$uid,
            'task_id' => $task_id,
            'account' => $request['account'],
            'friend_id' => $request['friend_id'],
            'friend_remark' => $request['friend_remark'],
            'device_code' => $request['device_code'],
            'message' => $request['message'],
            'message_id' => $request['message_id'],
            'chat_type' => AccountLogEnum::TOKENS_DEC_AI_WECHAT,
            'now'       => time(),
            'messages' => array_merge([['role' => 'system', 'content' => $keyword]], $logs),
            'knowledge' => $knowledge,
        ];

        // 任务数据
        $data = [
            'account' => $request['account'],
            'friend_id' => $request['friend_id'],
            'device_code' => $request['device_code'],
            'task_id' => $request['task_id'],
            'uid' => self::$uid,
            'request' => $request,
        ];

        // 推送到队列
        Queue::push('app\common\Jobs\SvAIMessageJob@handle', $data);*/
    }


    /**
     * @desc 解析消息
     * @param array $request
     * @param array $content
     * @return void
     */
    protected static function parseMessage(array $request, array $content)
    {
        foreach ($content as $item) {

            $send = true;

            switch ((int)$item['type']) {

                case 0: //文本

                    // 推送消息
                    $request['message_type'] = 1;
                    $request['message'] = str_replace('${remark}', $request['friend_remark'], $item['content']);
                    break;

                case 1: //图片

                    // 推送消息
                    $request['message'] = FileService::getFileUrl($item['content']);
                    $request['message_type'] = 2;
                    break;

                default:
                    $send = false;
            }

            if ($send) {
                self::send($request);
            }
        }
    }

    /**
     * @desc 正则匹配停止AI回复
     * @param SvRobot $robot
     * @param string $message
     * @return bool
     */
    protected static function regularMatchStopAI(SvReplyStrategy $reply, array $request)
    {

        $stop = false;

        $keywords = explode(';', $reply->stop_keywords);

        // 获取机器人设置的正关键词
        foreach ($keywords as $keyword) {

            if ((string)$keyword === $request['message']) {

                $stop = true;

                break;
            }
        }

        return $stop;
    }

    /**
     * @desc 发送消息
     * @param array $request
     * @return boov
     */
    protected static function send(array $request)
    {
        sleep(5);
        \app\common\service\ToolsService::Sv()->push($request);
    }
}