mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-03-29 04:23:34 +01:00
100 lines
3.7 KiB
Python
100 lines
3.7 KiB
Python
import os
|
|
from sphinx.util import logging
|
|
from .nav_utils import extract_headings_from_file
|
|
MAX_HEADING_LEVEL = 0
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def collect_folder_tree(dir_path, base_url):
|
|
"""
|
|
Recursively collects the folder tree starting from the given directory.
|
|
|
|
For each folder:
|
|
- It is ignored if it is hidden.
|
|
- If a representative file (index.rst/index.md or readme.md/readme.rst) exists,
|
|
its first heading is used as the folder title.
|
|
- Folders without such a representative file are skipped.
|
|
- All Markdown and reStructuredText files (except the representative file)
|
|
are listed without sub-headings, using the first heading as their title.
|
|
"""
|
|
# Ignore hidden directories
|
|
if os.path.basename(dir_path).startswith('.'):
|
|
return None
|
|
|
|
# List all files in current directory with .md or .rst extension
|
|
files = [f for f in os.listdir(dir_path)
|
|
if os.path.isfile(os.path.join(dir_path, f))
|
|
and (f.endswith('.md') or f.endswith('.rst'))]
|
|
|
|
# Find representative file for folder title using index or readme
|
|
rep_file = None
|
|
for candidate in ['index.rst', 'index.md', 'readme.md', 'readme.rst']:
|
|
for f in files:
|
|
if f.lower() == candidate:
|
|
rep_file = f
|
|
break
|
|
if rep_file:
|
|
break
|
|
|
|
# If no representative file, skip this folder
|
|
if not rep_file:
|
|
return None
|
|
|
|
rep_path = os.path.join(dir_path, rep_file)
|
|
# If MAX_HEADING_LEVEL is 0, use an effectively infinite level (e.g., 9999)
|
|
effective_max = MAX_HEADING_LEVEL if MAX_HEADING_LEVEL != 0 else 9999
|
|
headings = extract_headings_from_file(rep_path, max_level=effective_max)
|
|
folder_title = headings[0]['text'] if headings else os.path.basename(dir_path)
|
|
folder_link = os.path.join(base_url, os.path.splitext(rep_file)[0])
|
|
# Remove the representative file from the list to avoid duplication
|
|
files.remove(rep_file)
|
|
|
|
# Process the remaining files in the current directory
|
|
file_items = []
|
|
for file in sorted(files, key=lambda s: s.lower()):
|
|
file_path = os.path.join(dir_path, file)
|
|
file_headings = extract_headings_from_file(file_path, max_level=effective_max)
|
|
file_title = file_headings[0]['text'] if file_headings else file
|
|
file_base = os.path.splitext(file)[0]
|
|
file_link = os.path.join(base_url, file_base)
|
|
file_items.append({
|
|
'level': 1,
|
|
'text': file_title,
|
|
'link': file_link,
|
|
'anchor': '',
|
|
'priority': 1,
|
|
'filename': file
|
|
})
|
|
|
|
# Process subdirectories (ignoring hidden ones)
|
|
dir_items = []
|
|
for item in sorted(os.listdir(dir_path), key=lambda s: s.lower()):
|
|
full_path = os.path.join(dir_path, item)
|
|
if os.path.isdir(full_path) and not item.startswith('.'):
|
|
subtree = collect_folder_tree(full_path, os.path.join(base_url, item))
|
|
if subtree:
|
|
dir_items.append(subtree)
|
|
|
|
# Combine files and subdirectories as children of the current folder
|
|
children = file_items + dir_items
|
|
|
|
return {
|
|
'text': folder_title,
|
|
'link': folder_link,
|
|
'children': children,
|
|
'filename': os.path.basename(dir_path)
|
|
}
|
|
|
|
def add_local_subfolders(app, pagename, templatename, context, doctree):
|
|
"""
|
|
Sets the 'local_subfolders' context variable with the entire folder tree
|
|
starting from app.srcdir.
|
|
"""
|
|
root_dir = app.srcdir
|
|
folder_tree = collect_folder_tree(root_dir, '')
|
|
context['local_subfolders'] = [folder_tree] if folder_tree else []
|
|
|
|
def setup(app):
|
|
app.connect('html-page-context', add_local_subfolders)
|
|
return {'version': '0.1', 'parallel_read_safe': True}
|