#pragma once

#include "mboxd_pnor_partition_table.h"
#include <fcntl.h>
#include <string>
#include <unistd.h>
#include <experimental/filesystem>

namespace openpower
{
namespace file
{

class Descriptor
{
    private:
        /** default value */
        int fd = -1;

    public:
        Descriptor() = default;
        Descriptor(const Descriptor&) = delete;
        Descriptor& operator=(const Descriptor&) = delete;
        Descriptor(Descriptor&&) = delete;
        Descriptor& operator=(Descriptor &&) = delete;

        Descriptor(int fd) : fd(fd) {}

        ~Descriptor()
        {
            if (fd >= 0)
            {
                close(fd);
            }
        }

        int operator()() const
        {
            return fd;
        }

        void set(int descriptor)
        {
            fd = descriptor;
        }
};

}// namespace file

namespace virtual_pnor
{

namespace fs = std::experimental::filesystem;

enum class ReturnCode : uint8_t
{
    FILE_NOT_FOUND = 0,
    PARTITION_NOT_FOUND = 1,
    PARTITION_READ_ONLY = 2,
    FILE_OPEN_FAILURE = 3,
    SUCCESS = 4,
};

class Request
{
    public:

        Request() = default;
        Request(const Request&) = delete;
        Request& operator=(const Request&) = delete;
        Request(Request&&) = default;
        Request& operator=(Request&&) = default;
        ~Request() = default;

        openpower::file::Descriptor fd;

    protected:
        /** @brief opens the partition file
         *
         *  @param[in] filePath - Absolute file path.
         *  @param[in] mode - File open mode.
         */
        ReturnCode open(const std::string& filePath, int mode);

        /** @brief returns the partition file path associated with the offset.
         *
         *  @param[in] context - The mbox context pointer.
         *  @param[in] offset - The pnor offset(bytes).
         */

        std::string getPartitionFilePath(struct mbox_context* context,
                                         uint32_t offset);

        const pnor_partition* partition = nullptr;
};

/** @class RORequest
 *  @brief Represent the read request of the partition.
 *         Stores the partition meta data.
 */
class RORequest : public Request
{
    public:
        RORequest() = default;
        RORequest(const RORequest&) = delete;
        RORequest& operator=(const RORequest&) = delete;
        RORequest(RORequest&&) = default;
        RORequest& operator=(RORequest&&) = default;
        ~RORequest(){};

        /** @brief opens the partition file associated with the offset
         *         in read only mode and gets the partition details.
         *
         *  1.  Depending on the partition type,tries to open the file
         *      from the associated partition(RW/PRSV/RO).
         *  1a. if file not found in the corresponding
         *      partition(RW/PRSV/RO) then tries to read the file from
         *      the read only partition.
         *  1b. if the file not found in the read only partition then
         *      throw exception.
         *
         *  @param[in] context - The mbox context pointer.
         *  @param[in] offset - The pnor offset(bytes).
         */
        const pnor_partition* getPartitionInfo(struct mbox_context* context,
                                               uint32_t offset);
};

/** @class RWRequest
 *  @brief Represent the write request of the partition.
 *         Stores the partition meta data.
 */
class RWRequest : public Request
{
    public:

        RWRequest() = default;
        RWRequest(const RWRequest&) = delete;
        RWRequest& operator=(const RWRequest&) = delete;
        RWRequest(RWRequest&&) = default;
        RWRequest& operator=(RWRequest&&) = default;
        ~RWRequest() {};

        /** @brief opens the partition file associated with the offset
         *         in write mode and gets the parttition details.
         *
         *  1.  Depending on the partition type tries to open the file
         *      from the associated partition.
         *  1a. if file not found in the corresponding partition(RW/PRSV)
         *      then copy the file from the read only partition to the (RW/PRSV)
         *      partition depending on the partition type.
         *  1b. if the file not found in the read only partition then throw exception.
         *
         *  @param[in] context - The mbox context pointer.
         *  @param[in] offset - The pnor offset(bytes).
         */
        const pnor_partition* getPartitionInfo(struct mbox_context* context,
                                               uint32_t offset);
};

}// namespace virtual_pnor
}// namespace openpower