################################################################################
#  Licensed to the Apache Software Foundation (ASF) under one
#  or more contributor license agreements.  See the NOTICE file
#  distributed with this work for additional information
#  regarding copyright ownership.  The ASF licenses this file
#  to you under the Apache License, Version 2.0 (the
#  "License"); you may not use this file except in compliance
#  with the License.  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
# limitations under the License.
################################################################################

"""
Commit rollback to rollback 'COMPACT' commits for resolving conflicts.
"""

from pypaimon.table.instant import Instant


class CommitRollback:
    """Rollback COMPACT commits to resolve conflicts.

    When a conflict is detected during commit, if the latest snapshot is a
    COMPACT commit, it can be rolled back via TableRollback.
    """

    def __init__(self, table_rollback):
        """Initialize CommitRollback.

        Args:
            table_rollback: A TableRollback instance used to perform the rollback.
        """
        self._table_rollback = table_rollback

    def try_to_rollback(self, latest_snapshot):
        """Try to rollback a COMPACT commit to resolve conflicts.

        Only rolls back COMPACT type commits. Delegates to TableRollback
        to rollback to the previous snapshot (latest - 1), passing the
        latest snapshot ID as from_snapshot.

        Args:
            latest_snapshot: The latest snapshot that may need to be rolled back.

        Returns:
            True if rollback succeeded, False otherwise.
        """
        if latest_snapshot.commit_kind == "COMPACT":
            latest_id = latest_snapshot.id
            try:
                self._table_rollback.rollback_to(
                    Instant.snapshot(latest_id - 1), latest_id)
                return True
            except Exception:
                pass
        return False