summaryrefslogtreecommitdiff
path: root/ext/spl/examples/sub_dir.inc
blob: b18ddbfe4126d4fd650dcd2ebc5c80b1888891c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
<?php

/**
 * @brief   Subdirectory aware directory iterator.
 * @author  Marcus Boerger
 * @version 1.0
 *
 * This directory iterator recursively returns all files and directories
 * within a given path.
 */
class sub_dir implements spl_sequence
{
	protected $adir = array();
	protected $cnt  = 0;
	protected $path = "";
	protected $curr = "";
	protected $nodots = true;

	/**
	 * Construct a directory from a path. 
	 *
	 * @param path    The path to iterate.
	 * @param nodots  Whether or not to display the entries '.' and '..'.
	 */
	function __construct($path, $nodots = true) {
		$this->cnt = 0;
		$this->path = $path;
	}
	
	/**
	 * Rewind the directory.
	 *
	 * @return void
	 */
	function rewind() {
	    while($this->cnt) {
	    	unset($this->adir[$this->cnt--]);
	    }
		$dir = new spl_dir($this->path);
		$dir->path = "";
		$this->adir[1] = $dir;
		$this->cnt = 1;
		if ($this->nodots) {
			while ($this->has_more()) {
				$ent = $this->current();
				if ($ent != '.' && $ent != '..') {
					break;
				}
				$this->next();
			}
		}
	}

	/**
	 * Move to net dir or file entry.
	 *
	 * @return void
	 */
	function next() {
		if ($this->cnt) {
			$dir = $this->adir[$this->cnt];
			$ent = $dir->current();
			$path = $dir->get_path().'/'.$ent;
			if ($ent != '.' && $ent != '..' && is_dir($path)) {
				$new = new spl_dir($path);
				$new->path = $dir->path.$ent.'/';
				$new->cnt = $this->cnt++;
				$this->adir[$this->cnt] = $new;
				if ($this->nodots) {
					while ($new->has_more()) {
						$ent = $new->current();
						if ($ent != '.' && $ent != '..') {
							break;
						}
						$new->next();
					}
				}
			}
			$dir->next();
		}
	}

	/**
	 * @return Whether more dirs or files entries are available.
	 */
	function has_more() {
		while ($this->cnt) {
			$dir = $this->adir[$this->cnt];
			if ($dir->has_more()) {
				return true;
			}
			unset($this->adir[$this->cnt--]);
		}
		return false;
	}

	/**
	 * @return The current dir or file entry.
	 */
	function current() {
		if ($this->cnt) {
			$dir = $this->adir[$this->cnt];
			return $dir->path . $dir->current();
		}
		throw new exception("No more elements available");
	}

	/**
	 * Hidden __clone
	 */
	protected function __clone() {
		// disallow clone 
	}
}

?>