package gudusoft.gsqlparser.nodes;

import gudusoft.gsqlparser.nodes.flink.TFlinkTableProperty;

/**
 * AST node for StarRocks BACKUP SNAPSHOT statement.
 *
 * Syntax (v3.4.0+):
 * BACKUP [DATABASE <db_name>] SNAPSHOT [<db_name>.]<snapshot_name>
 * TO <repository_name>
 * [ON ( backup_object [, ...] )]
 * [PROPERTIES ("key"="value" [, ...])]
 *
 * backup_object:
 *   { ALL TABLES | TABLES <table_name>[, ...] } |
 *   { ALL MATERIALIZED VIEWS | MATERIALIZED VIEWS <mv_name>[, ...] } |
 *   { ALL VIEWS | VIEWS <view_name>[, ...] } |
 *   { ALL FUNCTIONS | FUNCTIONS <udf_name>[, ...] } |
 *   <table_name> PARTITION (<partition_name>[, ...])
 *
 * Example:
 * BACKUP DATABASE sr_hub SNAPSHOT sr_hub_backup TO test_repo
 * ON (TABLE sr_member PARTITION (p1, p2, p3))
 * PROPERTIES ("type" = "full");
 */
public class TBackupSnapshotSqlNode extends TParseTreeNode {
    // DATABASE clause - optional database name
    private TObjectName databaseName;

    // Has explicit DATABASE keyword
    private boolean hasDatabase;

    // SNAPSHOT name (may be qualified as db.snapshot_name)
    private TObjectName snapshotName;

    // TO clause - repository name
    private TObjectName repositoryName;

    // ON clause - objects to backup
    // Can contain: ALL TABLES, TABLES list, ALL MATERIALIZED VIEWS, MATERIALIZED VIEWS list,
    // ALL VIEWS, VIEWS list, ALL FUNCTIONS, FUNCTIONS list, or table with partitions
    private TParseTreeNode backupObjects;

    // Flags for ALL object types
    private boolean allTables;
    private boolean allMaterializedViews;
    private boolean allViews;
    private boolean allFunctions;

    // Object type for ON clause (tables, materialized_views, views, functions)
    private String objectType;

    // List of specific objects (tables, views, mvs, functions)
    private TObjectNameList objectList;

    // Table name for partition backup
    private TObjectName partitionTableName;

    // Partition list for specific table backup
    private TObjectNameList partitionList;

    // PROPERTIES clause
    private TParseTreeNode properties;

    // Getters and setters
    public TObjectName getDatabaseName() {
        return databaseName;
    }

    public void setDatabaseName(TObjectName databaseName) {
        this.databaseName = databaseName;
    }

    public boolean hasDatabase() {
        return hasDatabase;
    }

    public void setHasDatabase(boolean hasDatabase) {
        this.hasDatabase = hasDatabase;
    }

    public TObjectName getSnapshotName() {
        return snapshotName;
    }

    public void setSnapshotName(TObjectName snapshotName) {
        this.snapshotName = snapshotName;
    }

    public TObjectName getRepositoryName() {
        return repositoryName;
    }

    public void setRepositoryName(TObjectName repositoryName) {
        this.repositoryName = repositoryName;
    }

    public TParseTreeNode getBackupObjects() {
        return backupObjects;
    }

    public void setBackupObjects(TParseTreeNode backupObjects) {
        this.backupObjects = backupObjects;
    }

    public boolean isAllTables() {
        return allTables;
    }

    public void setAllTables(boolean allTables) {
        this.allTables = allTables;
    }

    public boolean isAllMaterializedViews() {
        return allMaterializedViews;
    }

    public void setAllMaterializedViews(boolean allMaterializedViews) {
        this.allMaterializedViews = allMaterializedViews;
    }

    public boolean isAllViews() {
        return allViews;
    }

    public void setAllViews(boolean allViews) {
        this.allViews = allViews;
    }

    public boolean isAllFunctions() {
        return allFunctions;
    }

    public void setAllFunctions(boolean allFunctions) {
        this.allFunctions = allFunctions;
    }

    public String getObjectType() {
        return objectType;
    }

    public void setObjectType(String objectType) {
        this.objectType = objectType;
    }

    public TObjectNameList getObjectList() {
        return objectList;
    }

    public void setObjectList(TObjectNameList objectList) {
        this.objectList = objectList;
    }

    public TObjectName getPartitionTableName() {
        return partitionTableName;
    }

    public void setPartitionTableName(TObjectName partitionTableName) {
        this.partitionTableName = partitionTableName;
    }

    public TObjectNameList getPartitionList() {
        return partitionList;
    }

    public void setPartitionList(TObjectNameList partitionList) {
        this.partitionList = partitionList;
    }

    @SuppressWarnings("unchecked")
    public TPTNodeList<TFlinkTableProperty> getProperties() {
        return (TPTNodeList<TFlinkTableProperty>) properties;
    }

    public void setProperties(TParseTreeNode properties) {
        this.properties = properties;
    }

    public void init(Object arg1) {
        this.snapshotName = (TObjectName) arg1;
    }
}