src/Util/ReversedFileReader.php

   1 <?php
   2 /**
   3  * @copyright Copyright (C) 2021, Friendica
   4  *
   5  * @license GNU AGPL version 3 or any later version
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as
   9  * published by the Free Software Foundation, either version 3 of the
  10  * License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22 namespace Friendica\Util;
  23
  24 /**
  25  * An iterator which returns lines from file in reversed order
  26  *
  27  * original code https://stackoverflow.com/a/10494801
  28  */
  29 class ReversedFileReader implements \Iterator
  30 {
  31         const BUFFER_SIZE = 4096;
  32         const SEPARATOR   = "\n";
  33
  34         /** @var resource */
  35         private $fh = null;
  36
  37         /** @var int */
  38         private $filesize = -1;
  39
  40         /** @var int */
  41         private $pos = -1;
  42
  43         /** @var array */
  44         private $buffer = null;
  45
  46         /** @var int */
  47         private $key = -1;
  48
  49         /** @var string */
  50         private $value = null;
  51
  52         /**
  53          * Open $filename for read and reset iterator
  54          *
  55          * @param string $filename      File to open
  56          * @return $this
  57          */
  58         public function open(string $filename)
  59         {
  60                 $this->fh = fopen($filename, 'r');
  61                 if (!$this->fh) {
  62                         // this should use a custom exception.
  63                         throw \Exception("Unable to open $filename");
  64                 }
  65                 $this->filesize = filesize($filename);
  66                 $this->pos      = -1;
  67                 $this->buffer   = null;
  68                 $this->key      = -1;
  69                 $this->value    = null;
  70                 return $this;
  71         }
  72
  73         private function _read($size)
  74         {
  75                 $this->pos -= $size;
  76                 fseek($this->fh, $this->pos);
  77                 return fread($this->fh, $size);
  78         }
  79
  80         private function _readline()
  81         {
  82                 $buffer = & $this->buffer;
  83                 while (true) {
  84                         if ($this->pos == 0) {
  85                                 return array_pop($buffer);
  86                         }
  87                         if (count($buffer) > 1) {
  88                                 return array_pop($buffer);
  89                         }
  90                         $buffer = explode(self::SEPARATOR, $this->_read(self::BUFFER_SIZE) . $buffer[0]);
  91                 }
  92         }
  93
  94         public function next()
  95         {
  96                 ++$this->key;
  97                 $this->value = $this->_readline();
  98         }
  99
 100         public function rewind()
 101         {
 102                 if ($this->filesize > 0) {
 103                         $this->pos    = $this->filesize;
 104                         $this->value  = null;
 105                         $this->key    = -1;
 106                         $this->buffer = explode(self::SEPARATOR, $this->_read($this->filesize % self::BUFFER_SIZE ?: self::BUFFER_SIZE));
 107                         $this->next();
 108                 }
 109         }
 110
 111         public function key()
 112         {
 113                 return $this->key;
 114         }
 115
 116         public function current()
 117         {
 118                 return $this->value;
 119         }
 120
 121         public function valid()
 122         {
 123                 return ! is_null($this->value);
 124         }
 125 }