// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#pragma once

#include <ddktl/device-internal.h>
#include <ddktl/device.h>
#include <ddktl/protocol/empty-protocol.h>
#include <fuchsia/hardware/camera/c/fidl.h>
#include <lib/async-loop/cpp/loop.h>
#include <lib/async/default.h>
#include <lib/async/wait.h>
#include <lib/fidl-utils/bind.h>
#include <lib/fzl/vmo-pool.h>

namespace virtual_camera {

class VirtualCameraDevice;
using VirtualCameraDeviceType =
    ddk::Device<VirtualCameraDevice, ddk::Unbindable, ddk::Messageable>;

class VirtualCameraDevice : public VirtualCameraDeviceType,
                            public ddk::EmptyProtocol<ZX_PROTOCOL_CAMERA> {
public:
    DISALLOW_COPY_ASSIGN_AND_MOVE(VirtualCameraDevice);

    static zx_status_t Create(void* ctx, zx_device_t* parent);
    VirtualCameraDevice(zx_device_t* parent)
        : VirtualCameraDeviceType(parent) {}

    ~VirtualCameraDevice() {}

    // DDK device implementation
    void DdkRelease();
    void DdkUnbind();
    zx_status_t DdkMessage(fidl_msg_t* msg, fidl_txn_t* txn);

private:
    // DDK Helper Functions.
    zx_status_t StartStreaming();
    zx_status_t StopStreaming();
    zx_status_t ReleaseFrame(uint32_t buffer_id);
    zx_status_t GetFormats(uint32_t index, fidl_txn_t* txn);
    zx_status_t CreateStream(
        const fuchsia_sysmem_BufferCollectionInfo* buffer_collection_info,
        const fuchsia_hardware_camera_FrameRate* rate, zx_handle_t stream,
        zx_handle_t stream_token);
    zx_status_t GetDeviceInfo(fidl_txn_t* txn);

    static constexpr fuchsia_hardware_camera_Stream_ops_t stream_ops = {
        .Start = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::StartStreaming>,
        .Stop = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::StopStreaming>,
        .ReleaseFrame = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::ReleaseFrame>,
    };

    static constexpr fuchsia_hardware_camera_Control_ops_t control_ops = {
        .GetFormats = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::GetFormats>,
        .CreateStream = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::CreateStream>,
        .GetDeviceInfo = fidl::Binder<VirtualCameraDevice>::BindMember<
            &VirtualCameraDevice::GetDeviceInfo>,
    };

    fzl::VmoPool buffers_;
};

} // namespace virtual_camera