mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-11-03 19:17:02 -05:00
97 lines
2.3 KiB
Python
97 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
|
|
from itertools import count
|
|
|
|
from docutils import nodes
|
|
from sphinx.environment.adapters.toctree import TocTree
|
|
|
|
id_counter = count()
|
|
ID = 'sidebar-collapsible-toc'
|
|
|
|
CSS = r'''
|
|
ID li {
|
|
list-style: none;
|
|
margin-left: 0;
|
|
padding-left: 0.2em;
|
|
text-indent: -0.7em;
|
|
}
|
|
|
|
ID li.leaf-node {
|
|
text-indent: 0;
|
|
}
|
|
|
|
ID li input[type=checkbox] {
|
|
display: none;
|
|
}
|
|
|
|
ID li > label {
|
|
cursor: pointer;
|
|
}
|
|
|
|
ID li > input[type=checkbox] ~ ul > li {
|
|
display: none;
|
|
}
|
|
|
|
ID li > input[type=checkbox]:checked ~ ul > li {
|
|
display: block;
|
|
}
|
|
|
|
ID li > input[type=checkbox]:checked + label:before {
|
|
content: "\025bf";
|
|
}
|
|
|
|
ID li > input[type=checkbox]:not(:checked) + label:before {
|
|
content: "\025b8";
|
|
}
|
|
'''.replace('ID', 'ul#' + ID)
|
|
|
|
|
|
class checkbox(nodes.Element):
|
|
pass
|
|
|
|
|
|
def visit_checkbox(self, node):
|
|
cid = node['ids'][0]
|
|
node['classes'] = []
|
|
self.body.append('<input id="{0}" type="checkbox" />'
|
|
'<label for="{0}"> </label>'.format(cid))
|
|
|
|
|
|
def modify_li(li):
|
|
sublist = li.first_child_matching_class(nodes.bullet_list)
|
|
if sublist is None or li[sublist].first_child_matching_class(nodes.list_item) is None:
|
|
if not li.get('classes'):
|
|
li['classes'] = []
|
|
li['classes'].append('leaf-node')
|
|
else:
|
|
c = checkbox()
|
|
c['ids'] = ['collapse-checkbox-{}'.format(next(id_counter))]
|
|
li.insert(0, c)
|
|
|
|
|
|
def create_toc(app, pagename):
|
|
tt = TocTree(app.env)
|
|
toctree = tt.get_toc_for(pagename, app.builder)
|
|
if toctree is not None:
|
|
subtree = toctree[toctree.first_child_matching_class(nodes.list_item)]
|
|
bl = subtree.first_child_matching_class(nodes.bullet_list)
|
|
if bl is None:
|
|
return # Empty ToC
|
|
subtree = subtree[bl]
|
|
for li in subtree.traverse(nodes.list_item):
|
|
modify_li(li)
|
|
subtree['ids'] = [ID]
|
|
return '<style>' + CSS + '</style>' + app.builder.render_partial(
|
|
subtree)['fragment']
|
|
|
|
|
|
def add_html_context(app, pagename, templatename, context, *args):
|
|
if 'toc' in context:
|
|
context['toc'] = create_toc(app, pagename) or context['toc']
|
|
|
|
|
|
def setup(app):
|
|
app.add_node(checkbox, html=(visit_checkbox, lambda *x: None))
|
|
app.connect('html-page-context', add_html_context)
|