#!/usr/bin/env bash

# Rosé Pine color palette
BASE=#191724
SURFACE=#1f1d2e
OVERLAY=#26233a

MUTED=#6e6a86
SUBTLE=#908caa
TEXT=#e0def4

LOVE=#eb6f92
GOLD=#f6c177
ROSE=#ebbcba
PINE=#31748f
FOAM=#9ccfd8
IRIS=#c4a7e7

HIGHLIGHT_LOW=#21202e
HIGHLIGHT_MED=#403d52
HIGHLIGHT_HIGH=#524f67

# Function to get CPU usage
get_cpu_usage() {
    cpu_usage=$(grep 'cpu ' /proc/stat | awk '{usage=($2+$4)*100/($2+$4+$5)} END {printf "%.0f%%\n", usage}')
    echo "^c$FOAM^CPU: $cpu_usage"
}

# Function to get memory usage
get_memory_usage() {
    # Extract used and total memory in kilobytes
    mem_info=$(free -k | awk '/^Mem:/ {print $3, $2}')
    used_mem=$(echo $mem_info | awk '{print $1}')
    total_mem=$(echo $mem_info | awk '{print $2}')

    # Calculate memory usage percentage
    mem_percent=$(awk "BEGIN {printf \"%.0f%%\", ($used_mem / $total_mem) * 100}")

    # Output the result with color
    echo "^c$IRIS^MEM: $mem_percent"
}

# Function to get volume level
get_volume() {
    # Get the default sink (output device) and its volume/mute status
    default_sink=$(pactl get-default-sink)
    volume=$(pactl get-sink-volume $default_sink | grep -o '[0-9]*%' | head -1)
    is_muted=$(pactl get-sink-mute $default_sink | grep -o 'yes\|no')

    # Replace "yes" with "MUT" and "no" with the volume level
    if [ "$is_muted" = "yes" ]; then
        volume="MUT"
    fi

    # Get the default source (input device) and its volume/mute status
    default_source=$(pactl get-default-source)
    mic_level=$(pactl get-source-volume $default_source | grep -o '[0-9]*%' | head -1)
    is_mic_muted=$(pactl get-source-mute $default_source | grep -o 'yes\|no')

    # Replace "yes" with "MUT" and "no" with the microphone level
    if [ "$is_mic_muted" = "yes" ]; then
        mic_level="MUT"
    fi

    # Output the results with colors
    echo "^c$PINE^VOL: $volume MIC: $mic_level"
}

# Function to get date and time in American 12-hour format
get_datetime() {
    datetime=$(date +"%a %d %b %I:%M %p")
    echo "^c$GOLD^$datetime"
}

# Main loop to update xsetroot
while true; do
    # Combine all status components
    status="$(get_volume) ^c$TEXT^| $(get_cpu_usage) ^c$TEXT^| $(get_memory_usage) ^c$TEXT^| $(get_datetime)"

    # Update the root window name with a darker background
    xsetroot -name "$status"

    # Sleep for 250 milis before updating
    sleep 0.25
done