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

package com.zt.core.db.annotation;

import cn.hutool.core.date.DateUtil;
import com.zt.common.annotation.QueryParam;
import com.zt.common.constant.Constant;
import com.zt.common.db.query.QueryFilter;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import org.apache.commons.lang3.StringUtils;
import org.springframework.core.MethodParameter;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.annotation.RequestParamMapMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

/**
 * 有@LoginUser注解的方法参数，注入当前登录用户
 *
 * @author hehz
 */
@Component("queryParamResolver")
public class QueryParamMethodArgumentResolver extends RequestParamMapMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.getParameterType().isAssignableFrom(QueryFilter.class)
                && parameter.hasParameterAnnotation(QueryParam.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer container, NativeWebRequest request,
            WebDataBinderFactory factory) throws Exception {
        Map<String, Object> map = (Map<String, Object>) super.resolveArgument(parameter, container, request, factory);

        List<com.zt.common.db.query.QueryParam> params = new ArrayList<>();

        List<ApiImplicitParam> apiImplicitParamList = new ArrayList<>();
        // 1、获取ApiImplicitParams和ApiImplicitParam注解
        for (Annotation annotation : parameter.getMethod().getDeclaredAnnotations()) {
            if (annotation instanceof ApiImplicitParams) {
                ApiImplicitParams apiImplicitParams = (ApiImplicitParams) annotation;
                apiImplicitParamList.addAll(Arrays.asList(apiImplicitParams.value()));
                break;
            } else if (annotation instanceof ApiImplicitParam) {
                apiImplicitParamList.add((ApiImplicitParam) annotation);
            }
        }
        // 2、获取ApiImplicitParam指定的参数
        for (ApiImplicitParam implicitParam : apiImplicitParamList) {
            String type = implicitParam.dataType();
            Object value = map.get(implicitParam.name());
            if (value != null && StringUtils.isNotBlank(value.toString())) {
                value = getValue(type, value.toString().trim());
            }
            params.add(new com.zt.common.db.query.QueryParam(implicitParam.name(), value, implicitParam.format()));
        }
        return new QueryFilter(params);
    }

    private static Object getValue(String type, String valStr) {
        Object o = null;
        if (type.equals(Constant.QT.INT)) {
            o = Integer.parseInt(valStr);
        } else if (type.equalsIgnoreCase("short")) {
            o = Short.parseShort(valStr);
        } else if (type.equalsIgnoreCase(Constant.QT.LONG)) {
            o = Long.parseLong(valStr);
        } else if (type.equalsIgnoreCase("float")) {
            o = Float.parseFloat(valStr);
        } else if (type.equalsIgnoreCase(Constant.QT.DOUBLE)) {
            o = Double.parseDouble(valStr);
        } else if (type.equalsIgnoreCase("boolean")) {
            o = Boolean.parseBoolean(valStr);
        } else if (type.equals(Constant.QT.STRING)) {
            o = valStr;
        } else if ("varchar".equalsIgnoreCase(type)) {
            o = valStr;
        } else if ("number".equalsIgnoreCase(type)) {
            o = Double.parseDouble(valStr);
        } else if ("date".equalsIgnoreCase(type)) {
            try {
                o = DateUtil.parse(valStr);
            } catch (Exception e) {
            }
        } else {
            o = valStr;
        }
        return o;
    }
}