Sphinx: create sphinx-vyos extension

1 files changed, 204 insertions, 0 deletions

diff --git a/docs/_ext/vyos.py b/docs/_ext/vyos.py
new file mode 100644
index 00000000..7ef88d0a
--- /dev/null
+++ b/docs/_ext/vyos.py
@@ -0,0 +1,204 @@
+from docutils import nodes, utils
+from docutils.parsers.rst.roles import set_classes
+from docutils.parsers.rst import Directive
+from sphinx.util.docutils import SphinxDirective
+
+
+def setup(app):
+
+    app.add_config_value(
+        'vyos_phabricator_url',
+        'https://phabricator.vyos.net/', ''
+    )
+    app.add_role('vytask', vytask_role)
+
+    print(app.config.vyos_phabricator_url)
+
+    app.add_node(
+        CmdDiv,
+        html=(CmdDiv.visit_div, CmdDiv.depart_div),
+        latex=(CmdDiv.visit_div, CmdDiv.depart_div),
+        text=(CmdDiv.visit_div, CmdDiv.depart_div)
+    )
+    app.add_node(
+        CmdHeader,
+        html=(CmdHeader.visit_div, CmdHeader.depart_div),
+        latex=(CmdHeader.visit_div, CmdHeader.depart_div),
+        text=(CmdHeader.visit_div, CmdHeader.depart_div)
+    )
+    app.add_node(CfgcmdList)
+    app.add_directive('cfgcmdlist', CfgcmdlistDirective)
+
+    app.add_node(OpcmdList)
+    app.add_directive('opcmdlist', OpcmdlistDirective)
+
+    app.add_directive('cfgcmd', CfgCmdDirective)
+    app.add_directive('opcmd', OpCmdDirective)
+    app.connect('doctree-resolved', process_cmd_nodes)
+
+
+class CfgcmdList(nodes.General, nodes.Element):
+    pass
+
+
+class OpcmdList(nodes.General, nodes.Element):
+    pass
+
+
+class CmdHeader(nodes.General, nodes.Element):
+
+    @staticmethod
+    def visit_div(self, node):
+        self.body.append(self.starttag(node, 'div'))
+
+    @staticmethod
+    def depart_div(self, node=None):
+        self.body.append('</div>\n')
+
+
+class CmdDiv(nodes.General, nodes.Element):
+
+    @staticmethod
+    def visit_div(self, node):
+        self.body.append(self.starttag(node, 'div'))
+
+    @staticmethod
+    def depart_div(self, node=None):
+        self.body.append('</div>\n')
+
+
+class CfgcmdlistDirective(Directive):
+
+    def run(self):
+        return [CfgcmdList('')]
+
+
+class OpcmdlistDirective(Directive):
+
+    def run(self):
+        return [OpcmdList('')]
+
+
+class CmdDirective(SphinxDirective):
+
+    has_content = True
+    custom_class = ''
+
+    def run(self):
+        title_list = []
+        content_list = []
+        title_text = ''
+        content_text = ''
+        has_body = False
+
+        cfgmode = self.custom_class + "cmd"
+
+        if '' in self.content:
+            index = self.content.index('')
+            title_list = self.content[0:index]
+            content_list = self.content[index + 1:]
+            title_text = ' '.join(title_list)
+            content_text = '\n'.join(content_list)
+            has_body = True
+        else:
+            title_text = ' '.join(self.content)
+
+        anchor_id = nodes.make_id(self.custom_class + "cmd-" + title_text)
+        target = nodes.target(ids=[anchor_id])
+
+        panel_name = 'cmd-{}'.format(self.custom_class)
+        panel_element = CmdDiv()
+        panel_element['classes'] += ['cmd', panel_name]
+
+        heading_element = CmdHeader(title_text)
+        title_nodes, messages = self.state.inline_text(title_text,
+                                                       self.lineno)
+
+        title = nodes.inline(title_text, '', *title_nodes)
+        target['classes'] += []
+        title['classes'] += [cfgmode]
+        heading_element.append(target)
+        heading_element.append(title)
+
+        heading_element['classes'] += [self.custom_class + 'cmd-heading']
+
+        panel_element.append(heading_element)
+
+        append_list = {
+            'docname': self.env.docname,
+            'cmdnode': title.deepcopy(),
+            'cmd': title_text,
+            'target': target,
+        }
+
+        if cfgmode == 'opcmd':
+            if not hasattr(self.env, "vyos_opcmd"):
+                self.env.vyos_opcmd = []
+            self.env.vyos_opcmd.append(append_list)
+
+        if cfgmode == 'cfgcmd':
+            if not hasattr(self.env, "vyos_cfgcmd"):
+                self.env.vyos_cfgcmd = []
+            self.env.vyos_cfgcmd.append(append_list)
+
+        if has_body:
+            body_element = CmdDiv(content_text)
+            self.state.nested_parse(
+                content_list,
+                self.content_offset,
+                body_element
+            )
+
+            body_element['classes'] += [self.custom_class + 'cmd-body']
+            panel_element.append(body_element)
+        return [panel_element]
+
+
+class OpCmdDirective(CmdDirective):
+    custom_class = 'op'
+
+
+class CfgCmdDirective(CmdDirective):
+    custom_class = 'cfg'
+
+
+def process_cmd_node(app, cmd, fromdocname):
+    para = nodes.paragraph()
+    newnode = nodes.reference(' ', ' ')
+    innernode = cmd['cmdnode']
+    newnode['refdocname'] = cmd['docname']
+    newnode['refuri'] = app.builder.get_relative_uri(
+        fromdocname, cmd['docname'])
+    newnode['refuri'] += '#' + cmd['target']['refid']
+    newnode['classes'] += ['cmdlink']
+    newnode.append(innernode)
+    para += newnode
+    return para
+
+
+def process_cmd_nodes(app, doctree, fromdocname):
+    env = app.builder.env
+
+    for node in doctree.traverse(CfgcmdList):
+        content = []
+
+        for cmd in sorted(env.vyos_cfgcmd, key=lambda i: i['cmd']):
+            content.append(process_cmd_node(app, cmd, fromdocname))
+        node.replace_self(content)
+
+    for node in doctree.traverse(OpcmdList):
+        content = []
+
+        for cmd in sorted(env.vyos_opcmd, key=lambda i: i['cmd']):
+            content.append(process_cmd_node(app, cmd, fromdocname))
+        node.replace_self(content)
+
+
+def vytask_role(name, rawtext, text, lineno, inliner, options={}, content=[]):
+    app = inliner.document.settings.env.app
+    base = app.config.vyos_phabricator_url
+    ref = base + str(text)
+    set_classes(options)
+    node = nodes.reference(
+        rawtext, utils.unescape(str(text)), refuri=ref, **options)
+    return [node], []