Repaired register by discord #70

kdb-bot/src/bot_api/abc/auth_service_abc.py

@@ -50,7 +50,7 @@ class AuthServiceABC(ABC):
     async def add_auth_user_by_oauth_async(self, dto: OAuthDTO): pass
 
     @abstractmethod
-    async def add_auth_user_by_discord_async(self, user_dto: AuthUserDTO) -> OAuthDTO: pass
+    async def add_auth_user_by_discord_async(self, user_dto: AuthUserDTO, dc_id: int) -> OAuthDTO: pass
 
     @abstractmethod
     async def update_user_async(self, update_user_dto: UpdateAuthUserDTO): pass

kdb-bot/src/bot_api/config/apisettings.edrafts-pc-ubuntu.json

@@ -1 +1,28 @@
-{}
+{
+  "Api": {
+    "Port": 5000,
+    "Host": "0.0.0.0",
+    "RedirectToHTTPS": false
+  },
+  "Authentication": {
+    "SecretKey": "RjNiNUxEeisjSnZ6Zz1XIUBnc2EleHNG",
+    "Issuer": "http://localhost:5000",
+    "Audience": "http://localhost:4200",
+    "TokenExpireTime": 1,
+    "RefreshTokenExpireTime": 7
+  },
+  "DiscordAuthentication": {
+    "ClientSecret": "V3FTb3JYVFBiVktEeHZxdWJDWW4xcnBCbXRwdmpwcy0=",
+    "_RedirectURL": "http://localhost:5000/api/auth/discord/register",
+    "RedirectURL": "http://localhost:4200/auth/register",
+    "Scope": [
+      "identify",
+      "email"
+    ],
+    "TokenURL": "https://discordapp.com/api/oauth2/token",
+    "AuthURL": "https://discordapp.com/api/oauth2/authorize"
+  },
+  "Frontend": {
+    "URL": "http://localhost:4200/"
+  }
+}

kdb-bot/src/bot_api/controller/auth_discord_controller.py

@@ -78,9 +78,8 @@ class AuthDiscordController:
             response['email'],
             str(uuid.uuid4()),
             None,
-            AuthRoleEnum.normal,
-            response['id']
-        ))
+            AuthRoleEnum.normal
+        ), response['id'])
         return jsonify(result.to_dict())
 
     @Route.post(f'{BasePath}/register')

kdb-bot/src/bot_api/service/auth_service.py

@@ -8,8 +8,9 @@ from typing import Optional
 import jwt
 from cpl_core.database.context import DatabaseContextABC
 from cpl_core.environment import ApplicationEnvironmentABC
-from cpl_core.mailing import EMailClientABC, EMail
+from cpl_core.mailing import EMail
 from cpl_core.utils import CredentialManager
+from cpl_discord.service import DiscordBotServiceABC
 from cpl_query.extension import List
 from cpl_translation import TranslatePipe
 from flask import request
@@ -31,9 +32,12 @@ from bot_api.model.token_dto import TokenDTO
 from bot_api.model.update_auth_user_dto import UpdateAuthUserDTO
 from bot_api.transformer.auth_user_transformer import AuthUserTransformer as AUT
 from bot_data.abc.auth_user_repository_abc import AuthUserRepositoryABC
+from bot_data.abc.server_repository_abc import ServerRepositoryABC
 from bot_data.abc.user_repository_abc import UserRepositoryABC
 from bot_data.model.auth_role_enum import AuthRoleEnum
 from bot_data.model.auth_user import AuthUser
+from bot_data.model.auth_user_users_relation import AuthUserUsersRelation
+from bot_data.model.user import User
 
 _email_regex = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
 
@@ -44,9 +48,11 @@ class AuthService(AuthServiceABC):
             self,
             env: ApplicationEnvironmentABC,
             logger: ApiLogger,
+            bot: DiscordBotServiceABC,
+            db: DatabaseContextABC,
             auth_users: AuthUserRepositoryABC,
             users: UserRepositoryABC,
-            db: DatabaseContextABC,
+            servers: ServerRepositoryABC,
             mailer: MailThread,
             t: TranslatePipe,
             auth_settings: AuthenticationSettings,
@@ -57,9 +63,11 @@ class AuthService(AuthServiceABC):
 
         self._environment = env
         self._logger = logger
+        self._bot = bot
+        self._db = db
         self._auth_users = auth_users
         self._users = users
-        self._db = db
+        self._servers = servers
         self._mailer = mailer
         self._t = t
         self._auth_settings = auth_settings
@@ -134,7 +142,7 @@ class AuthService(AuthServiceABC):
     def _create_and_save_refresh_token(self, user: AuthUser) -> str:
         token = str(uuid.uuid4())
         user.refresh_token = token
-        user.refresh_token_expire_time = datetime.now(tz=timezone.utc) + timedelta(days=self._auth_settings.refresh_token_expire_time)
+        user.refresh_token_expire_time = datetime.now() + timedelta(days=self._auth_settings.refresh_token_expire_time)
         self._auth_users.update_auth_user(user)
         self._db.save_changes()
         return token
@@ -244,20 +252,21 @@ class AuthService(AuthServiceABC):
         self._auth_users.update_auth_user(db_user)
         self._db.save_changes()
 
-    async def add_auth_user_by_discord_async(self, user_dto: AuthUserDTO) -> OAuthDTO:
-        db_user = self._auth_users.find_auth_user_by_email(user_dto.email)
+    async def add_auth_user_by_discord_async(self, user_dto: AuthUserDTO, dc_id: int) -> OAuthDTO:
+        db_auth_user = self._auth_users.find_auth_user_by_email(user_dto.email)
+
         # user exists
-        if db_user is not None and db_user.user_id is not None:
+        if db_auth_user is not None and db_auth_user.users.count() > 0:
             # raise ServiceException(ServiceErrorCode.InvalidUser, 'User already exists')
             self._logger.debug(__name__, f'Discord user already exists')
-            return OAuthDTO(AUT.to_dto(db_user), None)
+            return OAuthDTO(AUT.to_dto(db_auth_user), None)
 
         # user exists but discord user id not set
-        elif db_user is not None and db_user.user_id is None:
+        elif db_auth_user is not None and db_auth_user.users.count() == 0:
             self._logger.debug(__name__, f'Auth user exists but not linked with discord')
-            user = self._users.get_users_by_discord_id(user_dto.user_id).single()
-            db_user.user_id = user.user_id
-            db_user.oauth_id = None
+            # users = self._users.get_users_by_discord_id(user_dto.user_id)
+            # add auth_user to user refs
+            db_auth_user.oauth_id = None
 
         else:
             # user does not exists
@@ -269,12 +278,24 @@ class AuthService(AuthServiceABC):
                 user_dto.user_id = None
 
             await self.add_auth_user_async(user_dto)
-            db_user = self._auth_users.get_auth_user_by_email(user_dto.email)
-            db_user.oauth_id = uuid.uuid4()
+            db_auth_user = self._auth_users.get_auth_user_by_email(user_dto.email)
+            db_auth_user.oauth_id = uuid.uuid4()
 
-        self._auth_users.update_auth_user(db_user)
+        for g in self._bot.guilds:
+            member = g.get_member(int(dc_id))
+            if member is None:
+                continue
+
+            server = self._servers.get_server_by_discord_id(g.id)
+            users = self._users.get_users_by_discord_id(dc_id)
+            for user in users:
+                if user.server.server_id != server.server_id:
+                    continue
+                self._auth_users.add_auth_user_user_rel(AuthUserUsersRelation(db_auth_user, user))
+
+        self._auth_users.update_auth_user(db_auth_user)
         self._db.save_changes()
-        return OAuthDTO(AUT.to_dto(db_user), db_user.oauth_id)
+        return OAuthDTO(AUT.to_dto(db_auth_user), db_auth_user.oauth_id)
 
     async def update_user_async(self, update_user_dto: UpdateAuthUserDTO):
         if update_user_dto is None:

kdb-bot/src/bot_api/transformer/auth_user_transformer.py

@@ -20,7 +20,7 @@ class AuthUserTransformer(TransformerABC):
             None,
             None,
             None,
-            datetime.now(tz=timezone.utc),
+            datetime.now(),
             AuthRoleEnum.normal if dto.auth_role is None else AuthRoleEnum(dto.auth_role),
             dto.user_id,
             auth_user_id=0 if dto.id is None else dto.id