/*
 * MIT License
 *
 * Copyright (c) 2022 Joey Castillo
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <stdlib.h>
#include <string.h>
#include "day_one_face.h"
#include "watch.h"

static uint32_t _day_one_face_juliandaynum(uint16_t year, uint16_t month, uint16_t day) {
    // from here: https://en.wikipedia.org/wiki/Julian_day#Julian_day_number_calculation
    return (1461 * (year + 4800 + (month - 14) / 12)) / 4 + (367 * (month - 2 - 12 * ((month - 14) / 12))) / 12 - (3 * ((year + 4900 + (month - 14) / 12) / 100))/4 + day - 32075;
}

static void _day_one_face_update(day_one_state_t state) {
    char buf[14];
    watch_date_time date_time = watch_rtc_get_date_time();
    uint32_t julian_date = _day_one_face_juliandaynum(date_time.unit.year + WATCH_RTC_REFERENCE_YEAR, date_time.unit.month, date_time.unit.day);
    uint32_t julian_birthdate = _day_one_face_juliandaynum(state.birth_year, state.birth_month, state.birth_day);
    sprintf(buf, "DA  %6lu", julian_date - julian_birthdate);
    watch_display_string(buf, 0);
}

void day_one_face_setup(movement_settings_t *settings, uint8_t watch_face_index, void ** context_ptr) {
    (void) settings;
    (void) watch_face_index;
    if (*context_ptr == NULL) {
        *context_ptr = malloc(sizeof(day_one_state_t));
        memset(*context_ptr, 0, sizeof(day_one_state_t));
        movement_birthdate_t movement_birthdate = (movement_birthdate_t) watch_get_backup_data(2);
        if (movement_birthdate.reg == 0) {
            // if birth date is totally blank, set a reasonable starting date. this works well for anyone under 63, but
            // you can keep pressing to go back to 1900; just pass the current year. also picked this date because if you
            // set it to 1959-01-02, it counts up from the launch of Luna-1, the first spacecraft to leave the well.
            movement_birthdate.bit.year = 1959;
            movement_birthdate.bit.month = 1;
            movement_birthdate.bit.day = 1;
            watch_store_backup_data(movement_birthdate.reg, 2);
        }
    }
}

void day_one_face_activate(movement_settings_t *settings, void *context) {
    (void) settings;
    day_one_state_t *state = (day_one_state_t *)context;

    // stash the current year, useful in birthday setting mode.
    watch_date_time date_time = watch_rtc_get_date_time();
    state->current_year = date_time.unit.year + WATCH_RTC_REFERENCE_YEAR;
    // reset the current page to 0, display days alive.
    state->current_page = 0;

    // fetch the user's birth date from the birthday register.
    movement_birthdate_t movement_birthdate = (movement_birthdate_t) watch_get_backup_data(2);
    state->birth_year = movement_birthdate.bit.year;
    state->birth_month = movement_birthdate.bit.month;
    state->birth_day = movement_birthdate.bit.day;
}

bool day_one_face_loop(movement_event_t event, movement_settings_t *settings, void *context) {
    (void) settings;
    day_one_state_t *state = (day_one_state_t *)context;

    const uint8_t days_in_month[12] = {31, 29, 31, 30, 31, 30, 30, 31, 30, 31, 30, 31};
    char buf[6];

    switch (event.event_type) {
        case EVENT_ACTIVATE:
            _day_one_face_update(*state);
            break;
        case EVENT_LOW_ENERGY_UPDATE:
        case EVENT_TICK:
            if (state->current_page != 0) {
                // if in settings mode, update whatever the current page is
                switch (state->current_page) {
                    case 1:
                        watch_display_string("YR        ", 0);
                        if (event.subsecond % 2) {
                            sprintf(buf, "%4d", state->birth_year);
                            watch_display_string(buf, 4);
                        }
                        break;
                    case 2:
                        watch_display_string("MO        ", 0);
                        if (event.subsecond % 2) {
                            sprintf(buf, "%2d", state->birth_month);
                            watch_display_string(buf, 4);
                        }
                        break;
                    case 3:
                        watch_display_string("DA        ", 0);
                        if (event.subsecond % 2) {
                            sprintf(buf, "%2d", state->birth_day);
                            watch_display_string(buf, 6);
                        }
                        break;
                }
            } else {
                // otherwise, check if we have to update. the display only needs to change at midnight!
                watch_date_time date_time = watch_rtc_get_date_time();
                if (date_time.unit.hour == 0 &&  date_time.unit.minute == 0 && date_time.unit.second == 0) {
                    _day_one_face_update(*state);
                }
            }
            break;
        case EVENT_MODE_BUTTON_UP:
            movement_move_to_next_face();
            break;
        case EVENT_LIGHT_BUTTON_DOWN:
            // only illuminate if we're in display mode
            if (state->current_page == 0) movement_illuminate_led();
            break;
        case EVENT_LIGHT_BUTTON_UP:
            // otherwise use the light button to advance settings pages.
            if (state->current_page != 0) {
                // go to next setting page...
                state->current_page = (state->current_page + 1) % 4;
                if (state->current_page == 0) {
                    // ...unless we've been pushed back to display mode.
                    movement_request_tick_frequency(1);
                    // force display since it normally won't update til midnight.
                    _day_one_face_update(*state);
                }
            }
            break;
        case EVENT_ALARM_BUTTON_UP:
            // if we are on a settings page, increment whatever value we're setting.
            if (state->current_page != 0) {
                state->birthday_changed = true;
                switch (state->current_page) {
                    case 1:
                        state->birth_year = state->birth_year + 1;
                        if (state->birth_year > state->current_year) state->birth_year = 1900;
                        break;
                    case 2:
                        state->birth_month = (state->birth_month % 12) + 1;
                        break;
                    case 3:
                        state->birth_day = state->birth_day + 1;
                        if (state->birth_day == 0 || state->birth_day > days_in_month[state->birth_month - 1]) {
                            state->birth_day = 1;
                        }
                        break;
                }
            }
            break;
        case EVENT_ALARM_LONG_PRESS:
            // if we aren't already in settings mode, put us there.
            if (state->current_page == 0) {
                state->current_page++;
                movement_request_tick_frequency(4);
            }
            break;
        case EVENT_TIMEOUT:
            // return home if we're on a settings page (this saves our changes when we resign).
            if (state->current_page != 0) {
                movement_move_to_face(0);
            }
        default:
            break;
    }

    return true;
}

void day_one_face_resign(movement_settings_t *settings, void *context) {
    (void) settings;
    day_one_state_t *state = (day_one_state_t *)context;

    // if the user changed their birth date, store it to the birth date register
    if (state->birthday_changed) {
        day_one_state_t *state = (day_one_state_t *)context;
        movement_birthdate_t movement_birthdate = (movement_birthdate_t) watch_get_backup_data(2);
        movement_birthdate.bit.year = state->birth_year;
        movement_birthdate.bit.month = state->birth_month;
        movement_birthdate.bit.day = state->birth_day;
        watch_store_backup_data(movement_birthdate.reg, 2);
        state->birthday_changed = false;
    }
}