1 files changed, 69 insertions, 0 deletions
diff --git a/src/epy_reader/ebooks/mobi.py b/src/epy_reader/ebooks/mobi.py
new file mode 100644
index 0000000..39f3be4
--- /dev/null
+++ b/src/epy_reader/ebooks/mobi.py
@@ -0,0 +1,69 @@
+import contextlib
+import os
+import shutil
+import tempfile
+import xml.etree.ElementTree as ET
+from typing import Tuple, Union
+
+from epy_reader.ebooks.epub import Epub
+from epy_reader.models import BookMetadata
+from epy_reader.tools import unpack_kindle_book
+
+
+class Mobi(Epub):
+    def __init__(self, filemobi: str):
+        self.path = os.path.abspath(filemobi)
+        self.file = tempfile.mkdtemp(prefix="epy-")
+
+        # populate these attribute
+        # by calling self.initialize()
+        self.root_filepath: str
+        self.root_dirpath: str
+
+    def get_meta(self) -> BookMetadata:
+        # why self.file.read(self.root_filepath) problematic
+        with open(os.path.join(self.root_dirpath, "content.opf")) as f:
+            content_opf = ET.parse(f)  # .getroot()
+        return Epub._get_metadata(content_opf)
+
+    def initialize(self) -> None:
+        assert isinstance(self.file, str)
+
+        with contextlib.redirect_stdout(None):
+            unpack_kindle_book(self.path, self.file, epubver="A", use_hd=True)
+            # TODO: add cleanup here
+
+        self.root_dirpath = os.path.join(self.file, "mobi7")
+        self.toc_path = os.path.join(self.root_dirpath, "toc.ncx")
+        version = "2.0"
+
+        with open(os.path.join(self.root_dirpath, "content.opf")) as f:
+            content_opf = ET.parse(f)  # .getroot()
+
+        contents = Epub._get_contents(content_opf)
+        self.contents = tuple(os.path.join(self.root_dirpath, content) for content in contents)
+
+        with open(self.toc_path) as f:
+            toc = ET.parse(f).getroot()
+        self.toc_entries = Epub._get_tocs(toc, version, contents)  # *self.contents (absolute path)
+
+    def get_raw_text(self, content_path: Union[str, ET.Element]) -> str:
+        assert isinstance(content_path, str)
+        with open(content_path, encoding="utf8") as f:
+            content = f.read()
+        # return content.decode("utf-8")
+        return content
+
+    def get_img_bytestr(self, impath: str) -> Tuple[str, bytes]:
+        # TODO: test on windows
+        # if impath "Images/asdf.png" is problematic
+        image_abspath = os.path.join(self.root_dirpath, impath)
+        image_abspath = os.path.normpath(image_abspath)  # handle crossplatform path
+        with open(image_abspath, "rb") as f:
+            src = f.read()
+        return impath, src
+
+    def cleanup(self) -> None:
+        assert isinstance(self.file, str)
+        shutil.rmtree(self.file)
+        return