<?php

namespace app\common\model;

use think\Model;

/**
 * 流量池分组成员表模型类
 * 表名：ck_traffic_pool_group_member
 * 用途：手动添加到分组的成员（ruleType=2时使用）
 */
class TrafficPoolGroupMember extends Model
{
    // 设置数据表名
    protected $name = 'traffic_pool_group_member';

    // 主键
    protected $pk = 'id';

    // 自动写入时间戳
    protected $autoWriteTimestamp = 'createTime';
    protected $createTime = 'createTime';
    protected $updateTime = false;

    // 添加方式常量
    const ADD_TYPE_MANUAL = 1;  // 手动
    const ADD_TYPE_IMPORT = 2;  // 批量导入

    /**
     * 关联分组
     */
    public function group()
    {
        return $this->belongsTo(TrafficPoolGroup::class, 'groupId', 'id');
    }

    /**
     * 关联公司流量详情
     */
    public function poolCompany()
    {
        return $this->belongsTo(TrafficPoolCompany::class, 'poolCompanyId', 'id');
    }

    /**
     * 批量添加成员到分组
     * @param int $groupId
     * @param array $poolCompanyIds
     * @param int $companyId
     * @param int $operatorId
     * @param int $addType
     * @return int 成功添加的数量
     */
    public static function batchAddMembers(int $groupId, array $poolCompanyIds, int $companyId, int $operatorId = null, int $addType = self::ADD_TYPE_MANUAL)
    {
        $count = 0;
        $time = time();
        
        // 获取已存在的成员
        $existIds = self::where('groupId', $groupId)
            ->whereIn('poolCompanyId', $poolCompanyIds)
            ->where('isDel', 0)
            ->column('poolCompanyId');
        
        // 获取要添加的流量详情
        $poolCompanies = TrafficPoolCompany::whereIn('id', $poolCompanyIds)
            ->where('companyId', $companyId)
            ->where('isDel', 0)
            ->column('identifier', 'id');
        
        $insertData = [];
        foreach ($poolCompanyIds as $poolCompanyId) {
            if (in_array($poolCompanyId, $existIds)) {
                continue; // 跳过已存在的
            }
            if (!isset($poolCompanies[$poolCompanyId])) {
                continue; // 跳过不存在的
            }
            
            $insertData[] = [
                'groupId' => $groupId,
                'poolCompanyId' => $poolCompanyId,
                'identifier' => $poolCompanies[$poolCompanyId],
                'companyId' => $companyId,
                'addType' => $addType,
                'operatorId' => $operatorId,
                'createTime' => $time,
                'isDel' => 0,
            ];
        }
        
        if (!empty($insertData)) {
            (new self())->saveAll($insertData);
            $count = count($insertData);
            
            // 更新分组成员数量缓存
            TrafficPoolGroup::where('id', $groupId)->setInc('memberCount', $count);
        }
        
        return $count;
    }

    /**
     * 批量移除成员
     * @param int $groupId
     * @param array $poolCompanyIds
     * @return int
     */
    public static function batchRemoveMembers(int $groupId, array $poolCompanyIds)
    {
        $count = self::where('groupId', $groupId)
            ->whereIn('poolCompanyId', $poolCompanyIds)
            ->where('isDel', 0)
            ->update([
                'isDel' => 1,
                'deleteTime' => time()
            ]);
        
        if ($count > 0) {
            // 更新分组成员数量缓存
            TrafficPoolGroup::where('id', $groupId)->setDec('memberCount', $count);
        }
        
        return $count;
    }
}