package com.vincent.rsf.framework.generators.domain;

import com.vincent.rsf.framework.common.Cools;
import com.vincent.rsf.framework.generators.CoolGenerator;
import com.vincent.rsf.framework.generators.constant.SqlOsType;
import com.vincent.rsf.framework.generators.utils.GeneratorUtils;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Created by vincent on 2019-06-18
 */
public class Column {

    private String name; // 表字段名
    private String type; // 类型
    private String comment; // 备注
    private String humpName; // 小驼峰
    private boolean primaryKey; // 唯一主键
    private boolean mainKey;  // 普通主键
    private boolean notNull; // 非空
    private boolean major; // 主要
    private boolean image; // 图片
    private boolean checkBox; // 复选框
    private String foreignKey; // 外健实例名(大驼峰,如 sys_user ==> User)
    private String foreignKeyMajor; // 外键显示字段(大驼峰,如 name ==> Name)
    private List<Map<String, Object>> enums; // 枚举值
    private Integer length; // 字段长度

    public Column(Connection conn, String name, String type, String comment, boolean primaryKey, boolean mainKey, boolean notNull, Integer length, boolean init, SqlOsType sqlOsType) {
        this.name = name;
        this.type = type;
        this.comment = "";
        if (!Cools.isEmpty(comment)){
            // 枚举
            Pattern pattern1 = Pattern.compile("(.+?)(?=\\{)");
            Matcher matcher1 = pattern1.matcher(comment);
            // 外键
            Pattern pattern11 = Pattern.compile("(.+?)(?=\\[)");
            Matcher matcher11 = pattern11.matcher(comment);
            // 枚举
            if (matcher1.find()) {
                this.comment = matcher1.group();
                Pattern pattern2 = Pattern.compile("(?<=\\{)(.+?)(?=})");
                Matcher matcher2 = pattern2.matcher(comment);
                if (matcher2.find()) {
                    String group = matcher2.group();
                    if (!Cools.isEmpty(group)) {
                        String[] values = group.split(",");
                        this.enums = new ArrayList<>();
                        for (String val : values) {
                            Map<String, Object> map = new HashMap<>();
                            String[] split = val.split(":");
                            map.put(split[0], split[1]);
                            enums.add(map);
                        }
                    }
                }
            //  外键
            } else if (matcher11.find()){
                this.comment = matcher11.group();
                Pattern pattern22 = Pattern.compile("(?<=\\[)(.+?)(?=])");
                Matcher matcher22 = pattern22.matcher(comment);
                if (matcher22.find()) {
                    String group = matcher22.group();
                    if (!Cools.isEmpty(group)) {
                        this.foreignKey = GeneratorUtils.getNameSpace(group);
                        List<Column> foreignColumns = new ArrayList<>();
                        if (init) {
                            try {
                                switch (sqlOsType) {
                                    case MYSQL:
                                        foreignColumns = CoolGenerator.getMysqlColumns(conn, group, false, sqlOsType);
                                        break;
                                    case SQL_SERVER:
                                        foreignColumns = CoolGenerator.getSqlServerColumns(conn, group, false, sqlOsType);
                                        break;
                                    default:
                                        break;
                                }

                            } catch (Exception e){
                                e.printStackTrace();
                            }
                        }
                        if (!Cools.isEmpty(foreignColumns)){
                            for (Column column : foreignColumns){
                                if (column.isMajor()){
                                    this.foreignKeyMajor = GeneratorUtils.firstCharConvert(column.getHumpName(), false);
                                }
                            }
                            if (Cools.isEmpty(this.foreignKeyMajor)){
                                this.foreignKeyMajor = "Id";
                            }
                        }
                    }
                }
            } else {
                this.comment = comment;
            }
            // 主要字段
            if (comment.endsWith("(*)")){
                this.comment = comment.substring(0, comment.length()-3);
                this.major = true;
            }
            // 图片字段
            if (comment.endsWith("(img)")){
                this.comment = comment.substring(0, comment.length()-5);
                this.image = true;
            }
            // 复选框
            if (comment.endsWith("(checkbox)")){
                this.comment = comment.substring(0, comment.length()-10);
                this.checkBox = true;
            }
        }
//        if (primaryKey || mainKey){
//            this.primaryKey = true;
//        } else {
//            this.primaryKey = false;
//        }
        this.primaryKey = primaryKey;
        this.mainKey = mainKey;
        this.notNull = notNull;
        this.length = length;
        this.humpName = GeneratorUtils._convert(name);
    }

    public String getName() {
        return name;
    }

    public void setName(final String name) {
        this.name = name;
    }

    public String getType() {
        return type;
    }

    public void setType(final String type) {
        this.type = type;
    }

    public String getComment() {
        return comment;
    }

    public String getWholeComment() {
        if (!Cools.isEmpty(this.enums)){
            StringBuilder sb = new StringBuilder(" ");
            for (Map<String, Object> val : enums){
                for (Map.Entry<String, Object> entry : val.entrySet()){
                    sb.append(entry.getKey())
                            .append(": ")
                            .append(entry.getValue())
                            .append("  ");
                }
            }
            return comment + sb.toString();
        }
        return comment;
    }

    public void setComment(final String comment) {
        this.comment = comment;
    }

    public String getHumpName() {
        return humpName;
    }

//    public boolean isPrimaryKey() {
//        return primaryKey || mainKey;
//    }

    public boolean isPrimaryKey() {
        return primaryKey;
    }

    public boolean isOnly(){
        return primaryKey;
    }

    public void setPrimaryKey(boolean primaryKey) {
        this.primaryKey = primaryKey;
    }

    public boolean isNotNull() {
        return notNull;
    }

    public void setNotNull(final boolean notNull) {
        this.notNull = notNull;
    }

    public String getForeignKey() {
        return foreignKey;
    }

    public void setForeignKey(final String foreignKey) {
        this.foreignKey = foreignKey;
    }

    public String getForeignKeyMajor() {
        return foreignKeyMajor;
    }

    public void setForeignKeyMajor(final String foreignKeyMajor) {
        this.foreignKeyMajor = foreignKeyMajor;
    }

    public List<Map<String, Object>> getEnums() {
        return enums;
    }

    public void setEnums(List<Map<String, Object>> enums) {
        this.enums = enums;
    }

    public Integer getLength() {
        return length;
    }

    public void setLength(final Integer length) {
        this.length = length;
    }

    public boolean isMajor() {
        return major;
    }

    public void setMajor(final boolean major) {
        this.major = major;
    }

    public boolean isImage() {
        return image;
    }

    public void setImage(final boolean image) {
        this.image = image;
    }

    public boolean isCheckBox() {
        return checkBox;
    }

    public void setCheckBox(boolean checkBox) {
        this.checkBox = checkBox;
    }

    public boolean isMainKey() {
        return mainKey;
    }

    public void setMainKey(boolean mainKey) {
        this.mainKey = mainKey;
    }

    @Override
    public String toString() {
        return "Column{" +
                "name='" + name + '\'' +
                ", type='" + type + '\'' +
                ", comment='" + comment + '\'' +
                ", humpName='" + humpName + '\'' +
                ", primaryKey=" + primaryKey +
                ", notNull=" + notNull +
                ", major=" + major +
                ", image=" + image +
                ", foreignKey='" + foreignKey + '\'' +
                ", foreignKeyMajor='" + foreignKeyMajor + '\'' +
                ", enums=" + enums +
                ", length=" + length +
                '}';
    }

}