package org.yeshi.utils.generater.vo.service;

import org.springframework.stereotype.Service;
import org.yeshi.utils.FreemarkerUtils;
import org.yeshi.utils.StringUtil;
import org.yeshi.utils.bean.BeanUtil;
import org.yeshi.utils.generater.entity.BaseData;
import org.yeshi.utils.generater.entity.ClassInfo;
import org.yeshi.utils.generater.util.EntityUtil;

import javax.annotation.Resource;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

public class ServiceImplInfoVO extends BaseData {

    private List<String> importList;
    private ClassInfo service;
    private ClassInfo dao;
    //方法列表
    private List<ServiceMetodInfoVO> metodInfoList;
    private ClassInfo daoQuery;
    private ClassInfo query;


    public static class Builder {
        private Class entity;
        private String packageName;
        private ClassInfo dao;
        private ClassInfo service;
        private ClassInfo query;
        private ClassInfo daoQuery;
        private ClassInfo exception;

        public Builder setException(ClassInfo exception) {
            this.exception = exception;
            return this;
        }

        public Builder setEntity(Class entity) {
            this.entity = entity;
            return this;
        }

        public Builder setPackageName(String packageName) {
            this.packageName = packageName;
            return this;
        }

        public Builder setDao(ClassInfo dao) {
            this.dao = dao;
            return this;
        }

        public Builder setService(ClassInfo service) {
            this.service = service;
            return this;
        }

        public Builder setQuery(ClassInfo query) {
            this.query = query;
            return this;
        }

        public Builder setDaoQuery(ClassInfo daoQuery) {
            this.daoQuery = daoQuery;
            return this;
        }

        private void validParams() throws Exception {
            if (entity == null) {
                throw new Exception("entity不能为空");
            }
            if (packageName == null) {
                throw new Exception("packageName不能为空");
            }
            if (dao == null) {
                throw new Exception("dao不能为空");
            }
            if (service == null) {
                throw new Exception("service不能为空");
            }
            if (query == null) {
                throw new Exception("query不能为空");
            }
            if (daoQuery == null) {
                throw new Exception("daoQuery不能为空");
            }
        }

        private String getMethodContentTemplatePath(String type) {
            String basePath = "generater/service/method/impl/";
            if (dao.getName().endsWith("Mapper")) {
                basePath += "mybatis/";
            } else {
                basePath += "mongo/";
            }

            switch (type) {
                case "list":
                    return basePath + "list.ftl";
                case "count":
                    return basePath + "count.ftl";
                case "add":
                    return basePath + "add.ftl";
                case "save":
                    return basePath + "save.ftl";
                case "get":
                    return basePath + "selectByPrimaryKey.ftl";
                case "update":
                    return basePath + "updateSlective.ftl";
                case "delete":
                    return basePath + "delete.ftl";
            }
            return null;
        }

        public ServiceImplInfoVO build() throws Exception {
            validParams();
            Field identifyId = EntityUtil.getIdentifyId(entity);
            if (identifyId == null) {
                throw new Exception("尚未找到主键属性");
            }

            ServiceImplInfoVO serviceData = new ServiceImplInfoVO();
            serviceData.setDao(dao);
            serviceData.setService(service);
            serviceData.setEntity(new ClassInfo(entity.getSimpleName(), entity.getName()));
            serviceData.setPackageName(packageName);
            serviceData.setDaoQuery(daoQuery);
            serviceData.setQuery(query);
            List<String> importList = new ArrayList<>();
            //设置接口
            List<ServiceMetodInfoVO> metodInfoVOList = new ArrayList<>();
            /*******添加方法开始*******/
            List<String> params = null;
            ServiceMetodInfoVO metodInfo = null;
            //list方法
            params = new ArrayList<>();
            params.add(query.getName() + " " + StringUtil.firstCharToLower(query.getName()));
            params.add(" int page");
            params.add(" int pageSize");
            metodInfo = new ServiceMetodInfoVO("public", String.format("List<%s>", serviceData.getEntity().getName()), "list", StringUtil.concat(params, ","));
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("list"))));
            metodInfoVOList.add(metodInfo);
            //count方法
            params = new ArrayList<>();
            params.add(query.getName() + " " + StringUtil.firstCharToLower(query.getName()));
            metodInfo = new ServiceMetodInfoVO("public", "long", "count", StringUtil.concat(params, ","));
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("count"))));
            metodInfoVOList.add(metodInfo);
            //get方法
            params = new ArrayList<>();
            params.add(identifyId.getType().getSimpleName() + " id");
            metodInfo = new ServiceMetodInfoVO("public", serviceData.getEntity().getName(), "get", StringUtil.concat(params, ","));
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("get"))));
            metodInfoVOList.add(metodInfo);
            //add方法
            params = new ArrayList<>();
            params.add(serviceData.getEntity().getName() + " " + StringUtil.firstCharToLower(serviceData.getEntity().getName()));
            metodInfo = new ServiceMetodInfoVO("public", "void", "add", StringUtil.concat(params, ","));
            if (exception != null) {
                metodInfo.setExceptions(exception.getName());
                importList.add(exception.getClazz());
            } else {
                metodInfo.setExceptions("Exception");
                importList.add(Exception.class.getName());
            }
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("add"))));
            metodInfoVOList.add(metodInfo);
            //update方法
            params = new ArrayList<>();
            params.add(serviceData.getEntity().getName() + " " + StringUtil.firstCharToLower(serviceData.getEntity().getName()));
            metodInfo = new ServiceMetodInfoVO("public", "void", "update", StringUtil.concat(params, ","));
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("update"))));
            metodInfoVOList.add(metodInfo);
            //delete方法
            params = new ArrayList<>();
            params.add(String.format("List<%s> idList", identifyId.getType().getSimpleName()));
            metodInfo = new ServiceMetodInfoVO("public", String.format("void", serviceData.getEntity().getName()), "delete", StringUtil.concat(params, ","));
            metodInfo.setContent(FreemarkerUtils.generateInputStream(serviceData, getClass().getClassLoader().getResourceAsStream(getMethodContentTemplatePath("delete"))));
            metodInfoVOList.add(metodInfo);

            /*******添加方法结束*******/


            importList.add(Resource.class.getName());
            importList.add(Service.class.getName());
            importList.add(Date.class.getName());
            importList.add(BeanUtil.class.getName());
            importList.add(List.class.getName());
            importList.add(serviceData.getDao().getClazz());
            importList.add(entity.getName());
            importList.add(service.getClazz());
            importList.add(query.getClazz());
            importList.add(daoQuery.getClazz());

            serviceData.setImportList(importList);
            serviceData.setMetodInfoList(metodInfoVOList);
            return serviceData;
        }

    }

    public ClassInfo getService() {
        return service;
    }

    public void setService(ClassInfo service) {
        this.service = service;
    }

    public ClassInfo getDao() {
        return dao;
    }

    public void setDao(ClassInfo dao) {
        this.dao = dao;
    }

    public List<ServiceMetodInfoVO> getMetodInfoList() {
        return metodInfoList;
    }

    public void setMetodInfoList(List<ServiceMetodInfoVO> metodInfoList) {
        this.metodInfoList = metodInfoList;
    }

    public ClassInfo getDaoQuery() {
        return daoQuery;
    }

    public void setDaoQuery(ClassInfo daoQuery) {
        this.daoQuery = daoQuery;
    }

    public ClassInfo getQuery() {
        return query;
    }

    public void setQuery(ClassInfo query) {
        this.query = query;
    }

    public List<String> getImportList() {
        return importList;
    }

    public void setImportList(List<String> importList) {
        this.importList = importList;
    }
}