#!/bin/bash
# this will superceed the ssh binary in order to source all the config files
module_load file # loads find and build_file

[[ ! $SSH_CONFIG ]] && export SSH_CONFIG="$BASH_SHELL_ANY_NETWORK/ssh/_config"

function ssh_config() {
  local CDIRS
  local CDIR
  local DIRS
  local DIR
  local PDIRS

  declare OPTION
  declare OPTARG
  declare OPTIND
  while getopts 'd:' OPTION; do
    # echo $OPTION $OPTARG
    case "$OPTION" in
    d)
      PDIRS=$OPTARG
      # echo option d: $DIRS
      ;;
    *)
      echo unknown option $OPTION
      ;;
    esac
  done

  shift $((OPTIND - 1))

  [[ $PDIRS ]] && DIRS=($PDIRS) || DIRS=(${BASH_SHELL_DIRS} "$HOME/$BASH_SHELL_USER")
  # echo DIRS "${DIRS[@]}"
  # echo $SSH_CONFIG
  CDIRS=()
  j=0
  cnt=${#DIRS[@]}
  for ((i = 0; i < cnt; i++)); do
    # echo $i of $cnt
    # looks in ssh/config subdirectory of each DIRS is not passed
    DIR="${DIRS[i]}$([[ ! $PDIRS ]] && echo /ssh/config)"
    # echo ----- trying $DIR
    [ -d $DIR ] && CDIRS[j]=$DIR
    j+=1 || echo no directory $DIR
  done
  # CDIRS=("${CDIRS[@]}")
  # echo ${CDIRS[@]}

  local HEADER="##############################################################
# THIS FILE IS GENERATED BY function ssh_config. Do not edit #
# It is created by combination of ssh configuration files    #
# which are listed in a comment line before each             #
# It is used by the ssh function which then calls ssh binary #
##############################################################"

  debug ssh config file at: $SSH_CONFIG
  mkdir -p "$(dirname "$SSH_CONFIG")"
  echo -e "$HEADER" >$SSH_CONFIG
  # build_file appends the given file to output file cleanly with checks
  # append any system config
  build_file "/etc/ssh/ssh_config" $SSH_CONFIG
  # echo existing dirs ${CDIRS[@]}
  # will append any .cfg file found in ssh/config subdir of any BASH_SHELL_DIRS, including home shell
  for CDIR in "${CDIRS[@]}"; do
    # FILES=$(find -n '*.cfg' -d 0 $CDIR)
    for f in $(_find -n '*.cfg' -p 'archive off' -d 0 $CDIR); do
      # echo "Processing $f";
      [[ $f ]] && build_file "$f" $SSH_CONFIG
    done
  done
  # append any tradtional home config
  build_file "$HOME/.ssh/config" $SSH_CONFIG
}

ssh() {
  if [[ $SSH_CONFIG ]]; then
    [[ ! -f "$SSH_CONFIG" ]] && ssh_config "$SSH_CONFIG"
    command ssh -F $SSH_CONFIG "$@"
  else
    command ssh "$@"
  fi
}

sshp() {
  local opts="-o PreferredAuthentications=password -o PubkeyAuthentication=no"
  if [[ $SSH_CONFIG ]]; then
    [[ ! -f "$SSH_CONFIG" ]] && ssh_config "$SSH_CONFIG"
    command ssh $opts -F $SSH_CONFIG "$@"
  else
    command ssh $opts "$@"
  fi
}

 ssh_test() {
   echo running non-interactive ssh test on $2 with user $1
  ssh $1@$2 "cat .bashrc"
  # env | grep -E 'SHELL|BASH|SSH';type module_load;module_load helpers;adirname .
 }

function rrem() {
  ssh -X -t "$@"
}

ssh_script () {

  local SUDO=""

  declare OPTION
  declare OPTARG
  declare OPTIND
  while getopts 'ps:' OPTION; do
    # echo $OPTION $OPTARG
    case "$OPTION" in
    p)
      PASS=true
      # echo option password login only: $PASS
      ;;
    s)
      SUDO="sudo"
      # echo option s: $SUDO
      ;;
    *)
      echo unknown option $OPTION
      ;;
    esac
  done

  shift $((OPTIND - 1))

  HOST=$1@$2
  SCRIPT=$3

  # ssh  < 'bash -s' << local_script.sh
  
  scp $PATHtoSCRIPT$SCRIPTname $HOSTtoCONTROL:/tmp/
  ssh -t $HOSTtoCONTROL "sudo -s bash /tmp/$SCRIPTname"
  ssh root@MachineB 'bash -s' < local_script.sh
}