#!/bin/bash
# echo loading module library
module_confirm () {
local FILE
local DIR
local NAME
DIR=$1
NAME=$2
# echo finding $NAME in $DIR
FILE="$(find ${DIR} -type f -name "${NAME}.mod" -o -name "${NAME}".lib -o -name "${NAME}".func -o -name "${NAME}".sh)"
# echo files found $FILE
COUNT=0
if [ "$FILE" ]; then
  COUNT=$(echo $FILE | xargs -n 1 | wc -l)
fi
# echo Number: $COUNT
if [ $COUNT -gt 1 ]; then
  echo two or more modules of same name found, aborting
  echo $FILE | xargs -n 1
  return 1
fi
[ $COUNT == 1 ] && echo $FILE && return 0
return 1
}

# if succesfull returns the path
function module_find () {

[ ! $1 ] && echo "no module specified" && return 1

local MDIRS
local MDIR
local DIRS
local MODULE=$1
# Precidence is user. current hosts, all hosts, current network, all networks, base
DIRS=( \
$([ $BASH_SHELL_USER ] && echo $HOME/$BASH_SHELL_USER || echo $HOME/bash/shell) \
$BASH_SHELL_HOST/$(hostname)
$BASH_SHELL_HOST/all \
$BASH_SHELL_NETWORK/$NETWORKNAME \
$BASH_SHELL_NETWORK/all \
$BASH_SHELL_BASE \
)

# echo ${DIRS[@]}

MDIRS=()
j=0
cnt=${#DIRS[@]}
for ((i=0;i<cnt;i++)); do
    DIR="${DIRS[i]}/modules"
    [ -d $DIR ] && MDIRS[j]=$DIR && j+=1
done

[ $MODULE_DIRS ] && MDIRS=("$MODULE_DIRS" "${MDIRS[@]}")

# echo Module Directories ${MDIRS[@]}

for MDIR in "${MDIRS[@]}"
   do
     # echo mdir $MDIR name $MODULE
     RES=$(module_confirm "$MDIR" "$MODULE")
     [ $? -eq 0 ] && echo $RES && return 0
  done
  return 1
}

module_load () {
[ ! $1 ] && echo "no module specified" && return 1
# (return 0 2>/dev/null) && echo "module_load was sourced" || echo "module_log was executed"
local FILE
FILE=$(module_find $1)
[ $? -ne 0 ] && echo no module $1 found && return 1
# source $FILE "$0"
[[ $BASHPID -eq $$ ]] || echo $FILE
source $FILE
return 0
}

module_loaded () {
  [ ! $1 ] && echo "no module specified" && return 1
  local func
  local ret
  local file
  ret=0
  # function_list | grep module
  file=$(module_find $1)
  [ ! $file ] && echo no module $1 found && return 1
  # echo "$(function_list $file)"
  for func in $(function_list $file)
    do
       [ ! $(function_list | grep $func) ] && ret=1
       [[ $2 = -vv ]] && ([[ $ret = 0 ]] && echo "$func loaded" || echo "$func not loaded")
    done
    [[ $2 = -v ]] && echo "module: $1" && ([[ $ret = 0 ]] && echo "all functions " || echo "some or all functions not loaded")
    [[ $ret = 0 ]] && echo "loaded"
   return $ret
}


# exported for login environment
declare -f -x module_load
declare -f -x module_find
declare -f -x module_confirm


function test_modules {
RED='\033[0;31m'
NC='\033[0m' # No Color
echo module to test: $1
echo --------------------
echo -e "testing: ${RED}module_find${NC}"
FILE=$(module_find $1)
[ $? -ne 0 ] && echo no module $1 found && return 1
echo module $FILE found
# . $FILE
echo ---------------
echo -e "testing: ${RED}module_load${NC}"
module_load $1
[ $? -ne 0 ] && echo no module $1 found && return 1
echo loaded module $1
echo ----------------------
echo -e "testing: ${RED}module_loaded${NC}"
module_loaded $1
}

#comment this out if testing
unset -f test_modules