/*
 * Copyright (C) 2020  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 <cinttypes>
#include "printf.h"

#include "Bsp/Drivers/GpioDriver.h"
#include "Bsp/Drivers/RtcDriver.h"
#include "Bsp/Drivers/UsartDriver.h"
#include "Bsp/LowPowerTaskScheduler.h"
#include "Bsp/macros.h"

#include "test.h"

#include "Mcu.h"

using BSP::Time;
using BSP::ReturnCode;
using BSP::SystemTimer;
using BSP::time_t;

static BSP::Schedule::LowPowerTaskScheduler<1> g_sched;
static BSP::GpioDriver g_gpioa(GPIOA);
#if defined(BOARD_WATCH)
static BSP::UsartDriver g_test_uart(USART2, g_sched);
static BSP::GpioPin g_tx_pin(g_gpioa, 9);
#elif defined(BOARD_DEVBOARD)
static BSP::UsartDriver g_test_uart(USART1, g_sched);
#endif

static time_t get_time() {
    time_t time;
    ReturnCode rc = SystemTimer::get_time(time);
    if (rc != ReturnCode::OK) {
        g_test_uart.tx_blocking("Failed while getting the time\r\n");
        g_test_uart.tx_blocking(test_fail_text);
        TEST_SPIN();
    }
    return time;
}

static void stop_for(time_t delay) {

    ReturnCode rc = BSP::RtcDriver::set_wakeup_in(delay);
    if (rc != ReturnCode::OK) {
        g_test_uart.tx_blocking("Failed to set wakeup delay\r\n");
        g_test_uart.tx_blocking(test_fail_text);
        TEST_SPIN();
    }

    uint32_t pre_wakeups = BSP::RtcDriver::get_wakeup_count();
    time_t before = get_time();
    while (BSP::RtcDriver::get_wakeup_count() == pre_wakeups) {
    }
    time_t after = get_time();
    uint32_t post_wakeups = BSP::RtcDriver::get_wakeup_count();

    static char buffer[128] = { 0 };
    snprintf(buffer, sizeof(buffer),
             "Requested=%" PRIu32 " Actual=%" PRIu32
             " Wakeups=%" PRId32 "\r\n",
             (uint32_t) Time::to_millis(delay),
             (uint32_t) Time::to_millis(after - before),
             post_wakeups - pre_wakeups);

    g_test_uart.tx_blocking(buffer);
}

static void fail_if_wakeup(time_t delay) {
    uint32_t pre_wakeups = BSP::RtcDriver::get_wakeup_count();
    time_t before = get_time();
    while (before + delay > get_time() && BSP::RtcDriver::get_wakeup_count() == pre_wakeups) {}

    if (BSP::RtcDriver::get_wakeup_count() != pre_wakeups) {
        g_test_uart.tx_blocking("Got unexpected wakeup IRQ\r\n");
        g_test_uart.tx_blocking(test_fail_text);
        TEST_SPIN();
    }
}

[[noreturn]] void main() {

    BSP::RtcDriver::init();
    SystemTimer::set_timer(BSP::RtcDriver::get_system_timer());

    g_gpioa.enable();

#if defined(BOARD_WATCH)
    g_tx_pin.configure_alternate_function(4);
#endif

    g_test_uart.init();
    g_test_uart.tx_blocking(test_start_text);

    for (uint32_t i = 0; i <= 100; i++) {
        stop_for(Time::millis(5));
    }

    for (uint32_t delay_millis = 4; delay_millis <= 1024; delay_millis *= 2) {
        stop_for(Time::millis(delay_millis));
    }

    stop_for(Time::seconds(2));
    stop_for(Time::seconds(4));

    stop_for(Time::millis(5));
    fail_if_wakeup(Time::seconds(1));

    g_test_uart.tx_blocking(test_pass_text);

    TEST_SPIN();
}