/*
 * Copyright 2021 The Android Open Source Project
 *
 * 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.
 */

#define LOG_TAG "BTAudioProviderLeAudioOffload"

#include "LeAudioOffloadAudioProvider.h"

#include <android-base/logging.h>

#include "AudioPort_2_0_to_2_2_Wrapper.h"
#include "BluetoothAudioSessionReport_2_2.h"
#include "BluetoothAudioSupportedCodecsDB_2_2.h"

namespace android {
namespace hardware {
namespace bluetooth {
namespace audio {
namespace V2_2 {
namespace implementation {

using ::android::bluetooth::audio::BluetoothAudioSessionReport_2_2;
using ::android::hardware::Void;
using ::android::hardware::bluetooth::audio::V2_0::BitsPerSample;
using ::android::hardware::bluetooth::audio::V2_0::ChannelMode;
using ::android::hardware::bluetooth::audio::V2_1::SampleRate;

using DataMQ = MessageQueue<uint8_t, kSynchronizedReadWrite>;

LeAudioOffloadOutputAudioProvider::LeAudioOffloadOutputAudioProvider()
    : LeAudioOffloadAudioProvider() {
  session_type_ =
      V2_1::SessionType::LE_AUDIO_HARDWARE_OFFLOAD_ENCODING_DATAPATH;
}

LeAudioOffloadInputAudioProvider::LeAudioOffloadInputAudioProvider()
    : LeAudioOffloadAudioProvider() {
  session_type_ =
      V2_1::SessionType::LE_AUDIO_HARDWARE_OFFLOAD_DECODING_DATAPATH;
}

LeAudioOffloadAudioProvider::LeAudioOffloadAudioProvider()
    : BluetoothAudioProvider() {}

bool LeAudioOffloadAudioProvider::isValid(
    const V2_0::SessionType& sessionType) {
  LOG(ERROR) << __func__
             << ", invalid session type for Offloaded Le Audio provider: "
             << toString(sessionType);

  return false;
}

bool LeAudioOffloadAudioProvider::isValid(
    const V2_1::SessionType& sessionType) {
  return (sessionType == session_type_);
}

Return<void> LeAudioOffloadAudioProvider::startSession_2_1(
    const sp<V2_0::IBluetoothAudioPort>& hostIf,
    const V2_1::AudioConfiguration& audioConfig, startSession_cb _hidl_cb) {
  if (audioConfig.getDiscriminator() !=
      V2_1::AudioConfiguration::hidl_discriminator::leAudioCodecConfig) {
    LOG(WARNING) << __func__
                 << " - Invalid Audio Configuration=" << toString(audioConfig);
    _hidl_cb(BluetoothAudioStatus::UNSUPPORTED_CODEC_CONFIGURATION,
             DataMQ::Descriptor());
    return Void();
  }

  AudioConfiguration audioConfig_2_2;
  audioConfig_2_2.leAudioConfig().mode = LeAudioMode::UNKNOWN;
  audioConfig_2_2.leAudioConfig().config.unicastConfig() = {
      .streamMap = {{
          .streamHandle = 0xFFFF,
          .audioChannelAllocation =
              audioConfig.leAudioCodecConfig().audioChannelAllocation,
      }},
      .peerDelay = 0,
      .lc3Config = audioConfig.leAudioCodecConfig().lc3Config};

  sp<V2_2::IBluetoothAudioPort> hostIf_2_2 =
      new AudioPort_2_0_to_2_2_Wrapper(hostIf);
  return startSession_2_2(hostIf_2_2, audioConfig_2_2, _hidl_cb);
}

Return<void> LeAudioOffloadAudioProvider::startSession_2_2(
    const sp<V2_2::IBluetoothAudioPort>& hostIf,
    const AudioConfiguration& audioConfig, startSession_cb _hidl_cb) {
  /**
   * Initialize the audio platform if audioConfiguration is supported.
   * Save the IBluetoothAudioPort interface, so that it can be used
   * later to send stream control commands to the HAL client, based on
   * interaction with Audio framework.
   */
  if (audioConfig.getDiscriminator() !=
      AudioConfiguration::hidl_discriminator::leAudioConfig) {
    LOG(WARNING) << __func__
                 << " - Invalid Audio Configuration=" << toString(audioConfig);
    _hidl_cb(BluetoothAudioStatus::UNSUPPORTED_CODEC_CONFIGURATION,
             DataMQ::Descriptor());
    return Void();
  }

  if (!android::bluetooth::audio::IsOffloadLeAudioConfigurationValid(
          session_type_, audioConfig.leAudioConfig())) {
    LOG(WARNING) << __func__ << " - Unsupported LC3 Offloaded Configuration="
                 << toString(audioConfig.leAudioConfig());
    _hidl_cb(BluetoothAudioStatus::UNSUPPORTED_CODEC_CONFIGURATION,
             DataMQ::Descriptor());
    return Void();
  }

  return BluetoothAudioProvider::startSession_2_2(hostIf, audioConfig,
                                                  _hidl_cb);
}

Return<void> LeAudioOffloadAudioProvider::onSessionReady(
    startSession_cb _hidl_cb) {
  BluetoothAudioSessionReport_2_2::OnSessionStarted(session_type_, stack_iface_,
                                                    nullptr, audio_config_);
  _hidl_cb(BluetoothAudioStatus::SUCCESS, DataMQ::Descriptor());
  return Void();
}

}  // namespace implementation
}  // namespace V2_2
}  // namespace audio
}  // namespace bluetooth
}  // namespace hardware
}  // namespace android