golang_install.sh

lang-Install
# Project Home Page:
# https://github.com/skiy/golang-install
#
# Author: Skiychan <dev@skiy.net>
# Link: https://www.skiy.net

# Script file name
SCRIPT_NAME=$0

# Release link
RELEASE_URL="https://golang.org/dl/"

# Downlaod link
DOWNLOAD_URL="https://dl.google.com/go/"
# DOWNLOAD_URL="http://127.0.0.1/"

# GOPROXY
GOPROXY_TEXT="https://proxy.golang.org,https://athens.azurefd.net"

# Set environmental for golang
PROFILE="/etc/profile"

# Set GOPATH PATH
GO_PATH="/data/go"

# Is GWF
IN_CHINA=0

# Check if user is root
checkRoot() {
    ROOT=$(id -u)
    case $ROOT in
        0) ROOT='root';;
        *) printf "\e[1;31mError: You must be root to run this script\e[0m\n"; exit 1;;
    esac
}

# Chenck GWF
checkGWF() {
    urlstatus=$(curl -s -m 5 -IL $RELEASE_URL | grep 200)
    if [ "$urlstatus" == "" ]; then
        IN_CHINA=1
        RELEASE_URL="https://golang.google.cn/dl/"
        GOPROXY_TEXT="https://goproxy.cn,https://goproxy.io"   
    fi
}

# Get OS bit
initArch() {
    ARCH=$(uname -m)
    BIT=$ARCH
    case $ARCH in
        amd64) ARCH="amd64";;
        x86_64) ARCH="amd64";;
        i386) ARCH="386";;
        armv6l) ARCH="armv6l";; 
        armv7l) ARCH="armv6l";; 
        *) printf "\e[1;31mArchitecture %s is not supported by this installation script\e[0m\n" $ARCH; exit 1;;
    esac
    echo "ARCH = ${ARCH}"
}

# Get OS version
initOS() {
    OS=$(uname | tr '[:upper:]' '[:lower:]')
    case $OS in
        darwin) OS='darwin';;
        linux) OS='linux';;
        freebsd) OS='freebsd';;
#        mingw*) OS='windows';;
#        msys*) OS='windows';;
        *) printf "\e[1;31mOS %s is not supported by this installation script\e[0m\n" $OS; exit 1;;
    esac
    echo "OS = ${OS}"
}

initArgs() {
    key=""

    for arg in "$@" 
    do
        if test "-h" = $arg; then
            showHelpMessage
        fi

        if test -z $key; then 
            key=$arg
        else 
            if test "-v" = $key; then
                customVersion $arg
            elif test "-d" = $key; then
                GO_PATH=$arg
                createGOPATHFolder
            fi

            key=""
        fi
    done
}

# DIY version
customVersion() {
    if [ -n "${1}" ] ;then
        RELEASE_TAG="go${1}"
        echo "Custom Version = ${RELEASE_TAG}"
    fi
}

# if RELEASE_TAG was not provided, assume latest
latestVersion() {
    if [ -z "${RELEASE_TAG}" ]; then
        RELEASE_TAG="$(curl -sL ${RELEASE_URL} | sed -n '/toggleVisible/p' | head -n 1 | cut -d '"' -f 4)"
        echo "Latest Version = ${RELEASE_TAG}"
    fi
}

# Compare version
compareVersion() {
    OLD_VERSION="none"
    NEW_VERSION="${RELEASE_TAG}"
    if test -x "$(command -v go)"; then
        OLD_VERSION="$(go version | awk '{print $3}')"
    fi
    if [ "$OLD_VERSION" = "$NEW_VERSION" ]; then
       printf "\n\e[1;31mYou have installed this version: %s\e[0m\n" $OLD_VERSION; exit 1;
    fi

printf "
Current version: \e[1;33m %s \e[0m 
Target version: \e[1;33m %s \e[0m
" $OLD_VERSION $NEW_VERSION
}

# Compare version size 
versionGE() { test "$(echo "$@" | tr " " "\n" | sort -rV | head -n 1)" == "$1"; }

# Install curl command
installCURLCommand() {
    if !(test -x "$(command -v curl)"); then
        if test -x "$(command -v yum)"; then
            yum install -y curl
        elif test -x "$(command -v apt)"; then
            apt install -y curl
        else 
            printf "\e[1;31mYou must pre-install the curl tool\e[0m\n"
            exit 1
        fi
    fi  
}

# Download go file
downloadFile() {
    url="${1}"
    destination="${2}"

    printf "Fetching ${url} \n\n"

    if test -x "$(command -v curl)"; then
        code=$(curl --connect-timeout 15 -w '%{http_code}' -L "${url}" -o "${destination}")
    elif test -x "$(command -v wget)"; then
        code=$(wget -t2 -T15 -O "${destination}" --server-response "${url}" 2>&1 | awk '/^  HTTP/{print $2}' | tail -1)
    else
        printf "\e[1;31mNeither curl nor wget was available to perform http requests.\e[0m\n"
        exit 1
    fi

    if [ "${code}" != 200 ]; then
        printf "\e[1;31mRequest failed with code %s\e[0m\n" $code
        exit 1
    else 
	    printf "\n\e[1;33mDownload succeeded\e[0m\n"
    fi
}

# Set golang environment
setEnvironment() {
    profile="${1}"
    if [ -z "`grep 'export\sGOROOT' ${profile}`" ];then
        echo -e "\n## GOLANG" >> $profile
        echo "export GOROOT=/usr/local/go" >> $profile
    fi

    if [ -z "`grep 'export\sGOPATH' ${profile}`" ];then
        echo "export GOPATH=${GO_PATH}" >> $profile
    fi
    
    if [ -z "`grep 'export\sGOBIN' ${profile}`" ];then
        echo "export GOBIN=\$GOPATH/bin" >> $profile
    fi   

    if [ "${IN_CHINA}" == "1" ]; then 
        if [ -z "`grep 'export\sGOSUMDB' ${profile}`" ];then
            echo "export GOSUMDB=off" >> $profile
        fi      
    fi

    if [ -z "`grep 'export\sGOPROXY' ${profile}`" ];then
        if versionGE $RELEASE_TAG "go1.13"; then
            GOPROXY_TEXT="${GOPROXY_TEXT},direct"
        fi
        echo "export GOPROXY=${GOPROXY_TEXT}" >> $profile
    fi  

    if [ -z "`grep '\$GOROOT/bin:\$GOBIN' ${profile}`" ];then
        echo "export PATH=\$GOROOT/bin:\$GOBIN:\$PATH" >> $profile
    fi        
}

# Create GOPATH folder
createGOPATHFolder() {
    mkdir -p $GO_PATH
}

# Show copyright
showCopyright() {
    clear

printf "
###############################################################
###  Golang Install
###
###  Author:  Skiychan<dev@skiy.net>
###  Link:    https://www.skiy.net 
###  Project: https://github.com/skiy/golang-install
###############################################################
\n"
}

# Show system information
showSystemInformation() {
printf "
###############################################################
###  System: %s 
###  Bit: %s 
###  Version: %s 
###############################################################
\n" $OS $BIT $RELEASE_TAG
}

# Show success message
showSuccessMessage() {
printf "
###############################################################
# Install success, please execute again \e[1;33msource %s\e[0m
###############################################################
\n" $PROFILE
}

# Show help message
showHelpMessage() {
printf "
Go install
Usage: %s [-h] [-v version] [-d gopath]
Options:
  -h            : this help
  -v            : set go version (default: latest version)
  -d            : set go path (default: /data/go)
\n" $SCRIPT_NAME
exit 1
}

# identify platform based on uname output
initArgs $@

showCopyright

checkRoot

checkGWF

set -e

initArch

initOS

installCURLCommand

latestVersion

compareVersion

showSystemInformation

# Download File
BINARY_URL="${DOWNLOAD_URL}${RELEASE_TAG}.${OS}-${ARCH}.tar.gz"
DOWNLOAD_FILE="$(mktemp).tar.gz"
downloadFile $BINARY_URL $DOWNLOAD_FILE

# Tar file and move file
rm -rf /usr/local/go
tar -C /usr/local/ -zxf $DOWNLOAD_FILE && \
rm -rf $DOWNLOAD_FILE
 
setEnvironment $PROFILE
 
# Make environmental is enable
. $PROFILE

go env
go version
 
showSuccessMessage