<?php

namespace App\Service\common\ai;

use App\BaseApp\BaseService;
use App\Enum\AiGenerationSceneEnum;
use App\Enum\AiGenerationStatusEnum;
use App\Models\ai\AiGenerationModel;

/**
 * AI 生成历史落库：开始 / 成功 / 失败
 * 为什么单独抽：多场景复用同一流水表，业务只关心 begin/finish
 */
class AiGenerationLogService extends BaseService
{
    /**
     * 创建一条「进行中」记录，返回主键 ID
     *
     * @param array{
     *   scene?:string,
     *   name?:string,
     *   provider?:string,
     *   model?:string,
     *   api_key_id?:int,
     *   input_snapshot?:array|null,
     *   admin_id?:int
     * } $params
     */
    public function begin(array $params): int
    {
        $now = time();
        $scene = trim((string) ($params['scene'] ?? AiGenerationSceneEnum::CODE_GENERATION->value));
        $adminId = (int) ($params['admin_id'] ?? $this->userId);
        return (int) AiGenerationModel::insertGetId([
            'admin_id' => $adminId,
            'scene' => $scene,
            'name' => mb_substr(trim((string) ($params['name'] ?? '')), 0, 128),
            'provider' => trim((string) ($params['provider'] ?? '')),
            'model' => trim((string) ($params['model'] ?? '')),
            'status' => AiGenerationStatusEnum::RUNNING->value,
            'api_key_id' => (int) ($params['api_key_id'] ?? 0),
            'prompt_tokens' => 0,
            'completion_tokens' => 0,
            'total_tokens' => 0,
            'usage_json' => null,
            'started_at' => $now,
            'finished_at' => 0,
            'duration_ms' => 0,
            'error_msg' => '',
            'input_snapshot' => $this->encodeJson($params['input_snapshot'] ?? null),
            'result_json' => null,
            'raw_response' => null,
            'created_at' => $now,
            'updated_at' => 0,
            'deleted_at' => 0,
        ]);
    }

    /**
     * 标记成功并写入结果 / token
     *
     * @param array{
     *   provider?:string,
     *   model?:string,
     *   api_key_id?:int,
     *   usage?:array,
     *   result_json?:array|null,
     *   raw_response?:string|null,
     *   name?:string
     * } $payload
     */
    public function markSuccess(int $id, array $payload = []): bool
    {
        if ($id <= 0) {
            return false;
        }
        $row = AiGenerationModel::where('id', $id)->where('deleted_at', 0)->first();
        if (!$row) {
            return false;
        }
        $usage = is_array($payload['usage'] ?? null) ? $payload['usage'] : [];
        $finishedAt = time();
        $startedAt = (int) ($row->getRawOriginal('started_at') ?? 0);
        $data = [
            'status' => AiGenerationStatusEnum::SUCCESS->value,
            'finished_at' => $finishedAt,
            'duration_ms' => $this->calcDurationMs($startedAt, $finishedAt),
            'prompt_tokens' => (int) ($usage['prompt_tokens'] ?? 0),
            'completion_tokens' => (int) ($usage['completion_tokens'] ?? 0),
            'total_tokens' => (int) ($usage['total_tokens'] ?? 0),
            'usage_json' => $this->encodeJson($usage ?: null),
            'result_json' => $this->encodeJson($payload['result_json'] ?? null),
            'raw_response' => $this->clipText((string) ($payload['raw_response'] ?? ''), 65000),
            'error_msg' => '',
            'updated_at' => $finishedAt,
        ];
        if (!empty($payload['provider'])) {
            $data['provider'] = (string) $payload['provider'];
        }
        if (!empty($payload['model'])) {
            $data['model'] = (string) $payload['model'];
        }
        if (array_key_exists('api_key_id', $payload)) {
            $data['api_key_id'] = (int) $payload['api_key_id'];
        }
        if (!empty($payload['name'])) {
            $data['name'] = mb_substr(trim((string) $payload['name']), 0, 128);
        }
        return (bool) AiGenerationModel::where('id', $id)->update($data);
    }

    /**
     * 标记失败
     */
    public function markFail(int $id, string $errorMsg, array $payload = []): bool
    {
        if ($id <= 0) {
            return false;
        }
        $row = AiGenerationModel::where('id', $id)->where('deleted_at', 0)->first();
        if (!$row) {
            return false;
        }
        $finishedAt = time();
        $startedAt = (int) ($row->getRawOriginal('started_at') ?? 0);
        $data = [
            'status' => AiGenerationStatusEnum::FAIL->value,
            'finished_at' => $finishedAt,
            'duration_ms' => $this->calcDurationMs($startedAt, $finishedAt),
            'error_msg' => mb_substr(trim($errorMsg), 0, 1000),
            'raw_response' => $this->clipText((string) ($payload['raw_response'] ?? ''), 65000),
            'updated_at' => $finishedAt,
        ];
        if (!empty($payload['provider'])) {
            $data['provider'] = (string) $payload['provider'];
        }
        if (!empty($payload['model'])) {
            $data['model'] = (string) $payload['model'];
        }
        if (array_key_exists('api_key_id', $payload)) {
            $data['api_key_id'] = (int) $payload['api_key_id'];
        }
        $usage = is_array($payload['usage'] ?? null) ? $payload['usage'] : [];
        if (!empty($usage)) {
            $data['prompt_tokens'] = (int) ($usage['prompt_tokens'] ?? 0);
            $data['completion_tokens'] = (int) ($usage['completion_tokens'] ?? 0);
            $data['total_tokens'] = (int) ($usage['total_tokens'] ?? 0);
            $data['usage_json'] = $this->encodeJson($usage);
        }
        return (bool) AiGenerationModel::where('id', $id)->update($data);
    }

    /**
     * 分页列表（管理端生成记录）
     */
    public function getPageListForAdmin(): array
    {
        $this->model = AiGenerationModel::class;
        $this->selectField = [
            'id', 'admin_id', 'scene', 'name', 'provider', 'model', 'status',
            'api_key_id', 'prompt_tokens', 'completion_tokens', 'total_tokens',
            'started_at', 'finished_at', 'duration_ms', 'error_msg', 'created_at',
        ];
        $this->queryField = [
            'scene' => '=',
            'status' => '=',
            'provider' => '=',
            'name' => 'like',
        ];
        $this->orderBy = [
            'name' => 'id',
            'sort' => 'desc',
        ];
        // 重置 where，避免单例污染
        $this->where = [['deleted_at', '=', 0]];
        $list = $this->getPageList();
        foreach ($list['items'] as &$item) {
            $status = (int) ($item['status'] ?? 0);
            $scene = (string) ($item['scene'] ?? '');
            $item['status_text'] = AiGenerationStatusEnum::tryFrom($status)?->description() ?? '未知';
            $item['scene_text'] = AiGenerationSceneEnum::tryFrom($scene)?->description() ?? $scene;
        }
        unset($item);
        return $list;
    }

    /**
     * 详情（含 input/result，不含超大 raw 时可按需）
     */
    public function getDetailForAdmin(int $id): array
    {
        $row = AiGenerationModel::where('id', $id)->where('deleted_at', 0)->first();
        if (!$row) {
            $this->utils->notFound('记录不存在');
        }
        $data = $row->toArray();
        $status = (int) ($data['status'] ?? 0);
        $scene = (string) ($data['scene'] ?? '');
        $data['status_text'] = AiGenerationStatusEnum::tryFrom($status)?->description() ?? '未知';
        $data['scene_text'] = AiGenerationSceneEnum::tryFrom($scene)?->description() ?? $scene;
        return $data;
    }

    private function calcDurationMs(int $startedAt, int $finishedAt): int
    {
        if ($startedAt <= 0 || $finishedAt <= 0 || $finishedAt < $startedAt) {
            return 0;
        }
        return ($finishedAt - $startedAt) * 1000;
    }

    private function encodeJson(mixed $value): ?string
    {
        if ($value === null) {
            return null;
        }
        if (is_string($value)) {
            return $value;
        }
        return json_encode($value, JSON_UNESCAPED_UNICODE);
    }

    private function clipText(string $text, int $max): string
    {
        if ($text === '' || mb_strlen($text) <= $max) {
            return $text;
        }
        return mb_substr($text, 0, $max);
    }
}