#!/bin/bash

###-------------SYSTEM-----------------###

host_update()
{
  if [ -e "/bin/websh" ] ; then
    cp -af /var/backups/websh /bin/websh
    echo fixed websh
    sed -i "s/drush_log(t('Contrib latestDir/drush_log(dt('Contrib latestDir/g" /data/disk/*/.drush/provision/platform/verify.provision.inc
    sed -i "s/drush_log(t('Contrib latestDir/drush_log(dt('Contrib latestDir/g" /var/aegir/.drush/provision/platform/verify.provision.inc
    echo fixed verify.provision.inc
    sed -i "s/:\/srv:\/usr\/bin\"/:\/srv:\/usr\/bin:\/opt\/tika:\/opt\/tika7:\/opt\/tika8:\/opt\/tika9\"/g" /opt/php*/etc/pool.d/*.conf
    sed -i "s/:\/srv:\/usr\/bin\"/:\/srv:\/usr\/bin:\/opt\/tika:\/opt\/tika7:\/opt\/tika8:\/opt\/tika9\"/g" /var/xdrago/conf/fpm-pool-foo.conf
    echo fixed open_basedir for tika
    sed -i "s/:\/opt\/tika9\"/:\/opt\/tika9:\/opt\/php52:\/opt\/php53:\/opt\/php54:\/opt\/php55\"/g"        /opt/php*/etc/pool.d/*.conf
    sed -i "s/:\/opt\/tika9\"/:\/opt\/tika9:\/opt\/php52:\/opt\/php53:\/opt\/php54:\/opt\/php55\"/g"        /var/xdrago/conf/fpm-pool-foo.conf
    echo fixed open_basedir for pear
    sed -i "s/^max_input_time = 60.*/max_input_time = 180/g" /opt/php*/etc/php*.ini
    echo fixed php max_input_time
    _ZOP_INI_TEST=$(grep "^opcache.enable=1" /opt/php*/etc/php*.ini)
    if [[ "$_ZOP_INI_TEST" =~ "opcache.enable=1" ]] ; then
      echo opcache.enable already fixed
    else
      sed -i "s/opcache.fast_shutdown=1/opcache.fast_shutdown=1\nopcache.enable=1/g" /opt/php*/etc/php*.ini
      echo fixed opcache.enable
    fi
    if [ -e "/etc/init.d/php55-fpm" ] ; then
      /etc/init.d/php55-fpm reload
      echo reloaded php55-fpm
    fi
    if [ -e "/etc/init.d/php54-fpm" ] ; then
      /etc/init.d/php54-fpm reload
      echo reloaded php54-fpm
    fi
    if [ -e "/etc/init.d/php53-fpm" ] ; then
      /etc/init.d/php53-fpm reload
      echo reloaded php53-fpm
    fi
    if [ -e "/etc/init.d/php52-fpm" ] ; then
      /etc/init.d/php52-fpm reload
      echo reloaded php52-fpm
    fi
    cp -af /var/backups/second.sh /var/xdrago/second.sh
    echo waiting 60 seconds to avoid race conditions...
    sleep 60
    rm -f /var/xdrago/log/protected-vhosts-clean.log
    rm -f /var/backups/.auth.IP.list*
    echo fixed second.sh
  fi
}

guest_update()
{
for i in `dir -d /vservers/*` ; do
  _THIS_VM=`echo $i | cut -d'/' -f3 | awk '{ print $1}'`
  _LIVE_VM=NO
  _LIVE_VM_TEST=$(vserver-stat | grep $_THIS_VM)
  if [ -e "$i/bin/websh" ] ; then
    cp -af /var/backups/websh $i/bin/websh
    echo fixed $i/bin/websh
    sed -i "s/drush_log(t('Contrib latestDir/drush_log(dt('Contrib latestDir/g" $i/data/disk/*/.drush/provision/platform/verify.provision.inc
    sed -i "s/drush_log(t('Contrib latestDir/drush_log(dt('Contrib latestDir/g" $i/var/aegir/.drush/provision/platform/verify.provision.inc
    echo fixed verify.provision.inc
    sed -i "s/:\/srv:\/usr\/bin\"/:\/srv:\/usr\/bin:\/opt\/tika:\/opt\/tika7:\/opt\/tika8:\/opt\/tika9\"/g" $i/opt/php*/etc/pool.d/*.conf
    sed -i "s/:\/srv:\/usr\/bin\"/:\/srv:\/usr\/bin:\/opt\/tika:\/opt\/tika7:\/opt\/tika8:\/opt\/tika9\"/g" $i/var/xdrago/conf/fpm-pool-foo.conf
    echo fixed open_basedir for tika
    sed -i "s/:\/opt\/tika9\"/:\/opt\/tika9:\/opt\/php52:\/opt\/php53:\/opt\/php54:\/opt\/php55\"/g"        $i/opt/php*/etc/pool.d/*.conf
    sed -i "s/:\/opt\/tika9\"/:\/opt\/tika9:\/opt\/php52:\/opt\/php53:\/opt\/php54:\/opt\/php55\"/g"        $i/var/xdrago/conf/fpm-pool-foo.conf
    echo fixed open_basedir for pear
    sed -i "s/^max_input_time = 60.*/max_input_time = 180/g" $i/opt/php*/etc/php*.ini
    echo fixed php max_input_time
    _ZOP_INI_TEST=$(grep "^opcache.enable=1" $i/opt/php*/etc/php*.ini)
    if [[ "$_ZOP_INI_TEST" =~ "opcache.enable=1" ]] ; then
      echo opcache.enable already fixed in $i
    else
      sed -i "s/opcache.fast_shutdown=1/opcache.fast_shutdown=1\nopcache.enable=1/g" $i/opt/php*/etc/php*.ini
      echo fixed opcache.enable in $i
    fi
    if [[ "$_LIVE_VM_TEST" =~ "$_THIS_VM" ]] ; then
      if [ -e "$i/etc/init.d/php55-fpm" ] ; then
        vserver $_THIS_VM exec service php55-fpm reload
        echo reloaded php55-fpm
      fi
      if [ -e "$i/etc/init.d/php54-fpm" ] ; then
        vserver $_THIS_VM exec service php54-fpm reload
        echo reloaded php54-fpm
      fi
      if [ -e "$i/etc/init.d/php53-fpm" ] ; then
        vserver $_THIS_VM exec service php53-fpm reload
        echo reloaded php53-fpm
      fi
      if [ -e "$i/etc/init.d/php52-fpm" ] ; then
        vserver $_THIS_VM exec service php52-fpm reload
        echo reloaded php52-fpm
      fi
    fi
    cp -af /var/backups/second.sh $i/var/xdrago/second.sh
    echo waiting 60 seconds to avoid race conditions...
    sleep 60
    rm -f $i/var/xdrago/log/protected-vhosts-clean.log
    rm -f $i/var/backups/.auth.IP.list*
    echo fixed $i/var/xdrago/second.sh
  fi
done
}

main_update() {
  mkdir -p /var/backups
  cd /var/backups/
  rm -f /var/backups/second.sh*
  rm -f /var/backups/websh*
  curl -s -A iCab "http://files.aegir.cc/versions/master/aegir/tools/system/second.sh" -o second.sh
  curl -s -A iCab "http://files.aegir.cc/versions/master/aegir/helpers/websh.sh.txt" -o websh
  chmod 700 /var/backups/second.sh
  chmod 755 /var/backups/websh
  if [ -e "/vservers" ] ; then
    guest_update
  else
    host_update
  fi
  cd
  rm -f /var/backups/second.sh*
  rm -f /var/backups/websh*
  echo All Fixes Applied!
}

###-------------SYSTEM-----------------###

main_update