#!/bin/bash

###############################################################################
# 
# Скрипт использует пакет imagemagick для создания водяной марки и вставки ее 
# в изображения.
#
# Автор: digiwhite
# Дата создания: 02-06-2010
#
###############################################################################


###############################################################################
#
# Переменные скрипта. 
#

# Обходить директории рекурсивно
RECURSIVE=0

# Количество указанных обязательных параметров
REQ_PARAMS=0

# URL для водяной марки
URL=

# Папка для поиска изображений
FOLDER=

# Директория, куда будут складываться все измененные изображения
OUTPUT_FOLDER=

# Директория для измененных изображений по умолчанию.
DEFAULT_OUTPUT_FOLDER=~/watermarks


###############################################################################
# 
# Дополнительные настройки скрипта
#

# Шаблоны для поиска изображений с заданными расширениями
EXT="(*\.png$)|(*\.jpg$)"

# Место, где лежит сгенерированная водяная марка
WATERMARK_NAME="/tmp/watermark.gif"

# Место, где лежат шрифты, используемые для генерации водяной марки
FONT_FOLDER=/usr/share/fonts/truetype/freefont/

# Название шрифта
FONT_NAME=FreeSans.ttf

# Размер водяной марки - длинна
SIZE_X=200

# Размер водяной марки - ширина
SIZE_Y=30

###############################################################################

promt_help() {
  SCRIPT_NAME=`basename $0` 
   echo "$SCRIPT_NAME <-r> <-u URL> <-f FOLDER> <-o FOLDER>"
   echo
       echo "-r                Просматривать вложенные директории."
 echo "-u               URL, который используется для генерации водяной марки."
      echo "-f               Директория, в которой будут искаться файлы для вставки водяной марки."
       echo "-o    Директория, куда будет записан результат работы"
}



while getopts ru:f:o: OPT ; do
        case $OPT in 
                r)       let RECURSIVE=1
                      ;;
          u)       URL=$OPTARG
                          let REQ_PARAMS=$REQ_PARAMS+1
                 ;;
          f)       FOLDER=$OPTARG
                               let REQ_PARAMS=$REQ_PARAMS+1
                 ;;
          o)  OUTPUT_FOLDER=$OPTARG
                     ;;
          *)       
                    ;;
  esac
done

if < $REQ_PARAMS -eq 2 > ; then
       if << ! -d $FOLDER >> ;      then
         echo " Директория \"$FOLDER\" не существует."
                exit 3
      fi

       if < ! $URL > ; then
            echo " Указан пустой URL."
          exit 3
     fi
 
     CORRECT_URL=`echo $URL | egrep "^(http)|(https)|(ftp)://(www\.)?<0-9a-zA-Z.-_>+\\.<a-z>{2,5}"`
   if < ! $CORRECT_URL > ; then
           echo " Указанный URL не корректен."
         exit 3
     fi

     if << ! "$OUTPUT_FOLDER" >> ; then
               OUTPUT_FOLDER=$DEFAULT_OUTPUT_FOLDER
 fi

     if << ! -d "$OUTPUT_FOLDER" >> ; then
            mkdir -p "$OUTPUT_FOLDER"

              if << ! -d "$OUTPUT_FOLDER" >> ; then
                    echo " Не удалось создать директорию \"$OUTPUT_FOLDER\" для выходных файлов."
                       exit 3
             fi
  fi

     OLDIFS=$IFS
  IFS=`echo -en "\n\b"`
       IFS=$'\n'

      # Получаем список файлов для вставки водяной марки
  if < $RECURSIVE -eq 1 > ; then
         FILE_LIST=`find "$FOLDER" -type f -print | egrep $EXT`
       else
                FILE_LIST=`find "$FOLDER" -maxdepth 1 -type f | egrep $EXT`
  fi

     if << $FILE_LIST >> ; then
               # Создаем водяную марку.
            echo "<i> Создаем водяную марку $WATERMARK_NAME"
             convert -background 'rgba(255, 255, 255, 0.0)' -fill black -gravity center -font $FONT_FOLDER/$FONT_NAME -size ${SIZE_X}x${SIZE_Y} -pointsize 24 label:"$CORRECT_URL" $WATERMARK_NAME

 
               for file in $FILE_LIST ; do
                  FILE_NAME=`basename $file`
                   FILE_DIR_NAME=`dirname $file`

                  echo "</i><i> Маркировка файла $file => $OUTPUT_FOLDER/$FILE_DIR_NAME/`basename $file`"

                        if << ! -d  "$OUTPUT_FOLDER/$FILE_DIR_NAME" >> ; then
                            mkdir -p "$OUTPUT_FOLDER/$FILE_DIR_NAME"

                               if << ! -d "$OUTPUT_FOLDER/$FILE_DIR_NAME" >> ; then
                                     echo " Не удалось создать директорию \"$OUTPUT_FOLDER/$FILE_DIR_NAME\" для файла \"$FILE_NAME\""
                                    exit 3
                             fi
                  fi
 
                     composite -gravity SouthEast $WATERMARK_NAME $file "$OUTPUT_FOLDER/$FILE_DIR_NAME/$FILE_NAME"
               done

           IFS=$OLDIFS
  else
                echo "</i><i> Не неайдено ни одного подходящего файла для вставки водяной марки."
               exit 0
     fi


else
        echo " Указано недостаточно параметров."
    echo
      promt_help

     exit 3
fi

</i></a-z>