ServerManagementUtils/custom-metrics/disk-usage.sh

#!/bin/bash

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
. "${SCRIPT_DIR}/../share/logging.sh"
. "${SCRIPT_DIR}/../share/os-detection.sh"

LOGFILE=/var/log/disk_usage.log
OS_TYPE=$(DetectOS)

# Create log file with proper error handling
if ! touch "$LOGFILE" 2>/dev/null; then
    LogError "Cannot create log file $LOGFILE"
    exit 1
fi
chmod 644 "$LOGFILE" 2>/dev/null

LogInformation "Starting disk usage monitoring (OS: $OS_TYPE)"

for i in {1..6}; do
    timestamp=$(GetTimestamp)
    
    # Use process substitution instead of pipe to avoid subshell issues
    if [[ "$OS_TYPE" == "macos" ]]; then
        # macOS df format: Filesystem 512-blocks Used Available Capacity iused ifree %iused  Mounted on
        # Skip pseudo-filesystems (map auto_home, map auto_nfs) and devfs
        df 2>/dev/null | tail -n +2 | while IFS= read -r line; do
            # Get the first field to check filesystem type
            filesystem=$(echo "$line" | awk '{print $1}')
            
            # Skip map entries and other pseudo-filesystems
            if [[ "$filesystem" == "map" || "$filesystem" == "devfs" ]]; then
                continue
            fi
            
            # For valid filesystems, extract: blocks (col 2), used (col 3), mount point (col 9+)
            blocks=$(echo "$line" | awk '{print $2}')
            used=$(echo "$line" | awk '{print $3}')
            mount=$(echo "$line" | awk '{for(i=9;i<=NF;i++) printf "%s%s", $i, (i<NF?" ":""); print ""}')
            
            # Skip entries with 0 blocks
            if [[ -n "$mount" && -n "$used" && -n "$blocks" && "$blocks" != "0" ]]; then
                used_bytes=$((used * 512))
                total_bytes=$((blocks * 512))
                echo "$timestamp $mount $used_bytes $total_bytes" >> "$LOGFILE"
            fi
        done
    else
        # Linux df with --output flag
        while read -r mount used total; do
            if [[ -n "$mount" && -n "$used" && -n "$total" ]]; then
                echo "$timestamp $mount $used $total" >> "$LOGFILE"
            fi
        done < <(df --output=target,used,size -B1 2>/dev/null | tail -n +2)
    fi
    
    LogInformation "Disk usage data collected (iteration $i/6)"
    
    if [ $i -lt 6 ]; then
        sleep 10
    fi
done

LogInformation "Disk usage monitoring completed"