package cc.mrbird.febs.server.hr.util;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import com.baomidou.mybatisplus.annotation.TableName;

/**
 * 实体的工具类
 * 
 * @author zhangshuaipeng
 *
 */
public class ModelUtil {
    public static Map<String, Field[]> modelFields = new HashMap<String, Field[]>();

    public static Map<String, String> modelJson = new HashMap<String, String>();

    /**
     * 判断实体不为空
     * 
     * @param obj
     * @return
     */
    public static Boolean isNotNull(Object obj) {
        if (obj != null) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * 获取实体对应的数据表名
     * 
     * @param clazz
     *            要获取表名的实体类
     * @param equalEntity
     *            为true时返回实体的名称，否则返回注解的表名
     * @return
     */
    public static String getTableName(Class<?> clazz, boolean equalEntity) {
        String tableName = clazz.getSimpleName();

        if (!equalEntity) {
        	TableName annotation = (TableName)clazz.getAnnotation(TableName.class);
            if (!StringUtils.isEmpty(annotation.value())) {
                tableName = annotation.value();
            }
        }
        return tableName;
    }

    /**
     * 获取指定字段映射的数据表的列名
     * 
     * @param f
     *            指定的字段
     * @param isAnnotation
     *            为true表示取字段使用@Column进行注解的名称，当没有注解时取字段名
     * @return
     */
//    public static String getColumnName(Field f, boolean isAnnotation) {
//        String columnName = f.getName();
//        if (isAnnotation) {
//            Annotation[] annotations = f.getAnnotations();
//            for (int i = 0; i < annotations.length; i++) {
//                if (annotations[i] instanceof Column) {
//                    // 数据列
//                    Column column = (Column) annotations[i];
//                    if (StringUtils.isNotEmpty(column.name())) {
//                        columnName = column.name();
//                    }
//                } else if (annotations[i] instanceof JoinColumn) {
//                    // 关联列
//                    JoinColumn joinColumn = (JoinColumn) annotations[i];
//                    if (StringUtils.isNotEmpty(joinColumn.name())) {
//                        columnName = joinColumn.name();
//                    }
//                }
//
//            }
//        }

//        return columnName;
//    }

    /**
     * 得到类的属性集合
     * 
     * @param c
     * @param itself
     *            是否是自身的字段
     * @return
     */
    public static Field[] getClassFields(Class<?> c, boolean itself) {
        if (itself) {
            if (modelFields.get(c.getName()) != null) {
                return modelFields.get(c.getName());
            } else {
                Field[] fields = c.getDeclaredFields();
                modelFields.put(c.getName(), fields);
                return fields;
            }
        } else {
            if (modelFields.get(c.getName()) != null) {
                return modelFields.get(c.getName());
            } else {
                List<Field> fields = new ArrayList<Field>();
                getAllDeclaredFields(c, fields);
                Field[] fies = new Field[fields.size()];
                fields.toArray(fies);
                modelFields.put(c.getName(), fies);
                return fies;
            }
        }
    }

    /**
     * 从c类中取得全部字段,包括父类
     * 
     * @param c
     * @param fields
     */
    public static void getAllDeclaredFields(Class<?> c, List<Field> fields) {
        Field[] fies = c.getDeclaredFields();
        Collections.addAll(fields, fies);
        Class<?> parent = c.getSuperclass();
        if (parent != Object.class) {
            getAllDeclaredFields(parent, fields);
        } else {
            return;
        }
    }

    /**
     * 得到类的主键字段
     * 
     * @param clazz
     * @return
     */
//    public static String getClassPkName(Class<?> clazz) {
//        Field[] fields = getClassFields(clazz, false);
//        String pkName = "";
//        for (Field f : fields) {
//            FieldInfo fieldInfo = f.getAnnotation(FieldInfo.class);
//            if (isNotNull(fieldInfo)) {
//                if ("ID".equals(fieldInfo.type())) {
//                    pkName = f.getName();
//                    break;
//                }
//            }
//        }
//        return pkName;
//    }

}