/*
 * 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_WAKEUP_CONTROLLER_H
#define POWERMGR_WAKEUP_CONTROLLER_H

#include <functional>
#include <memory>
#include <mutex>
#include <vector>

#ifdef HAS_MULTIMODALINPUT_INPUT_PART
#include "i_input_event_consumer.h"
#endif
#include "power_state_machine.h"
#include "system_ability.h"
#ifdef HAS_SENSORS_SENSOR_PART
#include "sensor_agent.h"
#endif
#include "wakeup_sources.h"
#include "wakeup_source_parser.h"

namespace OHOS {
namespace PowerMgr {

using WakeupListener = std::function<void(WakeupDeviceType)>;
using namespace OHOS::MMI;
class WakeupMonitor;
class WakeupController : public std::enable_shared_from_this<WakeupController> {
public:
    explicit WakeupController(std::shared_ptr<PowerStateMachine>& stateMachine);
    ~WakeupController();
    void Init();
    void Cancel();
    void RegisterSettingsObserver();
    void UnregisterSettingsObserver();
    void ExecWakeupMonitorByReason(WakeupDeviceType reason);
    void Wakeup();
    void NotifyDisplayActionDone(uint32_t event);
    void SetOriginSettingValue(WakeupSource& source);
#ifdef POWER_MANAGER_WAKEUP_ACTION
    bool IsLowCapacityWakeup(WakeupDeviceType reason);
    void ProcessLowCapacityWakeup();
#endif
    static int32_t SetWakeupDoubleClickSensor(bool enable);
    static void ChangeWakeupSourceConfig(bool updateEnable);
    static void ChangePickupWakeupSourceConfig(bool updataEnable);
    static void PickupConnectMotionConfig(bool databaseSwitchValue);
    static void ChangeLidWakeupSourceConfig(bool updataEnable);
    std::shared_ptr<PowerStateMachine> GetStateMachine()
    {
        return stateMachine_;
    }
    WakeupDeviceType GetLastReason() const
    {
        return wakeupReason_;
    }
    bool CheckEventReciveTime(WakeupDeviceType wakeupType);

private:
    class SleepGuard final {
    public:
        explicit SleepGuard(const sptr<PowerMgrService>& pms);
        ~SleepGuard();
        SleepGuard(const SleepGuard&) = delete;
        SleepGuard(SleepGuard&&) = delete;
        SleepGuard& operator=(const SleepGuard&) = delete;
        SleepGuard& operator=(SleepGuard&&) = delete;

    private:
        sptr<IRemoteObject> token_;
        sptr<PowerMgrService> pms_;
    };

    void ControlListener(WakeupDeviceType reason);
    std::vector<WakeupSource> sourceList_;
    std::map<WakeupDeviceType, std::shared_ptr<WakeupMonitor>> monitorMap_;
    std::map<WakeupDeviceType, int64_t> eventHandleMap_;
    std::shared_ptr<PowerStateMachine> stateMachine_;
    WakeupDeviceType wakeupReason_ {0};
    std::mutex mutex_;
    std::mutex monitorMutex_;
    std::mutex eventHandleMutex_;
    static std::mutex sourceUpdateMutex_;
    int32_t monitorId_ {-1};
};

#ifdef HAS_MULTIMODALINPUT_INPUT_PART
class InputCallback : public IInputEventConsumer {
public:
    virtual void OnInputEvent(std::shared_ptr<KeyEvent> keyEvent) const;
    virtual void OnInputEvent(std::shared_ptr<PointerEvent> pointerEvent) const;
    virtual void OnInputEvent(std::shared_ptr<AxisEvent> axisEvent) const;
    bool NonWindowEvent(const std::shared_ptr<PointerEvent>& pointerEvent) const;

private:
    bool IsRemoteEvent(std::shared_ptr<InputEvent> event) const;
    bool isKeyboardKeycode(int32_t keyCode) const;
};
#endif

class WakeupMonitor : public std::enable_shared_from_this<WakeupMonitor> {
public:
    static std::shared_ptr<WakeupMonitor> CreateMonitor(WakeupSource& source);
    static constexpr int32_t POWER_KEY_PRESS_DELAY_MS = 10000;

    virtual ~WakeupMonitor() = default;
    virtual bool Init() = 0;
    virtual void Cancel() = 0;
    virtual void HandleEvent(uint32_t eventId)
    {
        // do nothing in base class
    }

    WakeupDeviceType GetReason() const
    {
        return reason_;
    }

    void RegisterListener(WakeupListener listener)
    {
        listener_ = listener;
    }

    void Notify()
    {
        listener_(reason_);
    }

protected:
    explicit WakeupMonitor(WakeupSource& source)
    {
        reason_ = source.GetReason();
    }

    WakeupDeviceType reason_;
    WakeupListener listener_;
};

class PowerkeyWakeupMonitor : public WakeupMonitor {
public:
    explicit PowerkeyWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~PowerkeyWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;

private:
    int32_t powerkeyShortPressId_ {-1};
};

class KeyboardWakeupMonitor : public WakeupMonitor {
public:
    explicit KeyboardWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~KeyboardWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class MousekeyWakeupMonitor : public WakeupMonitor {
public:
    explicit MousekeyWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~MousekeyWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class TouchpadWakeupMonitor : public WakeupMonitor {
public:
    explicit TouchpadWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~TouchpadWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class PenWakeupMonitor : public WakeupMonitor {
public:
    explicit PenWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~PenWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class SingleClickWakeupMonitor : public WakeupMonitor {
public:
    explicit SingleClickWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~SingleClickWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class DoubleClickWakeupMonitor : public WakeupMonitor {
public:
    explicit DoubleClickWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~DoubleClickWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class LidWakeupMonitor : public WakeupMonitor {
public:
    explicit LidWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~LidWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class SwitchWakeupMonitor : public WakeupMonitor {
public:
    explicit SwitchWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~SwitchWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

class PickupWakeupMonitor : public WakeupMonitor {
public:
    explicit PickupWakeupMonitor(WakeupSource& source) : WakeupMonitor(source) {}
    ~PickupWakeupMonitor() override = default;
    bool Init() override;
    void Cancel() override;
};

} // namespace PowerMgr
} // namespace OHOS

#endif