/*
 * Copyright (c) 2021-2024 Huawei Device Co., Ltd.
 * 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.
 */

#ifndef SESSION_SERVICE_IMPL_H
#define SESSION_SERVICE_IMPL_H

#include "Session.h"

#include <map>
#include <memory>
#include <string>
#include "ISessionListener.h"
#include "ISessionService.h"

namespace Communication {
namespace SoftBus {
class SessionServiceImpl : public ISessionService, public std::enable_shared_from_this<SessionServiceImpl> {
public:
    SessionServiceImpl() = default;

    ~SessionServiceImpl() override = default;

    int32_t CreateSessionServer(const std::string &pkgName, const std::string &sessionName,
        std::shared_ptr<ISessionListener> listener) override;

    int32_t RemoveSessionServer(const std::string &pkgName, const std::string &sessionName) override;

    std::shared_ptr<Session> OpenSession(const std::string &mySessionName, const std::string &peerSessionName,
        const std::string &peerNetworkId, const std::string &groupId, int32_t flags) override;

    int32_t CloseSession(std::shared_ptr<Session> session) override;

    int32_t GrantPermission(int32_t uid, int32_t pid, const std::string &busName) override;

    int32_t RemovePermission(const std::string &busName) override;

    int32_t OpenSessionCallback(int32_t sessionId);

    void CloseSessionCallback(int32_t sessionId);

    void BytesReceivedCallback(int32_t sessionId, const void *data, uint32_t len);

    void MessageReceivedCallback(int32_t sessionId, const void *data, uint32_t len);

private:
    static std::mutex listenerMutex_;
    static std::map<std::string, std::shared_ptr<ISessionListener>> listenerMap_;

    static std::mutex sessionMutex_;
    static std::map<int32_t, std::shared_ptr<Session>> sessionMap_;

    int32_t GetSessionListener(int32_t sessionId, std::shared_ptr<ISessionListener> &listener,
        std::shared_ptr<Session> &session);

    int32_t GetSessionListenerOnSessionOpened(int32_t sessionId,
        std::shared_ptr<ISessionListener> &listener, std::shared_ptr<Session> &session);

    int32_t CreateSession(int32_t sessionId, const std::shared_ptr<Session> &session);
};
} // namespace SoftBus
} // namespace Communication

#endif // SESSION_SERVICE_IMPL_H