package gudusoft.gsqlparser.nodes;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.stmt.TCreateTableSqlStatement;
import gudusoft.gsqlparser.stmt.snowflake.TSnowflakeCopyIntoStmt;
import gudusoft.gsqlparser.stmt.snowflake.TCreateStageStmt;

public class TCreateTableOption extends TParseTreeNode {

    private TDistributeBy distributeBy;

    public TDistributeBy getDistributeBy() {
        return distributeBy;
    }

    private TBaseTablePartition partitionSpec;

    public TBaseTablePartition getPartitionSpec() {
        return partitionSpec;
    }

    public TObjectName getComment() {
        return comment;
    }

    private TObjectName comment;

    /*
     * BigQuery external_table_option_list
     *
     * @see https://cloud.google.com/bigquery/docs/reference/standard-sql/data-
     * definition-language#external_table_option_list
     */
    private Boolean allowJaggedRows;
    private Boolean allowQuotedNewlines;
    private String compression;
    private String description;
    private Boolean enableLogicalTypes;
    private String encoding;
    private String expirationTimestamp;
    private String fieldDelimiter;
    private String format;
    private List<String> decimalTargetTypes;
    private String hivePartitionUriPrefix;
    private Boolean ignoreUnknownValues;
    private Long maxBadRecords;
    private String nullMarker;
    private String projectionFields;
    private String quote;
    private Boolean requireHivePartitionFilter;
    private String sheetRange;
    private Long skipLeadingRows;
    private List<String> uris;

    private  TObjectNameList columnNamelist;

    public TObjectNameList getColumnNamelist() {
        return columnNamelist;
    }

    private String awsSnsTopic = null;
    private TObjectNameList partitionColumnList;
    private TExpression partitionByExpr;

    /**
     * Bigquery partition by expr
     * @return expr
     */
    public TExpression getPartitionByExpr() {
        return partitionByExpr;
    }

    private String fileFormatName = null;
    private String fileFormatType = null;

    public void setFileFormatName(TSourceToken fileFormatName) {
        this.fileFormatName = fileFormatName.toString();
    }
    public void setFileFormatName(TObjectName fileFormatName) {
        this.fileFormatName = fileFormatName.toString();
    }

    public void setFileFormatType(TSourceToken fileFormatType) {
        this.fileFormatType = fileFormatType.toString();
    }

    public String getFileFormatName() {
        return fileFormatName;
    }

    public String getFileFormatType() {
        return fileFormatType;
    }

    private TStageLocation stageLocation;

    public void setStageLocation(TStageLocation stageLocation) {
        this.stageLocation = stageLocation;
    }

    public TStageLocation getStageLocation() {
        return stageLocation;
    }



    public String getAwsSnsTopic() {
        return awsSnsTopic;
    }

    public void setAwsSnsTopic(String awsSnsTopic) {
        this.awsSnsTopic = awsSnsTopic;
    }

    public TObjectNameList getPartitionColumnList() {
        return partitionColumnList;
    }

    public void setPartitionColumnList(TObjectNameList partitionColumnList) {
        this.partitionColumnList = partitionColumnList;
    }

    public String getCompression() {
        return compression;
    }

    public void setCompression(String compression) {
        this.compression = compression;
    }

    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public String getExpirationTimestamp() {
        return expirationTimestamp;
    }

    public void setExpirationTimestamp(String expirationTimestamp) {
        this.expirationTimestamp = expirationTimestamp;
    }

    public String getFieldDelimiter() {
        return fieldDelimiter;
    }

    public void setFieldDelimiter(String fieldDelimiter) {
        this.fieldDelimiter = fieldDelimiter;
    }

    public String getFormat() {
        return format;
    }

    public void setFormat(String format) {
        this.format = format;
    }

    public List<String> getDecimalTargetTypes() {
        return decimalTargetTypes;
    }

    public void setDecimalTargetTypes(List<String> decimalTargetTypes) {
        this.decimalTargetTypes = decimalTargetTypes;
    }

    public String getHivePartitionUriPrefix() {
        return hivePartitionUriPrefix;
    }

    public void setHivePartitionUriPrefix(String hivePartitionUriPrefix) {
        this.hivePartitionUriPrefix = hivePartitionUriPrefix;
    }

    public Boolean getIgnoreUnknownValues() {
        return ignoreUnknownValues;
    }

    public void setIgnoreUnknownValues(Boolean ignoreUnknownValues) {
        this.ignoreUnknownValues = ignoreUnknownValues;
    }

    public Long getMaxBadRecords() {
        return maxBadRecords;
    }

    public void setMaxBadRecords(Long maxBadRecords) {
        this.maxBadRecords = maxBadRecords;
    }

    public void setRequireHivePartitionFilter(Boolean requireHivePartitionFilter) {
        this.requireHivePartitionFilter = requireHivePartitionFilter;
    }

    public String getNullMarker() {
        return nullMarker;
    }

    public void setNullMarker(String nullMarker) {
        this.nullMarker = nullMarker;
    }

    public String getProjectionFields() {
        return projectionFields;
    }

    public void setProjectionFields(String projectionFields) {
        this.projectionFields = projectionFields;
    }

    public String getQuote() {
        return quote;
    }

    public void setQuote(String quote) {
        this.quote = quote;
    }

    public Boolean getRequireHivePartitionFilter() {
        return requireHivePartitionFilter;
    }

    public String getSheetRange() {
        return sheetRange;
    }

    public void setSheetRange(String sheetRange) {
        this.sheetRange = sheetRange;
    }

    public Long getSkipLeadingRows() {
        return skipLeadingRows;
    }

    public void setSkipLeadingRows(Long skipLeadingRows) {
        this.skipLeadingRows = skipLeadingRows;
    }

    public List<String> getUris() {
        return uris;
    }

    public void setUris(List<String> uris) {
        this.uris = uris;
    }

    public Boolean getAllowJaggedRows() {
        return allowJaggedRows;
    }

    public void setAllowJaggedRows(Boolean allowJaggedRows) {
        this.allowJaggedRows = allowJaggedRows;
    }

    public Boolean getAllowQuotedNewlines() {
        return allowQuotedNewlines;
    }

    public void setAllowQuotedNewlines(Boolean allowQuotedNewlines) {
        this.allowQuotedNewlines = allowQuotedNewlines;
    }

    public Boolean getEnableLogicalTypes() {
        return enableLogicalTypes;
    }

    public void setEnableLogicalTypes(Boolean enableLogicalTypes) {
        this.enableLogicalTypes = enableLogicalTypes;
    }

    public String getEncoding() {
        return encoding;
    }

    public void setEncoding(String encoding) {
        this.encoding = encoding;
    }
    public void setEncoding(TSourceToken encoding) {
                this.encoding = encoding.toString();
    } // added by baffle

    public String getExternalStageURL() {
        return externalStageURL;
    }

    public void setExternalStageURL(String externalStageURL) {
        this.externalStageURL = externalStageURL;
    }

    public void setPartitionByExpr(TExpression partitionByExpr) {
        this.partitionByExpr = partitionByExpr;
    }

    public void setFileFormatName(String fileFormatName) {
        this.fileFormatName = fileFormatName;
    }

    public void setFileFormatType(String fileFormatType) {
        this.fileFormatType = fileFormatType;
    }

    public void setDateRetentionInDays(TSourceToken dateRetentionInDays) {
        this.dateRetentionInDays = dateRetentionInDays;
    }

    public void setCommentToken(TSourceToken commentToken) {
        this.commentToken = commentToken;
    }

    public void setStageFileFormat(TDummy stageFileFormat) {
        this.stageFileFormat = stageFileFormat;
    }

    public void setCopyOptions(TDummy copyOptions) {
        this.copyOptions = copyOptions;
    }

    public void setExpressionList(TExpressionList expressionList) {
        this.expressionList = expressionList;
    }

    private TSourceToken dateRetentionInDays;

    public TSourceToken getDateRetentionInDays() {
        return dateRetentionInDays;
    }

    private TSourceToken commentToken;

    /**
     * @deprecated since v 2.8.1.1, please use {@link #getComment()} instead
     * @return
     */
    public TSourceToken getCommentToken() {
        return commentToken;
    }

    private TDummy stageFileFormat;

    private TDummy copyOptions;

    public TDummy getCopyOptions() {
        return copyOptions;
    }

    protected ECreateTableOption createTableOptionType;

    public void setCreateTableOptionType(ECreateTableOption createTableOptionType) {
        this.createTableOptionType = createTableOptionType;
    }

    public ECreateTableOption getCreateTableOptionType() {
        return createTableOptionType;
    }

    public void  init(Object arg1){
        createTableOptionType = (ECreateTableOption)arg1;
    }

    private TExpressionList expressionList;

    public TExpressionList getExpressionList() {
        return expressionList;
    }

    public TDummy getStageFileFormat() {
        return stageFileFormat;
    }

    private String externalStageURL=null;

    public void init(Object arg1, Object arg2){
        init(arg1);
        switch(createTableOptionType){
            case etoClusterBy:
                this.expressionList = (TExpressionList)arg2;
                break;
            case etoStageCopyOptions:
                this.copyOptions = (TDummy)arg2;
                break;
            case etoStageFileFormat:
                this.stageFileFormat = (TDummy)arg2;
                break;
            case etoComment:
                //this.commentToken = (TSourceToken)arg2;
                this.comment = (TObjectName) arg2;
                break;
            case etoDateRetentionTimeInDays:
                this.dateRetentionInDays = (TSourceToken)arg2;
                break;
            case etoPartitionBy:
                if (arg2 instanceof TObjectNameList){
                    this.partitionColumnList = (TObjectNameList)arg2;
                }else if (arg2 instanceof  TExpression){
                    this.partitionByExpr = (TExpression)arg2;
                }
                break;
            case etoAWSSnsTopic:
                awsSnsTopic = ((TSourceToken)arg2).toString();
                break;
            case etoStageURL:
                externalStageURL = ((TSourceToken)arg2).toString();
                break;
            case etoFiles:
                this.expressionList = (TExpressionList)arg2;
                break;
            case etoDistributeOn:
            case etoOrganizeOn:
                this.columnNamelist =(TObjectNameList)arg2;
                break;
            case etoReloptions:
                this.attributeOptions  = (ArrayList<TAttributeOption>)arg2;
                break;
            case etoPartitionSpec:
                this.partitionSpec = (TBaseTablePartition)arg2;
                break;
            case etoDistributeBy:
                this.distributeBy = (TDistributeBy)arg2;
                break;
            case etoDistributeByHash:
                this.columnNamelist = (TObjectNameList)arg2;
                break;
            case etoPartitioningKey:
                this.columnNamelist = (TObjectNameList)arg2;
                break;
            case etoOrganizeBy:
            case etoOrganizeByDimensions:
                this.valueRowItemList = (TMultiTargetList)arg2;
                break;
        }
    }

    private TMultiTargetList valueRowItemList;

    public TMultiTargetList getValueRowItemList() {
        return valueRowItemList;
    }

    public ArrayList<TAttributeOption> getAttributeOptions() {
        return attributeOptions;
    }

    private ArrayList<TAttributeOption> attributeOptions ;

    private void parseBigQueryTableOption() {
        String options = this.toString().trim();
        if (!options.toUpperCase().startsWith("OPTIONS"))
            return;
        options = options.replaceFirst("(?i)OPTIONS", "").trim();
        options = options.substring(1, options.length() - 1).trim();
        if (!options.endsWith(",")) {
            options = options + ",";
        }

        this.allowJaggedRows = getPatternBoolean(options, "allow_jagged_rows");
        this.allowQuotedNewlines = getPatternBoolean(options, "allow_quoted_newlines");
        this.compression = getPatternString(options, "compression");
        this.description = getPatternString(options, "description");
        this.enableLogicalTypes = getPatternBoolean(options, "enable_logical_types");
        this.encoding = getPatternString(options, "encoding");
        this.expirationTimestamp = getTimestampString(options, "expiration_timestamp");
        this.fieldDelimiter = getPatternString(options, "field_delimiter");
        this.format = getPatternString(options, "format");
        this.decimalTargetTypes = getPatternArray(options, "decimal_target_types");
        this.hivePartitionUriPrefix = getPatternString(options, "hive_partition_uri_prefix");
        this.ignoreUnknownValues = getPatternBoolean(options, "ignore_unknown_values");
        this.maxBadRecords = getPatternLong(options, "max_bad_records");
        this.nullMarker = getPatternString(options, "null_marker");
        this.projectionFields = getPatternString(options, "projection_fields");
        this.quote = getPatternString(options, "quote");
        this.requireHivePartitionFilter = getPatternBoolean(options, "require_hive_partition_filter");
        this.sheetRange = getPatternString(options, "sheet_range");
        this.skipLeadingRows = getPatternLong(options, "skip_leading_rows");
        this.uris = getPatternArray(options, "uris");
    }

    private String getTimestampString(String content, String field) {
        content = content.replaceAll("(\\s*,)+", ",");
        if (content.matches("(?is).*" + field + "\\s*=(.+?)=.+")) {
            String patternExp = "(?is)" + field + "\\s*=(.+?)=";
            Pattern pattern = Pattern.compile(patternExp);
            Matcher matcher = pattern.matcher(content);
            if (matcher.find()) {
                String timestamp = matcher.group(1).trim();
                timestamp = timestamp.substring(0, timestamp.lastIndexOf(",")).trim();
                return timestamp;
            }
        }
        else if(content.matches("(?is).*" + field + "\\s*=(.+),")){
            String patternExp = "(?is)" + field + "\\s*=(.+),";
            Pattern pattern = Pattern.compile(patternExp);
            Matcher matcher = pattern.matcher(content);
            if (matcher.find()) {
                String timestamp = matcher.group(1).trim();
                return timestamp;
            }
        } 
        return null;
    }

    private String getPatternString(String content, String field) {
        String patternExp = "(?is)" + field + "\\s*=\\s*(\".+?\"),";
        Pattern pattern = Pattern.compile(patternExp);
        Matcher matcher = pattern.matcher(content);
        if (matcher.find()) {
            return matcher.group(1).trim();
        }
        patternExp = "(?is)" + field + "\\s*=\\s*('.+?\'),";
        pattern = Pattern.compile(patternExp);
        matcher = pattern.matcher(content);
        if (matcher.find()) {
            return matcher.group(1).trim();
        }
        return null;
    }

    private Boolean getPatternBoolean(String content, String field) {
        String patternExp = "(?is)" + field + "\\s*=(.+?),";
        Pattern pattern = Pattern.compile(patternExp);
        Matcher matcher = pattern.matcher(content);
        if (matcher.find()) {
            return Boolean.parseBoolean(matcher.group(1).trim());
        }
        return null;
    }

    private Long getPatternLong(String content, String field) {
        String patternExp = "(?is)" + field + "\\s*=(.+?),";
        Pattern pattern = Pattern.compile(patternExp);
        Matcher matcher = pattern.matcher(content);
        if (matcher.find()) {
            return Long.parseLong(matcher.group(1).trim());
        }
        return null;
    }

    private List<String> getPatternArray(String content, String field) {
        String patternExp = "(?is)" + field + "\\s*=\\s*\\[(.+?)\\]\\s*,";
        Pattern pattern = Pattern.compile(patternExp);
        Matcher matcher = pattern.matcher(content);
        if (matcher.find()) {
            String arrayItems = matcher.group(1).trim();
            if (arrayItems.startsWith("(") && arrayItems.endsWith(")")) {
                arrayItems = arrayItems.substring(1, arrayItems.length() - 1).trim();
            }
            return Arrays.asList(arrayItems.split("\\s*,\\s*"));
        }
        return null;
    }

    public void doParse(TCustomSqlStatement psql, ESqlClause plocation) {

        if (psql instanceof TCreateTableSqlStatement) {
            TCreateTableSqlStatement c = (TCreateTableSqlStatement) psql;

            switch (this.createTableOptionType){
                case etoPartitionBy:
                    if (this.partitionColumnList != null){
                        c.setPartitionColumnList(this.partitionColumnList);
                    }else if (this.partitionByExpr != null){
                        c.setPartitionByExpr(this.partitionByExpr);
                        this.partitionByExpr.doParse(c,ESqlClause.createTable);
                    }
                    break;
                case etoPattern:
                    c.setRegex_pattern(this.getEndToken().toString());
                    break;
                case etoComment:
                    c.setTableComment(TObjectName.createObjectName (psql.dbvendor, EDbObjectType.comment, this.commentToken));
                    break;
                case etoWithLocation:
                    c.setStageLocation(this.getStageLocation());
                    break;
                case etoFileFormat:
                    c.setFileFormatName(this.fileFormatName);
                    c.setFileFormatType(this.fileFormatType);
                    break;
                case etoAWSSnsTopic:
                    c.setAwsSnsTopic(this.awsSnsTopic);
                    break;
                case etoBigQueryExternal:
                    parseBigQueryTableOption();
                    break;
                case etoClusterBy:
                    if (expressionList != null){
                        expressionList.doParse(psql,plocation);
                    }
                    break;
                default:
                    break;

            }
        }else if(psql instanceof TCreateStageStmt){
            TCreateStageStmt c = (TCreateStageStmt)psql;
            switch (this.createTableOptionType){
                case etoFileFormat:
                    c.setFileFormatName(this.fileFormatName);
                    c.setFileFormatType(this.fileFormatType);
                    break;
                case etoStageURL:
                    c.setExternalStageURL(this.externalStageURL);
                    break;
                default:
                    break;
            }
        }else if(psql instanceof TSnowflakeCopyIntoStmt){
            TSnowflakeCopyIntoStmt c = (TSnowflakeCopyIntoStmt)psql;
            switch (this.createTableOptionType){
                case etoFileFormat:
                    c.setFileFormatName(this.fileFormatName);
                    c.setFileFormatType(this.fileFormatType);
                    break;
                case etoStageURL:
                    break;
                case etoPattern:
                    c.setRegex_pattern(this.getEndToken().toString());
                    break;
                case etoFiles:
                    for(TExpression e:this.expressionList){
                        c.getFileList().add(e.toString());
                    }
                    break;
                default:
                    break;
            }
        }
    }

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