#!/usr/bin/env python3
import time
import sys
import platform
import os
# import urllib.request
import urllib3
import requests
import http.client as http_client
import argparse
import json
import signal
import subprocess
from xml.etree import ElementTree as etree
from time import gmtime, strftime

def signal_handler(signal, frame):
        print('Exit...')
        sys.exit(0)

def sendmessage(message):
        global ran
        # mymessage=hostname.upper() + ': ' +message
        mymessage = ("%s on %s" % (message, hostname.upper()))

        urllib3.disable_warnings()
        # http_client.HTTPConnection.debuglevel = 1

        url = ("%s/webhook/%s" % (baseurl, webhook))

        r = requests.post(url, json = { 'message': mymessage })
        if r.status_code == 200:
            print("Message has been sent successfully")
            # print(message)
        else:
            print("Message sending FAILED (%s)" % (r.status_code))
            # print(r.data)

        ran = True


__version__ = "1.4.0"
versionstring='TellMe v' + __version__

signal.signal(signal.SIGINT, signal_handler)

hostname=platform.node()
home=os.getenv("HOME")

parser = argparse.ArgumentParser(description='TellMe command line client')
parser.add_argument("-m", "--message", default="Your process finished", help="Message to send")
parser.add_argument('-v', '--version', action='version', version=versionstring)
parser.add_argument('-p', '--pid', action="store", dest="pid", type=int, default=0)
parser.add_argument('-i', '--interval', action="store", dest="interval", type=int, default=5, help="Set the interval of commands like watch that use it")
parser.add_argument('-w', '--watch', action="store", dest="watchcommand", help="Run the given command every n or 5 minutes")
parser.add_argument('-c', '--config', default=home +"/.config/tellme/config.json" ,help="Path to config file")
parser.add_argument('-P', '--ping', action="store", dest="pinghost", type=str, help="Ping a host until it is up")
#print parser.parse_args()
args = parser.parse_args()

configfile = args.config
# Read config
if not os.path.isfile(configfile):
	print("No config file found in %s" %(configfile))
	exit(1)

config = None
with open(configfile) as cfgfile:
    config = json.load(cfgfile)

baseurl = config['url']
webhook = config['webhook']

message = args.message

ran = False

if args.pid != 0:
	path = "/proc/%i" %(args.pid)
	print("Monitoring process %i" %(args.pid))
	while os.path.exists(path):
		time.sleep(10)

	sendmessage(message)

if(args.watchcommand):
    print("Watching %s every %i minutes" %(args.watchcommand,int(args.interval)))
    while True:
        output = subprocess.Popen(args.watchcommand, shell=True, stdout=subprocess.PIPE).stdout.read()
        print(output)
        sendmessage("|%s|" %(output))
        time.sleep(60*int(args.interval))

if(args.pinghost):
    print("Pinging %s" %(args.pinghost))
    response = None

    while response != 0:
        response = os.system("ping -c 1 -w 1 %s >/dev/null" % (args.pinghost))
        if response == 0:
            print("Up")
            sendmessage("Host %s is Up" % (args.pinghost))
        else:
            print("Down")

            if args.interval:
                interval = args.interval
            else:
                interval = 60

            time.sleep(int(args.interval))

if ran is False:
    sendmessage(message)