package gudusoft.gsqlparser.stmt;

import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.ESqlStatementType;
import gudusoft.gsqlparser.TCustomSqlStatement;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.nodes.flink.TFlinkTableProperty;

/**
 * RESTORE SNAPSHOT statement for StarRocks.
 *
 * 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" [, ...])]
 *
 * This statement restores database objects from a repository backup.
 * The operation is asynchronous - use SHOW RESTORE to monitor progress.
 */
public class TStarrocksRestoreStmt extends TCustomSqlStatement {

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

    // 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 TPTNodeList<TFlinkTableProperty> properties;

    public TStarrocksRestoreStmt(EDbVendor dbvendor) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sststarrocksRestore;
    }

    // Getters
    public TObjectName getSnapshotName() {
        return snapshotName;
    }

    public TObjectName getRepositoryName() {
        return repositoryName;
    }

    public TObjectName getSourceDatabaseName() {
        return sourceDatabaseName;
    }

    public TObjectName getTargetDatabaseName() {
        return targetDatabaseName;
    }

    public boolean hasDatabase() {
        return hasDatabase;
    }

    public boolean isAllTables() {
        return allTables;
    }

    public boolean isAllMaterializedViews() {
        return allMaterializedViews;
    }

    public boolean isAllViews() {
        return allViews;
    }

    public boolean isAllFunctions() {
        return allFunctions;
    }

    public String getObjectType() {
        return objectType;
    }

    public TObjectNameList getObjectList() {
        return objectList;
    }

    public TObjectName getObjectAlias() {
        return objectAlias;
    }

    public TObjectName getPartitionTableName() {
        return partitionTableName;
    }

    public TObjectNameList getPartitionList() {
        return partitionList;
    }

    public TObjectName getPartitionTableAlias() {
        return partitionTableAlias;
    }

    public TPTNodeList<TFlinkTableProperty> getProperties() {
        return properties;
    }

    @Override
    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        super.doParseStatement(psql);

        TRestoreSnapshotSqlNode node = (TRestoreSnapshotSqlNode) rootNode;

        this.snapshotName = node.getSnapshotName();
        this.repositoryName = node.getRepositoryName();
        this.sourceDatabaseName = node.getSourceDatabaseName();
        this.targetDatabaseName = node.getTargetDatabaseName();
        this.hasDatabase = node.hasDatabase();
        this.allTables = node.isAllTables();
        this.allMaterializedViews = node.isAllMaterializedViews();
        this.allViews = node.isAllViews();
        this.allFunctions = node.isAllFunctions();
        this.objectType = node.getObjectType();
        this.objectList = node.getObjectList();
        this.objectAlias = node.getObjectAlias();
        this.partitionTableName = node.getPartitionTableName();
        this.partitionList = node.getPartitionList();
        this.partitionTableAlias = node.getPartitionTableAlias();
        this.properties = node.getProperties();

        return 0;
    }

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

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