bitbot-3.11-fork/modules/silence.py

#--depends-on commands
#--depends-on permissions

import time
from src import EventManager, ModuleManager, utils

SILENCE_TIME = 60*5 # 5 minutes

class Module(ModuleManager.BaseModule):
    def on_load(self):
        self.exports.add("is-silenced", self._is_silenced)

    def _is_silenced(self, target):
        silence_until = target.get_setting("silence-until", None)
        if silence_until and time.time()<silence_until:
            return True
        return False

    @utils.hook("received.command.silence", channel_only=True)
    @utils.kwarg("help", "Prevent me saying anything for a period of time "
        "(default: 5 minutes)")
    @utils.kwarg("usage", "[+time]")
    @utils.kwarg("require_mode", "high")
    @utils.kwarg("permission", "silence")
    def silence(self, event):
        """
        :help: Silence me for 5 minutes
        :require_mode: high
        :permission: silence
        """
        duration = SILENCE_TIME
        if event["args"] and event["args_split"][0].startswith("+"):
            duration = utils.from_pretty_time(event["args_split"][0][1:])
            if duration == None:
                raise utils.EventError("Invalid duration provided")

        silence_until = time.time()+duration
        event["target"].set_setting("silence-until", silence_until)
        event["stdout"].write("Ok, I'll be back")

    @utils.hook("received.command.unsilence")
    @utils.kwarg("help", "Unsilence me")
    @utils.kwarg("unsilence", True)
    @utils.kwarg("channel_only", True)
    @utils.kwarg("require_mode", "high")
    @utils.kwarg("permission", "unsilence")
    def unsiltence(self, event):
        silence_until = event["target"].get_setting("silence-until", None)
        if not silence_until == None:
            event["target"].del_setting("silence-until")
            event["stdout"].write("Ok. I've been unsilenced")
        else:
            event["stderr"].write("I am not silenced")

    @utils.hook("preprocess.command", priority=EventManager.PRIORITY_HIGH)
    def preprocess_command(self, event):
        if event["is_channel"] and not event["hook"].get_kwarg(
                "unsilence", False):
            silence_until = event["target"].get_setting("silence-until", None)
            if silence_until:
                if self._is_silenced(event["target"]):
                    return utils.consts.PERMISSION_HARD_FAIL
                else:
                    event["target"].del_setting("silence-until")