import batteryInfo, { BatteryChargeState } from '@ohos.batteryInfo'

import { MethodHandler } from '../utils/handler'

import type Taro from '@tarojs/taro/types'

// 电量

export const getBatteryInfoSync: typeof Taro.getBatteryInfoSync = () => ({
  // @ts-ignore
  isCharging: [BatteryChargeState.ENABLE, BatteryChargeState.FULL].includes(batteryInfo.chargingStatus),
  level: batteryInfo.batterySOC
})

export const getBatteryInfo: typeof Taro.getBatteryInfo = async ({ success, fail, complete } = {}) => {
  const handle = new MethodHandler<Taro.getBatteryInfo.SuccessCallbackResult>({ name: 'getBatteryInfo', success, fail, complete })
  try {
    return handle.success(getBatteryInfoSync())
  } catch (error) {
    return handle.fail({
      errMsg: error?.message || error
    })
  }
}