#pragma once

#include "EDeviceCommandStatus.h"
#include <memory>
#include <vector>

namespace Incart::Usb
{

struct DeviceCommandStatus
{
    DeviceCommand::EStatus status;

    DeviceCommandStatus(DeviceCommand::EStatus status) : status(status)
    {
    }

    virtual ~DeviceCommandStatus()
    {
    }
};

template<typename TInfo>
struct DeviceCommandStatusInfo final : DeviceCommandStatus
{
    TInfo info;

    DeviceCommandStatusInfo() : DeviceCommandStatus(DeviceCommand::EStatus::OK)
    {
    }

    DeviceCommandStatusInfo(DeviceCommand::EStatus status) : DeviceCommandStatus(status)
    {
    }

    DeviceCommandStatusInfo(DeviceCommand::EStatus status, TInfo info_) : DeviceCommandStatus(status)
        , info(info_)
    {
    }
};

struct DeviceCommandStatusCreator
{
    static std::shared_ptr<Usb::DeviceCommandStatus> create(const std::vector<uint8_t>& answer_, int32_t status_)
    {
        std::shared_ptr<DeviceCommandStatus> status;
        if (status_ == DeviceCommand::EStatus::WRONG_FRAME_FORMAT)
        {
            status = std::make_shared<DeviceCommandStatusInfo<std::vector<uint8_t>>>(DeviceCommand::EStatus::WRONG_FRAME_FORMAT, answer_);
        }
        else
        {
            status = std::make_shared<DeviceCommandStatus>((DeviceCommand::EStatus)status_);
        }
        return status;
    }
};

} // namespace Incart::Usb