#!/bin/bash

RunningGlideins() {

OLD_IFS=$IFS;IFS=" ";ilocalslot=( $@ );IFS=$OLD_IFS

MYNODE=${ilocalslot[0]}

echo "#!/bin/bash" > /tmp/findglides_$$
echo "#Get Working Dir" >> /tmp/findglides_$$
for i in $(seq 1 $((${#ilocalslot[@]} - 1)))
do
MYSLOT=${ilocalslot[$i]}
echo "MYGRIDID=" >> /tmp/findglides_$$
echo MYWNDIR="\`less /var/log/condor/StarterLog.$MYSLOT | grep IWD: |  tail -1 | cut -d\  -f4 \`" >> /tmp/findglides_$$
echo "[ -f \$MYWNDIR/glide_*/execute/dir_*/_condor_stdout ]&&"  >> /tmp/findglides_$$
echo "#Get GridID if exist" >> /tmp/findglides_$$
echo MYGRIDID="\`grep subject \$MYWNDIR/glide_*/execute/dir_*/_condor_stdout | tail -1 | sed -e s/subject\ \ \ :/glideinrun/ \`" >> /tmp/findglides_$$
#old#echo MYGRIDID="\`grep subject \$MYWNDIR/glide_*/execute/dir_*/_condor_stdout | tail -1 | cut -d:  -f2\`" >> /tmp/findglides_$$
echo "#No GridId, Check if glidein pilot exist" >> /tmp/findglides_$$
echo "[ \"\$MYGRIDID\" == \"\" ]&&[ -f \$MYWNDIR/glide_*/grid-mapfile ]&&" >> /tmp/findglides_$$
echo MYGRIDID="\`grep condor \$MYWNDIR/glide_*/grid-mapfile | sed -e s/condor/glideinpilots/ |sed -e s/\\\"//g\`" >> /tmp/findglides_$$
echo "[[ \"\$MYGRIDID\" == *glideinpilot* ]]&&" >> /tmp/findglides_$$
echo MYGRIDID=\"\`echo glideinpilot \${MYGRIDID%%glideinpilots*} \` \" >> /tmp/findglides_$$
echo "#No GridId, Other" >> /tmp/findglides_$$
echo "[ \"\$MYGRIDID\" == \"\" ]&&[ -f \$MYWNDIR/x509_user_proxy ]&&" >> /tmp/findglides_$$
echo MYGRIDID="\`openssl x509 -noout -issuer -in \$MYWNDIR/x509_user_proxy | sed -e s/issuer=/jobproxy/g \`" >> /tmp/findglides_$$
echo "echo \"\$MYGRIDID,\" " >> /tmp/findglides_$$
done
MYSCRIPT=`cat /tmp/findglides_$$`
rm /tmp/findglides_$$
MYGRIDID=`ssh -Y $MYNODE "$MYSCRIPT"` 

echo $MYGRIDID
}

#get output from condor
OLD_IFS=$IFS;IFS=$'\n';lines=( $(condor_q -run $1) );IFS=$OLD_IFS
 
#preprocess condor output (get nodes/slots info)
for idx in $(seq 0 $((${#lines[@]} - 1)))
do
    line="${lines[$idx]}"
    linesid[$idx]="list"
    #tag empty lines
    [ "$line" == "" ] && linesid[$idx]="blank"
    #tag info lines
    [[ "$line" == *"--"* ]]&&linesid[$idx]="title" 
    [[ "$line" == *"ID"* ]]&&[[ "$line" == *"OWNER"* ]]&&linesid[$idx]="title" 

    #get node/slot info
    words=( $line )
#    [[ "${linesid[$idx]}" == "list" ]]&&nodeslot=(( echo "${words[5]}" | tr "@" " " ))
    OLD_IFS=$IFS;IFS=$'@';nodeslot=( ${words[5]} );IFS=$OLD_IFS
    pairinfo[$idx]="${nodeslot[1]} ${nodeslot[0]}"

    #organize nodes/slots into array of strings
    slot=0
    for inode in $(seq 0 $((${#slots[@]} - 1)))
    do
      [[ "${slots[$inode]}" == *"${nodeslot[1]}"* ]]&&slots[$inode]="${slots[$inode]} ${nodeslot[0]}"&&slot=1
      [[ "$slot" == "1" ]]&&reverse[$idx]=$inode&&pair[$inode]="${pair[$inode]} $idx"&&break
    done
    [[  "${linesid[$idx]}" == "list" ]]&&[[ "$slot" == "0" ]]&&
     slots[${#slots[@]}]="${nodeslot[1]} ${nodeslot[0]}"&&
     reverse[$idx]=$((${#slots[@]} - 1))&&
     pair[${reverse[$idx]}]=$idx
    
#debug test
#    echo "${linesid[$idx]}($idx) - ${lines[$idx]} (${pair[${reverse[$idx]}]}) ${slots[${reverse[$idx]}]}"
done 

#ssh to running nodes
echo "Start ssh to ${#slots[@]} nodes "
for inode in $(seq 0 $((${#slots[@]} - 1)))
do
   OLD_IFS=$IFS;IFS=" ";islot=( ${slots[$inode]} );IFS=$OLD_IFS

   WNNODE=${islot[0]}
   echo "Node ${islot[@]} ($((${#islot[@]}-1)) slots)"
   #Get Certs of running user in node (ssh)
   certs=`RunningGlideins "${islot[@]}"`
   OLD_IFS=$IFS;IFS=$',';listcerts=( $certs );IFS=$OLD_IFS
   #debug#echo "Node $WNNODE ${listcerts[@]}"

   #debug#echo "pair=${pair[$inode]}"
   listpairs=( ${pair[$inode]} )

   #Sort certs and put it in a list by condor_q_line (listOfCerts)
   for ipair in $(seq 0 $((${#listpairs[@]} - 1)))
   do
      let  j=ipair+1
      listOfCerts[${listpairs[$ipair]}]=${listcerts[$ipair]}  
   done
done

#Print output condor_q like extended with certs
for idx in $(seq 0 $((${#lines[@]} - 1)))
do
#    echo "${linesid[$idx]}($idx) - ${lines[$idx]} ${listOfCerts[$idx]}"
    echo "${lines[$idx]} ${listOfCerts[$idx]}"
done
exit