#!/bin/bash

echo "========================================"
echo " AI VPS Hardware Check"
echo "========================================"

echo
echo "=== SYSTEM ==="
grep -E '^(PRETTY_NAME|VERSION_ID)=' /etc/os-release
echo "Kernel: $(uname -r)"
echo "Architecture: $(uname -m)"
echo "Virtualization: $(systemd-detect-virt 2>/dev/null)"

echo
echo "=== CPU ==="
lscpu | grep -E '^(CPU\(s\)|Model name|Thread\(s\)|Core\(s\)|Socket\(s\)|CPU max MHz|CPU min MHz):'

echo
echo "=== AI-RELEVANT CPU FLAGS ==="
FLAGS=$(grep -m1 '^flags' /proc/cpuinfo)
for flag in sse4_1 sse4_2 avx avx2 fma f16c avx512f avx512bw avx512vl avx512_vnni; do
    if echo "$FLAGS" | grep -qw "$flag"; then
        printf "%-12s YES\n" "$flag"
    else
        printf "%-12s NO\n" "$flag"
    fi
done

echo
echo "=== MEMORY ==="
free -h
grep -E 'MemTotal|MemAvailable|SwapTotal|SwapFree' /proc/meminfo

echo
echo "=== DISK ==="
df -hT /

echo
echo "=== LOAD ==="
uptime

echo
echo "=== CPU LOAD / STEAL ==="
top -bn1 | head -n 5

echo
echo "========================================"
echo " Check complete"
echo "========================================"