package org.yeshi.utils.generater.entity;

import org.springframework.data.annotation.Id;
import org.springframework.data.annotation.Transient;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

public class MongoDBDaoData extends BaseData {

    private String baseDaoClass;
    private String daoName;
    private String baseDaoClassName;
    private String entityClassName;

    private ColumnData identityColumn;
    private List<ColumnData> columnDataList;

    public static class Builder {

        private Class baseDaoClass;

        private Class entityClass;

        private String daoPackageName;

        private String daoName;


        public Builder setBaseDaoClass(Class clazz) {
            this.baseDaoClass = clazz;
            return this;
        }

        public Builder setEntityClass(Class clazz) {
            this.entityClass = clazz;
            return this;
        }

        public Builder setDaoPackageName(String daoPackageName) {
            this.daoPackageName = daoPackageName;
            return this;
        }

        public Builder setDaoName(String daoName) {
            this.daoName = daoName;
            return this;
        }


        public MongoDBDaoData create() {

            if (daoName == null) {
                daoName = entityClass.getSimpleName() + "Dao";
            }

            Field[] fields = entityClass.getDeclaredFields();
            String identityColumnName = null;
            List<String> columNamelist = new ArrayList<>();
            for (Field fd : fields) {
                //private/public 非final，static属性才会参与解析
                if (fd.getModifiers() != 1 && fd.getModifiers() != 2) {
                    continue;
                }

                String columName = fd.getName();
                Annotation[] as = fd.getAnnotations();


                if (columName != null) {
                    for (Annotation a : as) {
                        //主键
                        if (a instanceof Id) {
                            identityColumnName = new String(columName);
                            columName = null;
                            break;
                        }
                    }
                }

                for (Annotation a : as) {
                    //不参与解析
                    if (a instanceof Transient || a instanceof java.beans.Transient) {
                        columName = null;
                        break;
                    }
                }


                if (columName != null) {
                    columNamelist.add(columName);
                }
            }

            List<ColumnData> columnDataList = new ArrayList<>();
            for (String columnName : columNamelist) {
                //更新时间不参与更新
                if (!"updateTime".equalsIgnoreCase(columnName)) {
                    columnDataList.add(new ColumnData(columnName));
                }
            }

            MongoDBDaoData data = new MongoDBDaoData();
            data.setBaseDaoClass(baseDaoClass.getName());
            data.setBaseDaoClassName(baseDaoClass.getSimpleName());
            data.setColumnDataList(columnDataList);
            data.setDaoName(daoName);
            data.setEntityClass(entityClass.getName());
            data.setEntityClassName(entityClass.getSimpleName());
            data.setIdentityColumn(new ColumnData(identityColumnName));
            data.setPackageName(daoPackageName);


            return data;
        }
    }


    public String getBaseDaoClass() {
        return baseDaoClass;
    }

    public void setBaseDaoClass(String baseDaoClass) {
        this.baseDaoClass = baseDaoClass;
    }

    public String getDaoName() {
        return daoName;
    }

    public void setDaoName(String daoName) {
        this.daoName = daoName;
    }

    public String getBaseDaoClassName() {
        return baseDaoClassName;
    }

    public void setBaseDaoClassName(String baseDaoClassName) {
        this.baseDaoClassName = baseDaoClassName;
    }

    public String getEntityClassName() {
        return entityClassName;
    }

    public void setEntityClassName(String entityClassName) {
        this.entityClassName = entityClassName;
    }


    public ColumnData getIdentityColumn() {
        return identityColumn;
    }

    public void setIdentityColumn(ColumnData identityColumn) {
        this.identityColumn = identityColumn;
    }


    public List<ColumnData> getColumnDataList() {
        return columnDataList;
    }

    public void setColumnDataList(List<ColumnData> columnDataList) {
        this.columnDataList = columnDataList;
    }


}