from docutils import nodes
from docutils.parsers.rst import Directive


class PybricksClasslinkDirective(Directive):

    required_arguments = 1

    def run(self):
        # Get link name from sphinx-directive
        name = self.arguments[0]

        html = (
            '<a href="{0}.html">'.format(name.lower()) +
            '<dl class="py class">' +
            '<dt>' +
            '<em class="property">class </em>' +
            '<code class="sig-name descname">' + name + '</code>' +
            '</dt>' +
            '<dd></dd>' +
            '</dl>' +
            '</a>'
        )

        # Return the node
        node = nodes.raw('', html, format="html")
        return [node]


def setup(app):
    app.add_directive_to_domain('py', 'pybricks-classlink', PybricksClasslinkDirective)