|
12 | 12 | from mkdocstrings.handlers import base |
13 | 13 |
|
14 | 14 | from . import crystal_html |
15 | | -from .collector import CrystalCollector |
16 | 15 | from .items import DocItem, DocPath |
17 | 16 |
|
18 | 17 | T = TypeVar("T") |
@@ -55,7 +54,7 @@ def update_env(self, md: Markdown, config: dict) -> None: |
55 | 54 |
|
56 | 55 | base.ShiftHeadingsExtension().extendMarkdown(md) |
57 | 56 | base.PrefixIdsExtension().extendMarkdown(md) |
58 | | - XrefExtension(self.collector).extendMarkdown(md) |
| 57 | + md.treeprocessors.register(_RefInsertingTreeprocessor(md), "mkdocstrings_crystal_xref", 12) |
59 | 58 |
|
60 | 59 | self.env.trim_blocks = True |
61 | 60 | self.env.lstrip_blocks = True |
@@ -116,15 +115,6 @@ def _monkeypatch(obj, attr, func): |
116 | 115 | setattr(obj, attr, old) |
117 | 116 |
|
118 | 117 |
|
119 | | -class XrefExtension(Extension): |
120 | | - def __init__(self, collector: CrystalCollector, **kwargs) -> None: |
121 | | - super().__init__(**kwargs) |
122 | | - self.collector = collector |
123 | | - |
124 | | - def extendMarkdown(self, md: Markdown) -> None: |
125 | | - md.treeprocessors.register(_RefInsertingTreeprocessor(md), "mkdocstrings_crystal_xref", 12) |
126 | | - |
127 | | - |
128 | 118 | class _RefInsertingTreeprocessor(Treeprocessor): |
129 | 119 | context: Optional[DocItem] |
130 | 120 |
|
|
0 commit comments