__init__.py

# -*- coding: utf-8 -*-
# This file is public domain according to its author, Roberto Alsina

"""Emoji directive for reStructuredText."""

import glob
import json
import os

from nikola.plugin_categories import ShortcodePlugin
from nikola import utils

TABLE = {}

LOGGER = utils.get_logger('scan_posts', utils.STDERR_HANDLER)


def _populate():
    for fname in glob.glob(os.path.join(os.path.dirname(__file__), 'data', '*.json')):
        with open(fname) as inf:
            data = json.load(inf)
            data = data[list(data.keys())[0]]
            data = data[list(data.keys())[0]]
            for item in data:
                if item['key'] in TABLE:
                    LOGGER.warning('Repeated emoji {}'.format(item['key']))
                else:
                    TABLE[item['key']] = item['value']


class Plugin(ShortcodePlugin):
    """Plugin for gist directive."""

    name = "emoji"

    def handler(self, name, filename=None, site=None, data=None, lang=None, post=None):
        """Create HTML for emoji."""
        if not TABLE:
            self._populate()
        try:
            output = '''<span class="emoji">{}</span>'''.format(TABLE[name])
        except KeyError:
            LOGGER.warning('Unknown emoji {}'.format(name))
            output = '''<span class="emoji error">{}</span>'''.format(name)

        return output, []