#!/usr/bin/env bash

# pip plugin for KYMSU
# https://github.com/welcoMattic/kymsu

# upgrade pip:
# python -m pip install --upgrade pip

#export PIP_USE_FEATURE=2020-resolver
#export PIP_USE_FEATURE=fast-deps
# pip config edit

#########################################
#
# Settings:

#version: pip ou pip3
pip_version=pip3
#user: "" or "--user"
user=""
# No distract mode
no_distract=false
#add module to do_not_update array
declare -a do_not_update=()
#
#########################################

if [[ $1 == "--nodistract" ]]; then
	no_distract=true
fi

if ! [ -x "$(command -v $pip_version)" ]; then
	echo "Error: $pip_version is not installed." >&2
	exit 1
fi

echo -e "\033[1m🐍  $pip_version (Python 3) \033[0m"

echo ""
echo -e "\033[31mpip use features 2020-resolver & fast-deps for testings. pip 20.3 (10/2020) pip definitively this features.\033[0m"

echo ""
$pip_version install --upgrade pip
echo ""

pip_outdated=$($pip_version list --outdated --format columns)
upd=$(echo "$pip_outdated" | sed '1,2d' | awk '{print $1}')

if [ -n "$upd" ]; then

	echo -e "\033[4mAvailables updates:\033[0m"
	#echo $pip3_outdated_freeze | tr [:space:] '\n'
	echo "$pip_outdated"
	echo ""
	
	for i in $upd
		do
			info=$($pip_version show "$i")
			#info=$($pip_version show $i | sed -n 4q)
			#info=$($pip_version show $i | head -5)
			#info=$($pip_version show $i | tail -n +5)
			echo "$info" | head -4
			echo ''
			#echo "$i"
		done

	if [ -x "$(command -v pipdeptree)" ]; then
		echo -e "\033[4mCheck dependancies:\033[0m"
		echo "Be carefull!! This updates can be a dependancie for some modules. Check for any incompatible version."
	fi
	echo ""
	for i in $upd
		do
			if [ -x "$(command -v pipdeptree)" ]; then
				dependencies=$(echo "$i" | xargs pipdeptree -r -p | grep "$upd")

				z=0
				while read -r line; do
					if [[ $line = *"<"* ]]; then
						echo -e "\033[31m$line\033[0m"
					else
						if [ "$z" -eq 0 ]; then
							echo -e "\033[3m$line\033[0m"
						else
							echo "$line"
						fi
						z=$((z+1))
					fi
				done <<< "$dependencies"
				
			else
				c=$(echo -e "Do you want to install pipdeptree to check dependancies ? (y/n)")
  				read -pr "$c" choice
  				case "$choice" in
    				y|Y|o ) $pip_version install $user pipdeptree ;;
    				n|N ) echo "Ok, let's continue";;
    				* ) echo "invalid";;
  				esac
			
			fi
			
			# If the update is not in the do_not_update array, we install it.
			
			FOUND=`echo ${do_not_update[*]} | grep "$i"`
			if [ "${FOUND}" = "" ] && [ "$no_distract" = false ]; then
			
				b=$(echo -e "Do you wanna run \033[1m$pip_version install $user --upgrade $i\033[0m ? (y/n)")
  				read -p "$b" choice
  				case "$choice" in
    				y|Y|o ) echo "$i" | xargs $pip_version install $user --upgrade ;;
    				n|N ) echo "Ok, let's continue";;
    				* ) echo "invalid";;
  				esac
  				echo ""

			elif  [ "${FOUND}" = "" ]; then
			
				echo "$i" | xargs $pip_version install $user --upgrade
				#echo "$i" | xargs $pip_version install $user --use-feature=2020-resolver --upgrade
				
			fi			
		done

else
	echo -e "\033[4mNo availables updates.\033[0m"
fi


echo -e "🐍  Running \033[1mpip check\033[0m for checking that everything is ok."

$pip_version check

echo ""
echo ""