import os
import subprocess
import time
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler

WATCH_DIR = "/data"  # Mount your host Books folder here

class NewBookHandler(FileSystemEventHandler):
    def __init__(self):
        super().__init__()
        self.seen_dirs = set()  # avoid processing the same dir multiple times

    def on_created(self, event):
        if event.is_directory:
            self.process_new_dir(event.src_path)

    def on_moved(self, event):
        if event.is_directory:
            self.process_new_dir(event.dest_path)

    def process_new_dir(self, path):
        # avoid duplicate processing
        if path in self.seen_dirs:
            return
        self.seen_dirs.add(path)

        # small delay to allow files to finish copying
        time.sleep(2)

        try:
            m4b_files = [
                f for f in os.listdir(path)
                if f.lower().endswith(".m4b") and not f.startswith("._")
            ]
        except FileNotFoundError:
            return  # directory disappeared

        if not m4b_files:
            print(f"⚠️ Skipping {path} (no .m4b files yet)")
            return

        book_name = os.path.basename(path)
        print(f"📕 New book detected: {book_name} at {path}")

        try:
            subprocess.run(
                ["python3", "main.py", "-da", path],
                check=True
            )
            print(f"✅ Finished processing {book_name}")
        except subprocess.CalledProcessError as e:
            print(f"❌ Error processing {book_name}: {e}")

if __name__ == "__main__":
    observer = Observer()
    event_handler = NewBookHandler()
    observer.schedule(event_handler, WATCH_DIR, recursive=True)
    observer.start()
    print(f"👀 Watching for new book folders in {WATCH_DIR} ...")

    try:
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        observer.stop()
    observer.join()