From 2b8c6a87143318dcb0e5c61f9044a6c8f292b7c8 Mon Sep 17 00:00:00 2001
From: Dima Gerasimov <karlicoss@gmail.com>
Date: Sat, 26 Sep 2026 14:18:42 +0100
Subject: [PATCH] fix: correct loader return types and export node classes

---
 src/orgparse/__init__.py | 10 +++++-----
 src/orgparse/node.py     |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/orgparse/__init__.py b/src/orgparse/__init__.py
index c74e0c9..e3e2f06 100644
--- a/src/orgparse/__init__.py
+++ b/src/orgparse/__init__.py
@@ -4,12 +4,12 @@
 from pathlib import Path
 from typing import TextIO
 
-from .node import OrgEnv, OrgNode, parse_lines  # todo basenode??
+from .node import OrgBaseNode, OrgEnv, OrgNode, OrgRootNode, parse_lines
 
-__all__ = ["load", "loadi", "loads"]
+__all__ = ["OrgBaseNode", "OrgEnv", "OrgNode", "OrgRootNode", "load", "loadi", "loads"]
 
 
-def load(path: str | Path | TextIO, env: OrgEnv | None = None) -> OrgNode:
+def load(path: str | Path | TextIO, env: OrgEnv | None = None) -> OrgRootNode:
     """
     Load org-mode document from a file.
 
@@ -39,7 +39,7 @@ def load(path: str | Path | TextIO, env: OrgEnv | None = None) -> OrgNode:
     return loadi(all_lines, filename=filename, env=env)
 
 
-def loads(string: str, filename: str = '<string>', env: OrgEnv | None = None) -> OrgNode:
+def loads(string: str, filename: str = '<string>', env: OrgEnv | None = None) -> OrgRootNode:
     """
     Load org-mode document from a string.
 
@@ -49,7 +49,7 @@ def loads(string: str, filename: str = '<string>', env: OrgEnv | None = None) ->
     return loadi(string.splitlines(), filename=filename, env=env)
 
 
-def loadi(lines: Iterable[str], filename: str = '<lines>', env: OrgEnv | None = None) -> OrgNode:
+def loadi(lines: Iterable[str], filename: str = '<lines>', env: OrgEnv | None = None) -> OrgRootNode:
     """
     Load org-mode document from an iterative object.
 
diff --git a/src/orgparse/node.py b/src/orgparse/node.py
index 537b6fe..d31da07 100644
--- a/src/orgparse/node.py
+++ b/src/orgparse/node.py
@@ -1445,7 +1445,7 @@ def repeated_tasks(self):
         return self._repeated_tasks
 
 
-def parse_lines(lines: Iterable[str], filename, env=None) -> OrgNode:
+def parse_lines(lines: Iterable[str], filename, env=None) -> OrgRootNode:
     if not env:
         env = OrgEnv(filename=filename)
     elif env.filename != filename: