package gudusoft.gsqlparser.nodes;

import gudusoft.gsqlparser.EExpressionType;
import gudusoft.gsqlparser.TBaseType;
import gudusoft.gsqlparser.TSourceToken;
import gudusoft.gsqlparser.nodes.oracle.TErrorLoggingClause;

import java.util.ArrayList;

/**
 * Interal used by parser.
*/
public class TInsertSqlNode extends TParseTreeNode {

    private String fileFormat;
    private String fileOptions = null;

    public String getFileFormat() {
        return fileFormat;
    }

    public String getFileOptions() {
        return fileOptions;
    }

    public void setFileFormat(TDummy dummy){
        fileFormat = dummy.st1.toString();
        if (dummy.node1 != null){
            fileOptions = dummy.node1.toString();
        }
    }

    private TObjectName sourceValueTable;

    public TObjectName getSourceValueTable() {
        return sourceValueTable;
    }

    private boolean insertAll = false;
    private boolean insertFirst = false;

    public boolean isInsertAll() {
        return insertAll;
    }

    public boolean isInsertFirst() {
        return insertFirst;
    }

    public void setInsertAllOrFirst(TSourceToken st){
        if (st == null) return;
        insertAll = st.toString().equalsIgnoreCase("all");
        insertFirst = st.toString().equalsIgnoreCase("first");
    }

    private TErrorLoggingClause errorLoggingClause;

    public void setErrorLoggingClause(TErrorLoggingClause errorLoggingClause) {
        this.errorLoggingClause = errorLoggingClause;
    }

    public TErrorLoggingClause getErrorLoggingClause() {

        return errorLoggingClause;
    }

    public void setDirectoryName(TObjectName directoryName) {
        this.directoryName = directoryName;
    }


    public void setOverwrite(TSourceToken overwrite) {
        if (overwrite == null) return;
        if (overwrite.toString().equalsIgnoreCase("overwrite")) {
            this.hiveInsertType = EHiveInsertType.overwriteTable;
        }
        return;
    }


    public void setHiveInsertType(EHiveInsertType hiveInsertType) {

        this.hiveInsertType = hiveInsertType;
        this.valueType = TBaseType.vt_hive;
    }

    private EHiveInsertType hiveInsertType = EHiveInsertType.intoUnknown;
    private TObjectName directoryName;

    public EHiveInsertType getHiveInsertType() {
        return hiveInsertType;
    }

    public TObjectName getDirectoryName() {
        return directoryName;
    }
    //private THiveTableOrPartition tableOrPartition;

    private TPTNodeList<TInsertCondition>  insertConditions;

    public void init(Object arg1){
            directoryName = (TObjectName)arg1;
    }

    public TCTEList getCteList() {
        return cteList;
    }

    public TPTNodeList<TInsertCondition> getInsertConditions() {
        return insertConditions;
    }

    public TPTNodeList<TInsertIntoValue> getInsertIntoValues() {
        return insertIntoValues;
    }

    public void setInsertConditions(TPTNodeList<TInsertCondition> insertConditions) {
        this.insertConditions = insertConditions;
    }

    private TPTNodeList<TInsertIntoValue> elseIntoValues;

    public void setElseIntoValues(TPTNodeList<TInsertIntoValue> elseIntoValues) {
        this.elseIntoValues = elseIntoValues;
    }

    public TPTNodeList<TInsertIntoValue> getElseIntoValues() {

        return elseIntoValues;
    }

    private TPTNodeList<TInsertIntoValue> insertIntoValues;

    public void setInsertIntoValues(TPTNodeList<TInsertIntoValue> insertIntoValues) {
        this.insertIntoValues = insertIntoValues;
    }

    public void setInsertToken(TSourceToken insertToken) {
        this.insertToken = insertToken;
    }

    public TSourceToken getInsertToken() {

        return insertToken;
    }

    private TSourceToken insertToken = null;

    public TSelectSqlNode getSubQueryNode() {
        return subQueryNode;
    }

    public TMultiTargetList getValues() {

        return values;
    }

    public void setSubQueryNode(TSelectSqlNode subQueryNode) {
        this.subQueryNode = subQueryNode;
        this.valueType = TBaseType.vt_query;
        if (this.valueType == TBaseType.vt_hive){
            this.valueType = TBaseType.vt_hive_query;
        }
    }

    public int getValueType() {

        return valueType;
    }

    public void setValueType(int valueType) {
        this.valueType = valueType;
    }

    private int valueType = TBaseType.vt_values;

    public TCTEList cteList = null;
    private TTopClause topClause = null;

    public TReturningClause getReturningClause() {
        return returningClause;
    }

    public TObjectNameList getColumnList() {

        return columnList;
    }

    public TOutputClause getOutputClause() {

        return outputClause;
    }

    public TFromTable getTargetTable() {

        return targetTable;
    }

    public TTopClause getTopClause() {

        return topClause;
    }

    private TOutputClause outputClause = null;
    private TFromTable targetTable = null;
    private TObjectNameList columnList = null;
    private boolean invalidColumnList = false;
    private TReturningClause returningClause = null;
    private TResultColumnList onDuplicateKeyUpdate = null;

    public void setOnDuplicateKeyUpdate(TResultColumnList onDuplicateKeyUpdate) {
        this.onDuplicateKeyUpdate = onDuplicateKeyUpdate;
    }

    public TResultColumnList getOnDuplicateKeyUpdate() {

        return onDuplicateKeyUpdate;
    }

    /**
     * MySQL 8.0.20+ row alias for INSERT ... ON DUPLICATE KEY UPDATE
     * Allows referencing the inserted row values without using VALUES() function.
     * Example: INSERT INTO t1 (a,b,c) VALUES (1,2,3) AS new ON DUPLICATE KEY UPDATE c = new.a+new.b;
     */
    private TObjectName insertRowAlias = null;

    /**
     * MySQL 8.0.20+ optional column aliases for the row alias.
     * Example: INSERT INTO t1 (a,b,c) VALUES (1,2,3) AS new(m,n,p) ON DUPLICATE KEY UPDATE c = m+n;
     */
    private TObjectNameList insertRowAliasColumnList = null;

    public void setInsertRowAlias(TObjectName insertRowAlias) {
        this.insertRowAlias = insertRowAlias;
    }

    public TObjectName getInsertRowAlias() {
        return insertRowAlias;
    }

    public void setInsertRowAliasColumnList(TObjectNameList insertRowAliasColumnList) {
        this.insertRowAliasColumnList = insertRowAliasColumnList;
    }

    public TObjectNameList getInsertRowAliasColumnList() {
        return insertRowAliasColumnList;
    }

    /**
     * Set row alias from grammar TDummy node.
     * The dummy node contains: node1=alias name, objectNameList=column aliases (optional)
     */
    public void setInsertRowAlias(TDummy dummyNode) {
        if (dummyNode == null) return;
        this.insertRowAlias = (TObjectName) dummyNode.node1;
        this.insertRowAliasColumnList = dummyNode.objectNameList;
    }

    private TOnConflictClause onConflictClause;

    public TOnConflictClause getOnConflictClause() {
        return onConflictClause;
    }

    public void setOnConflictClause(TOnConflictClause onConflictClause) {
        this.onConflictClause = onConflictClause;
    }

    private TSourceToken ignore;
    private TSourceToken priority_delayed;

    public void setIgnore(TSourceToken ignore) {
        this.ignore = ignore;
    }

    public void setPriority_delayed(TSourceToken priority_delayed) {
        this.priority_delayed = priority_delayed;
    }

    public TSourceToken getIgnore() {

        return ignore;
    }

    public TSourceToken getPriority_delayed() {
        return priority_delayed;
    }

    public void setReturningClause(TReturningClause returningClause) {
        this.returningClause = returningClause;
    }

    /*
     * this function was used in teradata y file to support syntax like this:
      * insert into t(v1,v2,v3) 
     * @param columnList
     */
    public void setColumnListByResultColumnList(TResultColumnList columnList){
        // expression in resultcolumn must be type TExpression.SimpleObject
        this.columnList = new TObjectNameList();
        invalidColumnList = (columnList == null) || (columnList.size() == 0);
        if (invalidColumnList) {
            return;
        }
        for(int i=0; i<columnList.size();i++){
            TResultColumn column = columnList.getResultColumn(i);
            TExpression expr = column == null ? null : column.getExpr();
            if ((expr != null)
                    && (expr.getExpressionType() == EExpressionType.simple_object_name_t)
                    && (expr.getObjectOperand() != null)){
                this.columnList.addObjectName(expr.getObjectOperand());
            } else {
                invalidColumnList = true;
            }
        }
    }

    /**
     * Whether Teradata column-list recovery produced a missing or non-name
     * element that cannot be used as an INSERT target column.
     */
    public boolean hasInvalidColumnList() {
        return invalidColumnList;
    }

    public void setColumnList(TObjectNameList columnList) {
        this.columnList = columnList;
    }

    public void setTargetTable(TFromTable targetTable) {
        this.targetTable = targetTable;
    }

    public void setTargetTable(TDummy targetTableAndColumns) {
        this.targetTable = (TFromTable) targetTableAndColumns.node1;
        if (targetTableAndColumns.objectNameList != null){
            this.columnList = targetTableAndColumns.objectNameList;
        }
        if (targetTableAndColumns.node2 != null){
            this.subQueryNode = (TSelectSqlNode) targetTableAndColumns.node2;
            this.valueType = TBaseType.vt_query;
        }
    }

    public void setOutputClause(TOutputClause outputClause) {
        this.outputClause = outputClause;
    }

    public void setTopClause(TTopClause topClause) {
        this.topClause = topClause;
    }

    private TMultiTargetList values = null;
    private TSelectSqlNode subQueryNode = null;
    private TExecuteSqlNode executeSqlNode = null;

    public TFunctionCall getFunctionCall() {
        return functionCall;
    }

    private TFunctionCall functionCall = null;

    /*
     * this function was used in teradata y file to support syntax like this:
      * insert into t(v1,v2,v3)
     */

    public TObjectName getRecordName() {
        return recordName;
    }

    private TObjectName recordName = null;

    public void setValuesByMultiTarget(TMultiTarget mt){
       this.valueType = TBaseType.vt_values;
       values = new TMultiTargetList();
       values.addMultiTarget(mt);
    }

    public void setValuesByMultiTargetList(TMultiTargetList mtList){
        this.valueType = TBaseType.vt_values;
        values = mtList;
    }

    private TResultColumnList setColumnValues = null;

    public TResultColumnList getSetColumnValues() {
        return setColumnValues;
    }

    private TIsolationClause isolationClause = null;

    public void setIsolationClause(TIsolationClause isolationClause) {
        this.isolationClause = isolationClause;
    }

    public TIsolationClause getIsolationClause() {
        return isolationClause;
    }

    public TExecuteSqlNode getExecuteSqlNode() {
        return executeSqlNode;
    }

    public void setValues(TDummy dummyNode){
        valueType = dummyNode.int1;
        switch(valueType){
            case TBaseType.vt_values:
                values = (TMultiTargetList)dummyNode.list1;
                break;
            case TBaseType.vt_query:
                subQueryNode = (TSelectSqlNode)dummyNode.node1;
                break;
            case TBaseType.vt_default_values:
                break;
            case TBaseType.vt_execute:
                executeSqlNode = (TExecuteSqlNode)dummyNode.node1;
                break;
            case TBaseType.vt_values_function:

                functionCall = (TFunctionCall)dummyNode.node1;
                break;
            case TBaseType.vt_values_oracle_record:
                recordName = (TObjectName)dummyNode.node1;
                break;
            case TBaseType.vt_values_empty:
                break;
            case TBaseType.vt_set_column_value:
                this.setColumnValues = (TResultColumnList)dummyNode.list1;
                break;
            case TBaseType.vt_table:
                this.sourceValueTable = (TObjectName) dummyNode.node1;
                break;
        }
    }

    private ArrayList<TInsertSqlNode> insertSqlNodes;

    public ArrayList<TInsertSqlNode> getInsertSqlNodes() {
        return insertSqlNodes;
    }

    public void setInsertSqlNodes(ArrayList<TInsertSqlNode> insertSqlNodes) {
        this.insertSqlNodes = insertSqlNodes;
    }

    public static void setFromList(ArrayList<TInsertSqlNode> insertSqlNodes, TFromTableList fromTables){
        for(int i=0;i<insertSqlNodes.size();i++){
            TInsertSqlNode insertSqlNode = insertSqlNodes.get(i);
            insertSqlNode.getSubQueryNode().setFromTableList(fromTables);
        }
    }

    public static TInsertSqlNode getFirstInsertSqlNode(ArrayList<TInsertSqlNode> insertSqlNodes){
        return  insertSqlNodes.get(0);
    }
}