/*
 * Copyright (c) 2023 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 POWERMGR_RUNNING_LOCK_PROXY_H
#define POWERMGR_RUNNING_LOCK_PROXY_H

#include <map>
#include <vector>

#include <iremote_object.h>
#include <functional>

namespace OHOS {
namespace PowerMgr {
enum class RunningLockEvent : uint32_t {
    RUNNINGLOCK_UPDATE = 0,
    RUNNINGLOCK_PROXY,
    RUNNINGLOCK_UNPROXY
};
class RunningLockProxy {
public:
    using WksMap = std::map<int32_t, std::pair<std::string, bool>>;
    using TokenWorkSourceMap = std::map<const sptr<IRemoteObject>,
        std::pair<WksMap, int32_t>>;
    using RunningLockProxyMap = std::map<std::string, TokenWorkSourceMap>;
    RunningLockProxy() = default;
    ~RunningLockProxy() = default;

    void AddRunningLock(pid_t pid, pid_t uid, const sptr<IRemoteObject>& remoteObj);
    void RemoveRunningLock(pid_t pid, pid_t uid, const sptr<IRemoteObject>& remoteObj);
    bool UpdateWorkSource(pid_t pid, pid_t uid, const sptr<IRemoteObject>& remoteObj,
        const std::map<int32_t, std::string>& workSources);
    bool IncreaseProxyCnt(pid_t pid, pid_t uid);
    bool DecreaseProxyCnt(pid_t pid, pid_t uid);
    void Clear();
    std::string DumpProxyInfo();
    void ResetRunningLocks();
    bool UpdateProxyState(pid_t pid, pid_t uid, const sptr<IRemoteObject>& remoteObj, bool state);
private:
    std::string AssembleProxyKey(pid_t pid, pid_t uid);
    void ProxyInner(const sptr<IRemoteObject>& remoteObj, const std::string& bundleNames, RunningLockEvent event);
    std::string GetRunningLockName(const sptr<IRemoteObject>& remoteObj);
    std::string MergeBundleName(const WksMap& wksMap);
    RunningLockProxyMap proxyMap_;
};
} // namespace PowerMgr
} // namespace OHOS
#endif // POWERMGR_RUNNING_LOCK_PROXY_H