#!/bin/bash -

#===============================================================================

#

#          FILE: shell_pack.sh

#

#         USAGE: ./shell_pack.sh

#

#   DESCRIPTION:

#

#       OPTIONS: ---

#  REQUIREMENTS: ---

#          BUGS: ---

#         NOTES: ---

#        AUTHOR: lwq (), scue@vip.qq.com

#  ORGANIZATION:

#       CREATED: // :: PM CST

#      REVISION:  ---

#===============================================================================

#===  FUNCTION  ================================================================

#         NAME:  usage

#  DESCRIPTION:  Display usage information.

#===============================================================================

function usage ()

{

        cat <<- EOT

  Usage :  $ -p package -s script file1 file2 file3 ..

  Options:

  -h|help       Display this message

  -p|package    The output package name

  -s|script     The script will run when unpack package

  Other         The all files what you want to pack

EOT

}    # ----------  end of function usage  ----------

#-----------------------------------------------------------------------

#  Handle command line arguments

#-----------------------------------------------------------------------

while getopts ":hp:s:" opt

do

  case $opt in

    h|help    ) usage; exit    ;;

    p|package ) package_name=$OPTARG ;;

    s|script  ) install_script=$OPTARG ;;

    \?        ) echo -e "\n  Option does not exist : $OPTARG\n"

          usage; exit    ;;

  esac    # --- end of case ---

done

shift $(($OPTIND-))

if [[ -z $package_name ]]; then

    echo "package_name can't not be empty"

    usage

    exit

fi

if [[ -z $package_name ]]; then

    echo "install_script can't not be empty"

    usage

    exit

fi

files=$@

generate_wrapper_script(){

    local install_script=$

    local wrapper_script=$

    cat <<-'EOT' >$wrapper_script

#!/bin/sh

echo "begin ..."

unpackdir=/tmp/$(basename $)_unpack

rm -rf $unpackdir >/dev/null

mkdir -p $unpackdir

echo "unpacking ..."

sed '1, /^#__SCRIPTEND__/d' $ | tar zxf - -C $unpackdir

if [ $? -ne  ]; then

    echo "unpack package failed."

    exit

fi

echo ""

echo "installing ..."

cd $unpackdir

EOT

    cat <<-EOR >>$wrapper_script

chmod +x $install_script

./$install_script

EOR

    cat <<-'EOE' >>$wrapper_script

if [ $? -ne  ]; then

    echo "install failed."

    exit

elif [[ -d $unpackdir ]]; then

    rm -rf $unpackdir

fi

echo "install ok, enjoy!"

exit

#__SCRIPTEND__

EOE

}

tarfile=package_content_$$.tgz

wrapfile=wrap_$$.sh

echo -e "start packing ..\n"

tar zcvf $tarfile $files $install_script

generate_wrapper_script $install_script $wrapfile

cat $wrapfile $tarfile > $package_name

chmod +x $package_name

echo -e "\noutput: $package_name\n"

rm -f $tarfile

rm -f $wrapfile