153 lines
5.4 KiB
Python
153 lines
5.4 KiB
Python
#!/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)
|