#!/usr/bin/env bash

set -euo pipefail
shopt -s nullglob

THP_ENABLED_PATH="/sys/kernel/mm/transparent_hugepage/enabled"
THP_DEFRAG_PATH="/sys/kernel/mm/transparent_hugepage/defrag"
if [ -f "$THP_ENABLED_PATH" ]; then
  echo "Attempting to set THP enabled mode to 'madvise'..."
  if echo madvise > "$THP_ENABLED_PATH"; then
    echo "Successfully set THP enabled mode to madvise."
  else
    echo "ERROR: Failed to set THP enabled mode." >&2
  fi
else
  echo "WARNING: THP enabled file not found at $THP_ENABLED_PATH. THP may not be compiled into the kernel."
fi

if [ -f "$THP_DEFRAG_PATH" ]; then
  echo "Attempting to set THP defrag mode to 'madvise'..."
  if echo madvise > "$THP_DEFRAG_PATH"; then
    echo "Successfully set THP defrag mode to madvise."
  else
    echo "ERROR: Failed to set THP defrag mode." >&2
  fi
else
  echo "WARNING: THP defrag file not found at $THP_DEFRAG_PATH."
fi

echo "THP configuration script finished."