feat: rollback snapshot

Author: jooola

hcloud/storage_boxes/client.py

@@ -12,6 +12,7 @@
     StorageBox,
     StorageBoxAccessSettings,
     StorageBoxFoldersResponse,
+    StorageBoxSnapshot,
     StorageBoxSnapshotPlan,
     StorageBoxStats,
 )
@@ -492,3 +493,29 @@ def update_access_settings(
             json=data,
         )
         return BoundAction(self._parent.actions, response["action"])
+
+    def rollback_snapshot(
+        self,
+        storage_box: StorageBox | BoundStorageBox,
+        # TODO: Add BoundStorageBoxSnapshot
+        # TODO: snapshot or storage_box_snapshot argument name?
+        snapshot: StorageBoxSnapshot,
+    ) -> BoundAction:
+        """
+        Rollback the Storage Box to the given snapshot.
+
+        See https://docs.hetzner.cloud/reference/cloud#TODO
+
+        :param storage_box: Storage Box to update.
+        :param snapshot: Snapshot to rollback to.
+        """
+        data: dict[str, Any] = {
+            "snapshot": snapshot.id_or_name,
+        }
+
+        response = self._client.request(
+            method="POST",
+            url=f"{self._base_url}/{storage_box.id}/actions/rollback_snapshot",
+            json=data,
+        )
+        return BoundAction(self._parent.actions, response["action"])

hcloud/storage_boxes/domain.py

@@ -227,3 +227,28 @@ def __init__(
         folders: list[str],
     ):
         self.folders = folders
+
+
+# Snapshots
+###############################################################################
+
+
+class StorageBoxSnapshot(BaseDomain, DomainIdentityMixin):
+    """
+    Storage Box Snapshot Domain.
+    """
+
+    # TODO: full domain
+    __api_properties__ = (
+        "id",
+        "name",
+    )
+    __slots__ = __api_properties__
+
+    def __init__(
+        self,
+        id: int | None = None,
+        name: str | None = None,
+    ):
+        self.id = id
+        self.name = name

tests/unit/storage_boxes/test_client.py

@@ -15,7 +15,7 @@
     StorageBox,
     StorageBoxesClient,
 )
-from hcloud.storage_boxes.domain import StorageBoxAccessSettings
+from hcloud.storage_boxes.domain import StorageBoxAccessSettings, StorageBoxSnapshot
 
 from ..conftest import BoundModelTestCase, assert_bound_action1
 
@@ -398,3 +398,24 @@ def test_update_access_settings(
         )
 
         assert_bound_action1(action, resource_client._parent.actions)
+
+    def test_rollback_snapshot(
+        self,
+        request_mock: mock.MagicMock,
+        resource_client: StorageBoxesClient,
+        action_response,
+    ):
+        request_mock.return_value = action_response
+
+        action = resource_client.rollback_snapshot(
+            StorageBox(id=42),
+            StorageBoxSnapshot(id=32),
+        )
+
+        request_mock.assert_called_with(
+            method="POST",
+            url="/storage_boxes/42/actions/rollback_snapshot",
+            json={"snapshot": 32},
+        )
+
+        assert_bound_action1(action, resource_client._parent.actions)