open-dbtool/src/main/java/xyz/wbsite/dbtool/javafx/po/Table.java

package xyz.wbsite.dbtool.javafx.po;

import javafx.scene.control.TreeItem;
import xyz.wbsite.dbtool.javafx.manger.ManagerFactory;
import xyz.wbsite.dbtool.javafx.manger.ProjectManager;
import xyz.wbsite.dbtool.javafx.tool.Tool;

import java.util.ArrayList;
import java.util.List;

public class Table extends TreeItem {
    private Module dBhandle;

    public Table() {
    }

    public Table(String tableName) {
        this.tableName = tableName;
        setValue(tableName);
        setExpanded(true);
    }

    public Table(String tableName, String tableComment) {
        this.tableName = tableName;
        this.tableComment = tableComment;
        setValue(tableName);
        setExpanded(true);
    }

    /**
     * 对象名
     */
    private String tableName;

    /**
     * 对象注释
     */
    private String tableComment;

    private boolean create = true;
    private boolean delete = true;
    private boolean update = true;
    private boolean find = true;
    private boolean get = true;
    private boolean search = false;
    private boolean html = false;
    private boolean sys = true;

    /**
     * 对象字段
     */
    private List<Field> fields = new ArrayList();

    public boolean has(String type) {
        for (Field field : fields) {
            if (field.getFieldType().name().equals(type) && !field.getIsSystem()) {
                return true;
            }
        }
        return false;
    }

    public String getImport() {
        StringBuilder sb = new StringBuilder("");
        if (has("Date")) {
            sb.append("import java.util.Date;\n");
        }
        if (has("BigDecimal")) {
            sb.append("import java.math.BigDecimal;\n");
        }

        return sb.toString();
    }

    public boolean putField(Field field) {
        if (fields == null) {
            fields = new ArrayList<Field>();
        }
        fields.add(field);
        return true;
    }

    public boolean putFirstField(Field field) {
        if (fields == null) {
            fields = new ArrayList<Field>();
        }
        fields.add(0, field);
        return true;
    }

    public String getCName() {
        return Tool.ABB2Abb(this.tableName);
    }

    public String getFName() {
        return Tool.lineToFieldName(this.tableName);
    }

    public String getLName() {
        return Tool.lineToLPoint(this.tableName);
    }

    public String getTableName() {
        return tableName;
    }

    public void setTableName(String tableName) {
        this.tableName = tableName;
        setValue(tableName);
    }

    public String getTableComment() {
        return tableComment;
    }

    public void setTableComment(String tableComment) {
        this.tableComment = tableComment != null ? tableComment : "";
    }

    public List<Field> getFields() {
        return fields;
    }

    public void setFields(List<Field> fields) {
        this.fields = fields;
    }

    public Module getdBhandle() {
        return dBhandle;
    }

    public void setdBhandle(Module dBhandle) {
        this.dBhandle = dBhandle;
    }

    public boolean getCreate() {
        return create;
    }

    public void setCreate(boolean create) {
        this.create = create;
    }

    public boolean getDelete() {
        return delete;
    }

    public void setDelete(boolean delete) {
        this.delete = delete;
    }

    public boolean getUpdate() {
        return update;
    }

    public void setUpdate(boolean update) {
        this.update = update;
    }

    public boolean getFind() {
        return find;
    }

    public void setFind(boolean find) {
        this.find = find;
    }

    public boolean getGet() {
        return get;
    }

    public void setGet(boolean get) {
        this.get = get;
    }

    public boolean getSearch() {
        return search;
    }

    public void setSearch(boolean search) {
        this.search = search;
    }

    public boolean getHtml() {
        return html;
    }

    public void setHtml(boolean html) {
        this.html = html;
    }

    public boolean getSys() {
        return sys;
    }

    public void setSys(boolean sys) {
        this.sys = sys;
        ProjectManager projectManager = ManagerFactory.getProjectManager();
        if (projectManager != null) projectManager.checkSysFields(this);
    }

    public boolean hasPrimaryKey() {
        for (Field field : fields) {
            if (field.getIsPrimaryKey()) {
                return true;
            }
        }
        return false;
    }

    public List<Field> primaryKeyList() {
        ArrayList<Field> list = new ArrayList<>();
        for (Field field : this.fields) {
            if (field.getIsPrimaryKey()) {
                list.add(field);
            }
        }
        return list;
    }

    public List<Field> searchKeyList() {
        ArrayList<Field> list = new ArrayList<>();
        for (Field field : this.fields) {
            if (field.getIsSearch()) {
                list.add(field);
            }
        }
        return list;
    }

    public boolean hasSearchKey() {
        for (Field field : fields) {
            if (field.getIsSearch()) {
                return true;
            }
        }
        return false;
    }
}