<?php

declare(strict_types=1);
/**
 * +----------------------------------------------------------------------
 * | ThinkAdmin Plugin
 * +----------------------------------------------------------------------
 * | Copyright (c) 2014~2026 ThinkAdmin [ thinkadmin.top ]
 * +----------------------------------------------------------------------
 * | Official Website: https://thinkadmin.top
 * +----------------------------------------------------------------------
 * | Licensed: https://mit-license.org
 * | Disclaimer: https://thinkadmin.top/disclaimer
 * | Vip Rights: https://thinkadmin.top/vip-introduce
 * +----------------------------------------------------------------------
 * | Gitee Repository: https://gitee.com/zoujingli/ThinkAdmin
 * | Github Repository: https://github.com/zoujingli/ThinkAdmin
 * +----------------------------------------------------------------------
 */

namespace think\admin\helper;

use think\admin\Helper;
use think\admin\model\QueryFactory;
use think\db\BaseQuery;
use think\db\exception\DbException;
use think\db\Query;
use think\Model;

/**
 * 数据更新管理器.
 * @class SaveHelper
 */
class SaveHelper extends Helper
{
    /**
     * 逻辑器初始化.
     * @param array $edata 表单扩展数据
     * @param string $field 数据对象主键
     * @param mixed $where 额外更新条件
     * @return bool|void
     * @throws DbException
     */
    public function init(BaseQuery|Model|string $dbQuery, array $edata = [], string $field = '', $where = [])
    {
        $query = QueryFactory::build($dbQuery);
        if (!$query instanceof Query) {
            throw new \InvalidArgumentException('SaveHelper only supports relational Query instances.');
        }
        $field = $field ?: ($query->getPk() ?: 'id');
        $edata = $edata ?: $this->app->request->post();
        $value = $this->app->request->post($field);

        // 主键限制处理
        if (!isset($where[$field]) && !is_null($value)) {
            $query->whereIn($field, str2arr(strval($value)));
            if (isset($edata)) {
                unset($edata[$field]);
            }
        }

        // 前置回调处理
        if ($this->class->callback('_save_filter', $query, $edata) === false) {
            return false;
        }

        // 检查原始数据
        $query->master()->where($where)->update($edata);

        // 模型自定义事件回调
        $model = $query->getModel();
        if ($model instanceof \think\admin\Model) {
            $model->onAdminSave(strval($value));
        }

        // 结果回调处理
        $result = true;
        if ($this->class->callback('_save_result', $result, $model) === false) {
            return $result;
        }

        // 回复前端结果
        $this->class->success('数据保存成功！', '');
    }
}