#! /usr/bin/env python3
import time
import sys
import os
import subprocess
from subprocess import call 

def callWithShell(cmd):
    process = subprocess.Popen(cmd, shell=True)
    process.wait()
    #call(cmd,shell=True)

#if an argument is being passed @ 1, then try to set as _SUDO
_SUDO= ''
try:
    _SUDO=sys.argv[1]
except Exception:
    pass

# exit on non-zero return
callWithShell("set -e")


#BUILD_TYPE="AUTOTOOLS"
#BUILD_TYPE="CMAKE"
BUILD_TYPE="HEADER_ONLY"

LIBRARY_NAME="cppzmq"
LIBRARY_FOLDER_NAME="cppzmq"
SOURCE_ARCHIVE_FILE="v4.2.2.zip"
SOURCE_ARCHIVE_ADDRESS="https://github.com/zeromq/cppzmq/archive/"
SOURCE_FOLDER_NAME="cppzmq-4.2.2"



print("Making Dirs")
sys.stdout.flush()
CWD = os.getcwd()
callWithShell("mkdir -p ./{}".format(LIBRARY_FOLDER_NAME))
os.chdir(LIBRARY_FOLDER_NAME)

print("The current directory is {}".format(os.getcwd()))
sys.stdout.flush()
if (os.path.isfile(SOURCE_ARCHIVE_FILE)):
    print("*** {}:: Archive File ({}) Exists, Skipping Source Fetch! ***".format(LIBRARY_NAME,SOURCE_ARCHIVE_FILE))
    sys.stdout.flush()
else:
    print("Fetching Source")
    sys.stdout.flush()
    callWithShell("wget {}{}".format(SOURCE_ARCHIVE_ADDRESS, SOURCE_ARCHIVE_FILE))

print( "Unpacking..." )
sys.stdout.flush()
callWithShell("unzip -o {}".format(SOURCE_ARCHIVE_FILE))

# change to the source directory
os.chdir(SOURCE_FOLDER_NAME)

print("Building...")
sys.stdout.flush()

if(BUILD_TYPE == "CMAKE"):
    callWithShell("mkdir -p ./build")
    os.chdir("./build")
    callWithShell("make -j8; make")
    print("Installing...")
    sys.stdout.flush()
    callWithShell("{} make install".format(_SUDO))
elif(BUILD_TYPE == "AUTOTOOLS"):
    callWithShell("./autogen.sh")
    callWithShell("./configure && make check")
    print("Installing...")
    sys.stdout.flush()
    callWithShell("{} make install".format())
    callWithShell("{} ldconfig".format(_SUDO))
elif(BUILD_TYPE == "HEADER_ONLY"):
    callWithShell("{} cp ./zmp.hpp /usr/local/include/zmq.hpp".format(_SUDO))
else:
    print("!!! UNKNOWN BUILD TYPE [{}]".format(BUILD_TYPE))
    sys.stdout.flush()

print("Cleaning up...")
sys.stdout.flush()
os.chdir(CWD)

# uncomment the following line to remove source code
#callWithShell("rm -rf ./{}".format(LIBRARY_FOLDER_NAME))

print("Finished!")
sys.stdout.flush()