From 6c0df81ec683d69951a1617467815638ab876a80 Mon Sep 17 00:00:00 2001
From: Frederick <fkfd@fkfd.me>
Date: Wed, 21 Sep 2022 00:17:27 +0800
Subject: [PATCH] feat: Option to invite teaching team when creating MM
 channels (#17)

Also add config for mattermost_teaching_team
---
 joint_teapot/app.py                | 13 +++++++++----
 joint_teapot/config.py             |  3 +++
 joint_teapot/workers/mattermost.py |  4 +++-
 3 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/joint_teapot/app.py b/joint_teapot/app.py
index b7705c5..2e6f630 100644
--- a/joint_teapot/app.py
+++ b/joint_teapot/app.py
@@ -140,18 +140,23 @@ def upload_assignment_grades(assignments_dir: Path, assignment_name: str) -> Non
     help="create channels for student groups according to group information on"
     " gitea",
 )
-def create_channels_on_mm(prefix: str = Option(""), suffix: str = Option("")) -> None:
+def create_channels_on_mm(
+    prefix: str = Option(""),
+    suffix: str = Option(""),
+    invite_teaching_team: bool = Option(False)
+) -> None:
     groups = {
         group_name: members
         for group_name, members in tea.pot.gitea.get_all_teams().items()
         if group_name.startswith(prefix)
     }
     logger.info(
-        f"{len(groups)} channel(s) to be created "
-        + (f"with suffix {suffix}" if suffix else "")
+        f"{len(groups)} channel(s) to be created"
+        + (f" with suffix {suffix}" if suffix else "")
+        + (f", inviting teaching team" if invite_teaching_team else "")
         + f": {','.join(groups.keys())}"
     )
-    tea.pot.mattermost.create_channels_for_groups(groups, suffix)
+    tea.pot.mattermost.create_channels_for_groups(groups, suffix, invite_teaching_team)
 
 
 @app.command(
diff --git a/joint_teapot/config.py b/joint_teapot/config.py
index 2a6eb90..19e66ed 100644
--- a/joint_teapot/config.py
+++ b/joint_teapot/config.py
@@ -29,6 +29,9 @@ class Settings(BaseSettings):
     mattermost_suffix: str = "/mm"
     mattermost_access_token: str = ""
     mattermost_team: str = ""
+    mattermost_teaching_team = [
+        "manuel",
+    ]
 
     # sid
     joj_sid: str = ""
diff --git a/joint_teapot/workers/mattermost.py b/joint_teapot/workers/mattermost.py
index b3f94ba..98ddada 100644
--- a/joint_teapot/workers/mattermost.py
+++ b/joint_teapot/workers/mattermost.py
@@ -40,7 +40,7 @@ class Mattermost:
             return
 
     def create_channels_for_groups(
-        self, groups: Dict[str, List[str]], suffix: str = ""
+        self, groups: Dict[str, List[str]], suffix: str = "", invite_teaching_team: bool = False
     ) -> None:
         for group_name, members in groups.items():
             channel_name = group_name + suffix
@@ -59,6 +59,8 @@ class Mattermost:
                     f"Error when creating channel {channel_name}: {e} Perhaps channel already exists?"
                 )
                 continue
+            if invite_teaching_team:
+                members.extend(settings.mattermost_teaching_team)
             for member in members:
                 try:
                     mmuser = self.endpoint.users.get_user_by_username(member)