#!/usr/bin/env bash 

# Copyright 2017 Mycroft AI Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Verify and present the user with information about their installation of mycroft. 
# Should be run as the mycroft user. 
#
# To do: functionalize and allow for parametereized calls of each.  
# rs 2017-05-05

helpfunc() {
echo "Usage: ${0} [FUNCTION]"
echo " Functions include
  -v version 
  -P python
  -p permissions
  -i internet 
  -s system info
  -u audio 
  -r running
  -m mimic
  -a run all checks
"
}

if [[ $# -eq 0 ]] ; then
    helpfunc && exit 1
fi

MYCROFT_HOME=''
RUN_AS_ROOT=1
source $(locate virtualenvwrapper.sh)

# log stuff and things
LOG_FILE=/tmp/my-info.$$.out
touch ${LOG_FILE}
if [[ ! -w "${LOG_FILE}" ]] ; then
    echo "Unable to write log file, output will be to screen only!"
    LOG_FILE="/dev/null"
    sleep 3
else
    echo "Logging to ${LOG_FILE}"
fi

# it's big, it's heavy, it's wood!
mlog() {
    local timestamp="[$(date +"%Y-%m-%d %H:%M:%S")]"
    message="$*"
    echo "${timestamp} ${message}" |tee -a ${LOG_FILE}
}

# Sup big perm.
checkperms () {
if ! [[ -e "${1}" && -w "${1}" && -O "${1}"  ]] ; then
    if ! [[ -e "${1}" ]] ; then
        # doesn't exist?
        return 1
    fi
    if [[ -w "${1}" ]] ; then
        # lacks ownership
        return 2
    else
        # lacks write permissions
        return 3
    fi
else 
    if [[ -x "${1}" ]] ; then
        # executable and awesome.
        return 10
    else
        # merely awesome
        return 0
    fi
fi
echo "If you can read this, we may need glasses."
}

# Check before we wreck:
checkfiles() {
mlog "Permission checks..."
cat << EOF > /tmp/my-list.$$
${MYCROFT_HOME}
${MYCROFT_HOME}/scripts/logs
/tmp/mycroft/
/opt/mycroft/skills
EOF

if [[ ${RUN_AS_ROOT} -eq 1 ]] ; then
    while read CHECKFN; do
        checkperms "${CHECKFN}"
        case $? in
            '0') mlog " - ${CHECKFN} has viable permissions." ;;
            '1') mlog " = Error: ${CHECKFN} doesn't exist?" ;;
            '2') mlog " = Error: ${CHECKFN} not owned by ${UID}." ;;
            '3') mlog " = Error: ${CHECKFN} not writeable by ${UID}." ;;
            '10') mlog " - ${CHECKFN} is executable and has viable permissions." ;;
            *) mlog " = Error: unable to verify permissions on ${CHECKFN}." ;;
        esac

    done < /tmp/my-list.$$
else
    mlog " = Error: permission checks skipped while running as root."
fi
rm -f /tmp/my-list.$$ 
}
# random info of potential interest
checksysinfo () {
mlog "System info..."
mlog " - CPU: $(awk -F: '/model name/ {print $2;exit}' /proc/cpuinfo)"
mlog " - $(echo "RAM Utilization:" && free -h)"
mlog " - $(echo "Mycroft partition disk usage:" && df -h "${MYCROFT_HOME}")"
mlog " - $(echo "OS Info:" &&  cat /etc/*elease*)"
mlog " - $(echo "Kernel version:" && uname -a)"
}

# -v
checkversion () {
mlog "Mycroft version is $(grep -B3 'END_VERSION_BLOCK' ${MYCROFT_HOME}/mycroft/version/__init__.py | cut -d' ' -f3 | tr -s '\012' '\056')"
}

# do you want to do repeat?
checkmimic () {
mlog "Checking Mimic..."
if hash mimic ; then
    mlog " - Mimic$(mimic --version| grep mimic)"
    mlog " - $(mimic -lv)"
else
    mlog " = Error: Mimic binary not found. Mimic may not be installed?"
fi
}

# pythoning!
checkPIP () {
mlog "Python checks"
mlog " - Verifying ${MYCROFT_HOME}/requirements.txt:"
if workon mycroft ; then
    pip list > /tmp/mycroft-piplist.$$
    
    while read reqline; do
        IFS='==' read  -r -a PIPREQ <<< "$reqline"  
        PIPREQVER=$(grep -i ^"${PIPREQ[0]} " /tmp/mycroft-piplist.$$ | cut -d'(' -f2| tr -d '\051')
        if [[ "${PIPREQVER}" == "${PIPREQ[2]}" ]]; then
            mlog " -- pip ${PIPREQ[0]} version ${PIPREQ[2]}" 
        else
            mlog " ~~ Warn: can't find ${PIPREQ[0]} ${PIPREQ[2]} in pip. (found ${PIPREQVER})"
        fi
    done < "${MYCROFT_HOME}/requirements.txt"
    deactivate 
    mlog " - PIP list can be found at /tmp/mycroft-piplist.$$ to verify any issues."
    else 
        mlog " = Error: Unable to enter the mycroft virtualenv, skipping python checks."
fi
}

# a series of tubes
checktubes () {
 mlog "Internet connectivity..."
case "$(curl -s --max-time 2 -I http://home.mycroft.ai/ | sed 's/^[^ ]*  *\([0-9]\).*/\1/; 1q')" in
    [23]) mlog " - HTTP connectivity to https://home.mycroft.ai worked!";;
    5) mlog " = Error: The web proxy won't let us through to https://home.mycroft.ai";;
    *) mlog " = Error: The network is down or very slow getting to https://home.mycroft.ai";;
esac
}

# I prefer biking myself.
checkrunning () {
while read SCREEN_SESS ; do
    SESS_NAME=$(echo "${SCREEN_SESS}"| cut -d'(' -f1 | cut -d'.' -f2)
    SESS_ID=$(echo "${SCREEN_SESS}"| cut -d'.' -f1)
    if [[ $(ps flax| grep "$SESS_ID" | awk ' { print $4 } ' | grep -c "$SESS_ID") -eq 1 ]]; then
        mlog " - ${SESS_NAME} appears to be currently running."
    fi
done < <(screen -list | grep mycroft)
}

# He's dead, Jim.
checkpulse () {
mlog "Sound settings..."
if hash pactl && [[ ${RUN_AS_ROOT} -eq 1 ]] ; then
    mlog " - $(echo "Pulse Audio Defaults:" && pactl info | grep "Default S[i,o]")"
    mlog " - $(echo "Pulse Audio Sinks:" && pactl list sinks | grep -e ^Sink  -e 'Name:' -e 'device.description' -e 'product_name' -e udev.id -e 'State:')"
    mlog " - $(echo "Pulse Audio Sources:" && pactl list sources| grep -e ^Sourc -e 'Name:' -e 'device.description' -e 'product_name' -e udev.id -e 'State:')"
else 
    mlog " = Error: Can't run pactl, skipping audio checks."
fi
}

# ok, fine, go ahead and run this crazy thing
mlog "Starting ${0}"

# Who am i? Check if running sudo/as root/etc. 
if [[ ${EUID} -ne ${UID} ]] ; then
    if [[ ${EUID} -gt 0 ]] ; then
        mlog " - Running as ${EUID} from UID ${UID}"
    else
        mlog " - Running with root permissions from UID ${UID}"
        RUN_AS_ROOT=0
    fi

else
    mlog " - Running as UID ${UID}"
fi

# where are we?
RUNDIR=$(readlink -f "${0}"| tr -s '\057' '\012' | sed \$d | tr -s '\012' '\057')

# Where is mycroft installed?
if [[ -f "${RUNDIR}/mycroft-service.screen" && -f "${RUNDIR}/../mycroft/__init__.py" ]] ; then
    MYCROFT_HOME=$(cd "${RUNDIR}" && cd .. && pwd ) 
else
    if [[ -f "/opt/mycroft/mycroft/__init__.py" ]] ; then
        MYCROFT_HOME="/opt/mycroft/"
    else
        mlog " = Error: Having some difficulty in guessing the home Mycroft directory?"
        exit 200
    fi
fi

mlog " - Mycroft appears to be running in ${MYCROFT_HOME}"


while [[ $# -gt 0 ]] ; do
    opt="$1";
    shift;
    case "$opt" in
        "-v" ) checkversion ;;
        "-P") checkPIP ;;
        "-p") checkfiles ;;
        "-i") checktubes ;;
        "-s") checksysinfo ;;
        "-u") checkpulse;;
        "-r") checkrunning;;
        "-m") checkmimic;;
        "-a") checkversion ; checkrunning; checkPIP; checkfiles; checktubes; checksysinfo; checkpulse; checkmimic ;;
        *) helpfunc; exit 1;;
   esac
done

exit 0