setup.sh

#!/usr/bin/env bash

mkdir -p ~/.local/bin

# Create the rickroller script in the ~/.local/bin directory
cat <<EOF >~/.local/bin/rick
#!/usr/bin/env bash

roll_count=0
path=\$(realpath "\$0")

# Reset lyrics index to 0 if roll_count is 0
if [ "\$1" = '-r' ]; then
    sed -i "s/^roll_count=[0-9]\+\$/roll_count=0/" "\$path"
    exit
fi

lyrics=("Never gonna give you up"
    "Never gonna let you down"
    "Never gonna run around and desert you"
    "Never gonna make you cry"
    "Never gonna say goodbye"
    "Never gonna tell a lie and hurt you"
    "Never gonna give you up"
    "Never gonna let you down"
    "Never gonna run around and desert you"
    "Never gonna make you cry"
    "Never gonna say goodbye"
    "Never gonna tell a lie and hurt you"
    "(Ooh, give you up)"
    "(Ooh, give you up)"
    "Never gonna give, never gonna give"
    "(Give you up)"
    "Never gonna give, never gonna give"
    "(Give you up)"
    "We've known each other for so long"
    "Your heart's been aching, but"
    "You're too shy to say it"
    "Inside, we both know what's been going on"
    "We know the game and we're gonna play it"
    "I just wanna tell you how I'm feeling"
    "Gotta make you understand"
    "Never gonna give you up"
    "Never gonna let you down"
    "Never gonna run around and desert you"
    "Never gonna make you cry"
    "Never gonna say goodbye"
    "Never gonna tell a lie and hurt you"
    "Never gonna give you up"
    "Never gonna let you down"
    "Never gonna run around and desert you"
    "Never gonna make you cry"
    "Never gonna say goodbye"
    "Never gonna tell a lie and hurt you"
    "Never gonna give you up"
    "Never gonna let you down"
    "Never gonna run around and desert you"
    "Never gonna make you cry"
    "Never gonna say goodbye"
    "Never gonna tell a lie and hurt you")

max="\${#lyrics[@]}"

if [ -t 1 ]; then
    echo "\${lyrics[\$roll_count]}"
else
    echo -n "\${lyrics[\$roll_count]}"
fi

if [[ -n "\${lyrics[\$((\$roll_count + 1))]}" ]]; then
    sed -i "s/^roll_count=[0-9]\+\$/roll_count=\$((\$roll_count + 1))/" "\$path"
else
    sed -i "s/^roll_count=[0-9]\+\$/roll_count=0/" "\$path"
fi
EOF

# # Determin the shell type (bash, zsh, etc) and thus the right RC file
# shell_rc_file=~/.$(basename "$SHELL")rc
shell_rc_file=~/.bashrc

echo -e "\n\n# Inserted by rickroll-lyrics-PS1. Don't update" >>$shell_rc_file

if [[ ! "$PATH" =~ "$HOME/.local/bin" ]]; then
    export PATH="$HOME/.local/bin:$PATH"
    echo 'export PATH="$HOME/.local/bin:$PATH"' >>$shell_rc_file
fi

cat <<EOF >~/.local/bin/roll
#!/usr/bin/env bash

argmnt() {
    if [[ \$1 =~ ^-\$2 ]]; then
        return 0
    else
        return 1
    fi
}

if argmnt "\$1" r; then
    # Reset PS1 content from ~/.ps1.bk
    if [ -f ~/.ps1.bk ]; then
        export PS1="\$(cat ~/.ps1.bk)"
    fi

    # Reset PROMPT_COMMAND from ~/.pcmd.bk
    if [ -f ~/.pcmd.bk ]; then
        export PROMPT_COMMAND="\$(cat ~/.pcmd.bk)"
    fi

    # Remove rick script plus PROMPT_COMMAND and PS1 backups
    rm -f ~/.pcmd.bk ~/.ps1.bk

    # Remove code added by the script in the shell_rc_file
    sed -i '/^# Inserted by rickroll-lyrics-PS1/,/^# End of rickroll-lyrics-PS1/d' "$shell_rc_file"

    echo "Done. You might need to open a new terminal session"
else
    # Backup current PS1
    if [ ! -f ~/.ps1.bk ]; then
        echo "\$PS1" > ~/.ps1.bk
    fi

    # Backup current PROMPT_COMMAND
    if [ -n "\$PROMPT_COMMAND" ]; then
        echo "\$PROMPT_COMMAND" > ~/.pcmd.bk
    fi

    # Reset lyrics index whenever 'roll' is called
    ~/.local/bin/rick -r

    export PROMPT_COMMAND='PS1="\[\033[1;32m\]\$(~/.local/bin/rick)\[\033[0m\] \[\033[1;34m\]\w\[\033[0m\]$ "'
fi
EOF

chmod +x ~/.local/bin/rick
chmod +x ~/.local/bin/roll

echo "# Do not remove or comment the line below, instead run 'source roll -r' to reset to default" >>$shell_rc_file
echo "source roll" >>$shell_rc_file
echo "# End of rickroll-lyrics-PS1" >>$shell_rc_file

# if [[ ! "$PATH" =~ "$HOME/.local/bin" ]]; then
#     export PATH="$HOME/.local/bin:$PATH"
# fi

# Output sucess message if script is not sourced. Results will be visible if sourced
if [[ "$(basename -- "$0")" == "setup.sh" ]]; then
    echo -e "\e[32mYou can now open a new terminal session and type 'roll'\e[0m"
    exit
fi

source roll