feat(cli): enable token rotation via CLI

nejch · JohnVillalovos · commit 0cb817153d81 · 2024-11-26T09:33:21.000-08:00
diff --git a/gitlab/mixins.py b/gitlab/mixins.py
@@ -663,6 +663,14 @@ class RotateMixin(_RestManagerBase):
     _path: Optional[str]
     gitlab: gitlab.Gitlab
 
+    @cli.register_custom_action(
+        cls_names=(
+            "PersonalAccessTokenManager",
+            "GroupAccessTokenManager",
+            "ProjectAccessTokenManager",
+        ),
+        optional=("expires_at",),
+    )
     @exc.on_http_error(exc.GitlabRotateError)
     def rotate(
         self, id: Union[str, int], expires_at: Optional[str] = None, **kwargs: Any
@@ -696,7 +704,12 @@ class ObjectRotateMixin(_RestObjectBase):
     _updated_attrs: Dict[str, Any]
     manager: base.RESTManager
 
-    def rotate(self, **kwargs: Any) -> None:
+    @cli.register_custom_action(
+        cls_names=("PersonalAccessToken", "GroupAccessToken", "ProjectAccessToken"),
+        optional=("expires_at",),
+    )
+    @exc.on_http_error(exc.GitlabRotateError)
+    def rotate(self, **kwargs: Any) -> Dict[str, Any]:
         """Rotate the current access token object.
 
         Args:
@@ -711,6 +724,7 @@ def rotate(self, **kwargs: Any) -> None:
             assert self.encoded_id is not None
         server_data = self.manager.rotate(self.encoded_id, **kwargs)
         self._update_attrs(server_data)
+        return server_data
 
 
 class SubscribableMixin(_RestObjectBase):
