#!/usr/bin/env bash

export LC_ALL=en_US.UTF-8

current_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$current_dir"/utils.sh

get_percent() {
    case $(uname -s) in
    Linux)
        total_mem=$(free -g | awk '/^Mem/ {print $2}')
        used_mem=$(free -g | awk '/^Mem/ {print $3}')
        memory_percent=$(awk "BEGIN {printf \"%.2f\", ($used_mem / $total_mem * 100)}")
        echo "$memory_percent%"
        ;;
    Darwin)
        used_mem=$(vm_stat | grep ' active\|wired ' | sed 's/[^0-9]//g' | paste -sd ' ' - | \
            awk -v pagesize="$(pagesize)" '{printf "%d\n", ($1+$2) * pagesize / 1048576}')
        total_mem=$(system_profiler SPHardwareDataType | grep "Memory:" | awk '{print $2}')
        memory_percent=$(awk "BEGIN {printf \"%.2f\", ($used_mem / $total_mem / 10)}")
        echo "$memory_percent%"
        ;;
    FreeBSD)
        hw_pagesize="$(sysctl -n hw.pagesize)"
        mem_inactive="$(($(sysctl -n vm.stats.vm.v_inactive_count) * hw_pagesize))"
        mem_unused="$(($(sysctl -n vm.stats.vm.v_free_count) * hw_pagesize))"
        mem_cache="$(($(sysctl -n vm.stats.vm.v_cache_count) * hw_pagesize))"

        free_mem=$(((mem_inactive + mem_unused + mem_cache) / 1024 / 1024))
        total_mem=$(($(sysctl -n hw.physmem) / 1024 / 1024))
        used_mem=$((total_mem - free_mem))
        memory_percent=$(awk "BEGIN {printf \"%.2f\", ($used_mem / $total_mem * 100)}")
        echo "$memory_percent%"
        ;;
    CYGWIN* | MINGW32* | MSYS* | MINGW*) ;; # TODO - windows compatibility
    esac
}

main() {
    RATE=$(get_tmux_option "@tmux2k-refresh-rate" 5)
    ram_label=$(get_tmux_option "@tmux2k-ram-usage-label" "")
    ram_percent=$(get_percent)
    echo "$ram_label $ram_percent"
    sleep "$RATE"
}

main