#!/usr/bin/env python3
# kinds of goals:
# - overall
#     "i want to code at least {time} per {interval}"
# - in a project
# - in a language
# - in an editor

from datetime import timedelta, date, datetime, time
import sys
import json
import sqlite3
from config import timeout
from utils import window

conn = sqlite3.connect("data.db")

if len(sys.argv) < 2:
    print("provide a file", file=sys.stderr)
    sys.exit(1)

filename = sys.argv[1]
with open(filename) as f:
    data = json.load(f)

goal_type = data["type"]
if goal_type == "overall":
    # in seconds
    goal_min = int(data["goal_min"])

    # figure out what timestamps to pull from
    if data["interval"] == "day":
        # except!
        def convert_weekday(s):
            s = s.lower()
            return dict(
                sunday=0,
                monday=1,
                tuesday=2,
                wednesday=3,
                thursday=4,
                friday=5,
                saturday=6,
            ).get(s)
        excepts = list(map(convert_weekday, data["except"]))

        if len(sys.argv) > 2:
            # if a day is given, print that day
            d = datetime.strptime(sys.argv[2], "%Y-%m-%d").date()
            start_time = datetime.combine(d, time(hour=0, minute=0, second=0))
            end_time = datetime.combine(d, time(hour=23, minute=59, second=59))
        else:
            # find the current day!
            today = date.today()
            start_time = datetime.combine(today, time(hour=0, minute=0, second=0))
            end_time = datetime.now()
        print(f"from {start_time} to {end_time}")
        start_stamp = datetime.timestamp(start_time)
        end_stamp = datetime.timestamp(end_time)

        # get all the entries
        c = conn.cursor()
        curs = c.execute("""
            select time from heartbeats where time > ? and time < ? order by time;
        """, [start_stamp, end_stamp])

        coded_secs = 0
        for (first, second) in window(curs, 2):
            print(datetime.fromtimestamp(first[0]))
            diff = second[0] - first[0]
            if diff < timeout:
                coded_secs += diff
        print(coded_secs)

    elif data["interval"] == "week":
        # start of the week
        today = date.today()
        last_monday = today - timedelta(days=today.weekday())
        print(last_monday)

# vim: set ft=python: