|
1 | 1 | import os |
2 | 2 | import base64 |
3 | 3 | import markdown |
| 4 | +import markopolis.data_dantic as D |
4 | 5 | from typing import Dict, Optional, Any, Tuple, List |
5 | 6 | from .md_extensions import ( |
6 | 7 | CalloutExtension, |
|
20 | 21 | MDROOT = settings.md_path |
21 | 22 |
|
22 | 23 |
|
23 | | -def list_markdown_files(): |
24 | | - if not os.path.exists(MDROOT): |
25 | | - return None, f"The directory {MDROOT} does not exist." |
| 24 | +# Parse title from markdown file metadata |
| 25 | +def get_markdown_title(filepath: str) -> str: |
| 26 | + try: |
| 27 | + with open(filepath, "r", encoding="utf-8") as f: |
| 28 | + first_line = f.readline().strip() |
| 29 | + if first_line.startswith("---"): |
| 30 | + metadata = "" |
| 31 | + for line in f: |
| 32 | + if line.strip() == "---": |
| 33 | + break |
| 34 | + metadata += line |
| 35 | + |
| 36 | + metadata_dict = yaml.safe_load(metadata) |
| 37 | + if isinstance(metadata_dict, dict) and "title" in metadata_dict: |
| 38 | + return metadata_dict["title"] |
| 39 | + except Exception as e: |
| 40 | + logger.error(f"Error reading file {filepath}: {e}") |
| 41 | + |
| 42 | + # Fallback to filename if title metadata is not present |
| 43 | + return os.path.splitext(os.path.basename(filepath))[0] |
26 | 44 |
|
| 45 | + |
| 46 | +# Recursively list folders and files |
| 47 | +def list_markdown_files( |
| 48 | + directory: str = "/", root: str = MDROOT |
| 49 | +) -> Tuple[Optional[D.FolderItem], Optional[str]]: |
| 50 | + if not os.path.exists(root): |
| 51 | + return None, f"The directory {root} does not exist." |
| 52 | + |
| 53 | + folder_item = D.FolderItem(folder=directory, members=[]) |
| 54 | + folders = [] |
27 | 55 | files = [] |
28 | | - for root, _, filenames in os.walk(MDROOT): |
29 | | - for filename in filenames: |
30 | | - if filename.endswith(".md"): |
31 | | - # Get the relative path from MDROOT |
32 | | - rel_path = os.path.relpath(os.path.join(root, filename), MDROOT) |
33 | | - # Remove the .md extension |
34 | | - rel_path_without_ext = os.path.splitext(rel_path)[0] |
35 | | - files.append(rel_path_without_ext) |
36 | | - |
37 | | - return files, "" |
| 56 | + |
| 57 | + # Separate folders and markdown files |
| 58 | + for item in sorted(os.listdir(root)): |
| 59 | + item_path = os.path.join(root, item) |
| 60 | + rel_path = os.path.relpath(item_path, MDROOT) |
| 61 | + |
| 62 | + if os.path.isdir(item_path): |
| 63 | + # Recurse into subfolder |
| 64 | + subfolder_item, error = list_markdown_files(directory=item, root=item_path) |
| 65 | + if error: |
| 66 | + logger.error(f"Error in subfolder: {error}") |
| 67 | + continue |
| 68 | + if ( |
| 69 | + subfolder_item and subfolder_item.members |
| 70 | + ): # Only include non-empty folders |
| 71 | + folders.append(subfolder_item) |
| 72 | + elif item.endswith(".md"): |
| 73 | + # It's a markdown file |
| 74 | + title = get_markdown_title(item_path) |
| 75 | + link = os.path.splitext(rel_path)[0] # Remove the .md extension |
| 76 | + file_item = D.FileItem(title=title, link=link) |
| 77 | + files.append(file_item) |
| 78 | + |
| 79 | + # Sort folders and files lexicographically |
| 80 | + folders = sorted(folders, key=lambda x: x.folder) |
| 81 | + files = sorted(files, key=lambda x: x.title) |
| 82 | + |
| 83 | + # Combine sorted folders and files, with folders appearing first |
| 84 | + folder_item.members = folders + files |
| 85 | + |
| 86 | + if not folder_item.members: |
| 87 | + return None, None # Return None for empty folders |
| 88 | + |
| 89 | + return folder_item, None |
38 | 90 |
|
39 | 91 |
|
40 | 92 | def clean_filename(filename): |
|
0 commit comments