Docs: Nearly there

Author: domdfcoding

doc-source/latex_toctree_sections.py

@@ -1,60 +1,32 @@
 # stdlib
-from typing import Any, Dict, List, Tuple
+from typing import Any, Dict, List
 
 # 3rd party
 import sphinx.directives.other
+import sphinx.writers.latex
 from docutils import nodes
 from sphinx.application import Sphinx
-from sphinx.domains import IndexEntry
-from sphinx.writers.latex import LaTeXTranslator
 
 __all__ = ["TocTreePlusDirective", "setup"]
 
 
-def generate_indices(translator) -> str:
-	def generate(content: List[Tuple[str, List[IndexEntry]]], collapsed: bool) -> None:
-		ret.append("\\bookmarksetupnext{{level=part}}\n")
-		ret.append('\\begin{sphinxtheindex}\n')
-		ret.append('\\let\\bigletter\\sphinxstyleindexlettergroup\n')
-		for i, (letter, entries) in enumerate(content):
-			if i > 0:
-				ret.append('\\indexspace\n')
-			ret.append('\\bigletter{%s}\n' % translator.escape(letter))
-			for entry in entries:
-				if not entry[3]:
-					continue
-				ret.append('\\item\\relax\\sphinxstyleindexentry{%s}' %
-						   translator.encode(entry[0]))
-				if entry[4]:
-					# add "extra" info
-					ret.append('\\sphinxstyleindexextra{%s}' % translator.encode(entry[4]))
-				ret.append('\\sphinxstyleindexpageref{%s:%s}\n' %
-						   (entry[2], translator.idescape(entry[3])))
-		ret.append('\\end{sphinxtheindex}\n')
-
-	ret = []
-	# latex_domain_indices can be False/True or a list of index names
-	indices_config = translator.builder.config.latex_domain_indices
-	if indices_config:
-		for domain in translator.builder.env.domains.values():
-			for indexcls in domain.indices:
-				indexname = '%s-%s' % (domain.name, indexcls.name)
-				if isinstance(indices_config, list):
-					if indexname not in indices_config:
-						continue
-				content, collapsed = indexcls(domain).generate(
-						translator.builder.docnames)
-				if not content:
-					continue
-				ret.append('\\renewcommand{\\indexname}{%s}\n' %
-						   indexcls.localname)
-				generate(content, collapsed)
-
-	return ''.join(ret)
+class LaTeXTranslator(sphinx.writers.latex.LaTeXTranslator):
+
+	def generate_indices(self) -> str:
+
+		lines = super().generate_indices().splitlines()
+
+		return "\n".join([
+				"\\bookmarksetupnext{{level=part}}\n",
+				*lines,
+				'',
+				"\\bookmarksetupnext{{level=part}}\n",
+				])
 
 # TODO: The first section in a part has all sub sections nested under it in the sidebar,
 #  The numbering is correct, and its correct in the contents
 
+
 class TocTreePlusDirective(sphinx.directives.other.TocTree):
 
 	def run(self) -> List[nodes.Node]:
@@ -69,7 +41,11 @@ def run(self) -> List[nodes.Node]:
 
 			# TODO: \setcounter{section}{0}
 			# https://tex.stackexchange.com/questions/271075/reset-counter-section-in-part
-			latex_part_node = nodes.raw(text=f"\\setcounter{{section}}{{0}}\n\\bookmarksetupnext{{level=part}}\n\\part{{{caption}}}\n\\setcounter{{chapter}}{{1}}", format="latex")
+			latex_part_node = nodes.raw(
+					text=
+					f"\\setcounter{{section}}{{0}}\n\\bookmarksetupnext{{level=part}}\n\\part{{{caption}}}\n\\setcounter{{chapter}}{{1}}",
+					format="latex"
+					)
 			output.append(latex_part_node)
 			# self.state.nested_parse(StringList(), self.content_offset, latex_part_node)
 
@@ -85,7 +61,7 @@ def setup(app: Sphinx) -> Dict[str, Any]:
 	"""
 
 	app.add_directive("toctree", TocTreePlusDirective, override=True)
-	LaTeXTranslator.generate_indices = generate_indices
+	app.set_translator("latex", LaTeXTranslator, override=True)
 
 	return {
 			"parallel_read_safe": True,