Vendor drastikbot/drastikbot+modules v2.1

src/irc/modules/text.py

@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+# coding=utf-8
+
+# Text Module for Drastikbot
+#
+# Transform textual input to various other styles.
+
+'''
+Copyright (C) 2019 drastik.org
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+import random
+
+
+class Module:
+    def __init__(self):
+        self.commands = [
+            "ae", "text-c", "text-nc", "text-s", "text-ns", "flag", "cirrus",
+            "strike", "strikethrough"
+        ]
+        self.manual = {
+            "desc": "Text transformation tools",
+            "bot_commands": {
+                "ae": {"usage": lambda p: f"{p}ae <text>",
+                       "info": "Example: Ｈｅｌｌｏ，　Ｗｏｒｌｄ！"},
+                "text-c": {"usage": lambda p: f"{p}text-c <text>",
+                           "info": "Example: ⒽⒺⓁⓁⓄ, ⓌⓄⓇⓁⒹ!"},
+                "text-nc": {"usage": lambda p: f"{p}text-nc <text>",
+                            "info": "Example: 🅗🅔🅛🅛🅞, 🅦🅞🅡🅛🅓!"},
+                "text-s": {"usage": lambda p: f"{p}text-s <text>",
+                           "info": "Example: 🄷🄴🄻🄻🄾, 🅆🄾🅁🄻🄳!"},
+                "text-ns": {"usage": lambda p: f"{p}text-ns <text>",
+                            "info": "Example: 🅷🅴🅻🅻🅾, 🆆🅾🆁🅻🅳!"},
+                "flag": {"usage": lambda p: f"{p}flag <text>",
+                         "info": ("Transforms two letter country codes to"
+                                  " regional indicator symbols.")},
+                "cirrus": {"usage": lambda p: f"{p}cirrus <text>",
+                           "info": "Example: Hello, WWorld!"},
+                "strike": {"usage": lambda p: f"{p}strike <text>",
+                           "info": "Example: H̶e̶l̶l̶o̶,̶ ̶W̶o̶r̶l̶d̶!̶",
+                           "alias": ["strikethrough"]},
+                "strikethrough": {
+                    "usage": lambda p: f"{p}strikethrough <text>",
+                    "info": "Example: H̶e̶l̶l̶o̶,̶ ̶W̶o̶r̶l̶d̶!̶",
+                    "alias": ["strike"]
+                }
+            }
+        }
+
+
+# https://en.wikipedia.org/wiki/Halfwidth_and_fullwidth_forms_(Unicode_block)
+FULLWIDTH_MAP = dict((i, i + 0xFEE0) for i in range(0x21, 0x7F))
+FULLWIDTH_MAP[0x20] = 0x3000
+
+# https://en.wikipedia.org/wiki/Enclosed_Alphanumerics
+_CIRCLED_NUM_MAP = dict((i, (i - 0x31) + 0x2460) for i in range(0x30, 0x3A))
+_CIRCLED_NUM_MAP[0x30] = 0x24EA  # Set the actual Circled digit zero character
+_CIRCLED_ALP_U_MAP = dict((i, (i - 0x41) + 0x24B6) for i in range(0x41, 0x5B))
+_CIRCLED_ALP_L_MAP = dict((i, (i - 0x61) + 0x24B6) for i in range(0x61, 0x7B))
+CIRCLED_MAP = {**_CIRCLED_NUM_MAP, **_CIRCLED_ALP_U_MAP, **_CIRCLED_ALP_L_MAP}
+
+# https://en.wikipedia.org/wiki/Enclosed_Alphanumeric_Supplement
+_NEGATIVE_CIRCLED_ALP_U_MAP = dict(
+    (i, (i - 0x41) + 0x1F150) for i in range(0x41, 0x5B))
+_NEGATIVE_CIRCLED_ALP_L_MAP = dict(
+    (i, (i - 0x61) + 0x1F150) for i in range(0x61, 0x7B))
+NEGATIVE_CIRCLED_MAP = {
+    **_NEGATIVE_CIRCLED_ALP_U_MAP,
+    **_NEGATIVE_CIRCLED_ALP_L_MAP
+}
+
+_SQUARED_ALP_U_MAP = dict((i, (i - 0x41) + 0x1F130) for i in range(0x41, 0x5B))
+_SQUARED_ALP_L_MAP = dict((i, (i - 0x61) + 0x1F130) for i in range(0x61, 0x7B))
+SQUARED_MAP = {**_SQUARED_ALP_U_MAP, **_SQUARED_ALP_L_MAP}
+
+_NEGATIVE_SQUARED_ALP_U_MAP = dict(
+    (i, (i - 0x41) + 0x1F170) for i in range(0x41, 0x5B))
+_NEGATIVE_SQUARED_ALP_L_MAP = dict(
+    (i, (i - 0x61) + 0x1F170) for i in range(0x61, 0x7B))
+NEGATIVE_SQUARED_MAP = {
+    **_NEGATIVE_SQUARED_ALP_U_MAP,
+    **_NEGATIVE_SQUARED_ALP_L_MAP
+}
+
+# https://en.wikipedia.org/wiki/Regional_Indicator_Symbol
+_REGIONAL_INDICATOR_SYMBOL_U_MAP = dict(
+    (i, (i - 0x41) + 0x1F1E6) for i in range(0x41, 0x5B))
+_REGIONAL_INDICATOR_SYMBOL_L_MAP = dict(
+    (i, (i - 0x61) + 0x1F1E6) for i in range(0x61, 0x7B))
+REGIONAL_INDICATOR_SYMBOL_MAP = {
+    **_REGIONAL_INDICATOR_SYMBOL_U_MAP,
+    **_REGIONAL_INDICATOR_SYMBOL_L_MAP
+}
+
+command_map_d = {
+    "ae": FULLWIDTH_MAP,
+    "text-c": CIRCLED_MAP,
+    "text-nc": NEGATIVE_CIRCLED_MAP,
+    "text-s": SQUARED_MAP,
+    "text-ns": NEGATIVE_SQUARED_MAP,
+    "flag": REGIONAL_INDICATOR_SYMBOL_MAP,
+}
+
+
+def cirrus(text):
+    words = text.split()
+    wc = len(words)
+    cc = 0
+    for i in range(wc):
+        if random.uniform(0, 1) < 0.38:
+            cc += 1
+            words[i] = f"{words[i][0]}{words[i]}"
+
+    if cc == 0:
+        i = random.randint(0, wc - 1)
+        words[i] = f"{words[i][0]}{words[i]}"
+
+    return " ".join(words)
+
+
+def strikethrough(text):
+    return "\u0336".join(text) + '\u0336'
+
+
+def main(i, irc):
+    if not i.msg_nocmd:
+        return
+    s = i.msg_nocmd
+
+    if i.cmd == "cirrus":
+        return irc.privmsg(i.channel, cirrus(s))
+    if i.cmd == "strike" or i.cmd == "strikethrough":
+        return irc.privmsg(i.channel, strikethrough(s))
+
+    t = s.translate(command_map_d[i.cmd])
+    if i.cmd == "ae" and t == s:
+        t = s.replace("", " ")[1: -1]
+
+    irc.privmsg(i.channel, t)