diff --git a/plugins/hypvote.py b/plugins/hypvote.py new file mode 100644 index 0000000..27e7fb5 --- /dev/null +++ b/plugins/hypvote.py @@ -0,0 +1,48 @@ +import io +import re +from typing import TYPE_CHECKING, TypeVar + +if TYPE_CHECKING: + from ica_typing import IcaNewMessage, IcaClient +else: + IcaNewMessage = TypeVar("NewMessage") + IcaClient = TypeVar("IcaClient") + + +def on_ica_message(msg: IcaNewMessage, client: IcaClient) -> None: + if (not (msg.is_from_self or msg.is_reply)) and (msg.content.startswith("/hyp") or "any hyp" in msg.content): + hypvote(msg) + + +vote = dict([(i, []) for i in range(0,24)] + +def hypvote(msg): + arg = re.match("/hyp(.+)", msg.content + " ").group(1) + if arg[0] == "vote": + map(lambda x: vote[int(x) % 24] += msg.sender, arg[1:]) + else if arg[0] == "unvote": + map(lambda x: vote[int(x) % 24].remove(msg.sender), arg[1:]) + else if arg[0] == "clear": + vote = dict([(i, []) for i in range(0,24)] + else if arg == [] or arg[0] == "ls" or "any hyp"in msg.content: + res = "\n".join([f"{x}\t{vote[x]}" for x in vote]) + msg.reply_with(res) + else if arg[0] == "help": + msg.reply_with("""NAME + /hyp - 计划时间,高效开黑 +SYNOPSIS + /hyp [command] [args] +OPTIONS + vote + vote for time you want to play + unvote + unvote for time you want to play + clear + clear the vote (OP only) + ls + list voted time, equivalent to empty + help + show this help +AUTHOR + dongdigua +""")