icon_fits

#!/usr/bin/env bash
#
# App icon fitter
# A tool for creating images for application icons.
# Create by Poplax [linjiang9999<at>gmail.com]
# Date 2018-02-06
#

set -eu

declare -r TOOLNAME='AppIconFitter'
declare -r VERSION='0.1.0'
declare -r EXTENSION='.png'

err() {
	local red="\033[1;31m"
	local normal="\033[0m"

	local err_text
	err_text="$1"

	echo -e "<${red}Error${normal}> ${err_text}" >&2
}

message() {
	local green="\033[1;32m"
	local normal="\033[0m"

	local message_text
	message_text="$1"

	echo -e "[${green}Message${normal}] ${message_text}"
}

lowcase() {
	local word
	word="$1"

	echo "${word}" | tr '[:upper:]' '[:lower:]'
}

make_dir() {
	local target_dir
	target_dir="$1"

	if [[ -z "${target_dir}" ]]; then
		err "Target dir can't be empty."
		exit -1
	fi

	if [[ ! -d "${target_dir}" ]]; then
		mkdir -p "${target_dir}"
	fi
}

check_option() {
	local opt
	local first

	opt="$1"
	first="$2"

	if [[ ${first:0:1} == '-' ]]; then

		err "\nOption -${opt}\nRequires argument error: can't start with '-'. "
		exit -1
	fi

}

usage() {
	cat <<USAGE_DOC
Version: ${TOOLNAME}-v${VERSION}
Copyright: © 2018 Poplax [linjiang9999<at>gmail.com]
License: MIT <https://opensource.org/licenses/MIT>
Usage: 
    $0 [OPTION ...] 

    -f	Image of App icon file, recommand size '1024x1024' px, Support png,jpeg,tiff
    -o	Output directory. default is current directory 'output'.
    -t	Specify subtitle text in the icon, default is none.

    -k	Auto create all size of App icon. note: only support 'iOS' platform, this version.
    -s	Specify size, note: Only create an icon of the specified size.

    -c  Compress files.
    -v	Software current version.
    -h	Help.

Example	:
    
    + Resize 120(px) of the icon(gray_1024.png), add subtitle : 'DemoApp' as watermark, create one icon to the 'output' directory:
    1> 
    $0 -s 120 -f gray_1024.png -o iconFitResize -t "DemoApp"
    2> 
    $0 -c -s 120 -o iconFitResize -t "DemoApp" gray_1024.png

    + create all size of the platform(default: iOS) support.
    1> 
    $0 -k -f blue_1024.png -o iconFit
    2> 
    $0 -co iconFitCompress blue_1024.png
    
USAGE_DOC
}

# Create watermark.
make_watermark() {
	# message "make watermark."

	local subtitle
	local image_file

	subtitle="$1"
	image_file="$2"

	local image_watermark="wmark_${image_file}"

	convert "${image_file}" -font GeorgiaB -pointsize 120 \
		-draw "gravity south fill white text 0,0 '${subtitle}' fill DarkGray text 0,-2 '${subtitle}'" \
		"${image_watermark}"

	if [[ -f "${image_watermark}" ]]; then
		echo "${image_watermark}"
	fi

}

# Make image.
make_image() {
	local size
	local input_file
	local output_dir
	local filename="default_file${EXTENSION}"

	size="$1"
	input_file="$2"
	output_dir="$3"

	if [[ "$#" -ge 4 ]]; then

		filename="$4"
	fi

	if [[ "${size}" -lt 10 || "${size}" -gt 2048 ]]; then
		err "Out of image size, min = 10, max = 2048."
		exit -1
	fi

	local output_file="${output_dir}/${filename}"
	if convert "${input_file}" -resize ${size} "${output_file}"; then
		message "Created ${output_file} ."
	fi
}

compress_files() {
	local file_ext
	local file_dir
	local file_name
	file_ext="$1"
	file_dir="$2"

	[[ "$#" -ge 3 ]] &&
		file_name="$3"

	if [[ "${file_ext}" == ".png" ]]; then
		message "Compress start ..."
		if [[ -n "${file_name}" ]]; then
			pngcrush -rem allb -brute -nofilecheck -blacken -reduce -ow ${file_name}
		else
			find ${file_dir} -name "*${file_ext}" |
				xargs -I {} pngcrush -rem allb -brute -nofilecheck -blacken -reduce -ow {}
		fi
		message "Compress done ."
	else
		# @TODO: Compress .jpg ...
		err "Compress Not support ${file_ext} yet."
	fi

}

make_ios_icons() {
	local origin_file
	local output_dir

	origin_file="$1"
	output_dir="$2"

	# Sizes value
	local app_store_1x="1024"
	make_image "${app_store_1x}" "${origin_file}" "${output_dir}" "iTunesArtwork${EXTENSION}"

	local suffix

	local iphone_3x="20 29 40 60"
	for m_size in ${iphone_3x}; do
		suffix=''
		for i in {1..3}; do
			[[ $i -gt 1 ]] && suffix="@${i}x"
			make_image "$((${m_size} * ${i}))" "${origin_file}" "${output_dir}" "iphone_${m_size}${suffix}${EXTENSION}"
		done
	done

	local ipad_2x="20 29 40 76"
	for m_size in ${ipad_2x}; do
		suffix=''
		for i in {1..2}; do
			[[ $i -gt 1 ]] && suffix="@${i}x"
			make_image "$((${m_size} * ${i}))" "${origin_file}" "${output_dir}" "ipad_${m_size}${suffix}${EXTENSION}"
		done
	done

	# iPad Pro
	local ipad_pro_size="167"
	make_image "${ipad_pro_size}" "${origin_file}" "${output_dir}" "ipad_pro_${ipad_pro_size}${EXTENSION}"

}

make_ios_config() {
	local config_file='Contents.json'

	local output_dir
	output_dir="$1"

	cat <<CONFIG_IOS >"${output_dir}/${config_file}"
{
  "images" : [
    {
      "size" : "20x20",
      "idiom" : "iphone",
      "filename" : "iphone_20@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "20x20",
      "idiom" : "iphone",
      "filename" : "iphone_20@3x.png",
      "scale" : "3x"
    },
    {
      "size" : "29x29",
      "idiom" : "iphone",
      "filename" : "iphone_29@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "29x29",
      "idiom" : "iphone",
      "filename" : "iphone_29@3x.png",
      "scale" : "3x"
    },
    {
      "size" : "40x40",
      "idiom" : "iphone",
      "filename" : "iphone_40@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "40x40",
      "idiom" : "iphone",
      "filename" : "iphone_40@3x.png",
      "scale" : "3x"
    },
    {
      "size" : "60x60",
      "idiom" : "iphone",
      "filename" : "iphone_60@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "60x60",
      "idiom" : "iphone",
      "filename" : "iphone_60@3x.png",
      "scale" : "3x"
    },
    {
      "size" : "20x20",
      "idiom" : "ipad",
      "filename" : "ipad_20.png",
      "scale" : "1x"
    },
    {
      "size" : "20x20",
      "idiom" : "ipad",
      "filename" : "ipad_20@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "29x29",
      "idiom" : "ipad",
      "filename" : "ipad_29.png",
      "scale" : "1x"
    },
    {
      "size" : "29x29",
      "idiom" : "ipad",
      "filename" : "ipad_29@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "40x40",
      "idiom" : "ipad",
      "filename" : "ipad_40.png",
      "scale" : "1x"
    },
    {
      "size" : "40x40",
      "idiom" : "ipad",
      "filename" : "ipad_40@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "76x76",
      "idiom" : "ipad",
      "filename" : "ipad_76.png",
      "scale" : "1x"
    },
    {
      "size" : "76x76",
      "idiom" : "ipad",
      "filename" : "ipad_76@2x.png",
      "scale" : "2x"
    },
    {
      "size" : "83.5x83.5",
      "idiom" : "ipad",
      "filename" : "ipad_pro_167.png",
      "scale" : "2x"
    },
    {
      "size" : "1024x1024",
      "idiom" : "ios-marketing",
      "filename" : "iTunesArtwork.png",
      "scale" : "1x"
    }
  ],
  "info" : {
    "version" : 1,
    "author" : "xcode"
  }
}
CONFIG_IOS

}

# main
main() {

	# Check Required.
	command -v convert >/dev/null 2>&1 ||
		{
			cat <<EOF
This Tool Depend On 
  'ImageMagick' [https://www.imagemagick.org/script/index.php].

OSX:
    Homebrew install:
    brew install imagemagick

Linux:
    yum install:
    yum install -y ImageMagick

EOF
			exit -1
		}

	# Check Argument
	if [[ $# -lt 1 ]]; then
		usage
		exit 1
	fi

	# Default kflag='true' in this version.
	local kflag='true'
	local sflag=''
	local fflag=''
	local tflag=''
	local oflag=''
	local cflag=''

	# Default value
	local size_value=''
	local file_value=''
	local text_value=''
	local kind_value='ios'
	local output_value='output'

	local OPTIND

	while getopts ":cvhks:f:t:o:" opt; do
		case ${opt} in
		k)
			# Output default image files.
			kflag='true'
			# TODO: Android.
			# kind_value=$(lowcase ${kind_value})
			# message "-${opt} ${kind_value}"
			;;

		s)
			# Resize.
			sflag='true'
			check_option "${opt}" "${OPTARG}"

			size_value="${OPTARG}"

			local re_int='^[0-9]+$'
			if [[ ! "${size_value}" =~ $re_int ]]; then
				err "Size option -${opt}, only accept +integer.\nyour input is '${size_value}'."
				exit 1
			fi
			# message "-${opt} ${OPTARG}"
			;;

		f)
			# Icon path.
			fflag='true'
			check_option "${opt}" "${OPTARG}"

			file_value="${OPTARG}"
			# message "-${opt} ${OPTARG}"
			;;

		t)
			# Watermark text.
			tflag='true'
			check_option "${opt}" "${OPTARG}"

			text_value="${OPTARG}"
			# message "-${opt} ${OPTARG}"
			;;

		o)
			# Output directory.
			oflag='true'
			check_option "${opt}" "${OPTARG}"

			output_value="${OPTARG}"

			make_dir ${output_value}
			# message "-${opt} ${OPTARG}"
			;;

		c)
			# PNG compress.
			cflag='true'

			command -v pngcrush >/dev/null 2>&1 ||
				{
					cat <<EOF
This Tool Depend On 
  'pngcrush' [https://pmt.sourceforge.io/pngcrush/]

OSX:
    Homebrew install:
    brew install pngcrush

Linux:
    yum install:
    yum install -y pngcrush

EOF
					exit -1
				}
			;;

		v)
			echo "${TOOLNAME}-v${VERSION}"
			exit 1
			;;

		h)
			usage
			exit 1
			;;

		:)
			err "Option -${OPTARG} requires an argument."
			usage
			exit 1
			;;

		?)
			err "Undefined Option -${OPTARG}"
			usage
			exit 1
			;;

		esac

	done
	shift $((${OPTIND} - 1))

	# Create Logic ----

	# 1> Handle file input.
	if [[ -z "${fflag}" && "$#" -eq 0 ]]; then
		err "No input file given."
		usage
		exit 1
	fi

	# file input default value.
	local input_file
	if [[ -z ${fflag} ]]; then
		input_file="$1"
	fi

	if [[ -z "${file_value}" ]]; then
		file_value="${input_file}"
	fi

	# 2> Handle subtitle text.
	if [[ -n "${tflag}" ]]; then
		local watermark_file
		watermark_file=$(make_watermark "${text_value}" "${file_value}")

		if [[ -z "${watermark_file}" ]]; then

			err "Test watermark make failed."
			exit 1
		fi

		file_value="${watermark_file}"
	fi

	# 3> Handle type of input type.
	if [[ -n "${sflag}" ]]; then
		make_dir "${output_value}"
		make_image "${size_value}" "${file_value}" "${output_value}" "resize_${size_value}${EXTENSION}"

		# Compress
		if [[ -n "${cflag}" ]]; then
			compress_files "${EXTENSION}" "${output_value}"
		fi

		local resize_file="${output_value}/resize_${size_value}${EXTENSION}"
		message "Create job done. ${resize_file}"
	else
		# @TODO: android.
		if [[ "${kind_value}" == "ios" ]]; then
			output_value="${output_value}/AppIcon.appiconset"
			make_dir "${output_value}"
			make_ios_icons "${file_value}" "${output_value}"
			make_ios_config "${output_value}"
		fi

		# Compress
		if [[ -n "${cflag}" ]]; then
			compress_files "${EXTENSION}" "${output_value}"
		fi

		message "Create job done. you can find files at '${output_value}' directory."
	fi

	# Temp text image need to delete.
	if [[ -n "${tflag}" ]]; then
		rm -f ${file_value}
	fi
}

main "$@"