// Copyright (c) 2017-2020, Mudita Sp. z.o.o. All rights reserved.
// For licensing, see https://github.com/mudita/MuditaOS/LICENSE.md

#include "CalendarListView.hpp"

namespace gui
{

    CalendarListView::CalendarListView(
        Item *parent, uint32_t x, uint32_t y, uint32_t w, uint32_t h, std::shared_ptr<ListItemProvider> prov)
        : ListView(parent, x, y, w, h, prov)
    {
        assert(prov != nullptr);
    }

    void CalendarListView::addLabelMarker(gui::AllEventsItem *item)
    {
        assert(item != nullptr);
        if (direction == style::listview::Direction::Bottom) {
            if (!(labelMark == item->getLabelMarker())) {

                labelMark = item->getLabelMarker();

                auto *calendarAllEventsItem = new gui::AllEventsItem();
                calendarAllEventsItem->setMarkerItem(labelMark);

                body->addWidget(calendarAllEventsItem);
            }
        }
        if (direction == style::listview::Direction::Top) {

            if (currentPageSize == 0) {
                labelMark = item->getLabelMarker();
                return;
            }
            else if (!(labelMark == item->getLabelMarker())) {

                previousLabelMark = labelMark;
                labelMark         = item->getLabelMarker();

                auto *calendarAllEventsItem = new gui::AllEventsItem();
                calendarAllEventsItem->setMarkerItem(previousLabelMark);

                body->removeWidget(item);
                body->addWidget(calendarAllEventsItem);
                body->addWidget(item);

                previousItemIsLabel = true;
            }
        }
    }

    void CalendarListView::addItemsOnPage()
    {
        currentPageSize            = 0;
        ListItem *item             = nullptr;
        ListItem *previousListItem = nullptr;
        labelMark                  = "";
        previousLabelMark          = "";

        while ((item = provider->getItem(getOrderFromDirection())) != nullptr) {

            // if direction bot add label mark before adding item
            if (direction == style::listview::Direction::Bottom) {
                auto eventItem = dynamic_cast<gui::AllEventsItem *>(item);
                if (eventItem != nullptr) {
                    addLabelMarker(eventItem);
                }
            }

            body->addWidget(item);

            // if added item is not visible -> page is full.
            if (!item->visible) {

                // if page full and direction top remove last element and add floating label mark on top if there was no
                // one previously
                if (direction == style::listview::Direction::Top) {

                    if (previousItemIsLabel) {
                        break;
                    }
                    else {

                        body->erase(item);
                        body->erase(previousListItem);

                        auto *calendarAllEventsItem = new gui::AllEventsItem();

                        calendarAllEventsItem->setMarkerItem(labelMark);

                        body->addWidget(calendarAllEventsItem);
                        currentPageSize--;
                    }
                }
                break;
            }

            // if direction top add label mark after adding item
            if (direction == style::listview::Direction::Top) {
                previousItemIsLabel = false;
                auto eventItem      = dynamic_cast<gui::AllEventsItem *>(item);
                if (eventItem != nullptr) {
                    addLabelMarker(eventItem);
                }
            }

            previousListItem = item;

            currentPageSize++;
        }

        recalculateStartIndex();

        // Add element on top for first page purpose
        if (startIndex == 0 && direction == style::listview::Direction::Top) {
            auto *calendarAllEventsItem = new gui::AllEventsItem();
            calendarAllEventsItem->setMarkerItem(labelMark);
            body->addWidget(calendarAllEventsItem);
        }
    }

} /* namespace gui */