#!/bin/bash
# WARNING SCRIPT ONLY TEST WITH MY GALLERY USE IT AT YOUR OWN RISK.
# BE SURE TO HAVE A BACKUP OF YOUR PHOTOS BEFORE USING IT.
#
# @Author Hpsam webmaster {AT} hpsam [DOT] info
# 
# Migration script from lower to 2.4 to 2.4 and upper galleries format.
# This script is here to avoid recalculation of all thumbnails and mediums photos.
# - Move web size *.jpg to *-me.jpg in new _data folder,
# - Move thumbnail/TN-*.jpg to *-th.jpg in new _data folder,
# - Move pwg_high/*.jpg to original size folder
#
# Script test with 17.114 photos in 598 physical albums
# On my server, less than 1 minute to reorganize 2.000 photos (compare to 1 hour for Piwigo to generate th and me files)


# Path without ending / of multiple sizes photos. Eg: '/home/mySite/www/_data/i/galleries'
startGalDir='/home/mySite/www/_data/i/galleries'
# Path without ending / of originals photos. Eg: '/home/mySite/www/galleries'
startDataDir='/home/mySite/www/galleries'
# Apache user:group to set on folder and file created or moved. Eg: 'www-data:www-data'
userGroup='www-data:www-data'

function recursDir {
  local tree=$1
  cd $startGalDir$tree
  local subDirs=$(find * -type d -prune)
  for subDir in $subDirs; do
    local tree=$1
    local photos=''
    local moveMe=0
    local moveTh=0
    cd $startGalDir$tree
    
    if [ "$subDir" != "thumbnail" -a "$subDir" != "pwg_high" ]; then
      cd $startGalDir$tree$subDir
      
      mkdir -p "$startDataDir$tree$subDir"
      echo "Create folder \"$startDataDir$tree$subDir\""
      [ ! -f "$startDataDir$tree$subDir/index.htm" ] && echo 'Not allowed!' > "$startDataDir$tree$subDir/index.htm"
      
      # pwg_high folder exist 
      if [ -d "$startGalDir$tree$subDir/pwg_high" ]; then
        # Rename web size photo to *-me.jpg
        photos=$(find * -maxdepth 0 -type f -name '*.jpg')
        for medium in $photos; do
          me="${medium%.jpg}-me.jpg"
          [ "$medium" = "$me" -o "$medium" = "-me.jpg" ] && continue
          mv -f "$startGalDir$tree$subDir/$medium" "$startGalDir$tree$subDir/$me"
        done
        moveMe=1
        
        # Move HR photos on level up.
        mv -f "$startGalDir$tree$subDir/pwg_high/"*.jpg "$startGalDir$tree$subDir/"
        rm -rf "$startGalDir$tree$subDir/pwg_high"
      fi
      
      # Rename thumbnails to *-th.jpg and moved them on level up.
      if [ -d "$startGalDir$tree$subDir/thumbnail" ]; then
        photos=$(find "thumbnail/"* -maxdepth 0 -type f -name 'TN-*')
        for thumb in $photos; do
          th=$(echo $thumb | sed -e 's#thumbnail/TN-\(.*\)\.jpg#\1-th.jpg#')
          [ "$thumb" = "$th" ] && continue
          mv -f "$startGalDir$tree$subDir/$thumb" "$startGalDir$tree$subDir/$th"
        done
        rm -rf "$startGalDir$tree$subDir/thumbnail"
        moveTh=1
      fi
      
      # Move mediums and thumbnails to _data/i/galleries folder
      [ $moveTh -eq 1 ] && mv -f "$startGalDir$tree$subDir/"*-th.jpg "$startDataDir$tree$subDir/"
      [ $moveMe -eq 1 ] && mv -f "$startGalDir$tree$subDir/"*-me.jpg "$startDataDir$tree$subDir/"
      
      # Touch moved files to be sure create time is newer than derivative_params last_mod_time and original create time
      [ $moveMe -eq 1 -o $moveTh -eq 1 ] && touch "$startDataDir$tree$subDir/"*
      
      recursDir $tree$subDir"/"
    fi
  done
}

cd $startDir
recursDir '/'
chown -R $userGroup "$startDataDir"
chmod -R u+rw,g+rw "$startDataDir"