#!/usr/bin/env python3
"""CLI alarm clock for Raspberry Pi Zero (mpg123 + USB speaker).

Distinguishes weekday / weekend / holiday(JP) alarm times, and plays a
different sound per day of the week.
"""
import argparse
import json
import os
import signal
import subprocess
import sys
import time
from datetime import datetime

import jpholiday

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
CONFIG_FILE = os.path.join(BASE_DIR, "config.json")
PID_FILE = "/tmp/alarm_clock.pid"
DEFAULT_SOUND = os.path.join(BASE_DIR, "sounds", "alarm.mp3")
DAY_NAMES = ["mon", "tue", "wed", "thu", "fri", "sat", "sun"]
CATEGORIES = ["weekday", "weekend", "holiday"]

DEFAULT_CONFIG = {
    "schedules": {
        "weekday": {"time": "07:00", "enabled": True},
        "weekend": {"time": "09:00", "enabled": True},
        "holiday": {"time": "09:00", "enabled": True},
    },
    "sounds": {day: DEFAULT_SOUND for day in DAY_NAMES},
}


def load_config():
    if not os.path.exists(CONFIG_FILE):
        return json.loads(json.dumps(DEFAULT_CONFIG))
    with open(CONFIG_FILE, "r") as f:
        config = json.load(f)
    # backfill any missing keys (safe for older config files)
    for cat in CATEGORIES:
        config["schedules"].setdefault(cat, DEFAULT_CONFIG["schedules"][cat])
    for day in DAY_NAMES:
        config["sounds"].setdefault(day, DEFAULT_SOUND)
    return config


def save_config(config):
    with open(CONFIG_FILE, "w") as f:
        json.dump(config, f, indent=2, ensure_ascii=False)


def category_for_date(d):
    if jpholiday.is_holiday(d):
        return "holiday"
    return "weekday" if d.weekday() < 5 else "weekend"


def cmd_set(args):
    if args.category not in CATEGORIES:
        print(f"category must be one of {CATEGORIES}")
        sys.exit(1)
    try:
        datetime.strptime(args.time, "%H:%M")
    except ValueError:
        print("time must be HH:MM")
        sys.exit(1)

    config = load_config()
    config["schedules"][args.category]["time"] = args.time
    save_config(config)
    print(f"{args.category} alarm set to {args.time}")


def cmd_toggle(args, enabled):
    if args.category not in CATEGORIES:
        print(f"category must be one of {CATEGORIES}")
        sys.exit(1)
    config = load_config()
    config["schedules"][args.category]["enabled"] = enabled
    save_config(config)
    print(f"{args.category} alarm {'enabled' if enabled else 'disabled'}")


def cmd_sound(args):
    if args.day not in DAY_NAMES:
        print(f"day must be one of {DAY_NAMES}")
        sys.exit(1)
    config = load_config()
    config["sounds"][args.day] = args.path
    save_config(config)
    print(f"{args.day} sound set to {args.path}")


def cmd_show(args):
    config = load_config()
    print("schedules:")
    for cat in CATEGORIES:
        s = config["schedules"][cat]
        status = "ON" if s["enabled"] else "off"
        print(f"  {cat:8s} {s['time']} [{status}]")
    print("sounds:")
    for day in DAY_NAMES:
        print(f"  {day}: {config['sounds'][day]}")

    today = datetime.now().date()
    print(f"today ({today}) category: {category_for_date(today)}")


def cmd_stop(args):
    if not os.path.exists(PID_FILE):
        print("no alarm currently sounding")
        return
    with open(PID_FILE) as f:
        pid = int(f.read().strip())
    try:
        os.killpg(os.getpgid(pid), signal.SIGTERM)
        print("alarm stopped")
    except ProcessLookupError:
        print("alarm process already gone")
    os.remove(PID_FILE)


def play_sound(sound_path):
    proc = subprocess.Popen(
        ["mpg123", "--loop", "-1", sound_path],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL,
        preexec_fn=os.setsid,
    )
    with open(PID_FILE, "w") as f:
        f.write(str(proc.pid))
    return proc


def cmd_run(args):
    print("alarm loop started, checking every 20s. Ctrl+C to quit.")
    fired_this_minute = set()
    while True:
        now = datetime.now()
        current_time = now.strftime("%H:%M")
        minute_key = now.strftime("%Y%m%d%H%M")
        category = category_for_date(now.date())
        day_name = DAY_NAMES[now.weekday()]

        config = load_config()
        schedule = config["schedules"][category]
        if schedule["enabled"] and schedule["time"] == current_time:
            key = (category, minute_key)
            if key not in fired_this_minute:
                fired_this_minute.add(key)
                sound = config["sounds"][day_name]
                print(f"[{now}] firing {category} alarm, playing {sound}")
                play_sound(sound)

        fired_this_minute = {k for k in fired_this_minute if k[1] == minute_key}
        time.sleep(20)


def main():
    parser = argparse.ArgumentParser(description="Raspberry Pi alarm clock")
    sub = parser.add_subparsers(dest="command", required=True)

    p_set = sub.add_parser("set", help="set alarm time for a category")
    p_set.add_argument("category", choices=CATEGORIES)
    p_set.add_argument("time", help="HH:MM 24h format")
    p_set.set_defaults(func=cmd_set)

    p_enable = sub.add_parser("enable", help="enable a category's alarm")
    p_enable.add_argument("category", choices=CATEGORIES)
    p_enable.set_defaults(func=lambda args: cmd_toggle(args, True))

    p_disable = sub.add_parser("disable", help="disable a category's alarm")
    p_disable.add_argument("category", choices=CATEGORIES)
    p_disable.set_defaults(func=lambda args: cmd_toggle(args, False))

    p_sound = sub.add_parser("sound", help="set the sound file for a day of week")
    p_sound.add_argument("day", choices=DAY_NAMES)
    p_sound.add_argument("path", help="path to mp3")
    p_sound.set_defaults(func=cmd_sound)

    p_show = sub.add_parser("show", help="show current config")
    p_show.set_defaults(func=cmd_show)

    p_stop = sub.add_parser("stop", help="stop a currently sounding alarm")
    p_stop.set_defaults(func=cmd_stop)

    p_run = sub.add_parser("run", help="run the alarm checking loop (foreground)")
    p_run.set_defaults(func=cmd_run)

    args = parser.parse_args()
    args.func(args)


if __name__ == "__main__":
    main()

$ alarm_clock.py set weekday 06:30

{
    "weekday":{
        "time":"06:30"
    }
}

$ alarm_clock.py sound mon music.mp3