]> git.somenet.org - pub/jan/mattermost-bot.git/blob - modules/CommandJoinAll.py
core/AbstractCommand.py
[pub/jan/mattermost-bot.git] / modules / CommandJoinAll.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 datetime
6 import threading
7 from inspect import cleandoc
8
9 import logging
10 logger = logging.getLogger(__name__)
11
12 # pylint: disable=wrong-import-position
13 from AbstractCommand import AbstractCommand
14 class CommandJoinAll(AbstractCommand):
15     TRIGGER = "join-all"
16     CONFIG = {"display_name": "somebot-command", "auto_complete": True,
17               "auto_complete_hint": "[<weeks; default=3>]"
18              }
19     USEINFO = CONFIG["auto_complete_desc"] = CONFIG["description"] = AbstractCommand.ICON_PRIVATE+"Join active public channels. considers channels with 7 posts+ in the last x weeks. default 3 weeks. There is no undo."
20
21
22     def __init__(self, team_id, skip_chann_ids):
23         super().__init__(team_id)
24
25         self.skip_chann_ids = skip_chann_ids
26         self.channel_index = None
27         self.channel_index_th = None
28
29
30     def on_register(self):
31         self._create_slash_command()
32         self.channel_index_th = threading.Timer(10.0, self._index_channels)
33         self.channel_index_th.setName("CommandJoinActive::_index_channels()")
34         self.channel_index_th.setDaemon(True)
35         self.channel_index_th.start()
36
37
38     def _index_channels(self):
39         if self.channel_index_th != threading.current_thread():
40             logger.warning("_index_channels(): thread mismatch")
41             return
42
43         logger.info("_index_channels(): started")
44         new_dict = {}
45         for chan in self.bot.api.get_team_channels(self.TEAM_ID):
46             if chan["id"] in self.skip_chann_ids:
47                 continue
48
49             posts = []
50             for post in self.bot.api.get_posts_for_channel(chan["id"]):
51                 if not post["type"].startswith("system_"):
52                     posts.append(post)
53
54                 if len(posts) == 7:
55                     break
56
57             new_dict[chan["id"]] = min([x["create_at"] for x in posts])
58
59         self.channel_index = new_dict
60         self.channel_index_th = threading.Timer(3600.0, self._index_channels)
61         self.channel_index_th.setName("CommandJoinActive::_index_channels()")
62         self.channel_index_th.setDaemon(True)
63         self.channel_index_th.start()
64         logger.info("_index_channels(): done")
65
66
67     def on_POST(self, request, data):
68         if self.channel_index is None:
69             request.respond_cmd_err("``/"+self.TRIGGER+"`` The channel indexer didnt finish yet. try again in a few minutes")
70             return
71
72         try:
73             td = datetime.timedelta(weeks=int(data["text"].strip()))
74         except:
75             td = datetime.timedelta(weeks=3)
76
77         timestamp = int((datetime.datetime.now()-td).timestamp())*1000
78
79         msg = ""
80         for chan in self.bot.api.get_team_channels(self.TEAM_ID):
81             #check against ban list and get the last 7 posts, if all of them are newer than min_timestamp: join
82             if chan["id"] not in self.channel_index and chan["id"] not in self.skip_chann_ids and self.channel_index[chan["id"]] > timestamp:
83                 self.bot.api.add_user_to_channel(chan["id"], data["user_id"])
84                 msg += "\n + ~"+chan["name"]
85
86         request.respond_cmd_temp(cleandoc("""
87             ## :white_check_mark: Success! :)
88             By default it wont join you into channels that had less than 5 messages posted within the last 2 weeks.
89             If you want to be joined into less active channels give the command an integer number of weeks to to consider.
90             /join-active 8 will join you into channels that had less than 5 messages posted within the last 2 months.
91             Anyway: joined the following active public channels in the team (A reload might be needed to display the updated list of channels):
92             {}
93             """).format(msg))