/*
 * Copyright (c) 2018-2019 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY 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/>.
 */

#pragma once
#include <switch.h>
#include "hossynch.hpp"
#include <memory>

class HosMessageQueue {
    private:
        HosMutex queue_lock;
        HosCondVar cv_not_full;
        HosCondVar cv_not_empty;
        std::unique_ptr<uintptr_t[]> buffer;
        size_t capacity;

        size_t count = 0;
        size_t offset = 0;
    public:
        HosMessageQueue(size_t c) : capacity(c) {
            this->buffer = std::make_unique<uintptr_t[]>(this->capacity);
        }

        HosMessageQueue(std::unique_ptr<uintptr_t[]> buf, size_t c) : buffer(std::move(buf)), capacity(c) { }

        /* Sending (FIFO functionality) */
        void Send(uintptr_t data);
        bool TrySend(uintptr_t data);
        bool TimedSend(uintptr_t data, u64 timeout);

        /* Sending (LIFO functionality) */
        void SendNext(uintptr_t data);
        bool TrySendNext(uintptr_t data);
        bool TimedSendNext(uintptr_t data, u64 timeout);

        /* Receive functionality */
        void Receive(uintptr_t *out);
        bool TryReceive(uintptr_t *out);
        bool TimedReceive(uintptr_t *out, u64 timeout);

        /* Peek functionality */
        void Peek(uintptr_t *out);
        bool TryPeek(uintptr_t *out);
        bool TimedPeek(uintptr_t *out, u64 timeout);
    private:
        bool IsFull() {
            return this->count >= this->capacity;
        }

        bool IsEmpty() {
            return this->count == 0;
        }

        void SendInternal(uintptr_t data);
        void SendNextInternal(uintptr_t data);
        uintptr_t ReceiveInternal();
        uintptr_t PeekInternal();

};