Add catalog for Ukko

1 files changed, 141 insertions, 52 deletions

diff --git a/templates/themes/catalog/theme.php b/templates/themes/catalog/theme.php
index 34b28e48..9ecdb9d1 100644
--- a/templates/themes/catalog/theme.php
+++ b/templates/themes/catalog/theme.php
@@ -1,68 +1,144 @@
 <?php
 	require 'info.php';
-	
+
 	function catalog_build($action, $settings, $board) {
 		global $config;
-		
+
+		$b = new Catalog($settings);
+		$boards = explode(' ', $settings['boards']);
+
 		// Possible values for $action:
 		//	- all (rebuild everything, initialization)
 		//	- news (news has been updated)
 		//	- boards (board list changed)
 		//	- post (a reply has been made)
 		//	- post-thread (a thread has been made)
-		
-		$boards = explode(' ', $settings['boards']);
-				
-		if ($action == 'all') {
+		if ($action === 'all') {
 			foreach ($boards as $board) {
-				$b = new Catalog();
 				if ($config['smart_build']) {
 					file_unlink($config['dir']['home'] . $board . '/catalog.html');
-				}
-				else {
-					$b->build($settings, $board);
+				} else {
+					$b->build($board);
 				}
 			}
-		} elseif ($action == 'post-thread' || ($settings['update_on_posts'] && $action == 'post') || ($settings['update_on_posts'] && $action == 'post-delete') && in_array($board, $boards)) {
-			$b = new Catalog();
-
+		} elseif (in_array($board, $boards) &&
+			$action == 'post-thread' ||
+			($settings['update_on_posts'] && $action == 'post') ||
+			($settings['update_on_posts'] && $action == 'post-delete'))
+		{
 			if ($config['smart_build']) {
 				file_unlink($config['dir']['home'] . $board . '/catalog.html');
+			} else {
+				$b->build($board);
 			}
-			else {
-				$b->build($settings, $board);
-			}
+		}
+
+		// FIXME: Check that Ukko is actually enabled
+		if ($settings['enable_ukko'] && (
+			$action === 'all' || $action === 'post' ||
+			$action === 'post-thread' || $action === 'post-delete'))
+		{
+			$b->buildUkko();
 		}
 	}
-	
+
 	// Wrap functions in a class so they don't interfere with normal Tinyboard operations
 	class Catalog {
-		public function build($settings, $board_name) {
-			global $config, $board;
+		private $settings;
+		// Cache for threads from boards that have already been processed
+		private $threadsCache = array();
 
-			if ($board['uri'] != $board_name) {			
-				if (!openBoard($board_name)) {
-					error(sprintf(_("Board %s doesn't exist"), $board_name));
+		public function __construct($settings) {
+			$this->settings = $settings;
+		}
+
+		/**
+		 * Build and save the HTML of the catalog for the Ukko theme
+		 */
+		public function buildUkko() {
+			global $config;
+
+			$ukkoSettings = themeSettings('ukko');
+ 			$queries = array();
+			$threads = array();
+
+			$exclusions = explode(' ', $ukkoSettings['exclude']);
+			$boards = array_diff(listBoards(true), $exclusions);
+
+			foreach ($boards as $b) {
+				if (array_key_exists($b, $this->threadsCache)) {
+					$threads = array_merge($threads, $this->threadsCache[$b]);
+				} else {
+					$queries[] = $this->buildThreadsQuery($b);
 				}
 			}
-			
-			$recent_images = array();
-			$recent_posts = array();
-			$stats = array();
-			
-                        $query = query(sprintf("SELECT *, `id` AS `thread_id`,
-				(SELECT COUNT(`id`) FROM ``posts_%s`` WHERE `thread` = `thread_id`) AS `reply_count`,
-				(SELECT SUM(`num_files`) FROM ``posts_%s`` WHERE `thread` = `thread_id` AND `num_files` IS NOT NULL) AS `image_count`,
-				'%s' AS `board` FROM ``posts_%s`` WHERE `thread`  IS NULL ORDER BY `bump` DESC",
-			$board_name, $board_name, $board_name, $board_name, $board_name)) or error(db_error());
-			
-			while ($post = $query->fetch(PDO::FETCH_ASSOC)) {
+
+			// Fetch threads from boards that haven't beenp processed yet
+			if (!empty($queries)) {
+				$sql = implode(' UNION ALL ', $queries);
+				$res = query($sql) or error(db_error());
+				$threads = array_merge($threads, $res->fetchAll(PDO::FETCH_ASSOC));
+			}
+
+			// Sort in bump order
+			usort($threads, function($a, $b) {
+				return strcmp($b['bump'], $a['bump']);
+			});
+			// Generate data for the template
+			$recent_posts = $this->generateRecentPosts($threads);
+
+			$this->saveForBoard($ukkoSettings['uri'], $recent_posts,
+				$config['root'] . $ukkoSettings['uri']);
+		}
+
+		/**
+		 * Build and save the HTML of the catalog for the given board
+		 */
+		public function build($board_name) {
+			if (!openBoard($board_name)) {
+				error(sprintf(_("Board %s doesn't exist"), $post['board']));
+			}
+
+			if (array_key_exists($board_name, $this->threadsCache)) {
+				$threads = $this->threadsCache[$board_name];
+			} else {
+				$sql = $this->buildThreadsQuery($board_name);
+				$query = query($sql . ' ORDER BY `bump` DESC') or error(db_error());
+				$threads = $query->fetchAll(PDO::FETCH_ASSOC);
+				// Save for posterity
+				$this->threadsCache[$board_name] = $threads;
+			}
+
+			// Generate data for the template
+			$recent_posts = $this->generateRecentPosts($threads);
+
+			$this->saveForBoard($board_name, $recent_posts);
+		}
+
+		private function buildThreadsQuery($board) {
+			$sql  = "SELECT *, `id` AS `thread_id`, " .
+				"(SELECT COUNT(`id`) FROM ``posts_$board`` WHERE `thread` = `thread_id`) AS `reply_count`, " .
+				"(SELECT SUM(`num_files`) FROM ``posts_$board`` WHERE `thread` = `thread_id` AND `num_files` IS NOT NULL) AS `image_count`, " .
+				"'$board' AS `board` FROM ``posts_$board`` WHERE `thread` IS NULL";
+
+			return $sql;
+		}
+
+		private function generateRecentPosts($threads) {
+			global $config, $board;
+
+			$posts = array();
+			foreach ($threads as $post) {
+				if ($board['uri'] !== $post['board']) {
+					openBoard($post['board']);
+				}
+
 				$post['link'] = $config['root'] . $board['dir'] . $config['dir']['res'] . link_for($post);
 				$post['board_name'] = $board['name'];
 
 				if ($post['embed'] && preg_match('/^https?:\/\/(\w+\.)?(?:youtube\.com\/watch\?v=|youtu\.be\/)([a-zA-Z0-9\-_]{10,11})(&.+)?$/i', $post['embed'], $matches)) {
 					$post['youtube'] = $matches[2];
-				}				
+				}
 
 				if (isset($post['files']) && $post['files']) {
 					$files = json_decode($post['files']);
@@ -71,45 +147,58 @@
 						if ($files[0]->file == 'deleted') {
 							if (count($files) > 1) {
 								foreach ($files as $file) {
-									if (($file == $files[0]) || ($file->file == 'deleted')) continue;
+									if (($file == $files[0]) || ($file->file == 'deleted'))
+										continue;
 									$post['file'] = $config['uri_thumb'] . $file->thumb;
 								}
 
-								if (empty($post['file'])) $post['file'] = $config['image_deleted'];
-							}
-							else {
+								if (empty($post['file']))
+									$post['file'] = $config['image_deleted'];
+							} else {
 								$post['file'] = $config['image_deleted'];
 							}
-						}
-						else if($files[0]->thumb == 'spoiler') {
+						} else if($files[0]->thumb == 'spoiler') {
 							$post['file'] = '/' . $config['spoiler_image'];
-						}
-						else {
+						} else {
 							$post['file'] = $config['uri_thumb'] . $files[0]->thumb;
 						}
 					}
 				}
 
-				if (empty($post['image_count'])) $post['image_count'] = 0;
-				$recent_posts[] = $post;
+				if (empty($post['image_count']))
+					$post['image_count'] = 0;
+
+				$posts[] = $post;
+			}
+
+			return $posts;
+		}
+
+		private function saveForBoard($board_name, $recent_posts, $board_link = null) {
+			global $board, $config;
+
+			if ($board_link === null) {
+				$board_link = $config['root'] . $board['dir'];
 			}
-			
-			$required_scripts = array('js/jquery.min.js', 'js/jquery.mixitup.min.js', 'js/catalog.js');
 
+			$required_scripts = array('js/jquery.min.js', 'js/jquery.mixitup.min.js',
+				'js/catalog.js');
+
+			// Include scripts that haven't been yet included
 			foreach($required_scripts as $i => $s) {
 				if (!in_array($s, $config['additional_javascript']))
 					$config['additional_javascript'][] = $s;
 			}
 
 			file_write($config['dir']['home'] . $board_name . '/catalog.html', Element('themes/catalog/catalog.html', Array(
-				'settings' => $settings,
+				'settings' => $this->settings,
 				'config' => $config,
 				'boardlist' => createBoardlist(),
-				'recent_images' => $recent_images,
+				'recent_images' => array(),
 				'recent_posts' => $recent_posts,
-				'stats' => $stats,
+				'stats' => array(),
 				'board' => $board_name,
-				'link' => $config['root'] . $board['dir']
+				'link' => $board_link
 			)));
 		}
-	};
+	}