/*
 * Copyright 2015 Canonical Ltd.
 * Copyright 2021-2024 Robert Tari
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 3, as published
 * by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranties of
 * MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Authors:
 *   Charles Kerr <charles.kerr@canonical.com>
 *   Robert Tari <robert@tari.in>
 */

#include <algorithm>
#include <array>
#include <datetime/alarm-queue-simple.h>
#include <datetime/clock-mock.h>
#include <datetime/engine-eds.h>
#include <datetime/myself.h>
#include <datetime/planner-range.h>

#include <gtest/gtest.h>

#include "glib-fixture.h"
#include "timezone-mock.h"
#include "wakeup-timer-mock.h"

using namespace ayatana::indicator::datetime;
using VAlarmFixture = GlibFixture;

/***
****
***/

TEST_F(VAlarmFixture, MissingTriggers)
{
    // start the EDS engine
    auto engine = std::make_shared<EdsEngine>(std::make_shared<Myself>());

    // we need a consistent timezone for the planner and our local DateTimes
    constexpr char const * zone_str {"America/Chicago"};
    auto tz = std::make_shared<MockTimezone>(zone_str);
    #if GLIB_CHECK_VERSION(2, 68, 0)
    auto gtz = g_time_zone_new_identifier(zone_str);

    if (gtz == NULL)
    {
        gtz = g_time_zone_new_utc();
    }
    #else
    auto gtz = g_time_zone_new(zone_str);
    #endif

    // make a planner that looks at the first half of 2015 in EDS
    auto planner = std::make_shared<SimpleRangePlanner>(engine, tz);
    const DateTime range_begin {gtz, 2015,1, 1, 0, 0, 0.0};
    const DateTime range_end   {gtz, 2015,6,30,23,59,59.5};
    planner->range().set(std::make_pair(range_begin, range_end));

    // give EDS a moment to load
    if (planner->appointments().get().empty()) {
        g_message("waiting a moment for EDS to load...");
        auto on_appointments_changed = [this](const std::vector<Appointment>& appointments){
            g_message("ah, they loaded");
            if (!appointments.empty())
                g_main_loop_quit(loop);
        };
        core::ScopedConnection conn(planner->appointments().changed().connect(on_appointments_changed));
        constexpr int max_wait_sec = 10;
        wait_msec(max_wait_sec * G_TIME_SPAN_MILLISECOND);
    }

    // build expected: one-time alarm 1
    std::vector<Appointment> expected1;
    Appointment a1;
    a1.type = Appointment::ALARM;
    a1.uid = "20150617T211838Z-6217-32011-2036-1@lomiri-phablet";
    a1.color = "#becedd";
    a1.summary = "One Time Alarm";
    a1.begin = DateTime { gtz, 2015, 6, 18, 10, 0, 0};
    a1.end = a1.begin;
    a1.alarms.resize(1);
    a1.alarms[0].audio_url = "file://" ALARM_DEFAULT_SOUND;
    a1.alarms[0].time = a1.begin;
    a1.alarms[0].text = a1.summary;
    expected1.push_back(a1);

    // build expected: recurring alarm 1
    a1.uid = "20150617T211913Z-6217-32011-2036-5@lomiri-phablet";
    a1.summary = "Recurring Alarm";
    a1.alarms[0].text = a1.summary;
    std::array<DateTime,13> recurrences {
        DateTime{ gtz, 2015, 6, 18, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 19, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 20, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 21, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 22, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 23, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 24, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 25, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 26, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 27, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 28, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 29, 10, 1, 0 },
        DateTime{ gtz, 2015, 6, 30, 10, 1, 0 },
    };
    for (const auto& time : recurrences) {
        a1.begin = a1.end = a1.alarms[0].time = time;
        expected1.push_back(a1);
    }

    // build expected: one-time alarm 2
    std::vector<Appointment> expected2;
    Appointment a2;
    a2.type = Appointment::ALARM;
    a2.uid = "20150617T211838Z-6217-32011-2036-1@lomiri-phablet";
    a2.color = "#62a0ea";
    a2.summary = "One Time Alarm";
    a2.begin = DateTime { gtz, 2015, 6, 18, 10, 0, 0};
    a2.end = a2.begin;
    a2.alarms.resize(1);
    a2.alarms[0].audio_url = "file://" ALARM_DEFAULT_SOUND;
    a2.alarms[0].time = a2.begin;
    a2.alarms[0].text = a2.summary;
    expected2.push_back(a2);

    // build expected: recurring alarm 2
    a2.uid = "20150617T211913Z-6217-32011-2036-5@lomiri-phablet";
    a2.summary = "Recurring Alarm";
    a2.alarms[0].text = a2.summary;
    for (const auto& time : recurrences) {
        a2.begin = a2.end = a2.alarms[0].time = time;
        expected2.push_back(a2);
    }

    // the planner should match what we've got in the calendar.ics file
    const auto appts = planner->appointments().get();

    EXPECT_PRED3([](auto lAppointmentsIn, auto lAppointmentsExpected1, auto lAppointmentsExpected2)
    {
        return lAppointmentsIn == lAppointmentsExpected1 || lAppointmentsIn == lAppointmentsExpected2;
    }, appts, expected1, expected2);

    // cleanup
    g_time_zone_unref(gtz);
}