Show omitted image replies count

inc/display.php

@@ -194,7 +194,6 @@
 	};
 	
 	class Thread {
-		public $omitted = 0;
 		public function __construct($id, $subject, $email, $name, $trip, $body, $time, $thumb, $thumbx, $thumby, $file, $filex, $filey, $filesize, $filename, $ip, $sticky, $locked, $root=null, $mod=false) {
 			global $config;
 			if(!isset($root)) $root = $config['root'];
@@ -215,6 +214,7 @@
 			$this->filesize = $filesize;
 			$this->filename = $filename;
 			$this->omitted = 0;
+			$this->omitted_images = 0;
 			$this->posts = Array();
 			$this->ip = $ip;
 			$this->sticky = $sticky;
@@ -348,7 +348,15 @@
 			$built .= $this->body .
 			
 			// Omitted posts
-			($this->omitted ? '<span class="omitted">' . $this->omitted . ' post' . ($this->omitted==1?'':'s') . ' omitted. Click reply to view.</span>':'') .
+			($this->omitted || $this->omitted_images? '<span class="omitted">' .
+				($this->omitted ?
+					$this->omitted . ' post' . ($this->omitted==1?'':'s') .
+						($this->omitted_images ? ' and ' : '')
+				:'') .
+				($this->omitted_images ?
+					$this->omitted_images . ' image repl' . ($this->omitted_images==1?'y':'ies')
+				:'') .
+			' omitted. Click reply to view.</span>':'') .
 			
 			// End
 			'</div>';

inc/functions.php

@@ -440,22 +440,28 @@
 			$posts->bindValue(2, ($th['sticky'] ? $config['threads_preview_sticky'] : $config['threads_preview']), PDO::PARAM_INT);
 			$posts->execute() or error(db_error($posts));
 			
+			$num_images = 0;
+			while($po = $posts->fetch()) {
+				if($po['file'])
+					$num_images++;
+					
+				$thread->add(new Post($po['id'], $th['id'], $po['subject'], $po['email'], $po['name'], $po['trip'], $po['body'], $po['time'], $po['thumb'], $po['thumbwidth'], $po['thumbheight'], $po['file'], $po['filewidth'], $po['fileheight'], $po['filesize'], $po['filename'], $po['ip'], $mod ? '?/' : $config['root'], $mod));
+			}
+			
 			if($posts->rowCount() == ($th['sticky'] ? $config['threads_preview_sticky'] : $config['threads_preview'])) {
-				$count = prepare(sprintf("SELECT COUNT(`id`) as `num` FROM `posts_%s` WHERE `thread` = ?", $board['uri']));
+				$count = prepare(sprintf("SELECT COUNT(`id`) as `num` FROM `posts_%s` WHERE `thread` = :thread UNION ALL SELECT COUNT(`id`) FROM `posts_%s` WHERE `file` IS NOT NULL AND `thread` = :thread", $board['uri'], $board['uri']));
-				$count->bindValue(1, $th['id']);
+				$count->bindValue(':thread', $th['id'], PDO::PARAM_INT);
 				$count->execute() or error(db_error($count));
 				
-				$count = $count->fetch();
+				$c = $count->fetch();
-				$omitted = $count['num'] - ($th['sticky'] ? $config['threads_preview_sticky'] : $config['threads_preview']);
+				$thread->omitted = $c['num'] - ($th['sticky'] ? $config['threads_preview_sticky'] : $config['threads_preview']);
-				$thread->omitted = $omitted;
+				
-				unset($count);
+				$c = $count->fetch();
-				unset($omitted);
+				$thread->omitted_images = $c['num'] - $num_images;
+				
+				$thread->omitted -= $thread->omitted_images;
 			}
 			
-			while($po = $posts->fetch()) {
-				$thread->add(new Post($po['id'], $th['id'], $po['subject'], $po['email'], $po['name'], $po['trip'], $po['body'], $po['time'], $po['thumb'], $po['thumbwidth'], $po['thumbheight'], $po['file'], $po['filewidth'], $po['fileheight'], $po['filesize'], $po['filename'], $po['ip'], $mod ? '?/' : $config['root'], $mod));
-			}
-
 			$thread->posts = array_reverse($thread->posts);
 			$body .= $thread->build(true);
 		}