package com.ruoyi.common.handler;


import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
import com.ruoyi.common.config.IgnoreTableConfig;
import com.ruoyi.common.utils.SecurityUtils;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.LongValue;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.ObjectUtils;

import java.util.Set;

@Component
public class CustomTenantLineHandler implements TenantLineHandler {

    @Override
    public Expression getTenantId() {
        try {
            Long tenantId = SecurityUtils.getLoginUser().getTenantId();
            return new LongValue(tenantId);
        } catch (Exception e) {
            return null;
        }
    }

    /**
     * 线程级临时忽略开关：某些 insert 场景下 jsqlparser 解析会抛
     * ClassCastException: SetOperationList cannot be cast to PlainSelect，
     * 可在该线程内临时跳过租户插件（tenant_id 由 MybatisHandler 自动填充保证）。
     */
    public static final ThreadLocal<Boolean> TEMP_IGNORE = new ThreadLocal<>();

    @Override
    public boolean ignoreTable(String tableName) {
        if (Boolean.TRUE.equals(TEMP_IGNORE.get())) {
            return true;
        }
        // 指定不需要添加过滤条件的表
        Set<String> ignoreTables = IgnoreTableConfig.IGNORE_TABLES;
        return ignoreTables.contains(tableName);
    }

    @Override
    public String getTenantIdColumn() {
        // 指定租户 ID 字段名称，比如 "station_id"
        return "tenant_id";
    }
}