#!/usr/bin/env bash
set -e

BASHRC="$HOME/.bashrc"
PROMPT_STR="PS1='\${debian_chroot:+(\$debian_chroot)}\\[\\033[01;31m\\]\\u\\[\\033[01;33m\\]@\\[\\033[01;36m\\]\\h \\[\\033[01;33m\\]\\w \\[\\033[01;35m\\]\\$ \\[\\033[00m\\]'"

if [ ! -f "$BASHRC" ]; then
    touch "$BASHRC"
fi

# Check if the custom prompt is already configured
if grep -q "01;31m" "$BASHRC" && grep -q "01;35m" "$BASHRC"; then
    echo "==> Custom PS1 prompt is already configured in $BASHRC"
else
    echo "==> Setting custom PS1 prompt in $BASHRC..."
    echo "" >> "$BASHRC"
    echo "# Custom colorful PS1 prompt" >> "$BASHRC"
    echo "$PROMPT_STR" >> "$BASHRC"
    echo "Done!"
fi

echo "==> Run 'source ~/.bashrc' or restart your terminal to apply the new prompt."