]> git.somenet.org - pub/jan/mattermost-bot.git/blob - modules/WSAlliterationAssociation.py
modules/WSOnboarding.py
[pub/jan/mattermost-bot.git] / modules / WSAlliterationAssociation.py
1 # Mattermost Bot module.
2 #  Copyright (c) 2016-2022 by Someone <someone@somenet.org> (aka. Jan Vales <jan@jvales.net>)
3 #  published under MIT-License
4
5 import json
6 import re
7
8 # pylint: disable=wrong-import-position
9 from AbstractWSHandler import AbstractWSHandler
10 class WSAlliterationAssociation(AbstractWSHandler):
11     NAME = "alliterationassociation"
12
13     def __init__(self, channel_ids):
14         super().__init__()
15
16         self.channel_ids = channel_ids
17
18
19     def on_WS_EVENT(self, data):
20         post = json.loads(data["data"]["post"])
21
22         if post["channel_id"] not in self.channel_ids:
23             return False
24
25         if post["user_id"] == self.bot.api._my_user_id:
26             return False
27
28         message = re.sub(r"\s+", " ", re.sub(r"[^\w\.\:;\!\?~@\n ]+", " ", post["message"]))
29
30         sentences = [s.strip() for s in re.split(r"\.|\:|;|\!|\?|~|@|\n", message)]
31         for sentence in sentences:
32             words = sentence.lower().split()
33             for word in words:
34                 if word[0] != words[0][0] and post["type"] in ["", "me"]:
35                     if "root_id" not in post or post["root_id"] == "":
36                         self.bot.api.create_post(post["channel_id"], "``"+sentence+"`` is not an alliteration! Kicking...", root_id=post["id"])
37                     else:
38                         self.bot.api.create_post(post["channel_id"], "``"+sentence+"`` is not an alliteration! Kicking...", root_id=post["root_id"])
39                     self.bot.api.remove_user_from_channel(post["channel_id"], post["user_id"])
40                     return True
41
42
43         # make user to channel admin, once they manage to say one alliteration.
44         self.bot.api.update_channel_members_scheme_roles(post["channel_id"], post["user_id"], {"scheme_user": True, "scheme_admin": True})
45
46         return True