mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-09-20 05:27:31 -07:00
84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
|
|
"github.com/distatus/battery"
|
|
)
|
|
|
|
type batt struct {
|
|
props *properties
|
|
env environmentInfo
|
|
percentageText string
|
|
}
|
|
|
|
const (
|
|
// BatteryIcon to display in front of the battery
|
|
BatteryIcon Property = "battery_icon"
|
|
// DisplayError to display when an error occurs or not
|
|
DisplayError Property = "display_error"
|
|
// ChargingIcon to display when charging
|
|
ChargingIcon Property = "charging_icon"
|
|
// DischargingIcon o display when discharging
|
|
DischargingIcon Property = "discharging_icon"
|
|
// ChargedIcon to display when fully charged
|
|
ChargedIcon Property = "charged_icon"
|
|
// ChargedColor to display when fully charged
|
|
ChargedColor Property = "charged_color"
|
|
// ChargingColor to display when charging
|
|
ChargingColor Property = "charging_color"
|
|
// DischargingColor to display when discharging
|
|
DischargingColor Property = "discharging_color"
|
|
)
|
|
|
|
func (b *batt) enabled() bool {
|
|
bt, err := b.env.getBatteryInfo()
|
|
displayError := b.props.getBool(DisplayError, false)
|
|
if err != nil && !displayError {
|
|
b.percentageText = "100%"
|
|
return true
|
|
}
|
|
if err != nil {
|
|
b.percentageText = "BATT ERR"
|
|
return true
|
|
}
|
|
batteryPercentage := bt.Current / bt.Full * 100
|
|
batteryPercentage = math.Min(100, batteryPercentage)
|
|
percentageText := fmt.Sprintf("%.0f", batteryPercentage)
|
|
var icon string
|
|
var colorPorperty Property
|
|
switch bt.State {
|
|
case battery.Discharging:
|
|
colorPorperty = DischargingColor
|
|
icon = b.props.getString(DischargingIcon, "")
|
|
case battery.Charging:
|
|
colorPorperty = ChargingColor
|
|
icon = b.props.getString(ChargingIcon, "")
|
|
case battery.Full:
|
|
colorPorperty = ChargedColor
|
|
icon = b.props.getString(ChargedIcon, "")
|
|
case battery.Empty, battery.Unknown:
|
|
b.percentageText = percentageText
|
|
return true
|
|
}
|
|
colorBackground := b.props.getBool(ColorBackground, false)
|
|
if colorBackground {
|
|
b.props.background = b.props.getColor(colorPorperty, b.props.background)
|
|
} else {
|
|
b.props.foreground = b.props.getColor(colorPorperty, b.props.foreground)
|
|
}
|
|
batteryIcon := b.props.getString(BatteryIcon, "")
|
|
b.percentageText = fmt.Sprintf("%s%s%s", icon, batteryIcon, percentageText)
|
|
return true
|
|
}
|
|
|
|
func (b *batt) string() string {
|
|
return b.percentageText
|
|
}
|
|
|
|
func (b *batt) init(props *properties, env environmentInfo) {
|
|
b.props = props
|
|
b.env = env
|
|
}
|