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

#include <cstdio>
#include <cstring>
#include <limits.h>
#include <map>
#include <vector>
#include "batteryd_api.h"

namespace OHOS {
namespace HDI {
namespace Battery {
namespace V1_0 {
const int MAX_SYSFS_SIZE = 64;

class PowerSupplyProvider {
public:
    // Keep it same as the BatteryHealthState in battery_info.h
    enum BatteryHealthState {
        BATTERY_HEALTH_UNKNOWN = 0,
        BATTERY_HEALTH_GOOD,
        BATTERY_HEALTH_OVERHEAT,
        BATTERY_HEALTH_OVERVOLTAGE,
        BATTERY_HEALTH_COLD,
        BATTERY_HEALTH_DEAD,
        BATTERY_HEALTH_RESERVED,
    };

    // Keep it same as the BatteryChargeState in battery_info.h
    enum BatteryChargeState {
        CHARGE_STATE_NONE = 0,
        CHARGE_STATE_ENABLE,
        CHARGE_STATE_DISABLE,
        CHARGE_STATE_FULL,
        CHARGE_STATE_RESERVED,
    };

    // Keep it same as the BatteryPluggedType in battery_info.h
    enum BatteryPluggedType {
        PLUGGED_TYPE_NONE = 0,
        PLUGGED_TYPE_AC,
        PLUGGED_TYPE_USB,
        PLUGGED_TYPE_WIRELESS,
        PLUGGED_TYPE_BUTT
    };

    PowerSupplyProvider();
    virtual ~PowerSupplyProvider() {}

    int32_t InitPowerSupplySysfs(void);
    void InitDefaultSysfs(void);
    int32_t ParseCapacity(int32_t* capacity) const;
    int32_t ParseTotalEnergy(int32_t* capacity) const;
    int32_t ParseCurrentAverage(int32_t* curAverage) const;
    int32_t ParseCurrentNow(int32_t* curNow) const;
    int32_t ParseRemainEnergy(int32_t* remainEnergy) const;
    int32_t ParseVoltage(int32_t* voltage) const;
    int32_t ParseTemperature(int32_t* temperature) const;
    int32_t ParseHealthState(int32_t* healthState) const;
    int32_t ParsePluggedType(int32_t* pluggedType) const;
    int32_t ParseChargeState(int32_t* chargeState) const;
    int32_t ParseChargeCounter(int32_t* chargeCounter) const;
    int32_t ParsePresent(int8_t* present) const;
    int32_t ParseTechnology(std::string& technology) const;
    BatterydInfo GetBatteryInfo() const;
    void ParseUeventToBatterydInfo(const char* msg, struct BatterydInfo* info) const;
    void UpdateInfoByReadSysFile(struct BatterydInfo* info) const;
    void SetSysFilePath(const std::string& path);
    void InitBatteryPath();

private:
    struct PowerSupplySysfsInfo {
        char* name;
        std::string typePath;
        std::string onlinePath;
        std::string currentMaxPath;
        std::string voltageMaxPath;
    } powerSupplySysfsInfos_; // [MAX_SYSFS_SIZE];

    struct BatterySysfsInfo {
        char* name;
        std::string capacityPath;
        std::string voltagePath;
        std::string temperaturePath;
        std::string healthStatePath;
        std::string chargeStatePath;
        std::string presentPath;
        std::string technologyPath;
        std::string chargeCounterPath;
        std::string totalEnergyPath;
        std::string curAveragePath;
        std::string curNowPath;
        std::string remainEnergyPath;
    } batterySysfsInfo_;

    static inline int32_t ParseInt(const char* str);
    inline void Trim(char* str) const;
    static inline void CapacityAssigner(const char* str, struct BatterydInfo* info);
    static inline void TotalEnergyAssigner(const char* str, struct BatterydInfo* info);
    static inline void CurrentAverageAssigner(const char* str, struct BatterydInfo* info);
    static inline void CurrentNowAssigner(const char* str, struct BatterydInfo* info);
    static inline void RemainEnergyAssigner(const char* str, struct BatterydInfo* info);
    static inline void VoltageAssigner(const char* str, struct BatterydInfo* info);
    static inline void TemperatureAssigner(const char* str, struct BatterydInfo* info);
    static int32_t HealthStateEnumConverter(const char* str);
    static inline void HealthStateAssigner(const char* str, struct BatterydInfo* info);
    static int32_t ChargeStateEnumConverter(const char* str);
    static inline void ChargeStateAssigner(const char* str, struct BatterydInfo* info);
    static inline void PresentAssigner(const char* str, struct BatterydInfo* info);
    static inline void TechnologyAssigner(const char* str, struct BatterydInfo* info);
    static inline void ChargeCounterAssigner(const char* str, struct BatterydInfo* info);

    void TraversalNode();
    void CheckSubfolderNode(const std::string& path);
    void FormatPath(std::string& path, size_t size, const char* format, const char* basePath, const char* name) const;
    void FormatSysfsPaths(struct PowerSupplySysfsInfo* info);
    int32_t ReadSysfsFile(const char* path, char* buf, size_t size) const;
    int32_t ReadBatterySysfsToBuff(const char* path, char* buf, size_t size) const;
    void GetPluggedTypeName(char* buf, size_t size) const;
    int32_t PluggedTypeEnumConverter(const char* str) const;
    int32_t ParsePluggedMaxCurrent(int32_t* maxCurrent) const;
    int32_t ParsePluggedMaxVoltage(int32_t* maxVoltage) const;
    void CopyBatteryInfo(const struct BatterydInfo* info) const;
    std::string CreateFile(std::string path, std::string content);
    std::vector<std::string> filenodeName_;
    std::map<std::string, std::string> nodeInfo_;
    std::string path_;
    int32_t index_;
};
}  // namespace V1_0
}  // namespace Battery
}  // namespace HDI
}  // namespace OHOS

#endif // POWER_SUPPLY_PROVIDER_H