package gudusoft.gsqlparser.stmt;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.nodes.couchbase.TIndexKeyTerm;
import gudusoft.gsqlparser.nodes.couchbase.TKeyspaceRef;
import gudusoft.gsqlparser.nodes.hive.THiveIndexProperties;
import gudusoft.gsqlparser.nodes.hive.THiveRowFormat;
import gudusoft.gsqlparser.nodes.hive.THiveTableFileFormat;
import gudusoft.gsqlparser.nodes.hive.THiveTableProperties;

/**
 * create an index.
 */
public class TCreateIndexSqlStatement extends TCustomSqlStatement {

    private TWhereClause whereCondition;

    public void setWhereCondition(TWhereClause whereCondition) {
        this.whereCondition = whereCondition;
    }

    public TWhereClause getWhereCondition() {
        return whereCondition;
    }

    private TExpression filterPredicate;

    public void setFilterPredicate(TExpression filterPredicate) {
        this.filterPredicate = filterPredicate;
    }

    public TExpression getFilterPredicate() {

        return filterPredicate;
    }

    public TObjectName getAsTypeName() {
        return asTypeName;
    }

    public TCreateIndexSqlNode getCreateIndexNode() {
        return createIndexNode;
    }

    public boolean isDeferredRebuildIndex() {
        return deferredRebuildIndex;
    }

    public THiveIndexProperties getIndexProperties() {
        return indexProperties;
    }

    public TObjectName getIndexComment() {
        return indexComment;
    }

    public TObjectName getInTableName() {
        return inTableName;
    }

    public THiveTableFileFormat getTableFileFormat() {
        return tableFileFormat;
    }

    public TObjectName getTableLocation() {
        return tableLocation;
    }

    public THiveTableProperties getTableProperties() {
        return tableProperties;
    }

    public THiveRowFormat getTableRowFormat() {
        return tableRowFormat;
    }

    private TObjectName asTypeName; //hive
    private boolean deferredRebuildIndex;
    private THiveIndexProperties indexProperties;
    private TObjectName inTableName;
    private THiveRowFormat tableRowFormat;
    private THiveTableFileFormat tableFileFormat;
    private TObjectName tableLocation;
    private THiveTableProperties tableProperties;
    private TObjectName indexComment;
    private boolean isClustered = false;
    private boolean isNonClustered = false;

    private TObjectName tableName;

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

    public void setColumnNameList(TOrderByItemList columnNameList) {
        this.columnNameList = columnNameList;
    }

    private TOrderByItemList columnNameList;

    public void setClustered(boolean isClustered) {
        this.isClustered = isClustered;
    }

    public void setNonClustered(boolean isNonClustered) {
        this.isNonClustered = isNonClustered;
    }

    public boolean isClustered() {

        return isClustered;
    }

    public boolean isNonClustered() {
        return isNonClustered;
    }

    public TCreateIndexSqlStatement(EDbVendor dbvendor) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstcreateindex;
    }

    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        createIndexNode = (TCreateIndexSqlNode)rootNode;
        super.doParseStatement(psql);
        this.indexName = createIndexNode.getIndexName();
        if (getSqlEnv().getDefaultCatalogName() != null){
            if (indexName.getDatabaseToken() == null){
                indexName.setDatabaseToken(new TSourceToken(getSqlEnv().getDefaultCatalogName()));
            }
        }
        tableName =  createIndexNode.getTableName();
       // tableName.setDbObjectType(EDbObjectType.table);
        columnNameList = createIndexNode.getColumnNameList();

        if (tableName != null){
            TTable lcTable = new TTable(tableName);
            lcTable.setPropertyFromObjectName(tableName, ETableEffectType.tetCreateIndexRef);
            this.addToTables(lcTable);

            if (columnNameList != null){
                for(int i=0;i<columnNameList.size();i++){
                    if (columnNameList.getOrderByItem(i).getSortKey().getExpressionType() == EExpressionType.simple_object_name_t){
                        lcTable.getLinkedColumns().addObjectName(columnNameList.getOrderByItem(i).getSortKey().getObjectOperand());
                    }

                }
            }
        }

        this.asTypeName = createIndexNode.getAsTypeName();
        this.deferredRebuildIndex  = createIndexNode.isDeferredRebuildIndex();
        this.indexProperties = createIndexNode.getIndexProperties();
        this.inTableName = createIndexNode.getInTableName();
        if (inTableName != null){
            inTableName.setObjectType(TObjectName.ttobjTable);
        }
        this.tableRowFormat = createIndexNode.getTableRowFormat();
        this.tableFileFormat = createIndexNode.getTableFileFormat();
        this.tableLocation = createIndexNode.getTableLocation();
        this.tableProperties = createIndexNode.getTableProperties();
        this.indexComment = createIndexNode.getIndexComment();
        if (createIndexNode.getTableName() != null){
            createIndexNode.getTableName().setObjectType(TObjectName.ttobjTable);
        }

        indexType = createIndexNode.getIndexType();
        isClustered = createIndexNode.isClustered();
        isNonClustered = createIndexNode.isNonClustered();

        filterPredicate = createIndexNode.getFilterPredicate();

        if (createIndexNode.getOptionList() != null) {
            for(int i=0;i<createIndexNode.getOptionList().size();i++){
                TDummy dummy = createIndexNode.getOptionList().getDummyItem(0);
                switch (dummy.int1){
                    case 1:
                        includeColumns = (TObjectNameList)dummy.node1;
                        break;
                    case 2:
                        break;
                    case 3:
                        filegroupOrPartitionSchemeName = (TObjectName)dummy.node1;
                        if (dummy.list1 != null){
                            partitionSchemeColumns = (TObjectNameList)dummy.list1;
                        }
                        break;
                    default:
                        break;
                }
            }
        }

        //couchbase
        keyspaceRef = createIndexNode.getKeyspaceRef();
        indexWith = createIndexNode.getIndexWith();
        indexWhere = createIndexNode.getIndexWhere();
        indexTerms = createIndexNode.getIndexTerms();
        indexPartition = createIndexNode.getIndexPartition();
        whereCondition = createIndexNode.getWhereCondition();
        if (whereCondition != null){
            whereCondition.doParse(this,ESqlClause.where);
        }

        return 0;
    }

    private TObjectName filegroupOrPartitionSchemeName;
    private TObjectNameList partitionSchemeColumns;

    public void setFilegroupOrPartitionSchemeName(TObjectName filegroupOrPartitionSchemeName) {
        this.filegroupOrPartitionSchemeName = filegroupOrPartitionSchemeName;
    }

    public void setPartitionSchemeColumns(TObjectNameList partitionSchemeColumns) {
        this.partitionSchemeColumns = partitionSchemeColumns;
    }

    public TObjectName getFilegroupOrPartitionSchemeName() {

        return filegroupOrPartitionSchemeName;
    }

    public TObjectNameList getPartitionSchemeColumns() {
        return partitionSchemeColumns;
    }

    private TObjectNameList includeColumns;

    public void setIncludeColumns(TObjectNameList includeColumns) {
        this.includeColumns = includeColumns;
    }

    public TObjectNameList getIncludeColumns() {

        return includeColumns;
    }

    private TObjectName indexName = null;

    private TCreateIndexSqlNode createIndexNode;

    private EIndexType indexType;

    public void setIndexType(EIndexType indexType) {
        this.indexType = indexType;
    }

    public EIndexType getIndexType(){
        return indexType;
        //return createIndexNode.getIndexType();
    }

    public TObjectName getTableName(){
        return  tableName; //createIndexNode.getTableName();
    }

    public TOrderByItemList getColumnNameList(){
        return columnNameList; //createIndexNode.getColumnNameList();
    }

    /**
     *
     * @return the name of the index to be created.
     */
    public TObjectName getIndexName() {
        return indexName;
    }

    public void accept(TParseTreeVisitor v){
        v.preVisit(this);
        v.postVisit(this);
    }

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        v.postVisit(this);
    }

    public void setAsTypeName(TObjectName asTypeName) {
        this.asTypeName = asTypeName;
    }

    public void setDeferredRebuildIndex(boolean deferredRebuildIndex) {
        this.deferredRebuildIndex = deferredRebuildIndex;
    }

    public void setIndexProperties(THiveIndexProperties indexProperties) {
        this.indexProperties = indexProperties;
    }

    public void setInTableName(TObjectName inTableName) {
        this.inTableName = inTableName;
    }

    public void setTableRowFormat(THiveRowFormat tableRowFormat) {
        this.tableRowFormat = tableRowFormat;
    }

    public void setTableFileFormat(THiveTableFileFormat tableFileFormat) {
        this.tableFileFormat = tableFileFormat;
    }

    public void setTableLocation(TObjectName tableLocation) {
        this.tableLocation = tableLocation;
    }

    public void setTableProperties(THiveTableProperties tableProperties) {
        this.tableProperties = tableProperties;
    }

    public void setIndexComment(TObjectName indexComment) {
        this.indexComment = indexComment;
    }

    public void setIndexName(TObjectName indexName) {
        this.indexName = indexName;
    }


    private TKeyspaceRef keyspaceRef;//couchbase

    public void setKeyspaceRef(TKeyspaceRef keyspaceRef) {
        this.keyspaceRef = keyspaceRef;
    }

    public TKeyspaceRef getKeyspaceRef() {

        return keyspaceRef;
    }

    private TPTNodeList<TIndexKeyTerm> indexTerms;

    public void setIndexTerms(TPTNodeList<TIndexKeyTerm> indexTerms) {
        this.indexTerms = indexTerms;
    }

    public TPTNodeList<TIndexKeyTerm> getIndexTerms() {

        return indexTerms;
    }

    private TExpressionList indexPartition;

    public void setIndexPartition(TExpressionList indexPartition) {
        this.indexPartition = indexPartition;
    }

    public TExpressionList getIndexPartition() {

        return indexPartition;
    }

    private TExpression indexWhere;

    public void setIndexWhere(TExpression indexWhere) {
        this.indexWhere = indexWhere;
    }

    public TExpression getIndexWhere() {

        return indexWhere;
    }

    private TExpression indexWith;

    public void setIndexWith(TExpression indexWith) {
        this.indexWith = indexWith;
    }

    public TExpression getIndexWith() {

        return indexWith;
    }

}