from __future__ import annotations

import argparse
from datetime import date

from .config import read_config
from .db import Database
from .report_service import ReportService
from .robot_service import create_reminder_payload, create_summary_payload, send_webhook
from .workday import is_workday


def send_reminder() -> dict:
    config = read_config()
    if not is_workday(date.today(), config.workday_calendar_path):
        return {"skipped": True, "reason": "not a workday"}
    payload = create_reminder_payload(f"{config.base_url}/submit")
    return send_webhook(config.feishu_webhook_url, payload)


def send_summary(report_date: str | None = None) -> dict:
    config = read_config()
    selected_date = date.fromisoformat(report_date) if report_date else date.today()
    if not is_workday(selected_date, config.workday_calendar_path):
        return {"skipped": True, "reason": "not a workday", "date": selected_date.isoformat()}

    database = Database(config.database_path, config.employee_seed_path)
    try:
        service = ReportService(database)
        selected_date_text = selected_date.isoformat()
        summary = service.list_reports_for_date(selected_date_text)
        payload = create_summary_payload(f"{config.base_url}/manager?date={selected_date_text}", summary)
        return send_webhook(config.feishu_webhook_url, payload)
    finally:
        database.close()


def main() -> None:
    parser = argparse.ArgumentParser(description="发送飞书日报定时消息")
    parser.add_argument("action", choices=["reminder", "summary"], help="reminder 发送提醒，summary 发送汇总")
    parser.add_argument("--date", help="汇总日期，格式 YYYY-MM-DD；不传则使用当天")
    args = parser.parse_args()

    if args.action == "reminder":
        result = send_reminder()
    else:
        result = send_summary(args.date)
    print(result)


if __name__ == "__main__":
    main()