update recovery after restart

src/gradio_space_ci/webhook.py

@@ -206,6 +206,8 @@ def recover_after_restart(space_id: str) -> None:
                 # Found a PR that is not yet synced
                 print(f"Recovery. Found an open PR that is not synced: {discussion.url}. Syncing it.")
                 background_pool.submit(sync_ci_space, space_id=space_id, pr_num=discussion.num)
+                # Check if this pr should be trusted or not
+                handle_trust_status(space_id=space_id, pr_num=discussion.num)
         if discussion.status == "merged" or discussion.status == "closed":
             ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=discussion.num)
             if repo_exists(repo_id=ci_space_id, repo_type="space"):
@@ -214,6 +216,42 @@ def recover_after_restart(space_id: str) -> None:
                 background_pool.submit(delete_ci_space, space_id=space_id, pr_num=discussion.num)
 
 
+def handle_trust_status(space_id: str, pr_num: int) -> None:
+    """after recovering we check each pr if it is trusted or not"""
+    events = get_discussion_details(repo_id=space_id, repo_type="space", discussion_num=pr_num).events
+    # flip history, last event is the first one
+    events.reverse()
+    for event in events:
+        # a commit and no commands found yet
+        # is it by a trusted author or not ?
+        if event.type == "commit":
+            if event.author not in EPHEMERAL_SPACES_CONFIG["trusted_authors"]:
+                # a commit by untrusted author
+                unset_config(space_id=space_id, pr_num=pr_num)
+                return
+            else:
+                # a commit by a trusted author
+                set_config(space_id=space_id, pr_num=pr_num)
+        
+        # a comment by a trusted author
+        elif (
+            event.type == "comment"
+            and event.author in EPHEMERAL_SPACES_CONFIG["trusted_authors"]
+            and event._event["data"]["edited"] is False
+            and event._event["data"]["hidden"] is False
+        ):
+            comment = event._event["data"]["latest"]["raw"]
+            if comment == "/trust_pr":
+                set_config(space_id=space_id, pr_num=pr_num)
+                return
+            elif comment == "/untrust_pr":
+                unset_config(space_id=space_id, pr_num=pr_num)
+                return
+            elif "<em>(This is an automated message.)</em>" in comment:
+                # synced and no commands yet
+                return
+
+
 ###
 # Define webhook on the Hub logic
 ###
@@ -437,7 +475,9 @@ def delete_ci_space(space_id: str, pr_num: int) -> None:
 def notify_pr(
     space_id: str,
     pr_num: int,
-    action: Literal["created_not_configured", "created_and_configured", "updated", "deleted","trusted_pr","untrusted_pr"],
+    action: Literal[
+        "created_not_configured", "created_and_configured", "updated", "deleted", "trusted_pr", "untrusted_pr"
+    ],
 ) -> None:
     ci_space_id = _get_ci_space_id(space_id=space_id, pr_num=pr_num)
     if action == "created_not_configured":
@@ -448,7 +488,7 @@ def notify_pr(
         comment = NOTIFICATION_TEMPLATE_UPDATED.format(ci_space_id=ci_space_id)
     elif action == "deleted":
         comment = NOTIFICATION_TEMPLATE_DELETED
-    elif action == "trusted_pr": 
+    elif action == "trusted_pr":
         comment = NOTIFICATION_TEMPLATE_TRUSTED_PR
     elif action == "untrusted_pr":
         comment = NOTIFICATION_TEMPLATE_UNTRUSTED_PR
@@ -516,11 +556,10 @@ def handle_command(space_id: str, payload: WebhookPayload) -> None:
     if event_author in EPHEMERAL_SPACES_CONFIG["trusted_authors"]:
         if payload.comment.content == "/trust_pr":
             set_config(space_id=space_id, pr_num=pr_num)
-            notify_pr(space_id=space_id,pr_num=pr_num,action="trusted_pr")
+            notify_pr(space_id=space_id, pr_num=pr_num, action="trusted_pr")
         elif payload.comment.content == "/untrust_pr":
             unset_config(space_id=space_id, pr_num=pr_num)
-            notify_pr(space_id=space_id,pr_num=pr_num,action="untrusted_pr")
-            
+            notify_pr(space_id=space_id, pr_num=pr_num, action="untrusted_pr")
 
 
 NOTIFICATION_TEMPLATE_CREATED_AND_CONFIGURED = """\