design/designer-base/src/main/java/com/fr/design/data/datapane/TableDataCreatorProducer.java

/*
 * Copyright (c) 2001-2014,FineReport Inc, All Rights Reserved.
 */

package com.fr.design.data.datapane;

import com.fr.data.impl.*;
import com.fr.data.impl.storeproc.StoreProcedure;
import com.fr.design.ExtraDesignClassManager;
import com.fr.design.data.tabledata.tabledatapane.*;

import com.fr.stable.ArrayUtils;

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

/**
 * @author : richie
 * @since : 8.0
 */
public class TableDataCreatorProducer {

    private static TableDataCreatorProducer ourInstance = new TableDataCreatorProducer();

    public static TableDataCreatorProducer getInstance() {
        return ourInstance;
    }

    private TableDataCreatorProducer() {

    }

    public TableDataNameObjectCreator[] createReportTableDataCreator() {
        TableDataNameObjectCreator dataBase = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_DS_Database_Query"),
                "ds",
                "/com/fr/design/standard/database_normal.svg", DBTableData.class, DBTableDataPane.class);
        TableDataNameObjectCreator ds_Class = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Class"),
                "Class",
                "/com/fr/design/standard/class_table_data_normal.svg", ClassTableData.class, ClassTableDataPane.class);
        TableDataNameObjectCreator table = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Embedded"),
                "Embedded",
                "/com/fr/design/standard/data_table_normal.svg", EmbeddedTableData.class, EmbeddedTableDataPane.class);
        TableDataNameObjectCreator multiTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Relation"),
                "Multi",
                "/com/fr/design/standard/multi_normal.svg", ConditionTableData.class, MultiTDTableDataPane.class) {
            public boolean isNeedParameterWhenPopulateJControlPane() {
                return true;
            }
        };
        TableDataNameObjectCreator fileTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_File"),
                "File",
                "/com/fr/design/standard/file_normal.svg", FileTableData.class, FileTableDataSmallHeightPane.class);
        TableDataNameObjectCreator treeTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Tree"),
                "Tree",
                "/com/fr/design/standard/tree_normal.svg", RecursionTableData.class, TreeTableDataPane.class) {
            public boolean isNeedParameterWhenPopulateJControlPane() {
                return true;
            }
        };
        TableDataNameObjectCreator storeProcedure = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Datasource_Stored_Procedure"),
                "Proc",
                "/com/fr/design/standard/store_procedure_normal.svg",
                StoreProcedure.class, ProcedureDataPane.class) {
            @Override
            public boolean shouldInsertSeparator() {
                return true;
            }
        };
        TableDataNameObjectCreator[] creators = new TableDataNameObjectCreator[]{dataBase, ds_Class, table, fileTable, storeProcedure, multiTable, treeTable};
        return merge(creators, ExtraDesignClassManager.getInstance().getReportTableDataCreators());
    }

    public TableDataNameObjectCreator[] createServerTableDataCreator() {
        TableDataNameObjectCreator dataBase = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_DS_Database_Query"),
                "/com/fr/design/standard/server_database_normal.svg", DBTableData.class,
                DBTableDataPane.class);
        TableDataNameObjectCreator ds_Class = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Class"),
                "/com/fr/design/standard/class_table_data_normal.svg", ClassTableData.class,
                ClassTableDataPane.class);
        TableDataNameObjectCreator table = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Embedded"),
                "/com/fr/design/standard/data_table_normal.svg", EmbeddedTableData.class,
                EmbeddedTableDataPane.class);
        TableDataNameObjectCreator fileTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_File"),
                "/com/fr/design/standard/file_normal.svg", FileTableData.class,
                FileTableDataSmallPane.class);
        TableDataNameObjectCreator treeTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Tree"),
                "/com/fr/design/standard/tree_normal.svg",
                RecursionTableData.class, GlobalTreeTableDataPane.class) {
            public boolean isNeedParameterWhenPopulateJControlPane() {
                return true;
            }
        };
        TableDataNameObjectCreator multiTable = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Relation"),
                "/com/fr/design/standard/multi_normal.svg",
                ConditionTableData.class, GlobalMultiTDTableDataPane.class) {
            public boolean isNeedParameterWhenPopulateJControlPane() {
                return true;
            }
        };
        TableDataNameObjectCreator storeProcedure = new TableDataNameObjectCreator(com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Tabledata_Type_Stored_Procedure"),
                "/com/fr/design/standard/store_procedure_normal.svg",
                StoreProcedure.class, ProcedureDataPane.class) {
            @Override
            public boolean shouldInsertSeparator() {
                return true;
            }
        };
        TableDataNameObjectCreator[] creators = new TableDataNameObjectCreator[]{dataBase, ds_Class, table, fileTable, storeProcedure, multiTable, treeTable};
        return merge(creators, ExtraDesignClassManager.getInstance().getServerTableDataCreators());
    }

    private TableDataNameObjectCreator[] merge(TableDataNameObjectCreator[] creators, TableDataNameObjectCreator[] extraCreators) {
        if (ArrayUtils.isEmpty(extraCreators)) {
            return creators;
        }
        List<TableDataNameObjectCreator> list = new ArrayList<>();
        List<Integer> coverIndexList = new ArrayList<>();
        List<TableDataNameObjectCreator> coverCreators = new ArrayList<>();
        for (TableDataNameObjectCreator ex : extraCreators) {
            int index = ArrayUtils.indexOf(creators, ex);
            // 说明已经存在了一个同类型的数据集，需要用插件的覆盖掉原先的
            if (index != ArrayUtils.INDEX_NOT_FOUND) {
                // 需要替换到列表里的
                coverIndexList.add(index);
                coverCreators.add(ex);
            } else {
                // 需要添加到列表中去的
                list.add(ex);
            }
        }
        for (int i = 0; i < coverIndexList.size(); i ++) {
            creators[coverIndexList.get(i)] = coverCreators.get(i);
        }
        return ArrayUtils.addAll(creators, list.toArray(new TableDataNameObjectCreator[0]));
    }
}