seeddms-code/SeedDMS_Core/Core/inc.ClassIterator.php

<?php
namespace SeedDMS\Core;

/**
 * Implementation of the document iterartor
 *
 * @category   DMS
 * @package    SeedDMS_Core
 * @license    GPL 2
 * @version    @version@
 * @author     Uwe Steinmann <uwe@steinmann.cx>
 * @copyright  Copyright (C) 2010, Uwe Steinmann
 * @version    Release: @package_version@
 */

class DocumentIterator implements \Iterator {
	/**
	 * @var object folder
	 */
	protected $_folder;

	/**
	 * @var object dms
	 */
	protected $_dms;

	/**
	 * @var array documents
	 */
	protected $_documents;

	public function __construct($folder) {
		$this->_folder = $folder;
		$this->_dms = $folder->getDMS();
		$this->_documents = array();
		$this->_pointer = 0;
		$this->_cache = array();
		$this->populate();
	}

	public function rewind() {
		$this->_pointer = 0;
	}

	public function valid() {
		return isset($this->_documents[$this->_pointer]);
	}

	public function next() {
		$this->_pointer++;
	}

	public function key() {
		return $this->_folders[$this->_pointer];
	}

	public function current() {
		if($this->_documents[$this->_pointer]) {
			$documentid = $this->_documents[$this->_pointer]['id'];
			if(!isset($this->_cache[$documentid])) {
//				echo $documentid." not cached<br />";
				$this->_cache[$documentid] = $this->_dms->getdocument($documentid);
			}
			return $this->_cache[$documentid];
		}
		return null;
	}

	private function populate($orderby="", $dir="asc", $limit=0, $offset=0) { /* {{{ */
		$db = $this->_dms->getDB();

		$queryStr = "SELECT `id` FROM `tblDocuments` WHERE `folder` = " . $this->_folder->getID();

		if ($orderby && $orderby[0]=="n") $queryStr .= " ORDER BY `name`";
		elseif ($orderby && $orderby[0]=="s") $queryStr .= " ORDER BY `sequence`";
		elseif ($orderby && $orderby[0]=="d") $queryStr .= " ORDER BY `date`";
		if($dir == 'desc')
			$queryStr .= " DESC";
		if(is_int($limit) && $limit > 0) {
			$queryStr .= " LIMIT ".$limit;
			if(is_int($offset) && $offset > 0)
				$queryStr .= " OFFSET ".$offset;
		}

		$resArr = $db->getResultArray($queryStr);
		if (is_bool($resArr) && $resArr == false)
			return false;

		$this->_documents = $resArr;
	} /* }}} */
}

class FolderIterator implements \Iterator { /* {{{ */
	/**
	 * @var object folder
	 */
	protected $_folder;

	/**
	 * @var object dms
	 */
	protected $_dms;

	/**
	 * @var array documents
	 */
	protected $_folders;

	public function __construct($folder) { /* {{{ */
		$this->_folder = $folder;
		$this->_dms = $folder->getDMS();
		$this->_folders = array();
		$this->_pointer = 0;
		$this->_cache = array();
		$this->populate();
	} /* }}} */

	public function rewind() { /* {{{ */
		$this->_pointer = 0;
	} /* }}} */

	public function valid() { /* {{{ */
		return isset($this->_folders[$this->_pointer]);
	} /* }}} */

	public function next() { /* {{{ */
		$this->_pointer++;
	} /* }}} */

	public function key() { /* {{{ */
		return $this->_folders[$this->_pointer];
	} /* }}} */

	public function current() { /* {{{ */
		if($this->_folders[$this->_pointer]) {
			$folderid = $this->_folders[$this->_pointer]['id'];
			if(!isset($this->_cache[$folderid])) {
//				echo $folderid." not cached<br />";
				$this->_cache[$folderid] = $this->_dms->getFolder($folderid);
			}
			return $this->_cache[$folderid];
		}
		return null;
	} /* }}} */

	private function populate($orderby="", $dir="asc", $limit=0, $offset=0) { /* {{{ */
		$db = $this->_dms->getDB();

		$queryStr = "SELECT `id` FROM `tblFolders` WHERE `parent` = " . $this->_folder->getID();

		if ($orderby && $orderby[0]=="n") $queryStr .= " ORDER BY `name`";
		elseif ($orderby && $orderby[0]=="s") $queryStr .= " ORDER BY `sequence`";
		elseif ($orderby && $orderby[0]=="d") $queryStr .= " ORDER BY `date`";
		if($dir == 'desc')
			$queryStr .= " DESC";
		if(is_int($limit) && $limit > 0) {
			$queryStr .= " LIMIT ".$limit;
			if(is_int($offset) && $offset > 0)
				$queryStr .= " OFFSET ".$offset;
		}

		$resArr = $db->getResultArray($queryStr);
		if (is_bool($resArr) && $resArr == false)
			return false;

		$this->_folders = $resArr;
	} /* }}} */
} /* }}} */

/**
 * The FolderFilterIterator checks if the given user has access on
 * the current folder.
 * FilterIterator uses an inner iterator passed to the constructor
 * to iterate over the sub folders of a folder.
 *
		$iter = new FolderIterator($folder);
		$iter2 = new FolderFilterIterator($iter, $user);
		foreach($iter2 as $ff) {
			echo $ff->getName()."<br />";
		}
 */
class FolderFilterIterator extends \FilterIterator { /* {{{ */
	public function __construct(Iterator $iterator , $filter ) {
		parent::__construct($iterator);
		$this->userFilter = $filter;
	}
	public function accept() { /* {{{ */
		$folder = $this->getInnerIterator()->current();
		echo "accept() for ".$folder->getName()."<br />";
		return true;
	} /* }}} */
} /* }}} */

/**
		$iter = new RecursiveFolderIterator($folder);
		$iter2 = new RecursiveIteratorIterator($iter, RecursiveIteratorIterator::SELF_FIRST);
		foreach($iter2 as $ff) {
			echo $ff->getID().': '.$ff->getName()."<br />";
		}
 */
class RecursiveFolderIterator extends FolderIterator implements \RecursiveIterator { /* {{{ */

	public function hasChildren() { /* {{{ */
		$db = $this->_dms->getDB();
		$queryStr = "SELECT id FROM `tblFolders` WHERE `parent` = ".(int) $this->current()->getID();
		$resArr = $db->getResultArray($queryStr);
		if (is_bool($resArr) && !$resArr)
			return false;
		return true;
	} /* }}} */

	public function getChildren() { /* {{{ */
		return new RecursiveFolderIterator($this->current());
	} /* }}} */
} /* }}} */

class RecursiveFolderFilterIterator extends FolderFilterIterator { /* {{{ */
	public function hasChildren() { /* {{{ */
		$db = $this->_dms->getDB();
		$queryStr = "SELECT id FROM `tblFolders` WHERE `parent` = ".(int) $this->current()->getID();
		$resArr = $db->getResultArray($queryStr);
		if (is_bool($resArr) && !$resArr)
			return false;
		return true;
	} /* }}} */

	public function getChildren() { /* {{{ */
		return new RecursiveFolderIterator($this->current());
	} /* }}} */

} /* }}} */