package com.ruoyi.framework.aspectj;

import com.ruoyi.common.annotation.BranchScopeWjlz;
import com.ruoyi.common.annotation.TableScopeAjk;
import com.ruoyi.common.config.UploadConfig;
import com.ruoyi.common.core.domain.BaseEntity;
import com.ruoyi.common.core.domain.entity.SysRole;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.entity.BaseSJFZEntity;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.bean.ApplicationContextUtils;
import com.ruoyi.system.domain.SysConfig;
import com.ruoyi.system.mapper.SysConfigMapper;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;

import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 2025-07-25 15:22:00
 * 動態(tài)切換數(shù)據(jù)表
 *
 * @author wangwei
 */
@Aspect
@Component
@Service
public class TableScopeAspectAjk {
    public static final String DATA_SCOPE = "tableScopeAjk";


    // 配置織入點(diǎn)-登錄用戶的所屬組織
    @Pointcut("@annotation(com.ruoyi.common.annotation.TableScopeAjk)")
    public void branchScopePointCut() {
    }

    //登錄用戶的所屬組織
    @Before("branchScopePointCut()")
    public void doBefore(JoinPoint point) throws Throwable {
        handleDataScope(point);
    }


    /***
     * 當(dāng)前登錄組織
     * @param joinPoint
     */
    protected void handleDataScope(final JoinPoint joinPoint) {
        // 獲得注解
        TableScopeAjk controllerBranchScope = getAnnotationLog(joinPoint);
        if (controllerBranchScope == null) {
            return;
        }
        branchScopeFilter(joinPoint);
    }


    /**
     * 數(shù)據(jù)范圍過濾-當(dāng)前用戶所屬組織
     *
     * @param joinPoint 切點(diǎn)
     */
    public static void branchScopeFilter(JoinPoint joinPoint) {
        SysConfigMapper sysConfigMapper = ApplicationContextUtils.getBean(SysConfigMapper.class);
        String tableName = "";

        SysConfig sysConfig = new SysConfig();
        sysConfig.setRemark("AJKTABLENAME");
        List<SysConfig> sysConfigList = sysConfigMapper.selectConfigList(sysConfig);
        if (sysConfigList.size() < 1) {
            tableName = "默認(rèn)配置表";
        } else {
            List<SysConfig> filteredList = sysConfigList.stream()
                    .filter(config -> "Y".equals(config.getConfigType()))
                    .collect(Collectors.toList());
            if (filteredList.size() < 1) {
                tableName = "默認(rèn)配置表";
            } else {
                tableName = filteredList.get(0).getConfigType();
            }
        }

        Object params = joinPoint.getArgs()[0];
        if (StringUtils.isNotNull(params) && params instanceof BaseSJFZEntity) {
            BaseSJFZEntity baseEntity = (BaseSJFZEntity) params;
            baseEntity
                    .getParams()
                    .put(
                            DATA_SCOPE,
                            tableName
                            //TODO 動態(tài)獲取 傳參 baseEntity.getTABLENAME()
                    );
        }
    }

    /**
     * 是否存在注解，如果存在就獲取
     */
    private TableScopeAjk getAnnotationLog(JoinPoint joinPoint) {
        Signature signature = joinPoint.getSignature();
        MethodSignature methodSignature = (MethodSignature) signature;
        Method method = methodSignature.getMethod();
        if (method != null) {
            return method.getAnnotation(TableScopeAjk.class);
        }
        return null;
    }
}

package com.ruoyi.common.annotation;

import java.lang.annotation.*;

/** 組織過濾注解 */

/***
 *組織數(shù)據(jù)過濾
 * 1.當(dāng)前登錄組織
 * wangwei
 * 2023-11-07 09:08:00
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface TableScopeAjk {
    /**
     * 案件表 名稱
     */
    public String TABLENAME() default "";
}

@TableScopeAjk
public List<Tb....> selectTb....ByList(Tb.... tb....);

${params.tableScopeAjk}