#!/usr/bin/env python3

import notmuch
import os

ICON = "\U000f01ee"  # 󰇮


def main():
    # Workaround for bug in the notmuch module's default config resolution
    if os.environ.get("NOTMUCH_CONFIG") is None:
        os.environ["NOTMUCH_CONFIG"] = os.path.join(
            os.environ.get(
                "XDG_CONFIG_HOME", os.path.join(os.path.expanduser("~"), ".config")
            ),
            "notmuch",
            os.environ.get("NOTMUCH_PROFILE", "default"),
            "config",
        )

    count = notmuch.Database().create_query("tag:unread").count_messages()

    print(f" {ICON} {count} \n")
    if count > 50:
        print(os.environ.get("black"))
        print(os.environ.get("red"))
    elif count > 0:
        print(os.environ.get("black"))
        print(os.environ.get("yellow"))


if __name__ == "__main__":
    main()