#!/bin/bash
PLAYER=mplayer
PLAYER_ARGS="-fs -zoom -quiet -vo xv -input conf=/home/mythtv/mythtv1/input.conf"
SCENE_DIR=/media/lvm/movies2
MOVIE_DIR=/media/lvm/movies

parser() {
  NAME=${1%*.[0-9][0-9][0-9][0-9].*} 
  NAME=${NAME%*.DVDRip*}
  NAME=${NAME%*.DVDRiP*}
  NAME=${NAME%*.DVD*}
  NAME=${NAME%*.PROPER*}
  NAME=${NAME%*.UNRATED*}
  echo $NAME
}

build_scriptcd2() {
  echo "cd "$SCENE_DIR/$1"
       #test avi existence
       if [ -e CD1/*.avi ] ; then
         if [ -e CD2/*.avi ] ; then
           $PLAYER $PLAYER_ARGS CD1/*.avi CD2/*.avi
         else
           xterm -e \"cd CD2 && unrar x *rar && cd ..\"
           $PLAYER $PLAYER_ARGS CD1/*.avi CD2/*.avi
         fi
       else
         xterm -e \"cd CD1 && unrar x *rar && cd .. && cd CD2 && unrar x *rar && cd ..\"
         $PLAYER $PLAYER_ARGS CD1/*.avi CD2/*.avi
         fi"  > $MOVIE_DIR/$2.sh
}

build_scriptcd1() {
  echo "cd "$SCENE_DIR/$1"
       #test avi existence
       if [ -e *.avi ] ; then
         $PLAYER $PLAYER_ARGS *.avi
       else
         xterm -e \"unrar x *rar && cd ..\"
         $PLAYER $PLAYER_ARGS *.avi
         fi"  > $MOVIE_DIR/$2.sh 
}

sfv_check() {
  if [ -d CD1 ] && [ -d CD2 ] ; then
    if cd CD1 && cksfv -qf *.sfv && cd .. && cd CD2 cksfv -qf *.sfv ; then
      cd $SCENE_DIR
      return 0
    else
      cd $SCENE_DIR
      return 3
    fi
  else 
    if cksfv -qf *.sfv ; then
      cd $SCENE_DIR
      return 1
    else
      cd $SCENE_DIR
      return 2
    fi 
  fi     
}

IMDB_ID()  {
  unset ID
  ID=$(cat $SCENE_DIR/$2/*nfo 2>/dev/null | grep imdb.com/title/tt)
  if [ -n "$ID" ]; then
    ID=${ID#*\/tt*}
    ID=${ID//\//}
    ID=${ID//[^0-9]/}
  else
    QUERY=$(perl /usr/share/mythtv/mythvideo/scripts/imdb.pl -M $1)  
    if [ $(echo "$QUERY" | wc -l) -eq 1 ] ; then
       ID=${QUERY%*:*}
       ID=${ID//[^0-9]/}
    else 
       QUERY=$(echo "$QUERY" | grep "${1//./ }")
       QUERY=$(echo "$QUERY" | head -n 1)
       ID=${QUERY%*:*}
       ID=${ID//[^0-9]/} 
    fi 
  fi
  echo $ID
}
    
      
cd $SCENE_DIR
for i in $(ls) ; do
   cd $i 
   TITLE=$(parser $i)
   sfv_check
   case "$?" in 
     0) build_scriptcd2 $i $(parser $i)  ;;
     1) build_scriptcd1 $i $(parser $i) ;;
     2) echo ERROR: sfv failure with $i ;;
   esac
   IMDB_ID=$(IMDB_ID $TITLE $i)
   if [ -n "$IMDB_ID" ]; then
     #echo " "
     #echo "IMDB ID for $TITLE is $IMDB_ID" 
     #echo " "
     /usr/share/mythtv/mythvideo/scripts/find_meta.py -a $IMDB_ID $MOVIE_DIR/$TITLE.sh
   else
     echo "ERROR: couldn't find imdb ID for $TITLE" 
   fi
done