#!/bin/bash
# Authors: James Campbell, Alex McLean
# Licence: GPLv3
# Date: September 2019
# Last Updated: October 2019
# Contact: james@jamescampbell.us / alex@slab.org
# What: tidalcycles installer script for OSX and Linux

#### COLORS
COLOR_PURPLE='\033[0;35m'
normal='\033[0m'

get_distro_alias() {
    distro_name=$1
    distro_alias=unknown

    case "${distro_name}" in
        "Debian"|"Debian GNU/Linux"|"debian")
            distro_alias=debian
            ;;
        "Ubuntu"|"ubuntu")
            distro_alias=ubuntu
            ;;
        "Exherbo"|"exherbo")
            distro_alias=exherbo
            ;;
        "Fedora"|"fedora")
            distro_alias=fedora
            ;;
        "CentOS Linux"|"CentOS"|"centos"|"Red Hat Enterprise Linux"*)
            distro_alias=centos
            ;;
        "Alpine Linux"|"Alpine")
            distro_alias=alpine
            ;;
        "Linux Mint"|"LinuxMint")
            distro_alias=mint
            ;;
        "Amazon Linux AMI")
            distro_alias=amazonlinux
            ;;
        "AIX")
            distro_alias=aix
            ;;
        "FreeBSD")
            distro_alias=freebsd
            ;;
        "Darwin")
            distro_alias=darwin
            ;;
    esac

    printf "%s" "${distro_alias}"

    unset distro_name distro_alias
}

get_distro_name() {
    if [ -f /etc/os-release ]; then
        # freedesktop.org and systemd
        # shellcheck disable=SC1091
        . /etc/os-release
        printf "%s" "$NAME"
    elif command_exists lsb_release ; then
        # linuxbase.org
        printf "%s" "$(lsb_release -si)"
    elif [ -f /etc/lsb-release ]; then
        # For some versions of Debian/Ubuntu without lsb_release command
        # shellcheck disable=SC1091
        . /etc/lsb-release
        printf "%s" "$DISTRIB_ID"
    elif [ -f /etc/redhat-release ]; then
        case "$(cat /etc/redhat-release)" in
        # Older CentOS releases didn't have a /etc/centos-release file
        "CentOS release "*)
            printf "CentOS"
            ;;
        "CentOS Linux release "*)
            printf "CentOS Linux"
            ;;
        "Fedora release "*)
            printf "Fedora"
            ;;
        # Fallback to uname
        *)
            printf "%s" "$(uname -s)"
            ;;
        esac
    elif [ -f /etc/debian_version ]; then
        # Older Debian/Ubuntu/etc.
        printf "Debian"
    else
        # Fall back to uname, e.g. "Linux <version>", also works for BSD, etc.
        printf "%s" "$(uname -s)"
    fi
}

get_distro_ver() {
    if [ -f /etc/os-release ]; then
        # freedesktop.org and systemd
        # shellcheck disable=SC1091
        . /etc/os-release
        printf "%s" "$VERSION_ID"
    elif command_exists lsb_release ; then
        # linuxbase.org
        printf "%s" "$(lsb_release -sr)"
    elif [ -f /etc/lsb-release ]; then
        # For some versions of Debian/Ubuntu without lsb_release command
        # shellcheck disable=SC1091
        . /etc/lsb-release
        printf "%s" "$DISTRIB_RELEASE"
    elif [ -f /etc/redhat-release ]; then
        case "$(cat /etc/redhat-release)" in
        # NB: Older CentOS releases didn't have a /etc/centos-release file
        "CentOS release "*|"Fedora release "*)
            printf "%s" "$(awk 'NR==1 { split($3, a, "."); print a[1] }' /etc/redhat-release)"
            ;;
        "CentOS Linux release "*)
            printf "%s" "$(awk 'NR==1 { split($4, a, "."); print a[1] }' /etc/redhat-release)"
            ;;
        # Fallback to uname
        *)
            printf "%s" "$(uname -r)"
            ;;
        esac
    elif [ -f /etc/debian_version ]; then
        # Older Debian/Ubuntu/etc.
        printf "%s" "$(cat /etc/debian_version)"
    else
        case "$(uname -s)" in
        AIX)
            printf "%s" "$(uname -v)"
            ;;
        FreeBSD)
            # we only care about the major numeric version part left of
            # the '.' in "11.2-RELEASE".
            printf "%s" "$(uname -r | cut -d . -f 1)"
            ;;
        *)
            # Fall back to uname, e.g. "Linux <version>", also works for BSD, etc.
            printf "%s" "$(uname -r)"
        esac
    fi
}

get_arch() {
    myarch=$(uname -m)

    case "${myarch}" in
    x86_64|amd64)
        printf "x86_64"  # or AMD64 or Intel64 or whatever
        ;;
    i*86)
        printf "i386"  # or IA32 or Intel32 or whatever
        ;;
    *)
        case "$(uname -s)" in
        AIX)
            case "$(uname -p)" in
            powerpc)
                printf "powerpc"
                ;;
            *)
                die "Cannot figure out architecture on AIX (was: ${myarch})"
                ;;
            esac
            ;;
        *)
            die "Cannot figure out architecture (was: ${myarch})"
            ;;
        esac
    esac

    unset myarch
}


echo "Detected system information:"
echo "  Architecture:   $(get_arch)"
echo "  Distribution:   $(get_distro_name)"
echo "  Distro alias:   $(get_distro_alias "$(get_distro_name)")"
echo "  Distro version: $(get_distro_ver)"

mydistro=$(get_distro_alias "$(get_distro_name)")

#### CHECK FOR GIT
if test "${mydistro}" = "darwin"; then
    if command -v git 2>/dev/null; then
	printf "$COLOR_PURPLE[1]$normal 'git' command already installed.\n"
    else
	printf "$COLOR_PURPLE[1]$normal 'git' required, installing commandline tools..."
	printf "** Please click 'install' when a popup appears, and wait until it finishes installing. **\n"
	/usr/bin/xcode-select --install
        printf "\nWhen that's done, click on this window and press enter to continue."
        read -r answer </dev/tty
    fi
else
    # Don't bother checking to see if they're there or not. Apt will
    # do that!
    printf "$COLOR_PURPLE[1]$normal Installing build tools (via apt)...\n"
    sudo apt-get install -qqy git build-essential libgmp-dev
fi

#### CHECK FOR HASKELL
if [ -e ~/.ghcup/bin/cabal ]; then
	printf "$COLOR_PURPLE[2]$normal Haskell found, skipping install of that...\n"
else
	printf "$COLOR_PURPLE[2]$normal Installing Haskell (via 'ghcup')...\n"
        curl https://get-ghcup.haskell.org -sSf | BOOTSTRAP_HASKELL_GHC_VERSION=latest BOOTSTRAP_HASKELL_CABAL_VERSION=latest BOOTSTRAP_HASKELL_NONINTERACTIVE=1 sh 2>&1 > /tmp/ghcup-install.log
        if [ $(grep -c ghcup ~/.bashrc) -ne 0 ]; then
            printf "$COLOR_PURPLE[2.1]$normal Adding ghcup initialisation to ~/.bashrc and zshrc...\n"
            echo 'source $HOME/.ghcup/env' >> "$HOME/.bashrc"
	    echo 'source $HOME/.ghcup/env' >> "$HOME/.zshrc"
        fi
fi

#### INSTALL TIDALCYCLES
printf "$COLOR_PURPLE[3]$normal Congratulations, you have all the pre-reqs...\n"
echo "Installing tidalcycles haskell library (via cabal)..."
echo ""
. "$HOME/.ghcup/env"
cabal v2-update
cabal v2-install tidal --lib

#### INSTALL ATOM
if test "${mydistro}" = "darwin"; then
    if [ -d "/Applications/Atom.app" ]; then
	printf "$COLOR_PURPLE[4]$normal Atom already installed, skipping...\n"
    else
	printf "$COLOR_PURPLE[4]$normal Installing Atom...\n"
	curl -Lk https://atom.io/download/mac --output /tmp/atom.zip
	unzip -q "/tmp/atom.zip" -d /Applications
	rm /tmp/atom.zip
    fi
else
    if command -v atom 2>/dev/null; then
	printf "$COLOR_PURPLE[4]$normal Atom already installed.\n"
    else
	printf "$COLOR_PURPLE[4]$normal Downloading and installing the Atom editor...\n"
	curl -Lk https://atom.io/download/deb --output /tmp/atom.deb
	sudo apt -qqy install /tmp/atom.deb
	rm /tmp/atom.deb
    fi
fi

printf "$COLOR_PURPLE[6]$normal Installing atom TidalCycles plugin...\n"
if test "${mydistro}" = "darwin"; then
    /Applications/Atom.app/Contents/Resources/app/apm/bin/apm install tidalcycles
else
    apm install tidalcycles
fi

#### INSTALL SUPERCOLLIDER
if test "${mydistro}" = "darwin"; then
    if [ -d "/Applications/SuperCollider.app" ]; then
	printf "$COLOR_PURPLE[7]$normal SuperCollider already installed, skipping...\n"
    else
	printf "$COLOR_PURPLE[7]$normal Installing SuperCollider...\n"
	curl -Lk https://github.com/supercollider/supercollider/releases/download/Version-3.11.2/SuperCollider-3.11.2+BigSur.aed25fa.zip --output /tmp/sc3.zip
	unzip -q "/tmp/sc3.zip" "SuperCollider/SuperCollider.app/*" -d /tmp/testsc
	mv /tmp/testsc/SuperCollider/SuperCollider.app /Applications
	rm /tmp/sc3.zip
    fi

    if [ -e "~/Library/Application\ Support/SuperCollider/Extensions/StkInst.scx" ]; then
	printf "$COLOR_PURPLE[8]$normal sc3-plugins already installed, skipping...\n"
    else
	#### INSTALL PLUGINS
	printf "$COLOR_PURPLE[8]$normal Installing SuperCollider Plugins...\n"
	curl -Lk https://github.com/supercollider/sc3-plugins/releases/download/Version-3.11.1/sc3-plugins-3.11.1-macOS-signed.zip --output /tmp/sc3plugins.zip
	mkdir -p ~/Library/Application\ Support/SuperCollider/Extensions/
	unzip -nq /tmp/sc3plugins.zip -d ~/Library/Application\ Support/SuperCollider/Extensions/
	rm /tmp/sc3plugins.zip
    fi
else
    if command -v scide 2>/dev/null; then
	printf "$COLOR_PURPLE[7]$normal SuperCollider already installed, skipping...\n"
    else
	printf "$COLOR_PURPLE[7]$normal Downloading, compiling and installing SuperCollider and sc3plugins...\n"
	mkdir ~/tidal-tmp
	cd ~/tidal-tmp
	git clone https://github.com/lvm/build-supercollider
	cd build-supercollider
	./build-supercollider.sh
	./build-sc3-plugins.sh
    fi

    printf "$COLOR_PURPLE[7.1]$normal Adding user to the 'audio' group.\n"
    sudo adduser $USER audio
fi


#### INSTALL SUPERDIRT
echo "$COLOR_PURPLE[9]$normal Installing the SuperDirt synths and samples (will take some time..)"
if test "${mydistro}" = "darwin"; then
    echo 'include("SuperDirt");"SuperDirt installation complete!".postln;0.exit;' | /Applications/SuperCollider.app/Contents/MacOS/sclang
else
    echo 'include("SuperDirt");"SuperDirt installation complete!".postln;0.exit;' | sclang
fi

echo "Tidal and SuperDirt should now be installed!\n\n"

echo "Please log out and in again to complete the set up.\n\n"

echo "You can then follow the instructions here to start everything up for the first time:"
echo "  https://tidalcycles.org/docs/getting-started/tidal_start"
echo "Enjoy!"
exit 0