JobModel.php

<?php

/**
 *
 * bareos-webui - Bareos Web-Frontend
 *
 * @link      https://github.com/bareos/bareos-webui for the canonical source repository
 * @copyright Copyright (c) 2013-2017 Bareos GmbH & Co. KG (http://www.bareos.org/)
 * @license   GNU Affero General Public License (http://www.gnu.org/licenses/)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

namespace Job\Model;

class JobModel
{
   public function getJobs(&$bsock=null, $jobname=null, $days=null)
   {
      if(isset($bsock)) {
         if($days == "all") {
            if($jobname == "all") {
               $cmd = 'llist jobs';
            }
            else {
               $cmd = 'llist jobs jobname="'.$jobname.'"';
            }
         }
         else  {
            if($jobname == "all") {
               $cmd = 'llist jobs days='.$days;
            }
            else {
               $cmd = 'llist jobs jobname="'.$jobname.'" days='.$days;
            }
         }
         $result = $bsock->send_command($cmd, 2, null);
         if(preg_match('/Failed to send result as json. Maybe result message to long?/', $result)) {
            //return false;
            $error = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
            return $error['result']['error'];
         }
         else {
            $jobs = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
            return $jobs['result']['jobs'];
         }
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getJobsByStatus(&$bsock=null, $jobname=null, $status=null, $days=null, $hours=null)
   {
      if(isset($bsock, $status)) {
         if(isset($days)) {
            if($days == "all") {
               $cmd = 'llist jobs jobstatus='.$status.'';
            }
            else {
               $cmd = 'llist jobs jobstatus='.$status.' days='.$days.'';
            }
         }
         elseif(isset($hours)) {
            if($hours == "all") {
               $cmd = 'llist jobs jobstatus='.$status.'';
            }
            else {
               $cmd = 'llist jobs jobstatus='.$status.' hours='.$hours.'';
            }
         }
         else {
            $cmd = 'llist jobs jobstatus='.$status.'';
         }
         if($jobname != "all") {
            $cmd .= ' jobname="'.$jobname.'"';
         }
         $result = $bsock->send_command($cmd, 2, null);
         $jobs = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
         return array_reverse($jobs['result']['jobs']);
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getJob(&$bsock=null, $id=null)
   {
      if(isset($bsock, $id)) {
         $cmd = 'llist jobid='.$id.'';
         $result = $bsock->send_command($cmd, 2, null);
         $job = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
         return $job['result']['jobs'];
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getJobLog(&$bsock=null, $id=null)
   {
      if(isset($bsock, $id)) {
         $cmd = 'list joblog jobid='.$id.'';
         $result = $bsock->send_command($cmd, 2, null);
         if(preg_match('/Failed to send result as json. Maybe result message to long?/', $result)) {
            //return false;
            $error = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
            return $error['result']['error'];
         }
         else {
            $log = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
            return $log['result']['joblog'];
         }

      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getJobsByType(&$bsock=null, $type=null)
   {
      if(isset($bsock)) {
         if($type == null) {
            $cmd = '.jobs';
         }
         else {
            $cmd = '.jobs type="'.$type.'"';
         }
         $result = $bsock->send_command($cmd, 2, null);
         $jobs = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
         return $jobs['result']['jobs'];
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getJobsLastStatus(&$bsock=null)
   {
      if(isset($bsock)) {
         $cmd = 'llist jobs last current enabled';
         $result = $bsock->send_command($cmd, 2, null);
         $jobs = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
         return $jobs['result']['jobs'];
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getRunningJobsStatistics(&$bsock = null) {
      if(isset($bsock)) {

         $jobstats = [];
         $i = 0;

         // GET RUNNING JOBS
         $runningJobs = $this->getJobsByStatus($bsock, null, 'R');

         // COLLECT REQUIRED DATA FOR EACH RUNNING JOB
         foreach($runningJobs as $job) {

            // GET THE JOB STATS
            $cmd = 'list jobstatistics jobid=' . $job['jobid'];
            $result = $bsock->send_command($cmd, 2, null);
            $tmp = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);

            // JOBID, JOBNAME AND CLIENT
            $jobstats[$i]['jobid'] = $job['jobid'];
            $jobstats[$i]['name'] = $job['name'];
            $jobstats[$i]['client'] = $job['client'];
            $jobstats[$i]['level'] = $job['level'];

            if(count($tmp['result']['jobstats']) > 2) {

               // CALCULATE THE CURRENT TRANSFER SPEED OF THE INTERVAL
               $a = strtotime( $tmp['result']['jobstats'][count($tmp['result']['jobstats']) - 1]['sampletime'] );
               $b = strtotime( $tmp['result']['jobstats'][count($tmp['result']['jobstats']) - 2]['sampletime'] );
               $interval = $a - $b;

               if($interval > 0) {
                  $speed = ($tmp['result']['jobstats'][count($tmp['result']['jobstats']) - 1]['jobbytes'] - $tmp['result']['jobstats'][count($tmp['result']['jobstats']) - 2]['jobbytes']) / $interval;
                  $speed = round($speed, 2);
               }
               else {
                  $speed = 0;
               }

               $jobstats[$i]['speed'] = $speed;

               // JOBFILES
               $tmp = $tmp['result']['jobstats'][count($tmp['result']['jobstats']) - 1];
               if($tmp['jobfiles'] == null) {
                  $jobstats[$i]['jobfiles'] = 0;
               }
               else {
                  $jobstats[$i]['jobfiles'] = $tmp['jobfiles'];
               }

               // JOBBYTES
               $jobstats[$i]['jobbytes'] = $tmp['jobbytes'];

               // SAMPLETIME
               $jobstats[$i]['sampletime'] = $tmp['sampletime'];

               // LAST BACKUP SIZE
               $level = $jobstats[$i]['level'];
               $cmd = 'list jobs jobname=' . $job['name'] . ' client=' . $job['client'] . ' jobstatus=T joblevel=' . $level . ' last';

               $result = $bsock->send_command($cmd, 2, null);
               $tmp = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
               $jobstats[$i]['lastbackupsize'] = $tmp['result']['jobs'][0]['jobbytes'];
               if($jobstats[$i]['lastbackupsize'] > 0) {
                  if($jobstats[$i]['jobbytes'] > 0 && $tmp['result']['jobs'][0]['jobbytes'] > 0) {
                     $jobstats[$i]['progress'] = ceil( (($jobstats[$i]['jobbytes'] * 100) / $tmp['result']['jobs'][0]['jobbytes']));
                     if($jobstats[$i]['progress'] > 100) {
                        $jobstats[$i]['progress'] = 99;
                     }
                  }
                  else {
                     $jobstats[$i]['progress'] = 0;
                  }
               }
               else {
                  $jobstats[$i]['progress'] = 0;
               }
            }
            else {
               $jobstats[$i]['speed'] = 0;
               $jobstats[$i]['jobfiles'] = 0;
               $jobstats[$i]['jobbytes'] = 0;
               $jobstats[$i]['sampletime'] = null;
               $jobstats[$i]['progress'] = 0;
            }

            $i++;

         }
         return $jobstats;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function getRestoreJobs(&$bsock=null)
   {
      if(isset($bsock)) {
         $cmd = '.jobs type=R';
         $result = $bsock->send_command($cmd, 2, null);
         $restorejobs = \Zend\Json\Json::decode($result, \Zend\Json\Json::TYPE_ARRAY);
         return $restorejobs['result']['jobs'];
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function runJob(&$bsock=null, $name=null)
   {
      if(isset($bsock, $name)) {
         $cmd = 'run job="'.$name.'" yes';
         $result = $bsock->send_command($cmd, 0, null);
         return $result;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function rerunJob(&$bsock=null, $id=null)
   {
      if(isset($bsock, $id)) {
         $cmd = 'rerun jobid='.$id.' yes';
         $result = $bsock->send_command($cmd, 0, null);
         return $result;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function cancelJob(&$bsock=null, $id=null)
   {
      if(isset($bsock, $id)) {
         $cmd = 'cancel jobid='.$id.' yes';
         $result = $bsock->send_command($cmd, 0, null);
         return $result;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function enableJob(&$bsock=null, $name=null)
   {
      if(isset($bsock, $name)) {
         $cmd = 'enable job="'.$name.'" yes';
         $result = $bsock->send_command($cmd, 0, null);
         return $result;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }

   public function disableJob(&$bsock=null, $name=null)
   {
      if(isset($bsock, $name)) {
         $cmd = 'disable job="'.$name.'" yes';
         $result = $bsock->send_command($cmd, 0, null);
         return $result;
      }
      else {
         throw new \Exception('Missing argument.');
      }
   }
}