zoneminder/web/includes/Storage.php

<?php
require_once('database.php');

$storage_cache = array();
class Storage {
  public function __construct( $IdOrRow = NULL ) {
    global $storage_cache;
    
    $row = NULL;
    if ( $IdOrRow ) {
      if ( is_integer( $IdOrRow ) or is_numeric( $IdOrRow ) ) {
        $row = dbFetchOne( 'SELECT * FROM Storage WHERE Id=?', NULL, array( $IdOrRow ) );
        if ( ! $row ) {
          Error("Unable to load Storage record for Id=" . $IdOrRow );
        }
      } elseif ( is_array( $IdOrRow ) ) {
        $row = $IdOrRow;
      }
    }
    if ( $row ) {
      foreach ($row as $k => $v) {
        $this->{$k} = $v;
      }
      $storage_cache[$IdOrRow] = $this;
    } else {
      $this->{'Name'} = '';
      $this->{'Path'} = '';
      $this->{'Type'} = 'local';
    }
  }

  public function Path() {
    if ( isset( $this->{'Path'} ) and ( $this->{'Path'} != '' ) ) {
      return $this->{'Path'};
    } else if ( ! isset($this->{'Id'}) ) {
      $path = ZM_DIR_EVENTS;
      if ( $path[0] != '/' ) {
        $this->{'Path'} = ZM_PATH_WEB.'/'.ZM_DIR_EVENTS;
      } else {
        $this->{'Path'} = ZM_DIR_EVENTS;
      }
      return $this->{'Path'};
      
    }
    return $this->{'Name'};
  }
  public function Name() {
    if ( isset( $this->{'Name'} ) and ( $this->{'Name'} != '' ) ) {
      return $this->{'Name'};
    } else if ( ! isset($this->{'Id'}) ) {
      return 'Default';
    }
    return $this->{'Name'};
  }

  public function __call( $fn, array $args= NULL){
  if ( count( $args )  ) {
      $this->{$fn} = $args[0];
    }
    if ( array_key_exists( $fn, $this ) ) {
      return $this->{$fn};
        
        $backTrace = debug_backtrace();
        $file = $backTrace[1]['file'];
        $line = $backTrace[1]['line'];
        Warning( "Unknown function call Storage->$fn from $file:$line" );
    }
  }
  public static function find_one( $parameters = null, $options = null ) {
    global $storage_cache;
    if ( 
        ( count($parameters) == 1 ) and
        isset($parameters['Id']) and
        isset($storage_cache[$parameters['Id']]) ) {
      return $storage_cache[$parameters['Id']];
    }
    $results = Storage::find_all( $parameters, $options );
    if ( count($results) > 1 ) {
      Error("Storage Returned more than 1");
      return $results[0];
    } else if ( count($results) ) {
      return $results[0];
    } else {
      return null;
    }
  }
  public static function find_all( $parameters = null, $options = null ) {
    $filters = array();
    $sql = 'SELECT * FROM Storage ';
    $values = array();

    if ( $parameters ) {
      $fields = array();
      $sql .= 'WHERE ';
      foreach ( $parameters as $field => $value ) {
        if ( $value == null ) {
          $fields[] = $field.' IS NULL';
        } else if ( is_array( $value ) ) {
          $func = function(){return '?';};
          $fields[] = $field.' IN ('.implode(',', array_map( $func, $value ) ). ')';
          $values += $value;

        } else {
          $fields[] = $field.'=?';
          $values[] = $value;
        }
      }
      $sql .= implode(' AND ', $fields );
    }
    if ( $options and isset($options['order']) ) {
    $sql .= ' ORDER BY ' . $options['order'];
    }
    $result = dbQuery($sql, $values);
    if ( $result ) {
      $results = $result->fetchALL(PDO::FETCH_CLASS | PDO::FETCH_PROPS_LATE, 'Storage');
      foreach ( $results as $row => $obj ) {
        $filters[] = $obj;
      }
    }
    return $filters;
  }
  public function disk_usage_percent() {
    $path = $this->Path();
    if ( ! $path ) {
      Warning("Storage::disk_usage_percent: path is empty");
      return 0;
    } else if ( ! file_exists( $path ) ) {
      Warning("Storage::disk_usage_percent: path $path does not exist");
      return 0;
    }
      
    $total = $this->disk_total_space();
    if ( ! $total ) {
      Error("disk_total_space returned false for " . $path );
      return 0;
    }
    $used = $this->disk_used_space();
    $usage = round( ($used / $total) * 100);
    //Logger::Debug("Used $usage = round( ( $used / $total ) * 100 )");
    return $usage;
  }
  public function disk_total_space() {
    if ( ! array_key_exists( 'disk_total_space', $this ) ) {
      $this->{'disk_total_space'} = disk_total_space( $this->Path() );
    }
    return $this->{'disk_total_space'};
  }
  public function disk_used_space() {
    # This isn't a function like this in php, so we have to add up the space used in each event.
    if ( (! array_key_exists( 'DiskSpace', $this )) or (!$this->{'DiskSpace'}) ) {
      $used = 0;
      if ( $this->{'Type'} == 's3fs' ) {
        $used = dbFetchOne('SELECT SUM(DiskSpace) AS DiskSpace FROM Events WHERE StorageId=? AND DiskSpace IS NOT NULL', 'DiskSpace', array($this->Id()) );

        foreach ( Event::find_all( array( 'StorageId'=>$this->Id(), 'DiskSpace'=>null ) ) as $Event ) {
          $Event->Storage( $this ); // Prevent further db hit
          $used += $Event->DiskSpace();
        }
      } else { 
        $path = $this->Path();
        $used = disk_total_space( $path ) - disk_free_space( $path );;
      }
      $this->{'DiskSpace'} = $used;
    }
		
    return $this->{'DiskSpace'};
  }
}
?>