#!/bin/bash
#
# .---.                  .              .
# |                      |              |
# |--- .--. .-.  .-.  .-.|  .-. .--.--. |.-.  .-. .--.  .-.
# |    |   (.-' (.-' (   | (   )|  |  | |   )(   )|  | (.-'
# '    '     --'  --'  -' -  -' '  '   -' -'   -' '   -  --'
#
#                    Freedom in the Cloud
#
# Go functions
#
# License
# =======
#
# Copyright (C) 2014-2016 Bob Mottram <bob@robotics.uk.to>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# For reasons unknown we initially have to upgrade to an intermediate version
# before getting to the version we want
GO_INTERMEDIATE_VERSION=1.4.2

GO_VERSION=1.7

GO_REPO="https://go.googlesource.com/go"
GO_PACKAGE_MANAGER_REPO="https://github.com/gpmgo/gopm"
GVM_HOME=/home/git/gvm
GVM_REPO="https://github.com/moovweb/gvm"
GVM_COMMIT='25ea8ae158e2861c92e2b22c458e60840157832f'

function select_go_version {
    if [ ! -d $GVM_HOME/bin ]; then
        echo $'GVM was not installed'
        exit 629532
    fi
    export GVM_ROOT=$GVM_HOME
    if ! grep -q "GVM_ROOT=" ~/.bashrc; then
        echo "export GVM_ROOT=$GVM_ROOT" >> ~/.bashrc
    else
        sed -i "s|export GVM_ROOT=.*|export GVM_ROOT=$GVM_ROOT|g" ~/.bashrc
    fi
    cd $GVM_ROOT/bin
    [[ -s "$GVM_ROOT/scripts/gvm" ]] && source "$GVM_ROOT/scripts/gvm"
    gvm use go${GO_VERSION} --default

    if [ ${#GOPATH} -lt 2 ]; then
        echo $'GOPATH was not set'
        exit 629825
    fi

    systemctl set-environment GOPATH=$GOPATH
}

function mesh_upgrade_golang_from_source {
    chroot "$rootdir" adduser --disabled-login --gecos 'go' ipfs

    git clone $GO_REPO $rootdir/home/go/go${GO_VERSION}
    cd $rootdir/home/go/go${GO_VERSION}
    git checkout go${GO_VERSION} -b go${GO_VERSION}

    git clone $GO_REPO $rootdir/home/go/go${GO_INTERMEDIATE_VERSION}
    cd $rootdir/home/go/go${GO_INTERMEDIATE_VERSION}
    git checkout go${GO_INTERMEDIATE_VERSION} -b go${GO_INTERMEDIATE_VERSION}

    cat <<EOF > ${rootdir}/root/upgrade_golang.sh
#!/bin/bash

apt-get -y install build-essential libc6-dev
apt-get -y install gcc-multilib g++-multilib make
apt-get -y install curl git mercurial binutils bison

if [ -d /home/go/Maildir ]; then
    rm -rf /home/go/Maildir
fi

export GOROOT=/home/go
export GOPATH=/home/go/go${GO_INTERMEDIATE_VERSION}

cd /home/go/go${GO_INTERMEDIATE_VERSION}/src
./all.bash
if [ ! -f /home/go/go${GO_INTERMEDIATE_VERSION}/bin/go ]; then
    exit 63722
fi

export GOROOT_BOOTSTRAP=/home/go/go${GO_INTERMEDIATE_VERSION}

cd /home/go/go${GO_VERSION}/src
./all.bash
if [ ! -f /home/go/go${GO_VERSION}/bin/go ]; then
    exit 528352
fi

chown -R go:go /home/go

export GOPATH=/home/go/go${GO_VERSION}
export GOROOT=/home/go
echo "export GOPATH=/home/go/go${GO_VERSION}" >> /home/go/.bashrc
echo "export GOPATH=/home/go/go${GO_VERSION}" >> ~/.bashrc
echo "export GOROOT=/home/go" >> /home/go/.bashrc
echo "export GOROOT=/home/go" >> ~/.bashrc
echo "systemctl set-environment GOPATH=\$GOPATH" >> ~/.bashrc
echo "systemctl set-environment GOROOT=\$GOROOT" >> ~/.bashrc
echo "systemctl set-environment GOPATH=\$GOPATH" >> /home/go/.bashrc
echo "systemctl set-environment GOROOT=\$GOROOT" >> /home/go/.bashrc
systemctl set-environment GOPATH=\$GOPATH
systemctl set-environment GOROOT=\$GOROOT

exit 0
EOF
    chroot ${rootdir} chmod +x /root/upgrade_golang.sh
    chroot ${rootdir} /root/upgrade_golang.sh
    if [ ! -f ${rootdir}/home/go/go${GO_VERSION}/bin/go ]; then
        echo $'Failed to upgrade golang'
        cat ${rootdir}/root/upgrade_golang.sh
        rm -f ${rootdir}/root/upgrade_golang.sh
        exit 836535
    fi
    rm -f ${rootdir}/root/upgrade_golang.sh
}

function mesh_upgrade_golang {
    chroot "$rootdir" adduser --disabled-login --gecos 'go' go

    GOARCH=
    if [[ $ARCHITECTURE == *"386" || $ARCHITECTURE == *"686" ]]; then
        GOARCH=386
    fi
    if [[ $ARCHITECTURE == *"amd64" ]]; then
        GOARCH=amd64
    fi
    if [[ $ARCHITECTURE == *"arm"* ]]; then
        GOARCH=armv6l
    fi
    if [ ! $GOARCH ]; then
        return
    fi

    GO_SOURCE=https://storage.googleapis.com/golang/go${GO_VERSION}.linux-${GOARCH}.tar.gz

    if [ ! -d ${rootdir}${INSTALL_DIR} ]; then
        chroot "$rootdir" mkdir -p ${INSTALL_DIR}
    fi
    cd ${rootdir}${INSTALL_DIR}
    wget ${GO_SOURCE}
    if [ ! -f ${rootdir}${INSTALL_DIR}/go${GO_VERSION}.linux-${GOARCH}.tar.gz ]; then
        exit 26524
    fi
    chroot "$rootdir" tar -C /home/go -xzf ${INSTALL_DIR}/go${GO_VERSION}.linux-${GOARCH}.tar.gz
    if [ ! -d ${rootdir}/home/go/go/bin ]; then
        echo 'Go binary not installed'
        exit 763562
    fi
    mv ${rootdir}/home/go/go ${rootdir}/home/go/go${GO_VERSION}
    echo "export GOROOT=/home/go" >> $rootdir/root/.bashrc
    echo "export GOROOT=/home/go" >> $rootdir/home/$MY_USERNAME/.bashrc
    echo "export GOROOT=/home/go" >> $rootdir/home/go/.bashrc

    echo "export GOPATH=\$GOROOT/go${GO_VERSION}/bin" >> $rootdir/root/.bashrc
    echo "export GOPATH=\$GOROOT/go${GO_VERSION}/bin" >> $rootdir/home/$MY_USERNAME/.bashrc
    echo "export GOPATH=\$GOROOT/go${GO_VERSION}/bin" >> $rootdir/home/go/.bashrc

    echo 'export PATH=$PATH:$GOPATH' >> $rootdir/root/.bashrc
    echo 'export PATH=$PATH:$GOPATH' >> $rootdir/home/$MY_USERNAME/.bashrc
    echo 'export PATH=$PATH:$GOPATH' >> $rootdir/home/go/.bashrc
    chroot "$rootdir" chown -R $MY_USERNAME:$MY_USERNAME /home/$MY_USERNAME
    chroot "$rootdir" chown -R go:go /home/go
    cp ${rootdir}/home/go/go${GO_VERSION}/bin/* ${rootdir}/usr/bin
}

function upgrade_golang {
    if [[ $SYSTEM_TYPE == "$VARIANT_WRITER" || $SYSTEM_TYPE == "$VARIANT_MAILBOX" || $SYSTEM_TYPE == "$VARIANT_CLOUD" || $SYSTEM_TYPE == "$VARIANT_SOCIAL" || $SYSTEM_TYPE == "$VARIANT_MEDIA" || $SYSTEM_TYPE == "$VARIANT_CHAT" || $SYSTEM_TYPE == "$VARIANT_MESH" ]]; then
        return
    fi

    # NOTE: this is annoyingly hacky and going in the opposite
    # direction of a pure blend, but it's necessary if you want
    # to run the latest version of gogs

    # update to the next commit
    function_check set_repo_commit
    set_repo_commit $INSTALL_DIR/gvm "gvm commit" "$GVM_COMMIT" $GVM_REPO

    if grep -Fxq "upgrade_golang" $COMPLETION_FILE; then
        return
    fi

    export GVM_ROOT=$GVM_HOME

    apt-get -y install build-essential libc6-dev-i386 gcc-multilib g++-multilib
    apt-get -y install gcc curl git mercurial make binutils bison
    if [ ! -d $INSTALL_DIR ]; then
        mkdir $INSTALL_DIR
    fi
    cd $INSTALL_DIR
    function_check git_clone
    git_clone $GVM_REPO gvm
    cd $INSTALL_DIR/gvm
    git checkout $GVM_COMMIT -b $GVM_COMMIT
    if [ ! -f binscripts/gvm-installer ]; then
        echo $'gvm installer not found'
    fi
    chmod +x binscripts/gvm-installer

    if [ -d /root/.gvm ]; then
        rm -rf /root/.gvm
    fi
    if [ -d $GVM_ROOT ]; then
        rm -rf $GVM_ROOT
    fi
    sed -i "s|export GVM_ROOT=.*|export GVM_ROOT=$GVM_ROOT|g" ~/.bashrc

    if [ ! -d /home/git ]; then
        # add a gogs user account within which the gvm home directory will exist
        adduser --disabled-login --gecos 'Gogs' git
    fi
    if [ -d /home/git/Maildir ]; then
        rm -rf /home/git/Maildir
    fi

    # TODO: this script is all over the place
    # and contains hardcoded github. See if you can do better
    ./binscripts/gvm-installer master /home/git

    if [ ! -d $GVM_ROOT ]; then
        echo $'Unable to install gvm'
        exit 83537
    fi

    [[ -s "$GVM_ROOT/scripts/gvm" ]] && source "$GVM_ROOT/scripts/gvm"
    if [ ! -f $GVM_ROOT/bin/gvm ]; then
        echo $'gvm was not installed'
    fi
    if ! grep -q "export GVM_ROOT=" ~/.bashrc; then
        echo "export GVM_ROOT=$GVM_ROOT" >> ~/.bashrc
    fi
    if ! grep -q 'PATH=$PATH:$GVM_ROOT/bin' ~/.bashrc; then
        echo 'PATH=$PATH:$GVM_ROOT/bin' >> ~/.bashrc
    fi

    export GOROOT_BOOTSTRAP=$GOROOT

    cd $GVM_ROOT/bin
    gvm install go${GO_INTERMEDIATE_VERSION}
    gvm use go${GO_INTERMEDIATE_VERSION}
    gvm install go${GO_VERSION}
    if [ ! "$?" = "0" ]; then
        echo $'Unable to upgrade golang'
        exit 529252
    fi
    gvm use go${GO_VERSION} --default

    chown -R git:git $GVM_HOME

    if ! grep -q "gvm commit" $COMPLETION_FILE; then
        echo "gvm commit:$GVM_COMMIT" >> $COMPLETION_FILE
    else
        sed -i "s/gvm commit.*/gvm commit:$GVM_COMMIT/g" $COMPLETION_FILE
    fi

    echo 'upgrade_golang' >> $COMPLETION_FILE
}

# NOTE: deliberately there is no "exit 0"