#!/bin/bash

echo " starting cool5tar's checkra1n-installer"

echo "started..."

VERSION=1.0
# Terminal colors
BLACK=$(tput setaf 0)
RED=$(tput setaf 1)
GREEN=$(tput setaf 2)
YELLOW=$(tput setaf 3)
LIME_YELLOW=$(tput setaf 190)
BLUE=$(tput setaf 4)
MAGENTA=$(tput setaf 5)
CYAN=$(tput setaf 6)
WHITE=$(tput setaf 7)
BRIGHT=$(tput bold)
NORMAL=$(tput sgr0)
BLINK=$(tput blink)
REVERSE=$(tput smso)
UNDERLINE=$(tput smul)

# Prints a line with color using terminal codes
Print_Style () {
  printf "%s\n" "${2}$1${NORMAL}"
}

if [ "$BASH_VERSION" = '' ]; then
  whiptail --msgbox "ERROR: this script must be run in bash!" 20 30 --ok-button "Exit"
  exit
fi

if [ "$EUID" -ne 0 ]; then 
  whiptail --msgbox "YOU AREN'T RUNNING AS ROOT! This script needs root, use sudo!" $((LINES/2)) $((COLUMNS*7/10)) --ok-button "Exit"
  Print_Style "ERROR: You need to run this as root, use sudo!" $RED
  exit
fi

LINES=$(tput lines)
COLUMNS=$(tput cols)
LISTHEIGHT=$((LINES/3))

# get system architecture
CPUArch=$(uname -m)

function mainMenu() {
  CHOICE=$(whiptail \
  --title "Checkra1n GUI Installer on $(uname -m)" \
  --menu "Choose an option" $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT)) \
    "Install Repo" "Install the repo. x86_64 ONLY!" \
    "Direct Download" "Use on any architecture." \
    "Install Autostart Service" "Automatically start checkra1n on boot." \
    "Credits" "This tool is open-source!" \
    "Update/Reinstall" "Download latest version" 3>&1 1>&2 2>&3)
  case $CHOICE in
    "Install Repo")
    if [[ "$CPUArch" == *"x86_64"* ]]; then
      Print_Style "x86_64 detected!" $GREEN
    else
      Print_Style "ERROR:You aren't on x86_64! You can't use this! Exiting..." $RED
      exit
    fi
    echo "Adding repo..."
    echo "deb https://assets.checkra.in/debian /" | sudo tee -a /etc/apt/sources.list
    sudo apt-key adv --fetch-keys https://assets.checkra.in/debian/archive.key
    sudo apt update
    echo "Installing..."
    sudo apt install checkra1n
    echo "All done!"
    ;;
    "Direct Download")
    # Downloads checkra1n
    GetJB () {
      wget -O checkra1n "$DL_LINK"
      chmod 755 checkra1n
    }

    # Check system architecture
    Print_Style "System Architecture: $CPUArch" "$YELLOW"

    # Get Linux distribution
    # Copied from Stack Overflow lol
    if [ -f /etc/os-release ]; then
        # freedesktop.org and systemd
        . /etc/os-release
        OS=$NAME
        VER=$VERSION_ID
    elif type lsb_release >/dev/null 2>&1; then
        # linuxbase.org
        OS=$(lsb_release -si)
        VER=$(lsb_release -sr)
    elif [ -f /etc/lsb-release ]; then
        # For some versions of Debian/Ubuntu without lsb_release command
        . /etc/lsb-release
        OS=$DISTRIB_ID
        VER=$DISTRIB_RELEASE
    elif [ -f /etc/debian_version ]; then
        # Older Debian/Ubuntu/etc.
        OS=Debian
        VER=$(cat /etc/debian_version)
    elif [ -f /etc/SuSe-release ]; then
        # Older SuSE/etc.
        ...
    elif [ -f /etc/redhat-release ]; then
        # Older Red Hat, CentOS, etc.
        ...
    else
        # Fall back to uname, e.g. "Linux <version>", also works for BSD, etc.
        OS=$(uname -s)
        VER=$(uname -r)
    fi

    # Determine Linux distro dependencies

    if [[ "$OS" == *"Raspbian"* ]]; then
      DEPENDENCIES="usbmuxd libimobiledevice6"
  
    else
      Print_Style "I do not know what dependencies you need for this distro ($OS). Using defaults for Raspbian..." $RED
      DEPENDENCIES="usbmuxd libimobiledevice6"
    fi

    # Choose correct download link
    # TODO: dynamically fetch latest urls from checkra1n website
    if [[ "$CPUArch" == *"aarch64"* || "$CPUArch" == *"arm64"* ]]; then
      Print_Style "ARM64 detected!" $YELLOW
      DL_LINK=https://assets.checkra.in/downloads/linux/cli/arm64/32cc7d1bd687800d1b5bc5cfc4f4ca15f393dada7570e78742eab068b1f2a3e2/checkra1n
  
    elif [[ "$CPUArch" == *"armhf"* || "$CPUArch" == *"armv"* ]]; then
      Print_Style "ARM detected!" $YELLOW
      DL_LINK=https://assets.checkra.in/downloads/linux/cli/arm/dde0ee4255403a427636bb76e09e409487f8be128af4b7d89fac78548bd5b35a/checkra1n
  
    elif [[ "$CPUArch" == *"x86_64"* ]]; then
      Print_Style "x86_64 detected!" $YELLOW
      DL_LINK=https://assets.checkra.in/downloads/linux/cli/x86_64/607faa865e90e72834fce04468ae4f5119971b310ecf246128e3126db49e3d4f/checkra1n

    elif [[ "$CPUArch" == *"x86"* ]]; then
      Print_Style "x86 detected!" $YELLOW
      DL_LINK=https://assets.checkra.in/downloads/linux/cli/i486/53d45283b5616d9f0daa8a265362b65a33ce503b3088528cc2839544e166d4c6/checkra1n
    else
      Print_Style "ERROR: Unknown/Unsuported architecture! Make sure your architecture is supported by checkra1n." $RED
      DL_LINK=UNKNOWN
      exit
    fi

    Print_Style "Getting checkra1n..." $GREEN
    GetJB
    Print_Style "Done! Marked as executable!" $GREEN
    # Ask user if they want it copied to /usr/bin for easy $PATH access
    whiptail --yesno "Install checkra1n to /usr/bin/ so you can execute it anywhere?" $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT))
    if [ "$?" = "0" ]; then
      cp checkra1n /usr/bin
      Print_Style "Copied executable to /usr/bin" $GREEN
      whiptail --yesno "Delete downloaded file (no longer needed)?" $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT))
      if [ "$?" = "0" ]; then
          rm checkra1n
      fi
    fi
    Print_Style "Attenpting to install dependencies." $BLUE
    # TODO: detect if yum or others are needed
    apt install -y $DEPENDENCIES
    Print_Style "All done!" $BLUE
    ;;
    "Install Autostart Service")
    CHOICE=$(whiptail \
  --title "Checkra1n Service Installer on $(uname -m)" \
  --menu "Choose ONE OR THE OTHER. Having them both working is not possible since they interfere with each other (i think?)." $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT)) \
    "Install Automatic checkra1n" "Automatically jailbreaks DFU devices." \
    "Install Automatic webra1n" "Starts webra1n on port 8081." 3>&1 1>&2 2>&3)
    case $CHOICE in
    "Install Automatic checkra1n")
      whiptail --yesno "Install autostart service? This requires you to put your device into DFU mode manually it to work." $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT))
      if [ "$?" = "0" ]; then
        wget -O checkra1n-linux.service https://raw.githubusercontent.com/Cool5tar/checkra1n-installer/master/checkra1n-linux.service
        chmod 644 checkra1n-linux.service
        mv checkra1n-linux.service /lib/systemd/system/checkra1n-linux.service
        Print_Style "Moved service to /lib/systemd/system/" $GREEN
        Print_Style "Enabling service..." $GREEN
        systemctl daemon-reload
        systemctl enable checkra1n-linux
        systemctl restart checkra1n-linux
        Print_Style "Success!"$ $GREEN
        whiptail --title "Using Checkra1n Autostart Service" --msgbox "This installation is now configured to autostart checkra1n on boot. \nThis means that your device must be in DFU mode manually in order to jailbreak. The autostart service may also interfere with other instances of checkra1n and cause them to fail. \nInstructions of how to put your device into DFU mode are here: \nhttps://www.reddit.com/r/jailbreak/wiki/dfumode \nPlug your device in, push some buttons, and checkra1n will do its work."
      fi
      mainMenu
      ;;
      "Install Automatic webra1n")
      whiptail --yesno "Install autostart service? This requires you to use a web browser on a different device to jailbreak." $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT))
      if [ "$?" = "0" ]; then
        wget -O checkra1n-linux.service https://raw.githubusercontent.com/Cool5tar/checkra1n-installer/master/checkra1n-linux-webra1n.service
        chmod 644 checkra1n-linux.service
        mv checkra1n-linux.service /lib/systemd/system/checkra1n-linux.service
        Print_Style "Moved service to /lib/systemd/system/" $GREEN
        Print_Style "Enabling service..." $GREEN
        systemctl daemon-reload
        systemctl enable checkra1n-linux
        systemctl restart checkra1n-linux
        Print_Style "Success!"$ $GREEN
        NET_IP=`ifconfig | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -1`
        whiptail --title "Using Checkra1n Autostart Service" --msgbox "This installation is now configured to autostart webra1n on boot. \nThis means that you must use a web browser and access this device's local IP via WiFi or Ethernet at port 8081. \nCurrent local URL: \n$NET_IP:8081/ \nNote: this will change if the device connects to a different network or loses connection."
      fi
      mainMenu
    esac
    ;;
    "Credits")
    NET_IP=`ifconfig | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -1`
    whiptail --title "Checkra1n GUI Installer" --msgbox "Checkra1n GUI Installer made by Cool5tar.\nThis project is open source! Check out https://github.com/Cool5tar\nFollow me on Github @cool5tar! \nPlease report all bugs in the GitHub issue tracker and feel free to make pull requests! \nINFO: $OS $(uname -mo) \nVERSION: $VERSION \nLocal IP: $NET_IP" $((LINES/2)) $((COLUMNS*7/10)) $((LISTHEIGHT))
    mainMenu
    ;;
    "Update/Reinstall")
    whiptail --title "Checkra1n GUI Installer" --yesno "Update to latest version?" $((LINES/2)) $((COLUMNS*7/10))
    case $? in
      1)
      mainMenu
      ;;
      0)
      Print_Style "Updating..." $GREEN
      mkdir checkra1n-linux
      cd checkra1n-linux
      wget https://raw.githubusercontent.com/Cool5tar/checkra1n-installer/master/installer.sh
      chmod 755 *.sh
      mv -f * ..
      cd ..
      rm -R checkra1n-linux
      Print_Style "Completed!" $GREEN
     ./installer.sh
   esac
  esac
}
mainMenu