#!/bin/bash

#

# createmodule.sh - Takes the name of a environment init script and 

# produces a modulefile that duplicates the changes made by the init script

#

# Copyright (C) 2010-2012 by Orion E. Poplawski <orion@cora.nwra.com>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 2 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

usage="Usage: $0 [-p prefix] <initscript> [args]"

usage() {

  echo $usage 1>&2

  exit 1

}

while getopts "p:" opt

do

  case $opt in

    p) prefix=$OPTARG; shift 2;;

    *) usage;;

  esac

done

# Need a script name

[ -z "$1" ] && usage

# Need to be a readable script

if [ ! -r "$1" ]

then

  echo "ERROR: Cannot read $1" 1>&2

  exit 1 

fi

#Will print out array assignment list

printenvarray () {

  env | while read x

  do

    key=${x%%=*}

    value=${x#*=}

    echo [$key]="'$value'"

  done

}

#Apparently we need to declare the associative arrays

declare -A env1 env2

#Record starting environment

eval env1=(`printenvarray`)

#Source the environment script

. "$@"

#Record ending environment

eval env2=(`printenvarray`)

#Print out the modulefile

echo "#%Module 1.0"

#Prefix

[ -n "$prefix" ] && echo -e "\nset prefix $prefix\n"

#Subshell so we can sort the output

(

dedup() {

  list=`mktemp`

  echo $1 | sed -r -e 's,[^/]+/\.\./,,g' -e 's,[^/]+/\.\./,,g' -e 's/:/\n/g' |

    while read x

    do

      grep -Fx ${x} $list && continue

      if [ -n "$prefix" ]

      then

        echo $x | sed -e s,$prefix,\$prefix,

      else

        echo $x

      fi

      echo $x >> $list

    done | tr '\n' : | sed -e 's/:$//'

  rm $list

} 

#Keys that changed

for key in "${!env1[@]}"

do

  if [ "${env1[$key]}" != "${env2[$key]}" ]

  then

    #Working directory change

    if [ "$key" = PWD ]

    then

      if [ -n "$prefix" ]

      then

        echo -e "chdir\t\t${env2[PWD]}" | sed -e s,$prefix,\$prefix,g

      else

        echo -e "chdir\t\t${env2[PWD]}"

      fi

    #Test for delete

    elif [ -z "${env2[$key]}" ]

    then

      echo -e "unsetenv\t${key}\t${env2[$key]}"

    #Test for prepend

    elif [ "${env2[$key]%${env1[$key]}}" != "${env2[$key]}" ]

    then

      added=$(dedup ${env2[$key]%:${env1[$key]}})

      echo -e "prepend-path\t$key\t${added}"

    #Test for prepend plus : added at end (MANPATH)

    elif [ "${env2[$key]%${env1[$key]}:}" != "${env2[$key]}" ]

    then

      added=$(dedup ${env2[$key]%${env1[$key]}:})

      echo -e "prepend-path\t$key\t${added}"

    #Test for append

    elif [ "${env2[$key]#${env1[$key]}}" != "${env2[$key]}" ]

    then

      added=$(dedup ${env2[$key]#:${env1[$key]}})

      echo -e "append-path\t$key\t${added}"

    #Test for prepend plus append

    elif [ "${env2[$key]%${env1[$key]}:*}" != "${env2[$key]}" ]

    then

      added=$(dedup ${env2[$key]%:${env1[$key]}*})

      echo -e "prepend-path\t$key\t${added}"

      added=$(dedup ${env2[$key]#*${env1[$key]}:})

      echo -e "append-path\t$key\t${added}"

    else

      #Unhandled

      echo "Unhandled change of $key" 1>&2

      echo "Before <${env1[$key]}>" 1>&2

      echo "After  <${env2[$key]}>" 1>&2

    fi

  fi

  #Delete keys we've handled

  unset env1[$key]

  unset env2[$key]

done

#New keys

for key in "${!env2[@]}"

do

  if [ "$key" = OLDPWD ]

  then

    continue

  fi

  #Use prepend-path for new paths

  if [ "${key/PATH/}" != "$key" ]

  then

    # TODO - Need to handle stripping of default MANPATH

    echo -e "prepend-path\t${key}\t"$(dedup ${env2[$key]})

  else

    if [ -n "$prefix" ]

    then

      echo -e "setenv\t\t${key}\t${env2[$key]}" | sed -e s,$prefix,\$prefix,g

    else

      echo -e "setenv\t\t${key}\t${env2[$key]}"

    fi

  fi

done

) | sort