#!/usr/bin/env bash

# Copy MC files in the current directory to the data handling system.
# Assume that cataloging is done after the files are copied by the 
# script ran by Dmitri.
#
# (which is not true for stntuples)
#
# The file names should follow the CDF file naming conventions.
#
# example: copy_to_dh -d sewk00 -D /cdf/data01/sewk01 [-V]
#
#      -V: debug-only mode - print but do not execute
#
#  2nd mode: running on FCDFDATA031: 
#
#  cd cdfopr;
#  copy_to_dh -i bewk0d/183530_184067.ftp
#-----------------------------------------------------------------------
export    VERBOSE=0
export  INPUT_DIR=""
export INPUT_FILE=""
export    DATASET=""
export        CMD=""
       ftp_script=/var/tmp/a_$$.ftp
#-----------------------------------------------------------------------
usage() {
  echo usage: copy_to_dh -d sewk00 -D /cdf/data01/sewk01 [-V]
}
if [ .$1 == "." ] ; then usage ; exit 0 ; fi
#-----------------------------------------------------------------------
# create input script for FTP  (do not forget to fix file attributes on
# the destination node - by default they may be wrong!)
#
# call: create_ftp_script tmp_file_name
#-----------------------------------------------------------------------
create_ftp_script_1() {
  f=`echo $file | awk -F / '{print $NF}'`
  cat > $1 <<EOF
open fcdfdata031.fnal.gov
cdfmc
lcd $INPUT_DIR
cd litvinse/data/
put $f
chmod 644 $f
put $f.ok
chmod 644 $f.ok
EOF
}

#-----------------------------------------------------------------------
create_ftp_script_2() {
  f=`echo $file | awk -F / '{print $NF}'`
  cat > $1 <<EOF
open $REMOTE_HOST
$REMOTE_USER
cd  $REMOTE_DIR
lcd $HOME/litvinse/data/
get $f
chmod 644 $f
chmod 644 $f.ok
EOF
}

#-----------------------------------------------------------------------
#  parse input parameters
#-----------------------------------------------------------------------
export OPTIND=1	
while getopts :d:D:i:Vvx: OPT; do
    case $OPT in
        d)                                          # input dataset ID
            export DATASET=$OPTARG
            if [ .$VERBOSE != "." ] ; then 
              echo   DATASET=$OPTARG
            fi
	    ;;
        D)                                          # input directory
            export INPUT_DIR=$OPTARG
            if [ .$VERBOSE != "." ] ; then 
              echo   INPUT_DIR=$OPTARG
            fi
	    ;;
        i)                                          # input file
            export INPUT_FILE=$OPTARG ; 
            if [ .$VERBOSE != "." ] ; then 
              echo   INPUT_FILE=$OPTARG
            fi
	    ;;
        v)                                          # debug script
            export       VERBOSE=1
            ;;
        V)                                          # debug
            export       CMD=echo
            if [ .$VERBOSE != "." ] ; then 
              echo   CMD=$CMD
            fi
            ;;
        x)                                          # export
            export       $OPTARG
            echo  export $OPTARG
            REDEFINED_ENV_VARS=${REDEFINED_ENV_VARS}" "$OPTARG
	    ;;
        *)
           echo OTHER: $OPT $OPTARG
           usage
           ;;
     esac
done

#-----------------------------------------------------------------------
proc1() {

  if [ .$VERBOSE != "." ] ; then echo proc1 entered INPUT_DIR=$INPUT_DIR ; fi

  p1=`echo $DATASET | awk '{print substr($1,1,1)substr($1,6,1)}'`
  p2=`echo $DATASET | awk '{print substr($1,2,4)}'`

  if [ .$VERBOSE != "." ] ; then echo p1,p2= $p1,$p2 ; fi

  cd $INPUT_DIR

  for file in `ls | grep $p1 | grep $p2` ; do

    if [ -f $file ] ; then
#-----------------------------------------------------------------------
#  step 2: copy file to fcdfdata031
#-----------------------------------------------------------------------
      create_ftp_script_1 $ftp_script

      $CMD touch ${file}.ok
      if [ .$CMD == "." ] ; then
        ftp < $ftp_script
      else 
        echo ftp will execute the collowing commands:
        echo "         ----------------------------"
        cat  $ftp_script
        echo "         ----------------------------"
      fi
      $CMD rm ${file}.ok
      rm $ftp_script
    fi
  done
}

#-----------------------------------------------------------------------
#  runs on FCDFDATA031, FTP's file from remote directory
#  format of the INPUT_FILE: ????
#-----------------------------------------------------------------------
proc2() {

   REMOTE_DIR=`cat $INPUT_FILE | grep REMOTE_DIR  | awk '{print $3}'`
  REMOTE_HOST=`cat $INPUT_FILE | grep REMOTE_HOST | awk '{print $3}'`
  REMOTE_USER=`cat $INPUT_FILE | grep REMOTE_USER | awk '{print $3}'`

  if [ $VERBOSE != 0 ] ; then echo proc2 entered ; fi

  for file in `cat $INPUT_FILE | awk '{if (substr($0,1,1) != "#") print $0}'` ; do
#-----------------------------------------------------------------------
#  step 2: copy file to fcdfdata031
#-----------------------------------------------------------------------
    create_ftp_script_2 $ftp_script

    if [ .$CMD == "." ] ; then
      ftp < $ftp_script
      if [ $? == "0" ] ; then touch $HOME/litvinse/data/${file}.ok ; fi
    else 
      echo ftp will execute the collowing commands:
      echo "         ----------------------------"
      cat  $ftp_script
      echo "         ----------------------------"
    fi
    rm $ftp_script
  done
}


#-----------------------------------------------------------------------
#  execution starts here
#-----------------------------------------------------------------------
if [ .$INPUT_FILE == "." ] ; then 
  proc1
else
  proc2
fi