/*
 * Copyright 2020 HIMSA II K/S - www.himsa.com.
 * Represented by EHIMA - www.ehima.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#pragma once

#include <base/callback.h>
#include <gmock/gmock.h>

#include <array>

#include "btm_iso_api_types.h"
#include "hcimsgs.h"

namespace hcic {
class HcicInterface {
 public:
  // iso_manager::cig_create_params is a workaround for the 10 params function
  // limitation that gmock sets
  virtual void SetCigParams(
      uint8_t cig_id,
      struct bluetooth::hci::iso_manager::cig_create_params cig_params,
      base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  virtual void RemoveCig(uint8_t cig_id,
                         base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  virtual void CreateCis(uint8_t num_cis,
                         const EXT_CIS_CREATE_CFG* cis_create_cfg,
                         base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  virtual void Disconnect(uint16_t handle, uint8_t reason) = 0;

  virtual void SetupIsoDataPath(
      uint16_t iso_handle, uint8_t data_path_dir, uint8_t data_path_id,
      uint8_t codec_id_format, uint16_t codec_id_company,
      uint16_t codec_id_vendor, uint32_t controller_delay,
      std::vector<uint8_t> codec_conf,
      base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  virtual void RemoveIsoDataPath(
      uint16_t iso_handle, uint8_t data_path_dir,
      base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  virtual void ReadIsoLinkQuality(
      uint16_t iso_handle, base::OnceCallback<void(uint8_t*, uint16_t)> cb) = 0;

  // iso_manager::big_create_params is a workaround for the 10 params function
  // limitation that gmock sets
  virtual void CreateBig(
      uint8_t big_handle,
      struct bluetooth::hci::iso_manager::big_create_params big_params) = 0;

  virtual void TerminateBig(uint8_t big_handle, uint8_t reason) = 0;

  virtual ~HcicInterface() = default;
};

class MockHcicInterface : public HcicInterface {
 public:
  MOCK_METHOD((void), SetCigParams,
              (uint8_t cig_id,
               struct bluetooth::hci::iso_manager::cig_create_params cig_params,
               base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD((void), RemoveCig,
              (uint8_t cig_id, base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD((void), CreateCis,
              (uint8_t num_cis, const EXT_CIS_CREATE_CFG* cis_create_cfg,
               base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD((void), Disconnect, (uint16_t handle, uint8_t reason),
              (override));

  MOCK_METHOD((void), SetupIsoDataPath,
              (uint16_t iso_handle, uint8_t data_path_dir, uint8_t data_path_id,
               uint8_t codec_id_format, uint16_t codec_id_company,
               uint16_t codec_id_vendor, uint32_t controller_delay,
               std::vector<uint8_t> codec_conf,
               base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD((void), RemoveIsoDataPath,
              (uint16_t iso_handle, uint8_t data_path_dir,
               base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD((void), ReadIsoLinkQuality,
              (uint16_t iso_handle,
               base::OnceCallback<void(uint8_t*, uint16_t)> cb),
              (override));

  MOCK_METHOD(
      (void), CreateBig,
      (uint8_t big_handle,
       struct bluetooth::hci::iso_manager::big_create_params big_params),
      (override));

  MOCK_METHOD((void), TerminateBig, (uint8_t big_handle, uint8_t reason),
              (override));
};

void SetMockHcicInterface(MockHcicInterface* mock_hcic_interface);

}  // namespace hcic