import_safari_bookmarks

#!/usr/bin/env python3

"""
Import bookmarks to Safari from a Netscape Bookmark file.
Bookmark folders are ignored.

Usage: import_safari_bookmarks <path/to/bookmarks.html>
"""

import plistlib
import re
import sys
import uuid
from pathlib import Path


def main() -> None:
    if len(sys.argv) < 2:
        print(__doc__)
        sys.exit(1)

    input_path = Path(sys.argv[1])
    text = input_path.read_text()

    bookmarks: list[object] = []
    for match in re.finditer(r'<DT><A HREF="(.*?)".*>(.*)</A>', text):
        bookmarks.append(
            {
                "ReadingListNonSync": {"neverFetchMetadata": 0},
                "URIDictionary": {"title": match.group(2)},
                "URLString": match.group(1),
                "WebBookmarkType": "WebBookmarkTypeLeaf",
                "WebBookmarkUUID": random_uuid(),
            }
        )

    if len(bookmarks) == 0:
        print(f"no bookmarks found in {input_path}")
        sys.exit(0)

    output_path = Path("~/Library/Safari/Bookmarks.plist").expanduser()
    with open(output_path, mode="r+b") as f:
        data = plistlib.load(f, fmt=plistlib.FMT_BINARY)
        f.seek(0)

        data["WebBookmarkUUID"] = random_uuid()
        data["Children"][3:] = bookmarks

        plistlib.dump(data, f, fmt=plistlib.FMT_BINARY, sort_keys=False)
        f.truncate()


def random_uuid() -> str:
    return str(uuid.uuid4()).upper()


if __name__ == "__main__":
    main()