/*
 * Copyright (C) 2019  Max Regan
 *
 * 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 "SetTimeScreen.h"
#include "SystemTime.h"
#include "SystemFonts.h"
#include "RtcDriver.h"

using Common::ReturnCode;
using Common::Time;
using Common::Schedule::NextTime;

SetTimeScreen::SetTimeScreen(BSP::DisplayDriver &display,
                             ScreenManager &manager)
    : m_display(display)
    , m_manager(manager)
    , m_state(SetState::HOURS)
    , m_is_acked(false)
    , m_time()
{}

ReturnCode SetTimeScreen::init()
{
    return ReturnCode::OK;
}

NextTime SetTimeScreen::execute()
{
    //TODO: Fix this so it doesn't constantly refresh
    refresh();

    return NextTime::never();
}

static char get_char_for_digit(uint8_t bcd_digit)
{
    if (bcd_digit > 9) {
        return '0';
    }
    return bcd_digit + '0';
}

void SetTimeScreen::display_number(uint32_t *x, uint32_t y,
                                   uint32_t tens, uint32_t ones, const struct font &f)
{
    char time_str[3] = { 0 };
    time_str[0] = get_char_for_digit(tens);
    time_str[1] = get_char_for_digit(ones);
    time_str[2] = '\0';

    m_display.string_at(x, y, time_str, &f);
}

void SetTimeScreen::render_time()
{
    uint32_t x = 0;
    uint32_t y = 32;
    const struct font &font = default_font;

    display_number(&x, y,
                   m_time.get_hours_12_tens(), m_time.get_hours_12_ones(),
                   font);

    m_display.string_at(&x, y, ":", &font);

    display_number(&x, y,
                   m_time.get_minutes_tens(), m_time.get_minutes_ones(),
                   font);

    m_display.string_at(&x, y, ":", &font);

    display_number(&x, y,
                   m_time.get_seconds_tens(), m_time.get_seconds_ones(),
                   font);

    m_display.refresh();
}

void SetTimeScreen::draw_line(uint32_t x, uint32_t y, uint32_t width)
{
    for (uint32_t i = 0; i < width; i += 8) {
        m_display.set_byte(x + i, y, 0);
        m_display.set_byte(x + i, y + 1, 0);
    }
}

void SetTimeScreen::render_selection()
{
    uint32_t font_width = default_font_width;

    switch (m_state) {
    case SetState::HOURS:
        draw_line(0, 64, font_width * 2);
        break;
    case SetState::MINUTES:
        draw_line(font_width * 2, 64, font_width * 2);
        break;
    case SetState::SECONDS:
        draw_line(font_width * 4, 64, font_width * 2);
        break;
    case SetState::AM_PM:
        break;
    case SetState::ACK_NACK:
        break;
    }
}

void SetTimeScreen::refresh()
{
    m_display.clear();

    render_time();
    render_selection();

    m_display.refresh();
}

void SetTimeScreen::enable()
{
    BSP::RtcDriver::get_time(m_time);
    m_state = SetState::HOURS;
    m_is_acked = true;

    refresh();
}

void SetTimeScreen::disable()
{
    m_display.clear();
}

void SetTimeScreen::notify_up_button()
{

    switch (m_state) {
    case SetState::HOURS:
        m_time.increment_hours();
        break;
    case SetState::MINUTES:
        m_time.increment_minutes();
        break;
    case SetState::SECONDS:
        m_time.increment_seconds();
        break;
    case SetState::AM_PM:
        m_time.toggle_am_pm();
        break;
    case SetState::ACK_NACK:
        m_is_acked = !m_is_acked;
        break;
    }

    refresh();
}

void SetTimeScreen::notify_middle_button()
{
    switch (m_state) {
    case SetState::HOURS:
        m_state = SetState::MINUTES;
        break;
    case SetState::MINUTES:
        m_state = SetState::SECONDS;
        break;
    case SetState::SECONDS:
        m_state = SetState::AM_PM;
        break;
    case SetState::AM_PM:
        m_state = SetState::ACK_NACK;
        break;
    case SetState::ACK_NACK:
        if (m_is_acked) {
            BSP::RtcDriver::set_time(m_time);
        }
        m_manager.pop_screen();
        return;
    }

    refresh();
}

void SetTimeScreen::notify_down_button()
{
    switch(m_state) {
    case SetState::HOURS:
        m_time.decrement_hours();
        break;
    case SetState::MINUTES:
        m_time.decrement_minutes();
        break;
    case SetState::SECONDS:
        m_time.decrement_seconds();
        break;
    case SetState::AM_PM:
        m_time.toggle_am_pm();
        break;
    case SetState::ACK_NACK:
        m_is_acked = !m_is_acked;
        break;
    }

    refresh();
}