From a21efb99dffc7818dc8e0fdf8146ec06c8a37bff Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Tue, 28 Nov 2023 13:55:05 +0300
Subject: [PATCH] author-invitee-fix

---
 orm/invite.py        |  4 ++--
 resolvers/collab.py  | 16 ++++++++--------
 schemas/core.graphql | 17 +++++++++++++++++
 3 files changed, 27 insertions(+), 10 deletions(-)

diff --git a/orm/invite.py b/orm/invite.py
index 3032693f..e8c7fdfa 100644
--- a/orm/invite.py
+++ b/orm/invite.py
@@ -16,10 +16,10 @@ class Invite(Base):
     __tablename__ = "invite"
 
     inviter_id = Column(ForeignKey("author.id"), nullable=False, index=True)
-    invitee_id = Column(ForeignKey("author.id"), nullable=False, index=True)
+    author_id = Column(ForeignKey("author.id"), nullable=False, index=True)
     shout_id = Column(ForeignKey("shout.id"), nullable=False, index=True)
     status = Column(Enum(InviteStatus), default=InviteStatus.PENDING)
 
     inviter = relationship(Author, foreign_keys=[inviter_id])
-    invitee = relationship(Author, foreign_keys=[invitee_id])
+    author = relationship(Author, foreign_keys=[author_id])
     shout = relationship(Shout)
diff --git a/resolvers/collab.py b/resolvers/collab.py
index 3eb9b966..7a26e8a5 100644
--- a/resolvers/collab.py
+++ b/resolvers/collab.py
@@ -17,7 +17,7 @@ async def accept_invite(_, info, invite_id: int):
         if author:
             # Check if the invite exists
             invite = session.query(Invite).filter(Invite.id == invite_id).first()
-            if invite and invite.invitee_id == author.id and invite.status == InviteStatus.PENDING:
+            if invite and invite.author_id == author.id and invite.status == InviteStatus.PENDING:
                 # Add the user to the shout authors
                 shout = session.query(Shout).filter(Shout.id == invite.shout_id).first()
                 if shout:
@@ -44,7 +44,7 @@ async def reject_invite(_, info, invite_id: int):
         if author:
             # Check if the invite exists
             invite = session.query(Invite).filter(Invite.id == invite_id).first()
-            if invite and invite.invitee_id == author.id and invite.status == InviteStatus.PENDING:
+            if invite and invite.author_id == author.id and invite.status == InviteStatus.PENDING:
                 # Delete the invite
                 session.delete(invite)
                 session.commit()
@@ -64,15 +64,15 @@ async def create_invite(_, info, slug: str = "", author_id: int = None, user: st
     with local_session() as session:
         shout = session.query(Shout).filter(Shout.slug == slug).first()
         if shout and shout.authors and user_id in [author.id for author in shout.authors]:
-            # Check if the invitee is a valid author
-            invitee = session.query(Author).filter(Author.id == author_id).first()
-            if invitee:
+            # Check if the author is a valid author
+            author = session.query(Author).filter(Author.id == author_id).first()
+            if author:
                 # Check if an invite already exists
                 existing_invite = (
                     session.query(Invite)
                     .filter(
                         Invite.inviter_id == user_id,
-                        Invite.invitee_id == author_id,
+                        Invite.author_id == author_id,
                         Invite.shout_id == shout.id,
                         Invite.status == InviteStatus.PENDING,
                     )
@@ -83,14 +83,14 @@ async def create_invite(_, info, slug: str = "", author_id: int = None, user: st
 
                 # Create a new invite
                 new_invite = Invite(
-                    inviter_id=user_id, invitee_id=author_id, shout_id=shout.id, status=InviteStatus.PENDING
+                    inviter_id=user_id, author_id=author_id, shout_id=shout.id, status=InviteStatus.PENDING
                 )
                 session.add(new_invite)
                 session.commit()
 
                 return {"error": None, "invite": new_invite}
             else:
-                return {"error": "Invalid invitee"}
+                return {"error": "Invalid author"}
         else:
             return {"error": "Access denied"}
 
diff --git a/schemas/core.graphql b/schemas/core.graphql
index 9134d326..a1014cad 100644
--- a/schemas/core.graphql
+++ b/schemas/core.graphql
@@ -36,6 +36,14 @@ enum FollowingEntity {
   REACTIONS
 }
 
+
+enum InviteStatus {
+  PENDING
+  ACCEPTED
+  REJECTED
+}
+
+
 # Типы
 
 
@@ -173,6 +181,14 @@ type Topic {
   oid: String
 }
 
+type Invite {
+  id: Int!
+  inviter_id: Int!
+  author_id: Int!
+  shout_id: Int!
+  status: InviteStatus
+}
+
 # Входные типы
 
 input ShoutInput {
@@ -288,6 +304,7 @@ type Result {
   communities: [Community]
 }
 
+
 # Мутации
 
 type Mutation {