/*
 *
 *    Copyright (c) 2020 Project CHIP Authors
 *    All rights reserved.
 *
 *    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.
 */

/**
 *    @file
 *      This file defines the CHIP Connection object that maintains a BLE connection.
 *
 *    TODO
 *    In the current state, Transport::BLE relies on beeing a Friend class of
 *    ChipDeviceController in order to use its private members to fire
 *    Ble related callbacks.
 *
 */

#ifndef __TRANSPORT_BLE_H__
#define __TRANSPORT_BLE_H__

#include <ble/BleConfig.h>

#if CONFIG_NETWORK_LAYER_BLE

#include <utility>

#include <ble/BleError.h>
#include <ble/BleLayer.h>
#include <core/CHIPCore.h>
#include <transport/Base.h>

namespace chip {
namespace DeviceController {
class ChipDeviceController;
} // namespace DeviceController
} // namespace chip

namespace chip {
namespace Transport {

/** Defines connections parameters for setting up a BLE transport */
class BleConnectionParameters
{
public:
    explicit BleConnectionParameters(DeviceController::ChipDeviceController * dc, Ble::BleLayer * layer) :
        mDeviceController(dc), mLayer(layer)
    {}

    BleConnectionParameters(const BleConnectionParameters &) = default;
    BleConnectionParameters(BleConnectionParameters &&)      = default;

    DeviceController::ChipDeviceController * GetDeviceController() { return mDeviceController; }

    Ble::BleLayer * GetBleLayer() { return mLayer; }

    bool HasConnectionObject() const { return mConnectionObj; };
    BLE_CONNECTION_OBJECT GetConnectionObject() const { return mConnectionObj; };
    BleConnectionParameters & SetConnectionObject(BLE_CONNECTION_OBJECT connObj)
    {
        mConnectionObj = connObj;

        return *this;
    }

    bool HasConnectionDiscriminator() const { return mConnectionDiscriminator != 0; };
    uint16_t GetConnectionDiscriminator() const { return mConnectionDiscriminator; };
    BleConnectionParameters & SetConnectionDiscriminator(const uint16_t connDiscriminator)
    {
        mConnectionDiscriminator = connDiscriminator;

        return *this;
    }

private:
    DeviceController::ChipDeviceController * mDeviceController = nullptr; ///< Associated device controller
    Ble::BleLayer * mLayer                                     = nullptr; ///< Associated ble layer
    BLE_CONNECTION_OBJECT mConnectionObj                       = 0;       ///< the target peripheral BLE_CONNECTION_OBJECT
    uint16_t mConnectionDiscriminator                          = 0;       ///< the target peripheral discriminator
};

/** Implements a transport using BLE. */
class DLL_EXPORT BLE : public Base
{
    /**
     *  The State of the BLE connection
     *
     */
    enum class State
    {
        kNotReady    = 0, /**< State before initialization. */
        kInitialized = 1, /**< State after class is connected and ready. */
    };

public:
    virtual ~BLE();

    /**
     * Initialize a BLE transport to a given peripheral or a given device name.
     *
     * @param params        BLE configuration parameters for this transport
     */
    CHIP_ERROR Init(BleConnectionParameters & params);

    CHIP_ERROR SendMessage(const MessageHeader & header, const Transport::PeerAddress & address,
                           System::PacketBuffer * msgBuf) override;

    bool CanSendToPeer(const Transport::PeerAddress & address) override
    {
        return (mState == State::kInitialized) && (address.GetTransportType() == Type::kBle);
    }

private:
    CHIP_ERROR InitInternal(Ble::BleLayer * bleLayer, BLE_CONNECTION_OBJECT connObj);
    CHIP_ERROR DelegateConnection(Ble::BleLayer * bleLayer, const uint16_t connDiscriminator);

    // Those functions are BLEConnectionDelegate callbacks used when the connection
    // parameters used a name instead of a BLE_CONNECTION_OBJECT.
    static void OnBleConnectionComplete(void * appState, BLE_CONNECTION_OBJECT connObj);
    static void OnBleConnectionError(void * appState, BLE_ERROR err);

    // Those functions are BLEEndPoint callbacks
    static void OnBleEndPointReceive(BLEEndPoint * endPoint, PacketBuffer * buffer);
    static void OnBleEndPointConnectionComplete(BLEEndPoint * endPoint, BLE_ERROR err);
    static void OnBleEndPointConnectionClosed(BLEEndPoint * endPoint, BLE_ERROR err);

    DeviceController::ChipDeviceController * mDeviceController = nullptr;          ///< Associated device controller
    Ble::BleLayer * mBleLayer                                  = nullptr;          ///< Associated ble layer
    BLEEndPoint * mBleEndPoint                                 = nullptr;          ///< BLE endpoint used by the transport
    State mState                                               = State::kNotReady; ///< State of the BLE transport
};

} // namespace Transport
} // namespace chip

#endif // CONFIG_NETWORK_LAYER_BLE

#endif // __TRANSPORT_BLE_H__