/**
 * Copyright (c) 2018 人人开源 All rights reserved.
 *
 * https://www.renren.io
 *
 * 版权所有，侵权必究！
 */

package com.zt.core.aspect;

import com.zt.common.annotation.DataFilter;
import com.zt.common.constant.Constant;
import com.zt.common.db.query.DataScope;
import com.zt.common.exception.ErrorCode;
import com.zt.common.exception.RenException;
import com.zt.core.context.User;
import com.zt.core.context.UserContext;
import org.aspectj.lang.JoinPoint;
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.stereotype.Component;

import java.lang.reflect.Method;
import java.util.Map;

/**
 * 数据过滤，切面处理类
 *
 * @author Mark sunlightcs@gmail.com
 */
@Aspect
@Component("dataFilter")
public class DataFilterAspect {

    @Pointcut("@annotation(com.zt.common.annotation.DataFilter)")
    public void dataFilterCut() {

    }

    @Before("dataFilterCut()")
    public void dataFilter(JoinPoint point) {
        Object params = point.getArgs()[0];
        if (params != null && params instanceof Map) {
            User user = UserContext.getUser();

            // 如果是超级管理员，则不进行数据过滤
            if (user.isSuperAdmin()) {
                return;
            }

            try {
                // 否则进行数据过滤
                Map map = (Map) params;
                String sqlFilter = getSqlFilter(user, point);
                map.put(Constant.Q.SQL_FILTER, new DataScope(sqlFilter));
            } catch (Exception e) {

            }

            return;
        }

        throw new RenException(ErrorCode.DATA_SCOPE_PARAMS_ERROR);
    }

    /**
     * 获取数据过滤的SQL
     */
    private String getSqlFilter(User user, JoinPoint point) throws Exception {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = point.getTarget().getClass()
                .getDeclaredMethod(signature.getName(), signature.getParameterTypes());
        DataFilter dataFilter = method.getAnnotation(DataFilter.class);

        // 获取表的别名
        // String tableAlias = dataFilter.tableAlias();
        // if(StringUtils.isNotBlank(tableAlias)){
        // tableAlias += ".";
        // }

        StringBuilder sqlFilter = new StringBuilder();
        sqlFilter.append(" (");
        //
        // //部门ID列表
        // List<Long> deptIds = user.getDeptIdList();
        // if(CollUtil.isNotEmpty(deptIds)){
        // sqlFilter.append(tableAlias).append(dataFilter.deptId());
        //
        // sqlFilter.append(" in(").append(StringUtils.join(deptIds,
        // ",")).append(")");
        // }
        //
        // //查询本人数据
        // if(CollUtil.isNotEmpty(deptIds)){
        // sqlFilter.append(" or ");
        // }
        // sqlFilter.append(tableAlias).append(dataFilter.userId()).append("=").append(user.getId());
        //
        // sqlFilter.append(")");

        return sqlFilter.toString();
    }
}