#!/usr/bin/env bash

# battery_segment

# Set default symbols if not already defined in config
# Defaults should be standard symbols.
[[ -z ${PL_SYMBOLS[battery_charging]} ]] && PL_SYMBOLS[battery_charging]='■ '
[[ -z ${PL_SYMBOLS[battery_dicharging]} ]] && PL_SYMBOLS[battery_dicharging]='▬ '

# -----------------------------------------------------------------------------
# append to prompt: indicator for battery level
# arg: $1 background color
# arg: $2 foreground color
function battery_segment {
    local bg_color="$1"
    local fg_color="$2"
    local batt_dir
    local content
    local batt_dir="/sys/class/power_supply/BAT"
    if [ -d $batt_dir"0" ]; then
        batt_dir=$batt_dir"0"
    elif [ -d $batt_dir"1" ]; then
        batt_dir=$batt_dir"1"
    else
        return 1
    fi
    local cap
    cap="$(<"$batt_dir/capacity")"
    local status
    status="$(<"$batt_dir/status")"

    if [ "$status" == "Discharging" ]; then
        content="${PL_SYMBOLS[battery_discharging]} "
    else
        content="${PL_SYMBOLS[battery_charging]}"
    fi
    content="$content$cap%"

    PS1+="$(segment_end "$fg_color" "$bg_color")"
    PS1+="$(segment_content "$fg_color" "$bg_color" " $content ")"
    __last_color="$bg_color"
}
