#!/bin/bash

if [[ $EUID -ne 0 ]]; then
  echo "This script must be run as root"
  exit 1
fi
# Check dependencies
if type curl &>/dev/null; then
  echo "" &>/dev/null
else
  echo "You need to install 'curl' to use the chatgpt script."
  exit
fi
if type jq &>/dev/null; then
  echo "" &>/dev/null
else
  echo "You need to install 'jq' to use the chatgpt script."
  exit
fi

# Installing imgcat if using iTerm
if [[ "$TERM_PROGRAM" == "iTerm.app" ]]; then
  if [[ ! $(which imgcat) ]]; then
    curl -sS https://iterm2.com/utilities/imgcat -o /usr/local/bin/imgcat
    chmod +x /usr/local/bin/imgcat
    echo "Installed imgcat"
  fi
fi

# Installing magick if using kitty
if [[ "$TERM" == "xterm-kitty" ]]; then
  if [[ ! $(which magick) ]]; then
    curl -sS https://imagemagick.org/archive/binaries/magick -o /usr/local/bin/magick
    chmod +x /usr/local/bin/magick
    echo "Installed magick"
  fi
fi

# Installing chatgpt script
curl -sS https://raw.githubusercontent.com/0xacx/chatGPT-shell-cli/main/chatgpt.sh -o /usr/local/bin/chatgpt

# Replace open image command with xdg-open for linux systems
if [[ "$OSTYPE" == "linux"* ]] || [[ "$OSTYPE" == "freebsd"* ]]; then
  sed -i 's/open "\${image_url}"/xdg-open "\${image_url}"/g' '/usr/local/bin/chatgpt'
fi
chmod +x /usr/local/bin/chatgpt
echo "Installed chatgpt script to /usr/local/bin/chatgpt"

echo "The script will add the OPENAI_KEY environment variable to your shell profile and add /usr/local/bin to your PATH"
echo "Would you like to continue? (Yes/No)"
read -e answer
if [ "$answer" == "Yes" ] || [ "$answer" == "yes" ] || [ "$answer" == "y" ] || [ "$answer" == "Y" ] || [ "$answer" == "ok" ]; then

  read -p "Please enter your OpenAI API key: " key

  # Adding OpenAI key to shell profile
  # zsh profile
  if [ -f ~/.zprofile ]; then
    echo "export OPENAI_KEY=$key" >>~/.zprofile
    if [[ ":$PATH:" != *":/usr/local/bin:"* ]]; then
      echo 'export PATH=$PATH:/usr/local/bin' >>~/.zprofile
    fi
    echo "OpenAI key and chatgpt path added to ~/.zprofile"
    source ~/.zprofile
  # zshrc profile for debian
  elif [ -f ~/.zshrc ]; then
    echo "export OPENAI_KEY=$key" >>~/.zshrc
    if [[ ":$PATH:" == *":/usr/local/bin:"* ]]; then
      echo 'export PATH=$PATH:/usr/local/bin' >>~/.zshrc
    fi
    echo "OpenAI key and chatgpt path added to ~/.zshrc"
    source ~/.zshrc
  # bash profile mac
  elif [ -f ~/.bash_profile ]; then
    echo "export OPENAI_KEY=$key" >>~/.bash_profile
    if [[ ":$PATH:" != *":/usr/local/bin:"* ]]; then
      echo 'export PATH=$PATH:/usr/local/bin' >>~/.bash_profile
    fi
    echo "OpenAI key and chatgpt path added to ~/.bash_profile"
    source ~/.bash_profile
  # profile ubuntu
  elif [ -f ~/.profile ]; then
    echo "export OPENAI_KEY=$key" >>~/.profile
    if [[ ":$PATH:" != *":/usr/local/bin:"* ]]; then
      echo 'export PATH=$PATH:/usr/local/bin' >>~/.profile
    fi
    echo "OpenAI key and chatgpt path added to ~/.profile"
    source ~/.profile
  else
    export OPENAI_KEY=$key
    echo "You need to add this to your shell profile: export OPENAI_KEY=$key"
  fi
  echo "Installation complete"

else
  echo "Please take a look at the instructions to install manually: https://github.com/0xacx/chatGPT-shell-cli/tree/main#manual-installation "
  exit
fi