package gudusoft.gsqlparser.nodes.redshift;

import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.nodes.TObjectNameList;
import gudusoft.gsqlparser.nodes.TParseTreeNode;
import gudusoft.gsqlparser.nodes.TObjectName;
import gudusoft.gsqlparser.nodes.TParseTreeVisitor;

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

/**
 * Represents the table_attributes for Redshift materialized views
 * Based on: https://docs.aws.amazon.com/redshift/latest/dg/materialized-view-create-sql-command.html
 */ 
public class TRedshiftTableAttributes extends TParseTreeNode {

    /**
     * Possible distribution styles for Redshift materialized views
     */
    public enum EDistStyle {
        EVEN, // Distribute data evenly across the nodes (default)
        ALL,  // Distribute entire copy of data to each node
        KEY   // Distribute according to the values in DISTKEY column
    }
    
    // The type of table attribute (distribution style, distribution key, sort key)
    private ETableAttributeType attributeType = ETableAttributeType.ttUnknown;
    
    // Distribution style
    private EDistStyle distStyle = EDistStyle.EVEN; // Default is EVEN
    
    // Distribution key column identifier
    private TObjectName distKey;
    
    // Sort key columns
    private TObjectNameList sortKeyColumns ;
    
    /**
     * Default constructor
     */
    public TRedshiftTableAttributes() {
        this.dbvendor = EDbVendor.dbvredshift;
    }
    
    /**
     * Constructor with distribution style
     * @param distStyle The distribution style to use
     */
    public TRedshiftTableAttributes(EDistStyle distStyle) {
        this();
        this.distStyle = distStyle;
        this.attributeType = ETableAttributeType.ttDistStyle;
    }
    

    public void init(Object arg1) {
        this.attributeType = (ETableAttributeType)arg1;
    }

    public void init(Object arg1, Object arg2) {
        this.attributeType = (ETableAttributeType)arg1;
        switch (attributeType) {
            case ttDistStyle:
                this.distStyle = (EDistStyle)arg2;
                break;
            case ttDistKey:
                this.distKey = (TObjectName)arg2;
                break;
            case ttSortKey:
                this.sortKeyColumns = (TObjectNameList)arg2;
                break;
            default:
                break;
        }
    }

    /**
     * Constructor with distribution style and key
     * @param distStyle The distribution style to use
     * @param distKey The distribution key column
     */
    public TRedshiftTableAttributes(EDistStyle distStyle, TObjectName distKey) {
        this(distStyle);
        this.distKey = distKey;
        if (distStyle == EDistStyle.KEY) {
            this.attributeType = ETableAttributeType.ttDistKey;
        }
    }
    
    /**
     * Get the attribute type
     * @return The type of table attribute
     */
    public ETableAttributeType getAttributeType() {
        return attributeType;
    }
    
    /**
     * Set the attribute type
     * @param attributeType The type of table attribute
     */
    public void setAttributeType(ETableAttributeType attributeType) {
        this.attributeType = attributeType;
    }
    
    /**
     * Get the distribution style
     * @return The distribution style
     */
    public EDistStyle getDistStyle() {
        return distStyle;
    }
    
    /**
     * Set the distribution style
     * @param distStyle The distribution style
     * @return this object for method chaining
     */
    public TRedshiftTableAttributes setDistStyle(EDistStyle distStyle) {
        this.distStyle = distStyle;
        this.attributeType = ETableAttributeType.ttDistStyle;
        return this;
    }
    
    /**
     * Get the distribution key
     * @return The distribution key column identifier
     */
    public TObjectName getDistKey() {
        return distKey;
    }
    
    /**
     * Set the distribution key
     * @param distKey The column identifier to use as distribution key
     * @return this object for method chaining
     */
    public TRedshiftTableAttributes setDistKey(TObjectName distKey) {
        this.distKey = distKey;
        if (this.distStyle == EDistStyle.KEY) {
            this.attributeType = ETableAttributeType.ttDistKey;
        }
        return this;
    }
    
    /**
     * Get the sort key columns
     * @return List of sort key columns
     */
    public TObjectNameList getSortKeyColumns() {
        return sortKeyColumns;
    }
    

   
    /**
     * Generate the SQL fragment for table_attributes
     * @return SQL string representing the table attributes
     */
    public String toSql() {
        StringBuilder sql = new StringBuilder();
        
        // Add distribution style
        if (this.attributeType == ETableAttributeType.ttDistStyle || 
            this.attributeType == ETableAttributeType.ttDistKey) {
            sql.append("DISTSTYLE ").append(distStyle.name());
        }
        
        // Add distribution key if specified and style is KEY
        if (this.attributeType == ETableAttributeType.ttDistKey && 
            distStyle == EDistStyle.KEY && 
            distKey != null) {
            sql.append(" DISTKEY (").append(distKey).append(")");
        }
        
        // Add sort key if columns are specified
        if (this.attributeType == ETableAttributeType.ttSortKey && sortKeyColumns != null && sortKeyColumns.size() > 0) {
            if (sql.length() > 0) {
                sql.append(" ");
            }
            sql.append("SORTKEY (");
            sql.append(String.join(", ", sortKeyColumns.toString()));
            sql.append(")");
        }
        
        return sql.toString();
    }
    
    @Override
    public String toString() {
        return toSql();
    }

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

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