#!/usr/bin/env python
# Bot Wrapper by neersighted
import sys
import os
import subprocess
import re

usage = "usage: ./cloudbot {start|stop|restart|status}"
quit = "Goodbye! Thanks for using CloudBot!"

pwd = os.getcwd()
clearlog = ": >./bot.log && "
command = ":"

daemoncheck = subprocess.check_output("locate /usr/bin/daemon", shell=True)
daemon = re.match(r'^/usr/bin/daemon$', daemoncheck)
screencheck = subprocess.check_output("locate /usr/bin/screen", shell=True)
daemon = re.match(r'^/usr/bin/screen$', screencheck)

error1 = "Neither screen nor daemon is installed! "\
        "This script cannot run! {ERROR 1}"
error2 = "Could not find bot.py! Are you in the wrong folder? "\
        "(" + pwd + ") {ERROR 2}"
error3 = "Invalid choice, exiting! {ERROR 3}"
error4 = "Program killed by user! {ERROR 4}"
error5 = "Author error! We be derpin'! {ERROR 5}"

start = "echo " + error1
stop = "echo " + error1
restart = "echo " + error1
status = "ps aux|grep cloudbot|grep -v grep|grep -v ./cloudbot"

if daemon:
    start = "daemon -n cloudbot -O " + pwd + \
            "/bot.log -r python " + pwd + "/bot.py"
    stop = clearlog + "daemon -n cloudbot --stop"
    restart = stop + " && " + start
elif screen:
    start = "screen -S cloudbot -dm python" + pwd +\
            "/bot.py >>" + pwd + "/bot.log 2>&1"
    stop = clearlog + "kill `pidof /usr/bin/screen`"
    restart = stop + " && " + start
else:
    print error1

print "  ______  __        ______    __    __  "\
" _______  .______     ______   .___________."
print " /      ||  |      /  __  \  |  |  |  | "\
"|       \ |   _  \   /  __  \  |           |"
print "|  ,----'|  |     |  |  |  | |  |  |  | "\
"|  .--.  ||  |_)  | |  |  |  | `---|  |----`"
print "|  |     |  |     |  |  |  | |  |  |  | "\
"|  |  |  ||   _  <  |  |  |  |     |  |     "
print "|  `----.|  `----.|  `--'  | |  `--'  | "\
"|  '--'  ||  |_)  | |  `--'  |     |  |     "
print " \______||_______| \______/   \______/  "\
"|_______/ |______/   \______/      |__|     "
print "http://git.io/cloudbot                  "\
"                                 by lukeroge"
try:
    if (len(sys.argv) > 1):
        if (sys.argv[1] == 'start'):
            command = start
        elif (sys.argv[1] == 'stop'):
            command = stop
        elif (sys.argv[1] == 'restart'):
            command = restart
        elif (sys.argv[1] == 'status'):
            command = status
        else:
            print usage
            exit
    else:
        print "{1|start} {2|stop} {3|restart} {4|status} {5|exit}"
        read = int(raw_input('Please choose a option: '))
        if read == 1:
            command = start
        elif read == 2:
            command = stop
        elif read == 3:
            command = restart
        elif read == 4:
            command = status
        elif read == 5:
            exit
        else:
            print error3
            exit
except (TypeError, ValueError):
    print error3
    exit
except (KeyboardInterrupt):
    print error4
    exit
except (NameError, SyntaxError):
    print error5
    exit

if os.path.isfile("./bot.py"):
    command = command
else:
    print error2
    exit

subprocess.call(command, shell=True)
print quit
exit