package gudusoft.gsqlparser.nodes;

import gudusoft.gsqlparser.EParameterMode;
import gudusoft.gsqlparser.TBaseType;
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.nodes.teradata.TDataConversion;

import java.util.ArrayList;

/**
* Parameters in create procedure/function statements.
* Field declaration in plsql record type definition
* Parameter declaration in plsql cursor declaration.
*/

public class TParameterDeclaration extends TParseTreeNode {

    private int mode = TBaseType.parameter_mode_default;
    private boolean nocopy = false;
    private EParameterMode parameterMode = EParameterMode.defaultValue;

    /**
     *
     * @return Parameter modes that define the action of formal parameters:
     * <p>Oracle plsq:  in, out, in out.
     * <p>SQL Server: out, output, readonly
     *
     */
    public EParameterMode getParameterMode() {
        return parameterMode;
    }

    public void setNocopy(boolean nocopy) {
        this.nocopy = nocopy;
    }

    /**
     * Datatype of declared element.
     * @return
     */

    public TTypeName getDataType() {
        return dataType;
    }

    /**
     * Specify NOCOPY to instruct the database to pass this argument as fast as possible.

     * @return

     */
    public boolean isNocopy() {
        return nocopy;
    }

    public void setMode(int mode) {
        this.mode = mode;
        switch (this.mode){
            case TBaseType.parameter_mode_in:
                parameterMode = EParameterMode.in;
                break;
            case TBaseType.parameter_mode_out:
                parameterMode = EParameterMode.out;
                break;
            case TBaseType.parameter_mode_inout:
                parameterMode = EParameterMode.inout;
                break;
            case TBaseType.parameter_mode_output:
                parameterMode = EParameterMode.output;
                break;
            case TBaseType.parameter_mode_readonly:
                parameterMode = EParameterMode.readonly;
                break;
            case TBaseType.parameter_mode_aslocator:
                parameterMode = EParameterMode.aslocator;
                break;
            default:
                break;
        }
    }


    /**
     *
     * @return Parameter modes that define the action of formal parameters:
     * <p>Oracle plsq:  in, out, in out.
     * <p>SQL Server: out, output, readonly
     *
     * Use {@link #getParameterMode() instead}
     */
    public int getMode() {

        return mode;
    }

    /**
     * Specifies a default value for an IN parameter.
     * @return
     */
    public TExpression getDefaultValue() {
        return defaultValue;
    }

    /**
     * The name of the formal parameter that you are declaring, which you can reference in the body of the subprogram.
     * <p> Or, the name of the field in field declaration of record type definition.
     * @return
     */
    public TObjectName getParameterName() {
        return parameterName;
    }

    private TObjectName parameterName = null;
    private TTypeName dataType = null;

    public void setDefaultValue(TExpression defaultValue) {
        this.defaultValue = defaultValue;
    }

    private TExpression defaultValue = null;
    private int howtoSetValue = TBaseType.howtoSetValue_none;

    public void setNotNull(Boolean notNull) {
        NotNull = notNull;
    }

    /**
     * Is valid when this class represents field declaration in  plsql record type definition.
     */
    public Boolean getNotNull() {
        return NotNull;
    }


    private Boolean NotNull = false;



    /**
     * How deafult value of parameter was set.
     * <p> := | DEFAULT
     * <p> := {@link  TBaseType#howtoSetValue_assign}
     * <p> or DEFAULT keyword  {@link  TBaseType#howtoSetValue_default};
     * @return
     */

    public int getHowtoSetValue() {
        return howtoSetValue;
    }


    private boolean isVarying = false;
    private TConstant varyPrecision = null;

    /**
     *  SQL Server varying
     * @return
     */
    public boolean isVarying() {
        return isVarying;
    }

    public void setParameterMode(EParameterMode parameterMode) {
        this.parameterMode = parameterMode;
    }

    /**
     * SQL Server varying precision
     * @return

     */
    public TConstant getVaryPrecision() {
        return varyPrecision;
    }

    public void init(Object arg1) {
        parameterName = (TObjectName) arg1;
        parameterName.setObjectType(TObjectName.ttobjParameter);
    }

    private ArrayList<TDataConversion> dataConversions;

    public ArrayList<TDataConversion> getDataConversions() {
        return dataConversions;
    }

    public void init(Object arg1, Object arg2) {
        if (arg1 != null) { // it's possible that there is no parameter name in db2 create procedure
            parameterName = (TObjectName) arg1;
            //parameterName.setObjectType(TObjectName.ttobjParameter);
        }
        if (arg2 instanceof TTypeName){
            dataType = (TTypeName) arg2;
        }else{
            // teradata, BEGINDATE (DATE, DEFAULT DATE), bugs/mantisbt/view.php?id=2537
            dataConversions = ( ArrayList<TDataConversion>)arg2;
            dataType = dataConversions.get(0).getDataConversionItems().get(0).getDataType();
        }

    }

    public void init(Object arg1,Object arg2,Object arg3){
        init(arg1,arg2);

        if (arg3 != null){
            if (arg3 instanceof TParameterMode){
                parameterMode = ((TParameterMode)arg3).getParameterMode();
            }else{//tdummy
                TDummy dmy = (TDummy)arg3;
                if ((dmy.dbvendor == EDbVendor.dbvoracle)||(dmy.dbvendor == EDbVendor.dbvdb2)){
                    if (dmy.int1 == 1){
                        defaultValue = (TExpression)(dmy.node1);
                        howtoSetValue = dmy.int1;
                    }else if (dmy.int1 == 2){
                        defaultValue = (TExpression)(dmy.node1);
                        howtoSetValue = dmy.int1;
                    }
                }else if ((dmy.dbvendor == EDbVendor.dbvmssql)||(dmy.dbvendor == EDbVendor.dbvsybase)){
                  if (dmy.st2 != null){
                      this.isVarying = true;
                      if (dmy.node2 != null){
                          this.varyPrecision = (TConstant)(((TDummy)(dmy.node2)).node1);
                      }
                    }

                     this.howtoSetValue = dmy.int1;
                     if (dmy.node1 != null){
                          this.defaultValue = (TExpression)(dmy.node1);
                      }
                      if (dmy.st1 != null){
                         if (dmy.st1.toString().equalsIgnoreCase("out")){
                             this.mode = TBaseType.parameter_mode_out;
                             parameterMode = EParameterMode.out;
                         }else if (dmy.st1.toString().equalsIgnoreCase("in")){
                             this.mode = TBaseType.parameter_mode_in;
                             parameterMode = EParameterMode.in;
                         }else if (dmy.st1.toString().equalsIgnoreCase("output")){
                             this.mode = TBaseType.parameter_mode_output;
                             parameterMode = EParameterMode.output;
                         }else if (dmy.st1.toString().equalsIgnoreCase("inout")){
                             this.mode = TBaseType.parameter_mode_inout;
                             parameterMode = EParameterMode.inout;
                         }if (dmy.st1.toString().equalsIgnoreCase("readonly")){
                             this.mode = TBaseType.parameter_mode_readonly;
                              parameterMode = EParameterMode.readonly;
                         }
                      }

                  } // mssql

            }//dummy

        }
    }

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

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

    public void setParameterName(TObjectName parameterName) {
        this.parameterName = parameterName;
    }

    public void setDataType(TTypeName dataType) {
        this.dataType = dataType;
    }

    public void setHowtoSetValue(int howtoSetValue) {
        this.howtoSetValue = howtoSetValue;
    }

    public void setVarying(boolean isVarying) {
        this.isVarying = isVarying;
    }

    public void setVaryPrecision(TConstant varyPrecision) {
        this.varyPrecision = varyPrecision;
    }
}