package gudusoft.gsqlparser.nodes;

import gudusoft.gsqlparser.nodes.flink.TFlinkTableProperty;

/**
 * AST node for StarRocks RESTORE SNAPSHOT statement.
 *
 * Syntax (v3.4.0+):
 * RESTORE SNAPSHOT [<db_name>.]<snapshot_name>
 * FROM <repository_name>
 * [DATABASE <db_name_in_snapshot> [AS <target_db>]]
 * [ON ( restore_object [, ...] )]
 * [PROPERTIES ("key"="value" [, ...])]
 *
 * restore_object:
 *   { ALL TABLES | TABLES <table_name>[, ...] [AS <alias>] } |
 *   { ALL MATERIALIZED VIEWS | MATERIALIZED VIEWS <mv_name>[, ...] [AS <alias>] } |
 *   { ALL VIEWS | VIEWS <view_name>[, ...] [AS <alias>] } |
 *   { ALL FUNCTIONS | FUNCTIONS <udf_name>[, ...] [AS <alias>] } |
 *   <table_name> PARTITION (<partition_name>[, ...]) [AS <alias>]
 *
 * Example:
 * RESTORE SNAPSHOT sr_hub_backup
 * FROM test_repo
 * DATABASE sr_hub AS sr_hub_new
 * ON (TABLE sr_member AS new_member)
 * PROPERTIES ("backup_timestamp" = "2024-12-09-10-25-58-842");
 */
public class TRestoreSnapshotSqlNode extends TParseTreeNode {
    // SNAPSHOT name (may be qualified as db.snapshot_name)
    private TObjectName snapshotName;

    // FROM clause - repository name
    private TObjectName repositoryName;

    // DATABASE clause - source database name in snapshot
    private TObjectName sourceDatabaseName;

    // DATABASE ... AS clause - target database name for restore
    private TObjectName targetDatabaseName;

    // Has explicit DATABASE keyword
    private boolean hasDatabase;

    // ON clause - objects to restore
    private TParseTreeNode restoreObjects;

    // 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;

    // Alias for object list (AS clause)
    private TObjectName objectAlias;

    // Table name for partition restore
    private TObjectName partitionTableName;

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

    // Alias for partition table (AS clause)
    private TObjectName partitionTableAlias;

    // PROPERTIES clause
    private TParseTreeNode properties;

    // Getters and setters
    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 TObjectName getSourceDatabaseName() {
        return sourceDatabaseName;
    }

    public void setSourceDatabaseName(TObjectName sourceDatabaseName) {
        this.sourceDatabaseName = sourceDatabaseName;
    }

    public TObjectName getTargetDatabaseName() {
        return targetDatabaseName;
    }

    public void setTargetDatabaseName(TObjectName targetDatabaseName) {
        this.targetDatabaseName = targetDatabaseName;
    }

    public boolean hasDatabase() {
        return hasDatabase;
    }

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

    public TParseTreeNode getRestoreObjects() {
        return restoreObjects;
    }

    public void setRestoreObjects(TParseTreeNode restoreObjects) {
        this.restoreObjects = restoreObjects;
    }

    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 getObjectAlias() {
        return objectAlias;
    }

    public void setObjectAlias(TObjectName objectAlias) {
        this.objectAlias = objectAlias;
    }

    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;
    }

    public TObjectName getPartitionTableAlias() {
        return partitionTableAlias;
    }

    public void setPartitionTableAlias(TObjectName partitionTableAlias) {
        this.partitionTableAlias = partitionTableAlias;
    }

    @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;
    }
}