Resend a workspace member's invite

atst/domain/invitations.py

@@ -4,6 +4,8 @@ from sqlalchemy.orm.exc import NoResultFound
 from atst.database import db
 from atst.models.invitation import Invitation, Status as InvitationStatus
 from atst.domain.workspace_roles import WorkspaceRoles
+from atst.domain.authz import Authorization, Permissions
+from atst.domain.workspaces import Workspaces
 
 from .exceptions import NotFoundError
 
@@ -104,3 +106,18 @@ class Invitations(object):
     def revoke(cls, token):
         invite = Invitations._get(token)
         return Invitations._update_status(invite, InvitationStatus.REVOKED)
+
+    @classmethod
+    def resend(cls, user, workspace_id, token):
+        workspace = Workspaces.get(user, workspace_id)
+        Authorization.check_workspace_permission(
+            user,
+            workspace,
+            Permissions.ASSIGN_AND_UNASSIGN_ATAT_ROLE,
+            "resend a workspace invitation",
+        )
+
+        previous_invitation = Invitations._get(token)
+        Invitations._update_status(previous_invitation, InvitationStatus.REVOKED)
+
+        return Invitations.create(user, previous_invitation.workspace_role)

atst/models/workspace_role.py

@@ -109,6 +109,12 @@ class WorkspaceRole(Base, mixins.TimestampsMixin, mixins.AuditableMixin):
     def has_environment_roles(self):
         return self.num_environment_roles > 0
 
+    @property
+    def can_resend_invitation(self):
+        return self.latest_invitation and (
+            self.latest_invitation.is_rejected or self.latest_invitation.is_expired
+        )
+
 
 Index(
     "workspace_role_user_workspace",

atst/routes/workspaces.py

@@ -374,3 +374,9 @@ def revoke_invitation(workspace_id, token):
     Invitations.revoke(token)
 
     return redirect(url_for("workspaces.workspace_members", workspace_id=workspace.id))
+
+
+@bp.route("/workspaces/<workspace_id>/invitations/<token>/resend", methods=["POST"])
+def resend_invitation(workspace_id, token):
+    Invitations.resend(g.current_user, workspace_id, token)
+    return redirect(url_for("workspaces.workspace_members", workspace_id=workspace_id))