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

#pragma once

#include <Common.hpp>
#include <Service/Common.hpp>
#include <Service/Message.hpp>
#include <Service/Service.hpp>
#include <Service/Timer.hpp>

#include <service-db/DBServiceName.hpp>

#include "messages/RenderingFinished.hpp"

#include "ContextPool.hpp"
#include "DrawCommandsQueue.hpp"
#include "Common.hpp"

#include <cstdint>
#include <memory>
#include <string>

namespace gui
{
    class Context;
    class DrawCommand;
} // namespace gui

namespace service::gui
{
    class WorkerGUI;

    class ServiceGUI : public sys::Service
    {
        friend WorkerGUI;

      public:
        explicit ServiceGUI(const std::string &name = service::name::gui, std::string parent = {});
        ~ServiceGUI() noexcept override;

        sys::ReturnCodes InitHandler() override;
        sys::ReturnCodes DeinitHandler() override;
        sys::MessagePointer DataReceivedHandler(sys::DataMessage *msgl, sys::ResponseMessage *resp) override;
        sys::ReturnCodes SwitchPowerModeHandler(const sys::ServicePowerMode mode) override;

      private:
        struct CachedRender
        {
            int contextId;
            ::gui::RefreshModes refreshMode;
        };
        enum class State
        {
            NotInitialised,
            Idle,
            Busy,
            Suspended
        };

        static void initAssetManagers();
        void registerMessageHandlers();

        void cacheRender(int contextId, ::gui::RefreshModes refreshMode);
        void invalidateCache();

        void prepareDisplayEarly(::gui::RefreshModes refreshMode);
        void notifyRenderer(std::list<std::unique_ptr<::gui::DrawCommand>> &&commands, ::gui::RefreshModes refreshMode);
        void notifyRenderColorSchemeChange(::gui::ColorScheme &&scheme);
        void enqueueDrawCommands(DrawCommandsQueue::QueueItem &&item);
        void sendOnDisplay(::gui::Context *context, int contextId, ::gui::RefreshModes refreshMode);
        void scheduleContextRelease(int contextId);
        bool isNextFrameReady() const noexcept;
        bool isAnyFrameBeingRenderedOrDisplayed() const noexcept;
        void trySendNextFrame();

        void setState(State state) noexcept;
        bool isInState(State state) const noexcept;

        sys::MessagePointer handleDrawMessage(sys::Message *message);
        sys::MessagePointer handleGUIRenderingFinished(sys::Message *message);
        sys::MessagePointer handleEinkInitialized(sys::Message *message);
        sys::MessagePointer handleImageDisplayedNotification(sys::Message *message);
        sys::MessagePointer handleChangeColorScheme(sys::Message *message);

        std::unique_ptr<ContextPool> contextPool;
        std::unique_ptr<WorkerGUI> worker;
        std::unique_ptr<DrawCommandsQueue> commandsQueue;
        std::unique_ptr<::gui::ColorScheme> colorSchemeUpdate;
        std::optional<CachedRender> cachedRender;
        std::unique_ptr<sys::Timer> contextReleaseTimer;
        State currentState;
    };
} // namespace service::gui

namespace sys
{
    template <> struct ManifestTraits<service::gui::ServiceGUI>
    {
        static auto GetManifest() -> ServiceManifest
        {
            ServiceManifest manifest;
            manifest.name         = service::name::gui;
            manifest.dependencies = {service::name::db};
            return manifest;
        }
    };
} // namespace sys