package gudusoft.gsqlparser.nodes;

import gudusoft.gsqlparser.*;

import java.util.ArrayList;


/**
 * SQL constant (sometimes called a literal) specifies a value.
 * Constants are classified as:
 * <ul>
 * <li>string(text) constants</li>
 * <li>numeric constants,Numeric constants are further classified as integer, floating-point, or decimal</li>
 * <li>datetime constants</li>
 * <li>interval constants</li>
 * <li>boolean constants</li>
 * </ul>
 * {@link #getLiteralType()} returns the type of this constant, {@link #getValueToken()} returns the value of this constant.
 * {@link #toString} returns the text of this constant.
 *
 *
 */
public class TConstant extends TParseTreeNode {

    public Integer getIntegerValue(){
        Integer result = null;

        switch (getLiteralType()){
            case integer_et:
            case numeric_et:
            case etNumber:
                try {
                    result = Integer.parseInt(valueToken.toString());

                } catch (NumberFormatException e) {
                }
                break;
        }

        return result;
    }
    private ArrayList<TKeyValueSqlNode> keyValues;

    public ArrayList<TKeyValueSqlNode> getKeyValues() {
        return keyValues;
    }

    /**
     * Set the explicit type of this constant
     *
     * @param castType explicit type
     * @see #getCastType
     */
    public void setCastType(TTypeName castType) {
        this.castType = castType;
        switch (castType.getDataType()){
            case time_t:
                this.literalType = ELiteralType.etTime;
                break;
            case timestamp_t:
                this.literalType = ELiteralType.etTimestamp;
                break;
            case interval_t:
                this.literalType = ELiteralType.etInterval;
                break;
        }
    }

    public void setLiteralType(TSourceToken st){
        if (st == null) return;
        for(ELiteralType et:ELiteralType.values()){
            if (et.getText().equalsIgnoreCase(st.toString())){
                this.literalType = et;
                break;
            }
        }
    }

    /**
     * Constants can be specified with explicit types. This method returns the explicit type.
     * In PostgreSQL, a constant of an arbitrary type can be entered using any one of the following notations:
     * <pre>
     *     type 'string'
     *     'string'::type
     *     CAST 'string' AS type
     * </pre>
     *
     * @return explicit type for this constant
     */
    public TTypeName getCastType() {

        return castType;
    }

    private TTypeName castType;
    private TSourceToken valueToken;
    private TSourceToken signToken; // +,-

    public void setSignToken(TSourceToken signToken) {
        this.signToken = signToken;
    }

    public TSourceToken getSignToken() {
        return signToken;
    }

    private TSourceToken charSetName;
    private TSourceToken char_String_SetLiteral;
    private TSourceTokenList stringLiteralSequence;
    private ELiteralType literalType = ELiteralType.unknownType;
    private TSourceToken leadingPrecision;
    private EIntervalType intervalType;

    /**
     * Set the sub-type of interval constant
     *
     * @param intervalType sub-type of interval constant
     * @see #getIntervalType
     */
    public void setIntervalType(EIntervalType intervalType) {
        this.intervalType = intervalType;
    }

    /**
     * Oracle interval year to month and interval day to second constant.
     * This method return sub-type of interval constant such as year to month, minute to second.
     *
     * @return sub-type of interval constant
     */
    public EIntervalType getIntervalType() {

        return intervalType;
    }

    /**
     * set value of this constant
     * <pre>
     *     INTERVAL '30.12345' SECOND(2,4)
     * </pre>
     * '30.12345' in the above SQL in the valueToken
     *
     * @param valueToken value of this constant
     */
    public void setValueToken(TSourceToken valueToken) {
        this.valueToken = valueToken;
        stringValue = valueToken.toString();
    }

    /**
     * set the precision which is the maximum number of digits in the leading field of a interval constant.
     * <pre>
     *     INTERVAL '400 5' DAY(3) TO HOUR
     * </pre>
     * 3 in the above SQL is the <code>leadingPrecision</code>
     *
     *
     * @param leadingPrecision the precision of the leading field
     */
    public void setLeadingPrecision(TSourceToken leadingPrecision) {

        this.leadingPrecision = leadingPrecision;
    }

    /**
     * set the fractional precision which is the number of digits in the fractional part of the SECOND datetime field
     * in a interval day to second constant.
     * <pre>
     *     INTERVAL '30.12345' SECOND(2,4)
     * </pre>
     * 4 in the above SQL is the <code>fractionalSecondsPrecision</code>, 2 is the <code>leadingPrecision</code>
     *
     * @param fractionalSecondsPrecision
     */
    public void setFractionalSecondsPrecision(TSourceToken fractionalSecondsPrecision) {
        this.fractionalSecondsPrecision = fractionalSecondsPrecision;
    }

    /**
     * the maximum number of digits in the leading field of a interval constant.
     * <pre>
     *     INTERVAL '400 5' DAY(3) TO HOUR
     * </pre>
     * 3 in the above SQL is the <code>leadingPrecision</code>
     *
     * @return the leading precision
     */
    public TSourceToken getLeadingPrecision() {

        return leadingPrecision;
    }

    /**
     * the number of digits in the fractional part of the SECOND datetime field in a interval day to second constant.
     *
     * <pre>
     *     INTERVAL '30.12345' SECOND(2,4)
     * </pre>
     * 4 in the above SQL is the <code>fractionalSecondsPrecision</code>, 2 is the <code>leadingPrecision</code>
     *
     * @return the fractional seconds precision
     */
    public TSourceToken getFractionalSecondsPrecision() {
        return fractionalSecondsPrecision;
    }

    private TSourceToken fractionalSecondsPrecision;

    public TConstant(){}

    /**
     * Class constructor, set the type of this constant while creating a new instance of the constant
     *
     * @param pLiteralType type of this constant
     *
     */
    public TConstant(ELiteralType pLiteralType){
        this.literalType = pLiteralType;
    }

    /**
     * Class constructor, set the type and value of this constant when creating a new instance of the constant
     *
     * @param literalType type of this constant
     * @param valueToken  value of this constant
     */
    public TConstant (ELiteralType literalType, TSourceToken valueToken){
        this(literalType);
        this.valueToken = valueToken;
        this.stringValue = valueToken.toString();
        this.setStartToken(valueToken);
        this.setEndToken(valueToken);
    }

    /**
     * Class constructor, set the type and value of this constant when creating a new instance of the constant.
     * set the sub-type of an interval constant.
     *
     * @param literalType   type of this constant, must be etInterval
     * @param valueToken    value of this constant
     * @param intervalType  sub-type of an interval constant
     */
    public TConstant (ELiteralType literalType, TSourceToken valueToken,EIntervalType intervalType){
        this(literalType, valueToken);
        this.intervalType = intervalType;
    }

    /**
     * @deprecated use {@link #getValueToken} instead
     *
     * @param stringValue text value of this constant
     */
//    public void setStringValue(String stringValue) {
//        this.stringValue = stringValue;
//    }

    /**
     *
     * @return text value of this constant without quote characters
     */
    public String getValue(){
        //return getStringValue();
        String ret;
        switch (literalType){
            case etString:
            case etNumber:
            case etFloat:
                if (valueToken != null){
                    ret = TBaseType.removeQuoteChar(valueToken.toString());
                }else{
                    ret =  TBaseType.removeQuoteChar(toString());
                }

                break;
            default:
                ret = toString();
                break;
        }

        return ret;
    }

    /**
     * @deprecated since 1.8.2.4, use {@link #getValue()} instead.
     *
     * @return text value of this constant
     */
    public String getStringValue() {
        return getValue();
    }

    private String stringValue;

    /**
     * get type of this constant, can be strings, integers, and floating point numbers and more.
     * Please check {@link gudusoft.gsqlparser.ELiteralType} for more type of constant
     *
     * @return type of this constant
     */
    public ELiteralType getLiteralType() {
        return literalType;
    }

    public void setStringLiteralSequence(TSourceTokenList stringLiteralSequence) {
        this.stringLiteralSequence = stringLiteralSequence;
    }

    /**
     * String list constant in Hive.
     *
     * @return a sequence of source token
     */
    public TSourceTokenList getStringLiteralSequence() {

        return stringLiteralSequence;
    }

    /**
     * The '+' or '-' before this constant. Valid only when {@link #getLiteralType} is {@link ELiteralType#etNumber}
     *
     * @param sign the '+' or '-' token
     */
    public void setSign(TSourceToken sign) {
        this.sign = sign;
    }

    private TSourceToken sign;

    /**
     * The '+' or '-' before this constant. Valid only when {@link #getLiteralType} is {@link ELiteralType#etNumber}
     *
     * @return the '+' or '-' token
     */
    public TSourceToken getSign() {
        return sign;
    }

    /**
     * The value of this constant represented by a source token.
     *
     * @return the value of this constant
     */
    public TSourceToken getValueToken(){
        return valueToken;
    }

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

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

    public void init(Object arg1){
        literalType = (ELiteralType)arg1;
    }

    /**
     * The type of this constant.
     * <p>
     * The most used type are: etNumber,etFloat,etString,etTimestamp,etDate,etInterval
     * @param literalType the type of this constant
     */
    public void setLiteralType(ELiteralType literalType) {
        this.literalType = literalType;
    }

    public static TConstant createConstant(ELiteralType literalType, String valueStr){
        TConstant constant = new TConstant(literalType);
        constant.setValueToken(new TSourceToken(valueStr));
        return constant;
    }

    public void init(Object arg1, Object arg2){
        init(arg1);
        switch (literalType){
            case integer_et:
                valueToken = (TSourceToken)arg2;
                break;
            case etNumber:
                valueToken = (TSourceToken)arg2;
                break;
            case etFloat:
                valueToken = (TSourceToken)arg2;
                break;
            case etString:
                valueToken = (TSourceToken)arg2;
                break;
            case etTimestamp:
                valueToken = (TSourceToken)arg2;
                break;
            case etDate:
                valueToken = (TSourceToken)arg2;
                break;
            case etInterval:
                valueToken = (TSourceToken)arg2;
                break;
            case etStringLiteralSequence:
                stringLiteralSequence  = (TSourceTokenList)arg2;
                break;
            case etStar:
                valueToken = (TSourceToken)arg2;
                break;
            case interval:
                int64_expression = (TExpression)arg2;
                break;
            case etJson:
                keyValues = (ArrayList<TKeyValueSqlNode>)arg2;
                break;
            default:
                break;
        }

        if (arg2 instanceof TSourceToken){
            this.setStartToken((TSourceToken)arg2);
            this.setEndToken((TSourceToken)arg2);
        }

    }

    private TSourceToken bind1;
    private TSourceToken bind2;
    private TSourceToken indicator;

    public void setBind1(TSourceToken bind1) {
        this.bind1 = bind1;
    }

    public void setBind2(TSourceToken bind2) {
        this.bind2 = bind2;
    }

    public void setIndicator(TSourceToken indicator) {
        this.indicator = indicator;
    }

    public TSourceToken getBind1() {

        return bind1;
    }

    public TSourceToken getBind2() {
        return bind2;
    }

    public TSourceToken getIndicator() {
        return indicator;
    }

    EIntervalType getIntervalTypeByDummy(TDummy dummy){
        EIntervalType rtn = EIntervalType.itUnknown;
        for(EIntervalType t : EIntervalType.values()){
            if (t.getValue() == dummy.int1){
                rtn = t;
                break;
            }
        }
        return rtn;
    }

    private ArrayList<TSourceToken> intervalValueTokens = new ArrayList<>();

    public ArrayList<TSourceToken> getIntervalValueTokens() {
        return intervalValueTokens;
    }

    public ArrayList<EIntervalType> getIntervalTypes() {
        return intervalTypes;
    }

    private ArrayList<EIntervalType> intervalTypes = new ArrayList<>();

    public void addIntervalTypeByDummy(TDummy dummy){
        intervalTypes.add( getIntervalTypeByDummy(dummy) );
    }

    public void init(Object arg1, Object arg2,Object arg3){
        init(arg1);
        switch (literalType){
            case etInterval:
                valueToken = (TSourceToken)arg2;
                if (arg3 instanceof TDummy){ // databricks
                    intervalType = getIntervalTypeByDummy((TDummy) arg3);
                    intervalValueTokens.add(valueToken);
                    intervalTypes.add(intervalType);
                }else{
                    intervalType = (EIntervalType)arg3;
                }
                break;
            case etBindVar:
                bind1 = (TSourceToken)arg2;
                bind2 = (TSourceToken)arg3;
                break;
            default:
                break;
        }
    }


    public void init(Object arg1, Object arg2,Object arg3,Object arg4){
        init(arg1);
        switch (literalType){
            case etBindVar:
                bind1 = (TSourceToken)arg2;
                bind2 = (TSourceToken)arg3;
                indicator = (TSourceToken)arg4;
                break;
            default:
                break;
        }
    }

    private TExpression int64_expression;

    /**
     * BigQuery interval literal int64_expression
     *
     * @return BigQuery interval literal int64_expression
     */
    public TExpression getInt64_expression() {
        return int64_expression;
    }
}