Link new users to their backend profile

Author: AllenAnthes

pybot/endpoints/slack/events.py

@@ -6,6 +6,8 @@
 
 from pybot.endpoints.slack.utils.event_utils import (
     build_messages,
+    get_backend_auth_headers,
+    link_backend_user,
     send_community_notification,
     send_user_greetings,
 )
@@ -25,8 +27,9 @@ async def team_join(event: Event, app: SirBot) -> None:
     notifies the community channel of the new member.
     """
     slack_api = app.plugins["slack"].api
+    user_id = event["user"]["id"]
 
-    *user_messages, community_message = build_messages(event["user"]["id"])
+    *user_messages, community_message = build_messages(user_id)
     futures = [
         send_user_greetings(user_messages, slack_api),
         send_community_notification(community_message, slack_api),
@@ -35,3 +38,7 @@ async def team_join(event: Event, app: SirBot) -> None:
     logger.info(f"New team join event: {event}")
     await asyncio.sleep(30)
     await asyncio.wait(futures)
+
+    headers = await get_backend_auth_headers(app.http_session)
+    if headers:
+        await link_backend_user(user_id, headers, slack_api, app.http_session)

pybot/endpoints/slack/utils/__init__.py

@@ -12,6 +12,10 @@
 PORT = os.environ.get("SIRBOT_PORT", 5000)
 HOST = os.environ.get("SIRBOT_ADDR", "0.0.0.0")
 PYBOT_ENV = os.environ.get("PYBOT_ENV", "dev")
+BACKEND_URL = os.environ.get("BACKEND_URL", "https://api.operationcode.org")
+BACKEND_USERNAME = os.environ.get("BACKEND_USERNAME", "Pybot@test.test")
+BACKEND_PASS = os.environ.get("BACKEND_PASS", "fakePassword")
+
 BOT_URL = "https://github.com/OperationCode/operationcode-pybot"
 
 slack_configs = {

pybot/endpoints/slack/utils/event_utils.py

@@ -1,10 +1,17 @@
-from typing import List
+import logging
+from typing import Dict, List, Tuple
 
+from aiohttp import ClientSession
 from slack import methods
 from slack.events import Message
 from slack.io.abc import SlackAPI
 
-from pybot.endpoints.slack.utils import COMMUNITY_CHANNEL
+from pybot.endpoints.slack.utils import (
+    BACKEND_PASS,
+    BACKEND_URL,
+    BACKEND_USERNAME,
+    COMMUNITY_CHANNEL,
+)
 from pybot.endpoints.slack.utils.action_messages import not_greeted_attachment
 from pybot.endpoints.slack.utils.event_messages import (
     base_resources,
@@ -13,8 +20,17 @@
     team_join_initial_message,
 )
 
+logger = logging.getLogger(__name__)
+
 
-def build_messages(user_id):
+def base_user_message(user_id: str) -> Message:
+    message = Message()
+    message["channel"] = user_id
+    message["as_user"] = True
+    return message
+
+
+def build_messages(user_id) -> Tuple[Message, Message, Message, Message]:
     initial_message = base_user_message(user_id)
     initial_message["text"] = team_join_initial_message(user_id)
 
@@ -34,17 +50,56 @@ def build_messages(user_id):
     return initial_message, second_message, action_menu, community_message
 
 
-async def send_user_greetings(user_messages: List[Message], slack_api: SlackAPI):
+async def send_user_greetings(
+    user_messages: List[Message], slack_api: SlackAPI
+) -> None:
     for message in user_messages:
         await slack_api.query(url=methods.CHAT_POST_MESSAGE, data=message)
 
 
-async def send_community_notification(community_message: Message, slack_api: SlackAPI):
+async def send_community_notification(
+    community_message: Message, slack_api: SlackAPI
+) -> dict:
     return await slack_api.query(url=methods.CHAT_POST_MESSAGE, data=community_message)
 
 
-def base_user_message(user_id):
-    message = Message()
-    message["channel"] = user_id
-    message["as_user"] = True
-    return message
+async def link_backend_user(
+    slack_id: str,
+    auth_header: Dict[str, str],
+    slack_api: SlackAPI,
+    session: ClientSession,
+) -> None:
+    """
+    Updates the slack user with their profile in the backend
+    """
+
+    user_info = await slack_api.query(methods.USERS_INFO, {"user": slack_id})
+    email = user_info["user"]["profile"]["email"]
+
+    async with session.patch(
+        f"{BACKEND_URL}/auth/profile/admin/",
+        headers=auth_header,
+        params={"email": email},
+        json={"slackId": slack_id},
+    ) as response:
+        data = await response.json()
+        logger.info(f"Backend response from user linking: {data}")
+
+
+async def get_backend_auth_headers(session: ClientSession) -> Dict[str, str]:
+    """
+    Authenticates with the OC Backend server
+
+    :return:  Authorization header containing the returned JWT
+    """
+    async with session.post(
+        f"{BACKEND_URL}/auth/login/",
+        json={"email": BACKEND_USERNAME, "password": BACKEND_PASS},
+    ) as response:
+        if 400 <= response.status:
+            logger.exception("Failed to authenticate with backend")
+            return {}
+        response.raise_for_status()
+        data = await response.json()
+        headers = {"Authorization": f"Bearer {data['token']}"}
+    return headers