fix(picker): Always pick document

Author: SilverRainZ

src/sphinxnotes/snippet/ext.py

@@ -119,9 +119,11 @@ def on_doctree_resolved(app: Sphinx, doctree: nodes.document, docname: str) -> N
 
     doc = []
     snippets = pick(app, doctree, docname)
+    tags = []
     for s, n in snippets:
         # FIXME: Better filter logic.
-        if extract_tags(s) not in allowed_tags:
+        tags.append(extract_tags(s))
+        if tags[-1] not in allowed_tags:
             continue
         tpath = [x.astext() for x in titlepath.resolve(app.env, docname, n)]
         if isinstance(s, Section):
@@ -144,7 +146,12 @@ def on_doctree_resolved(app: Sphinx, doctree: nodes.document, docname: str) -> N
         del cache[cache_key]
 
     logger.debug(
-        '[snippet] picked %s/%s snippetes in %s', len(doc), len(snippets), docname
+        '[snippet] picked %s/%s snippets in %s, tags: %s, allowed tags: %s',
+        len(doc),
+        len(snippets),
+        docname,
+        tags,
+        allowed_tags,
     )
 
 

src/sphinxnotes/snippet/picker.py

@@ -35,15 +35,16 @@ def pick(
     """
     # FIXME: Why doctree.source is always None?
     if not doctree.attributes.get('source'):
-        logger.debug('Skip document without source')
+        logger.debug('Skip document %s: no source', docname)
         return []
 
     metadata = app.env.metadata.get(docname, {})
     if 'no-search' in metadata or 'nosearch' in metadata:
-        logger.debug('Skip document with nosearch metadata')
+        logger.debug('Skip document %s: have :no[-]search: metadata', docname)
         return []
 
     # Walk doctree and pick snippets.
+
     picker = SnippetPicker(doctree)
     doctree.walkabout(picker)
 
@@ -83,13 +84,14 @@ def depart_section(self, node: nodes.section) -> None:
         section = self._sections.pop()
         assert section == node
 
+        # Always pick document.
+        if len(self._sections) == 0:
+            self.snippets.append((Document(self.document), node))
+            return
         # Skip non-leaf section without content
         if self._is_empty_non_leaf_section(node):
             return
-        if len(self._sections) == 0:
-            self.snippets.append((Document(self.document), node))
-        else:
-            self.snippets.append((Section(node), node))
+        self.snippets.append((Section(node), node))
 
     def unknown_visit(self, node: nodes.Node) -> None:
         pass  # Ignore any unknown node