Correctly consume autolink_skip and improve error (#17)

Author: felix-hilden

docs/src/release_notes.rst

@@ -8,6 +8,10 @@ These release notes are based on
 sphinx-codeautolink adheres to
 `Semantic Versioning <https://semver.org>`_.
 
+Unreleased
+----------
+- Correctly consume :code:`autolink-skip:: next` (:issue:`17`)
+
 0.1.1 (2021-09-22)
 ------------------
 - Correctly filter out names from concatenated sources (:issue:`14`)

src/sphinx_codeautolink/extension/block.py

@@ -72,7 +72,11 @@ def unknown_visit(self, node):
             self.implicit_imports.append(node.content)
             node.parent.remove(node)
         elif isinstance(node, AutoLinkSkipMarker):
-            assert node.level in ('next', 'section', 'file', 'none')
+            if node.level not in ('next', 'section', 'file', 'none'):
+                raise UserError(
+                    f'Invalid skipping argument: `{node.level}` '
+                    f'in document "{self.current_document}"'
+                )
             self.autolink_skip = node.level if node.level != 'none' else None
             node.parent.remove(node)
 
@@ -103,6 +107,10 @@ def visit_literal_block(self, node: nodes.literal_block):
         implicit_imports = self.implicit_imports
         self.implicit_imports = []
 
+        skip = self.autolink_skip
+        if skip == 'next':
+            self.autolink_skip = None
+
         if (
             len(node.children) != 1
             or not isinstance(node.children[0], nodes.Text)
@@ -112,10 +120,8 @@ def visit_literal_block(self, node: nodes.literal_block):
 
         source = node.children[0].astext()
 
-        if self.autolink_skip:
+        if skip:
             self.source_transforms.append(SourceTransforms(source, []))
-            if self.autolink_skip == 'next':
-                self.autolink_skip = None
             return
 
         modified_source = '\n'.join(self.concat_sources + implicit_imports + [source])