/*
 *    Copyright (c) 2021 Project CHIP Authors
 *
 *    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 <lib/core/CHIPConfig.h>
#include <messaging/ReliableMessageProtocolConfig.h>
#include <transport/SessionHolder.h>
#include <transport/raw/PeerAddress.h>

namespace chip {
namespace Transport {

class SecureSession;
class UnauthenticatedSession;
class GroupSession;

class Session
{
public:
    virtual ~Session() {}

    enum class SessionType : uint8_t
    {
        kUndefined       = 0,
        kUnauthenticated = 1,
        kSecure          = 2,
        kGroup           = 3,
    };

    virtual SessionType GetSessionType() const = 0;
#if CHIP_PROGRESS_LOGGING
    virtual const char * GetSessionTypeString() const = 0;
#endif

    void AddHolder(SessionHolder & holder)
    {
        VerifyOrDie(!holder.IsInList());
        mHolders.PushBack(&holder);
    }

    void RemoveHolder(SessionHolder & holder)
    {
        VerifyOrDie(mHolders.Contains(&holder));
        mHolders.Remove(&holder);
    }

    // For types of sessions using reference counter, override these functions, otherwise leave it empty.
    virtual void Retain() {}
    virtual void Release() {}

    virtual Access::SubjectDescriptor GetSubjectDescriptor() const     = 0;
    virtual bool RequireMRP() const                                    = 0;
    virtual const ReliableMessageProtocolConfig & GetMRPConfig() const = 0;
    virtual System::Clock::Milliseconds32 GetAckTimeout() const        = 0;

    SecureSession * AsSecureSession();
    UnauthenticatedSession * AsUnauthenticatedSession();
    GroupSession * AsGroupSession();

    bool IsGroupSession() const { return GetSessionType() == SessionType::kGroup; }

protected:
    // This should be called by sub-classes at the very beginning of the destructor, before any data field is disposed, such that
    // the session is still functional during the callback.
    void NotifySessionReleased()
    {
        SessionHandle session(*this);
        while (!mHolders.Empty())
        {
            mHolders.begin()->OnSessionReleased(); // OnSessionReleased must remove the item from the linked list
        }
    }

private:
    IntrusiveList<SessionHolder> mHolders;
};

} // namespace Transport
} // namespace chip