1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# poezio emoji_ascii plugin
#
# Will translate received Emoji to :emoji: for better display on text terminals,
# and outgoing :emoji: into Emoji on the wire.
#
# Requires emojis.json.gz (MIT licensed) from:
#
# git clone https://github.com/vdurmont/emoji-java
# gzip -9 < ./src/main/resources/emojis.json > poezio/plugins/emojis.json.gz
# TODOs:
# 1. it messes up your log files (doesn't log original message, logs mutilated :emoji: instead)
# 2. Doesn't work on outgoing direct messages
# 3. Doesn't detect pastes, corrupts jabber:x:foobar
# 4. no auto-completion of emoji aliases
# 5. coloring of converted Emojis to be able to differentiate them from incoming ASCII
import gzip
import json
import os
import re
from poezio.plugin import BasePlugin
from typing import Dict
class Plugin(BasePlugin):
emoji_to_ascii: Dict[str, str] = {}
ascii_to_emoji: Dict[str, str] = {}
emoji_pattern = None
alias_pattern = None
def init(self):
emoji_map_file_name = os.path.abspath(os.path.dirname(__file__) + '/emojis.json.gz')
emoji_map_data = gzip.open(emoji_map_file_name, 'r').read().decode('utf-8')
emoji_map = json.loads(emoji_map_data)
for e in emoji_map:
self.emoji_to_ascii[e['emoji']] = ':%s:' % e['aliases'][0]
for alias in e['aliases']:
# work around :iq: and similar country code misdetection
flag = re.match('^[a-z][a-z]$', alias) and "flag" in e["tags"]
if not flag:
self.ascii_to_emoji[':%s:' % alias] = e['emoji']
self.emoji_pattern = re.compile('|'.join(self.emoji_to_ascii.keys()).replace('*', '\*'))
self.alias_pattern = re.compile('|'.join(self.ascii_to_emoji.keys()).replace('+', '\+'))
self.api.add_event_handler('muc_msg', self.emoji2alias)
self.api.add_event_handler('conversation_msg', self.emoji2alias)
self.api.add_event_handler('private_msg', self.emoji2alias)
self.api.add_event_handler('muc_say', self.alias2emoji)
self.api.add_event_handler('private_say', self.alias2emoji)
self.api.add_event_handler('conversation_say', self.alias2emoji)
def emoji2alias(self, msg, tab):
msg['body'] = self.emoji_pattern.sub(lambda m: self.emoji_to_ascii[m.group()], msg['body'])
def alias2emoji(self, msg, tab):
msg['body'] = self.alias_pattern.sub(lambda m: self.ascii_to_emoji[m.group()], msg['body'])
|
