package gudusoft.gsqlparser.nodes.snowflake;


import gudusoft.gsqlparser.EDbObjectType;
import gudusoft.gsqlparser.TSourceToken;
import gudusoft.gsqlparser.nodes.*;

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

public class TTaskOption extends TParseTreeNode {

    private String name;
    private boolean orReplace;
    private boolean ifNotExists;
    private String warehouse;
    private String userTaskManagedInitialWarehouseSize;
    private TSchedule schedule;
    private String scheduleString;

    public String getScheduleString() {
        return scheduleString;
    }

    private String config;
    private Boolean allowOverlappingExecution;
    private ArrayList<TNameValuePair> sessionParameters;
    private Integer userTaskTimeoutMs;
    private Integer suspendTaskAfterNumFailures;
    private String errorIntegration;
    private String comment;
    private String finalize;
    private Integer taskAutoRetryAttempts;
    private Map<String, String> tags;
    private TObjectNameList after;
//    private String when;
    private String sql;

    private ETaskOptionType taskOptionType;

    private TExpression whenCondition;

    public TExpression getWhenCondition() {
        return whenCondition;
    }

    public void init(Object arg1) {
        taskOptionType = (ETaskOptionType) arg1;
    }

    public void init(Object arg1, Object arg2){
        init(arg1);
        switch (taskOptionType){
            case WAREHOUSE:
            {
                if (arg2 instanceof TSourceToken){
                    warehouse = ((TSourceToken) arg2).toString();
                } else if (arg2 instanceof TConstant) {
                    warehouse = ((TConstant) arg2).toString();
                }else{
                    warehouse = (String) arg2;
                }
                    break;
                }
            case USER_TASK_MANAGED_INITIAL_WAREHOUSE_SIZE:
                userTaskManagedInitialWarehouseSize = ((TConstant) arg2).toString();
                break;
            case SCHEDULE:{
                if (arg2 instanceof TConstant){
                    scheduleString = ((TConstant) arg2).toString();
                }else if (arg2 instanceof TSourceToken){
                    scheduleString = ((TSourceToken) arg2).toString();
                }
                break;
            }
            case CONFIG:
                config = (String) arg2;
                break;
            case ALLOW_OVERLAPPING_EXECUTION:
                allowOverlappingExecution = (Boolean) arg2;
                break;
            case SESSION_PARAMETER:
                sessionParameters = (ArrayList<TNameValuePair>) arg2;
                break;
            case USER_TASK_TIMEOUT_MS:
                userTaskTimeoutMs = (Integer) arg2;
                break;
            case SUSPEND_TASK_AFTER_NUM_FAILURES:
                suspendTaskAfterNumFailures = (Integer) arg2;
                break;
            case ERROR_INTEGRATION:
                errorIntegration = (String) arg2;
                break;
            case COMMENT:
                comment = (String) arg2;
                break;
            case FINALIZE:
                finalize = (String) arg2;
                break;
            case TASK_AUTO_RETRY_ATTEMPTS:
                taskAutoRetryAttempts = (Integer) arg2;
                break;
            case TAG:
                tags = (Map<String, String>) arg2;
                break;
            case AFTER:
                after = (TObjectNameList) arg2;
                for(TObjectName objectName: after){
                    objectName.setDbObjectType(EDbObjectType.task);
                }
                break;
            case WHEN:
                whenCondition = (TExpression) arg2;
                break;
            default:
                throw new IllegalArgumentException("Unknown TaskOptionType: " + taskOptionType);
        }
    }

    public ETaskOptionType getTaskOptionType() {
        return taskOptionType;
    }
    // Getters and Setters



    // Getters and Setters for all fields

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public boolean isOrReplace() {
        return orReplace;
    }

    public void setOrReplace(boolean orReplace) {
        this.orReplace = orReplace;
    }

    public boolean isIfNotExists() {
        return ifNotExists;
    }

    public void setIfNotExists(boolean ifNotExists) {
        this.ifNotExists = ifNotExists;
    }

    public String getWarehouse() {
        return warehouse;
    }

    public void setWarehouse(String warehouse) {
        this.warehouse = warehouse;
    }

    public String getUserTaskManagedInitialWarehouseSize() {
        return userTaskManagedInitialWarehouseSize;
    }

    public void setUserTaskManagedInitialWarehouseSize(String userTaskManagedInitialWarehouseSize) {
        this.userTaskManagedInitialWarehouseSize = userTaskManagedInitialWarehouseSize;
    }

    public TSchedule getSchedule() {
        return schedule;
    }

    public void setSchedule(TSchedule schedule) {
        this.schedule = schedule;
    }

    public String getConfig() {
        return config;
    }

    public void setConfig(String config) {
        this.config = config;
    }

    public Boolean getAllowOverlappingExecution() {
        return allowOverlappingExecution;
    }

    public void setAllowOverlappingExecution(Boolean allowOverlappingExecution) {
        this.allowOverlappingExecution = allowOverlappingExecution;
    }

    public ArrayList<TNameValuePair> getSessionParameters() {
        return sessionParameters;
    }



    public Integer getUserTaskTimeoutMs() {
        return userTaskTimeoutMs;
    }

    public void setUserTaskTimeoutMs(Integer userTaskTimeoutMs) {
        this.userTaskTimeoutMs = userTaskTimeoutMs;
    }

    public Integer getSuspendTaskAfterNumFailures() {
        return suspendTaskAfterNumFailures;
    }

    public void setSuspendTaskAfterNumFailures(Integer suspendTaskAfterNumFailures) {
        this.suspendTaskAfterNumFailures = suspendTaskAfterNumFailures;
    }

    public String getErrorIntegration() {
        return errorIntegration;
    }

    public void setErrorIntegration(String errorIntegration) {
        this.errorIntegration = errorIntegration;
    }

    public String getComment() {
        return comment;
    }

    public void setComment(String comment) {
        this.comment = comment;
    }

    public String getFinalize() {
        return finalize;
    }

    public void setFinalize(String finalize) {
        this.finalize = finalize;
    }

    public Integer getTaskAutoRetryAttempts() {
        return taskAutoRetryAttempts;
    }

    public void setTaskAutoRetryAttempts(Integer taskAutoRetryAttempts) {
        this.taskAutoRetryAttempts = taskAutoRetryAttempts;
    }

    public Map<String, String> getTags() {
        return tags;
    }

    public void setTags(Map<String, String> tags) {
        this.tags = tags;
    }

    public TObjectNameList getAfter() {
        return after;
    }

    public void setAfter(TObjectNameList after) {
        this.after = after;
    }

//    public String getWhen() {
//        return when;
//    }
//
//    public void setWhen(String when) {
//        this.when = when;
//    }

    public String getSql() {
        return sql;
    }

    public void setSql(String sql) {
        this.sql = sql;
    }


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

    public void acceptChildren(TParseTreeVisitor v) {
        v.preVisit(this);
        v.postVisit(this);
    }
    // Inner class for Schedule
    public static class TSchedule {
        private Integer minutes;
        private String cronExpression;
        private String timeZone;

        public Integer getMinutes() {
            return minutes;
        }

        public void setMinutes(Integer minutes) {
            this.minutes = minutes;
        }

        public String getCronExpression() {
            return cronExpression;
        }

        public void setCronExpression(String cronExpression) {
            this.cronExpression = cronExpression;
        }

        public String getTimeZone() {
            return timeZone;
        }

        public void setTimeZone(String timeZone) {
            this.timeZone = timeZone;
        }
    }



}