#!/bin/bash

echo "Checking the environment..."
unameOut="$(uname -s)"
case "${unameOut}" in
    Linux*)     machine=linux;;
    Darwin*)    machine=mac;;
    CYGWIN*)    machine=cygwin;;
    MSYS*)      machine=msys;;
    MINGW32*)   machine=mingw32;;
    MINGW64*)   machine=mingw64;;
    *)          machine=unknown;;
esac
if [ "${machine}" = "unknown" ] ; then
	echo "* Warning! Unknown environment: ${unameOut}"
else
	echo "* ${machine} environment detected!"
fi

# $1 = package_name
function install_package {
	# check if the installer (package manager) exists
	if [ "${machine}" = "cygwin" ] ; then
		while true; do
			if [ ! -x "$(command -v apt-cyg --version)" ] ; then
				# we give it one more chance and try to download the installer
				echo "Do you want to download the cygwin package manager (apt-cyg) and install the required utilities? (Press 1 or 2)"
				select yn in "Yes" "No"; do
				    case $yn in
	        			Yes )
	        				url="rawgit.com/transcode-open/apt-cyg/master/apt-cyg"
	        				dst="/tmp/apt-cyg"
							# check if wget or curl is installed
							if [ -x "$(command -v wget)" ] ; then
								echo "Downloading apt-cyg using wget..."
								wget $url -O $dst
							elif [ -x "$(command -v curl)" ]; then
								echo "Downloading apt-cyg using curl..."
								curl -o $dst -sfL $url
							else
								echo "Could not find curl or wget! Cannot download and install a package manager!" >&2
								return
							fi
	        				install $dst /bin
	        				rm $dst
	        				break;;
	        			No ) 
							echo "Please install it manually using you package manager!" >&2
	        				exit 1;
					esac
				done
			else
				break
			fi
		done
	elif [ "${machine}" = "msys" ] ; then
		if [ ! -x "$(command -v pacman)" ] ; then
			echo "Cannot detect pacman manager. Please install it manually using you package manager!" >&2
        	exit 1;
		fi
	fi

	# now install
	echo "Do you want to install '$1' now? (Press 1 or 2)"
	select yn in "Yes" "No"; do
	    case $yn in
			Yes )
	        	break;;
			No ) 
				echo "Please install it manually using your package manager!" >&2
	        	exit 1;
		esac
	done

	if [ "${machine}" = "cygwin" ] ; then
		apt-cyg install $1
	elif [ "${machine}" = "msys" ] ; then
		pacman -S mingw-w64-x86_64-$1
	elif [ "${machine}" = "linux" ] ; then
		sudo apt-get install $1
	fi
}

echo "Checking the Python version..."
# check python version - should be 2.x ONLY
python_bin="python2.7"
python_ver=$($python_bin -V 2>&1 | grep -Po '(?<=Python )(.+)')
if [[ -z "$python_ver" ]] || [[ ! $python_ver =~ ^2\.7.* ]] ; then
    echo "Error! Python 2.7.x is required. It should be installed and added to the PATH!"
    install_package python2
fi

# $1 = name, $2 = library name, $3 = package name
function check_library {
	echo "* Checking $1..."
	while true; do
		lib=$(pkg-config --libs $2 2>&1 | grep -Po '(\-l'$1')')
		if [[ -z "$lib" ]] ; then
			echo "Error! Library $1 NOT FOUND!"
		else
			echo "* Library $1 FOUND!"
			return
		fi
		install_package $3
	done
}

echo "* Python $python_ver detected!"

echo "Updating git submodules for scripts..."
git submodule update --init -- bin/gerbmerge bin/python-combine-pdfs bin/InteractiveHtmlBom bin/pcb-tools

echo "Checking the Python modules..."
declare -A modules
modules[simpleparse]=simpleparse
modules[contextlib2]=contextlib2
modules[PyPDF2]=PyPDF2
modules[gerber]=gerber
modules[configparser]=configparser
modules[gzip]=gzip
modules[cairocffi]=cairocffi

# check modules
for module in "${!modules[@]}"; do
	pymodule=${modules[$module]}
	while true; do
		$python_bin -c "import sys, pkgutil; sys.path.append('./bin/pcb-tools'); sys.exit(0 if (pkgutil.find_loader('$module')) else 1)"
		if [ $? -eq 0 ]; then
		    echo "* Checking Python module '$pymodule': OK"
		    break
		else
		    echo "* Checking Python module '$pymodule': ERROR!"
		    echo "  Python module '$pymodule' is required and NOT found!"
		    
		    # some modules have dependencies
			if [ "$pymodule" = "cairocffi" ]; then
				check_library ffi libffi libffi-devel
				check_library cairo cairo libcairo-devel
			fi

		    if [ ]; then
		    	echo "Please use 'pip2 install $pymodule' to install it manually!"
		    	exit 1;
		    else
				echo "Do you want to download and install it now? (Press 1 or 2)"
				select yn in "Yes" "No"; do
				    case $yn in
	        			Yes ) pip2 install $pymodule; break;;
	        			No ) exit 1;
					esac
				done
			fi
		fi
	done
done

echo "Checking if Node.js is installed..."
node_bin="node"
node_ver=$($node_bin -v 2>&1 | grep -Po '(v[0-9]+.*)')
if [[ -z "$node_ver" ]] ; then
    echo "Error! This script requires Node.Js installed in PATH!"
    echo "Please download and install it from here: https://nodejs.org/en/download/"
    exit 1
fi

echo "* Node.js $node_ver detected!"

echo "Checking Node.js packages..."
pushd ./bin/render_vrml > /dev/null
for package in 'puppeteer' 'pngjs' 'fs' 'zlib'; do
	if [ `npm list --depth=0 | grep -c "\-\- ${package}"` -eq 1 ]; then
	    echo "* Checking Node.js module '$package': OK"
	else
	    echo "* Checking Node.js module '$package': ERROR!"
	    echo "  The module '$package' is required and NOT found! Please use 'npm install $package' to install it"
		echo "Do you want to download and install it now? (Press 1 or 2)"
		select yn in "Yes" "No"; do
		    case $yn in
	       		Yes ) npm install $package --no-shrinkwrap; break;;
	       		No ) exit 1;
			esac
		done
	fi
done
popd > /dev/null

echo "All checks done!"

exit 0