#!/bin/python

# check Cowthink in Python

import colorama, os, sys, subprocess, traceback
from colorama import Fore, Style
from datetime import datetime
colorama.init()


def checkOs():
    print("Started checking operating system at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))

    if sys.platform == "win32": 
        print(Fore.GREEN + "Operating System:", end=""); sys.stdout.flush()
        os.system('ver')
        print(Style.RESET_ALL, end="")
        operatingSystem = "Windows"

    elif sys.platform == "darwin": 
        print(Fore.GREEN + "Operating System: ")
        os.system('sw_vers')
        print(Style.RESET_ALL, end="")
        operatingSystem = "macOS"

    elif sys.platform == "linux": 
        print(Fore.GREEN + "Operating System: ")
        os.system('uname -r')
        print(Style.RESET_ALL, end="")
        operatingSystem = "Linux"

    print("Finished checking operating system at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))

    print("")
    return operatingSystem


def checkCowthink(): 
    print("\nCheck Cowthink in Python.\n")
    operatingSystem = checkOs()

    try:
        startDateTime = datetime.now()
        print("Started checking Cowthink at", startDateTime.strftime("%m-%d-%Y %I:%M %p"))

        FNULL = open(os.devnull,  'w')

        if operatingSystem == "macOS" or operatingSystem == "Linux":

            checkCowthinkOnMacOrLinux = subprocess.call(['which', 'cowthink'], stdout=FNULL)

            if checkCowthinkOnMacOrLinux == 0:
                os.system('cowthink "Cowthink is installed"')
                print(Fore.GREEN + "Successfully checked Cowthink." + Style.RESET_ALL)

                finishedDateTime = datetime.now()

                print("Finished checking Cowthink at", finishedDateTime.strftime("%m-%d-%Y %I:%M %p"))

                duration = finishedDateTime - startDateTime
                print("Total execution time: {0} second(s)".format(duration.seconds))
                print("")

            else: 
                raise Exception("Cowthink is not installed.")

        elif operatingSystem == "Windows": 
            checkCowthinkOnWindows = subprocess.call(['where', 'cowthink'], stdout=FNULL) 
            
            if checkCowthinkOnWindows == 0:
                os.system('cowthink "Cowthink is installed"')
                print(Fore.GREEN + "Successfully checked Cowthink." + Style.RESET_ALL)

                finishedDateTime = datetime.now()

                print("Finished checking Cowthink at", finishedDateTime.strftime("%m-%d-%Y %I:%M %p"))

                duration = finishedDateTime - startDateTime
                print("Total execution time: {0} second(s)".format(duration.seconds))
                print("")
                
            else: 
                raise Exception("Cowthink is not installed.")

    except Exception: 
        print(Fore.RED + "Failed to check Cowthink in Python.")
        traceback.print_exc()
        exit("" + Style.RESET_ALL)


checkCowthink()