#!/usr/bin/env bash

set -e

function finish {
    rv=$?

    if [ "$rv" != "0" ]
    then
        echo -e "\e[31mERROR!\e[39m"
    else
        echo -e "\e[32mSuccess!\e[39m"
    fi

    return $rv
}
trap finish EXIT

function matplotlib_backend_fix() {
    mkdir -p ~/.config/matplotlib
    if ! test ~/.config/matplotlib/matplotlibrc;
    then
        echo "backend : Agg" > ~/.config/matplotlib/matplotlibrc
    fi
}

function check_style() {
    flake8 --version
    if ! flake8;
    then
        (>&2 echo "Please, check your code using flake8.")
        exit 1
    fi

    if [ "${TRAVIS_OS_NAME}" == "linux" ];
    then
        rstcheck --version
        if ! rstcheck -r .;
        then
            (>&2 echo "Please, check your code using rstcheck.")
            exit 1
        fi
    fi

    if msg=$(grep --include=\*.{py,rst} -Rn -P "\t" 2> /dev/null);
    then
        (>&2 echo "Please, remove tab character from the following files.")
        (>&2 echo "$msg")
        exit 1
    fi
}

function check_black_format() {
    if [[ $MAJOR -gt 2 ]];
    then
        python3 -m pip install -U black -q
        git clean -xdf

        # Make sure the code has been formatted.
        find . -type f -name "*.py" -exec cksum "{}" \; | sort > checksum0.txt
        find . -type f -name "*.py" -exec black --quiet --fast {} \;
        find . -type f -name "*.py" -exec cksum "{}" \; | sort > checksum1.txt
        cat checksum0.txt
        cat checksum1.txt
        if ! diff checksum0.txt checksum1.txt;
        then
            err="Please, apply the black Python code formatter"
            (>&2 echo "$err on the following files:")
            msg=$(diff checksum0.txt checksum1.txt | sed '1d; n; d')
            echo $(echo $msg | awk -F ' ' '{print $4}' | uniq)
            rm checksum0.txt
            rm checksum1.txt
            exit 1
        else
            rm checksum0.txt
            rm checksum1.txt
        fi
    fi
}

function check_broken_links() {

    if type gem && gem install awesome_bot
    then
        if [ -f README.md ]
        then
            awesome_bot README.md --allow-dupe --allow-redirect --skip-save-results
            if [ $? -ne 0 ]
            then
                exit 1
            fi
        fi
    fi
}

function has_mkl() {
    cmd="python3 -c \"import numpy as np;"
    cmd="$cmd print(len(np.__config__.blas_mkl_info) > 0)\""
    has=$(eval $cmd)
    test $has = True
}

function has_conda() {
    type conda >/dev/null 2>&1
}

function install_deps() {
    python3 -m pip install -U setuptools pip pytest pytest-pycodestyle -q
    python3 -m pip install -U flake8 rstcheck pygments docutils -q
    python3 -m pip install -U rstcheck -q
    python3 -m pip install -U shell-timeit -q

    if has_conda;
    then
        conda install numpy scipy --yes -q
    else
        python3 -m pip install numpy scipy -q -U
    fi
}

function find_pkg_name {
    python3 -c "from setuptools import find_packages; print(find_packages()[0])"
}

function testit {
    orig_dir=$(pwd)
    cmd="python3 -c 'import platform; print(platform.python_version())'"
    cmd="$cmd | awk -F '.' '{print \$1}'"
    MAJOR=$(eval $cmd)

    if [ "${TRAVIS_OS_NAME}" == "osx" ]; then
        export PATH=$(brew --prefix)/opt/grep/libexec/gnubin:$PATH
    fi

    if [ -z "$PKG_NAME" ]
    then
        PKG_NAME=$(find_pkg_name)
    fi

    matplotlib_backend_fix
    install_deps
    check_style
    check_black_format
    check_broken_links

    python3 setup.py test
    git clean -xdfq
    python3 -m pip install -q . && git clean -xdfq
    cd ~/

    cmd="python3 -c \"import $PKG_NAME\""
    msg=$(timeit "$(echo $cmd)" | grep loop)
    elapsed=$(echo "$msg" | awk -F ' ' '{print $1}')
    unit=$(echo "$msg" | awk -F ' ' '{print $2}')
    if [[ $unit == "s" ]];
    then
        elapsed=$(bc <<< "$elapsed * 1000")
    fi
    elapsed=${elapsed%.*}

    echo "Importing time: $elapsed milliseconds"
    if [[ $elapsed -ge 1000 ]];
    then
        (>&2 echo "Too slow to import $PKG_NAME: more than a second.")
        (>&2 echo "Please, fix it as it is taking $elapsed ms.")
    fi

    cmd="import sys; import $PKG_NAME; sys.exit(not hasattr($PKG_NAME, 'test'))"
    if python3 -c "$cmd"
    then
        python3 -c "import sys; import $PKG_NAME; sys.exit($PKG_NAME.test())"
    fi
    python3 -m pip uninstall $PKG_NAME --yes
    cd $orig_dir && git clean -xdfq
    python3 -m pip install -r requirements.txt -q
    python3 -m pip install . && git clean -xdfq
    [ -d doc ] && cd doc && make html && cd $orig_dir
    [ -d docs ] && cd docs && make html && cd $orig_dir
    git clean -xdfq
    python3 -m pip uninstall $PKG_NAME --yes
    python3 setup.py sdist
    python3 -m pip install dist/$(ls dist | grep -i -E '\.(gz)$' | head -1)
    cd ~/
    cmd="import sys; import $PKG_NAME; sys.exit(not hasattr($PKG_NAME, 'test'))"
    if python3 -c "$cmd"
    then
        python3 -c "import sys; import $PKG_NAME; sys.exit($PKG_NAME.test())"
    fi
    cd $orig_dir
}

(set +x; testit)