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