mage2docker.plugin.zsh

# Mage2Docker
#
# Plugin for Oh-My-Zsh
_m2d_commands() {
	ids=("logs"
		"login"
		"root"
		"magento"
		"mage"
		"mage-cache"
		"mage-reindex"
		"mage-di"
		"mage-upgrade"
		"mage-report"
		"mage-log"
		"grunt"
		"rename"
		"rm"
		"restart"
		"stop"
		"inspect"
		"top"
		"mysqldump"
		"mysql"
		"ip"
		"vst"
		"varnish-purge"
		"redis-flushall"
		"nginx-reload"
		"stop-all"
	)

	echo ${ids[*]// /}
}

_zshm2d_usage() {
	print "Usage: m2d [containerName] [command] \n $(_m2d_commands) \n" |
		fold -s -w $COLUMNS >&2
}

_docker_get_container_name() {
	docker ps | awk '{if(NR>1) print $NF}'
}

_mage2docker_magento() {
	compadd $(docker exec $1 bin/magento list | sed 's/\x1b\[[0-9;]*m//g' | awk '{if(NR > 15 && /:/) print $1}')
}

_mage2docker_mage() {
	docker exec -it -u 1000 $1 bin/magento $2
}

_mage2docker_container_ip() {
	docker inspect -f '{{range .NetworkSettings.Networks}}{{.IPAddress}}{{end}}' $1
}

_mage2docker_report() {
	compadd $(docker exec -u 33 $1 ls -tr var/report)
}

_mage2docker_log() {
	compadd $(docker exec -u 33 $1 ls -tr var/log)
}

_shell_type_finder() {
  docker exec -i $1 cat /etc/passwd | grep root | awk -F: '{print $7}'| awk 'NR==1{print $1}'
}

_mage2docker_mysql_data() {
	printf '%s ' 'User name:'
	read user
	printf '%s ' 'User password:'
	read password
	printf '%s ' 'Database name:'
	read database
	printf '%s ' 'file name:'
	read file
}

_mage2docker() {

	local curcontext="$curcontext" state line
	typeset -A opt_args

	_arguments \
		'1: :->containerName' \
		'2: :->command' \
		'3: :->options'

	case $state in
	containerName)
		compadd $(_docker_get_container_name)
		;;
	command)
		compadd "$@" $(_m2d_commands)
		;;
	options)
		case $words[3] in
		mage)
			_mage2docker_magento $words[2]
			;;
		mage-report)
			_mage2docker_report $words[2]
			;;
		mage-log)
			_mage2docker_log $words[2]
			;;
		esac
		;;
	esac
}

mage2docker() {
	case $2 in
	restart | stop | inspect | rm | rename | top)
		echo cmd=$2
		docker $2 $1
		;;
	logs)
		docker logs -f $1
		;;
	root)
		docker exec -it -e LINES=$(tput lines) -e COLUMNS=$(tput cols) -u 0 $1 $(_shell_type_finder $1) -l
		;;
	login)
		docker exec -it -e LINES=$(tput lines) -e COLUMNS=$(tput cols) -u 1000 $1 $(_shell_type_finder $1) -l
		;;
	magento)
		docker exec -it -u 33 $1 bin/magento
		;;
	mage)
		_mage2docker_mage $1 $3
		;;
	mage-cache)
		_mage2docker_mage $1 cache:clean
		;;
	mage-reindex)
		_mage2docker_mage $1 indexer:reindex
		;;
	mage-upgrade)
		_mage2docker_mage $1 setup:upgrade
		;;
	mage-di)
		_mage2docker_mage $1 setup:di:compile
		;;
	mage-deploy)
		_mage2docker_mage $1 setup:static-content:deploy
		;;
	grunt)
		docker exec -it -u 1000 $1 grunt $3
		;;
	mage-report)
		docker exec -it $1 cat var/report/$3
		;;
	redis-flushall)
		docker exec -it $1 redis-cli flushall
		;;
	vst)
		docker exec -it -e LINES=$(tput lines) -e COLUMNS=$(tput cols) -u 0 $1 varnishstat
		;;
	varnish-purge)
		docker exec -it $1 varnishadm "ban req.url ~ /"
		;;
	nginx-reload)
		docker exec $1 nginx -s reload
		;;
	mage-log)
		docker exec -it -u 33 $1 tail -f var/log/$3
		;;
		#new informations
	ip)
		_mage2docker_container_ip $1
		;;
	mysqldump)
		_mage2docker_mysql_data
		docker exec $1 /usr/bin/mysqldump -u $user --password=$password $database > $file.sql
		echo "Success database backup was created"
		;;
	mysql)
		_mage2docker_mysql_data
		docker exec $1 /usr/bin/mysql -u $user --password=$password $database < $file.sql
		echo "Success database restore"
		;;
	help)
		_zshm2d_usage
		;;
	stop-all)
		docker stop $(docker ps -qa)
		;;
	*)
		if [ ! "$1" ]; then
			_zshm2d_usage
			docker ps
		else
			docker exec -it -e LINES=$(tput lines) -e COLUMNS=$(tput cols) -u 1000 $1 bash -l
		fi
		;;
	esac

}

if type "docker" >/dev/null; then

	compdef _mage2docker mage2docker
	alias m2d='mage2docker'
	alias stats_m2d='docker stats $(docker inspect -f '{{.Name}}' $(docker ps -q) | cut -c 2-)'
else
	echo "mage2docker - docker is not installed"
fi