From 49c01e636f3d7b7d27828c218631b3048e138c4b Mon Sep 17 00:00:00 2001
From: Someone <someone@somenet.org>
Date: Fri, 19 Jun 2020 01:32:13 +0200
Subject: [PATCH] [management/account_handling] convert accounts to guests and
 back to full-users.

---
 management/account_handling.py | 121 +++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100755 management/account_handling.py

diff --git a/management/account_handling.py b/management/account_handling.py
new file mode 100755
index 0000000..4874f90
--- /dev/null
+++ b/management/account_handling.py
@@ -0,0 +1,121 @@
+#!/usr/bin/env python3
+#
+# Someone's Mattermost scripts.
+#  Copyright (c) 2016-2020 by Someone <someone@somenet.org> (aka. Jan Vales <jan@jvales.net>)
+#  published under MIT-License
+#
+# consider running as cronjob
+#   0 * * * *  (cd /home/someone/mattermost/management; python3 -u account_handling.py)
+#
+
+import mattermost
+import re
+import config
+import time
+import pprint
+import sys
+
+from inspect import cleandoc
+
+import config
+
+
+def confine_guest(user):
+    #print("confine_guest()")
+    global mm, all_team_ids
+
+    for team_id in config.guest_team_ids:
+        mm.add_user_to_team(team_id, user["id"])
+
+    user_pub_chans = set()
+    for team_id in all_team_ids:
+        try:
+            user_pub_chans = user_pub_chans.union({c["id"] for c in mm.get_channels_for_user(user["id"], team_id, exc=True) if c["type"]=="O"})
+        except mattermost.ApiException as e:
+            pass
+            
+    to_add = config.guest_channels - user_pub_chans
+    #pprint.pprint(to_add)
+    for chan_id in to_add:
+        mm.add_user_to_channel(chan_id, user["id"])
+        
+    to_kick = user_pub_chans - config.guest_channels
+    #pprint.pprint(to_kick)
+    for chan_id in to_kick:
+        mm.remove_user_from_channel(chan_id, user["id"])
+
+
+cnt = 0
+cnt_g = 0
+cnt_u = 0
+cnt_u2g = 0
+cnt_g2u = 0
+TS_START = time.time()
+rex = re.compile(config.allowed_emails_regex)
+
+mm = mattermost.MMApi(config.mm_api_url)
+mm.login(config.mm_user, config.mm_user_pw)
+print("* ["+("%07.6g"%round(time.time() - TS_START, 5))+"] login")
+
+all_team_ids = [t["id"] for t in mm.get_teams()]
+print("* ["+("%07.6g"%round(time.time() - TS_START, 5))+"] list all team_ids")
+
+for user in mm.get_users():
+    cnt += 1
+    if "system_user" in user["roles"]:
+        cnt_u += 1
+
+        if not rex.match(user["email"]):
+            cnt_u2g += 1
+            u2g = user
+
+            #print("demote")
+            #pprint.pprint(mm.demote_a_user(user["id"]))
+            mm.demote_a_user(user["id"])
+            confine_guest(user)
+
+            c = mm.create_dm_channel_with(user["id"])
+            mm.create_post(c["id"], cleandoc("""
+                ### ``Confirm your student email to get full access to Mattermost``
+                As previously announced your account has been converted to a [guest account](https://docs.mattermost.com/deployment/guest-accounts.html), which means you will be restricted to three channels:
+                * Beratung
+                * Guest-Talk
+                * Aufnahmeverfahren
+
+                Mattermost has a dedicated channel for every course, to access them you need to go to your Account Settings and change your email to ``eXXXXXXXX@student.tuwien.ac.at`` and then click on the link in the confirmation email. You will then get full access within an hour.
+
+                Sorry for the inconvenience, we hope to see you soon!
+                If something seems off, contact my master @someone by mentioning him in the ``Guest-Talk`` channel.
+                """))
+
+        else:
+            pass
+            # todo: inform of guest status and to change email
+
+
+    elif "system_guest" in user["roles"]:
+        cnt_g += 1
+        if rex.match(user["email"]):
+            cnt_g2u += 1
+            g2u = user
+
+            # convert account to user.
+            #print("promote")
+            #pprint.pprint(mm.promote_a_guest(user["id"]))
+            mm.promote_a_guest(user["id"])
+            c = mm.create_dm_channel_with(user["id"])
+            mm.create_post(c["id"], cleandoc("""
+                ### ``Your account has been converted to a FULL-USER account!``
+                You are now able to use all the features mattermost and this instance has to offer!
+                """))
+        else:
+            # re-confine guests
+            confine_guest(user)
+            
+    else:
+        print("weird user")
+        pprint.pprint(user)
+        
+    print("* ["+("%07.6g"%round(time.time() - TS_START, 5))+"] cnt:", cnt, " cnt_u:", cnt_u, " cnt_g:", cnt_g, " cnt_u2g:", cnt_u2g, " cnt_g2u:", cnt_g2u)
+
+mm.logout()
-- 
2.43.0