#!/bin/bash

# NAME
#	strip-docker-image-export  - exports the bare essentials from a Docker image
#
# SYNOPSIS
#	strip-docker-image-export  [-d export-dir ] [-p package | -f file] [-v]
#
#
# OPTIONS
#	-d export-directory	to copy content to, defaults to /export.
#	-p package		package to include from image, multiple -p allowed.
#	-f file			file to include from image, multiple -f allowed.
#	-v			verbose
#
# DESCRIPTION
#   	this script copies all the files from an installed package and copies them
#	to an export directory. Additional files can be added. When an executable
#	is copied, all dynamic libraries required by the executed are included too.
#
# EXAMPLE
#	The following example strips the nginx installation from the default NGiNX docker image,
#	and allows the files in ./export to be added to a scratch image.
#
#        docker run -v $PWD/export/:/export \
#		   -v $PWD/bin:/mybin nginx \
#		/mybin/strip-image.sh \
#			-p nginx  \
#			-f /etc/passwd \
#			-f /etc/group \
#			-f '/lib/*/libnss*' \
#			-f /bin/ls \
#			-f /bin/cat \
#			-f /bin/sh \
#			-f /bin/mkdir \
#			-f /bin/ps \
#			-f /var/run \
#			-f /var/log/nginx \
#			-d /export
# CAVEATS
#	requires an image that has a bash, readlink and ldd  installed.
#
# AUTHOR
#  Mark van Holsteijn
#
# COPYRIGHT
#
#   Copyright 2015 Xebia Nederland B.V.
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
export EXPORT_DIR=/export

function usage() {
	echo "usage: $(basename $0) [-v] [-d export-dir ] [-p package | -f file]" >&2
	echo "	$@" >&2
}

function parse_commandline() {

	while getopts "vp:f:d:" OPT; do
	    case "$OPT" in
		v)
		    VERBOSE=v
		    ;;
		p)
		    PACKAGES="$PACKAGES $OPTARG"
		    ;;
		f)
		    FILES="$FILES $OPTARG"
		    ;;
		d)
		    EXPORT_DIR="$OPTARG"
		    ;;
		*)
		    usage
		    exit 1
		    ;;
	    esac
	done
	shift $((OPTIND-1))

	if [ -z "$PACKAGES" -a -z "$FILES" ] ; then
		usage "Missing -p or -f options"
		exit 1
	fi
	if [ ! -d $EXPORT_DIR ] ; then
		usage "$EXPORT_DIR is not a directory."
		exit 1
	fi
}

function print_file() {
		if [ "$1" = "/usr/bin/ldd" ]; then
			exit
		fi

		if [ -e "$1" ] ; then
			echo "$1"
		else
			test -n "$VERBOSE" && echo "INFO: ignoring not existent file '$1'" >&2
		fi

		if [ -s "$1" ] ; then
			TARGET=$(readlink "$1")
			if  [ -n "$TARGET" ] ; then
				if expr "$TARGET" : '^/' >/dev/null 2>&1 ; then
					list_dependencies "$TARGET"
				else
					list_dependencies $(dirname "$1")/"$TARGET"
				fi
			fi
		fi
}

function list_dependencies() {
	for FILE in $@ ; do
		if [ -e "$FILE" ] ; then
			print_file "$FILE"
			if /usr/bin/ldd "$FILE" >/dev/null 2>&1 ; then
				/usr/bin/ldd "$FILE" | \
				awk '/statically/{next;} /=>/ { print $3; next; } { print $1 }' | \
				while read LINE ; do
					test -n "$VERBOSE" && echo "INFO: including $LINE" >&2
					print_file "$LINE"
				done
			fi
		else
			test -n "$VERBOSE" && echo "INFO: ignoring not existent file $FILE" >&2
		fi
	done
}

function list_packages() {
		if test -e /usr/bin/dpkg; then
			DEPS=$(/usr/bin/dpkg -L $1)
		elif test -e /usr/bin/rpm; then
			DEPS=$(/usr/bin/rpm -ql $1)
		elif test -e /sbin/apk; then
			DEPS=$(/sbin/apk info -L $1 | grep -Ev '^$|contains:' | sed 's/^/\//g')
		else
			echo 'WARN: Unknown OS, aborted list_packages()'
			exit
		fi
		while read FILE ; do
			if [ ! -d "$FILE" ] ; then
				list_dependencies "$FILE"
			fi
    done <<< "$DEPS"
}

function list_all_packages() {
	for i in "$@" ; do
		list_packages "$i"
	done
}

parse_commandline "$@"


tar czf - $(

	(
	list_all_packages $PACKAGES
	list_dependencies $FILES
	)  | sort -u

) | ( cd $EXPORT_DIR ; tar -xzh${VERBOSE}f - )