forked from leftypol/leftypol
158 lines
6.4 KiB
PHP
158 lines
6.4 KiB
PHP
<?php
|
|
namespace Vichan\Data;
|
|
|
|
use Vichan\Functions\Net;
|
|
|
|
|
|
/**
|
|
* Browse user posts
|
|
*/
|
|
class UserPostQueries {
|
|
private const CURSOR_TYPE_PREV = 'p';
|
|
private const CURSOR_TYPE_NEXT = 'n';
|
|
|
|
private \PDO $pdo;
|
|
|
|
public function __construct(\PDO $pdo) {
|
|
$this->pdo = $pdo;
|
|
}
|
|
|
|
private function paginate(array $board_uris, int $page_size, ?string $cursor, callable $callback): PageFetchResult {
|
|
// Decode the cursor.
|
|
if ($cursor !== null) {
|
|
list($cursor_type, $uri_id_cursor_map) = Net\decode_cursor($cursor);
|
|
} else {
|
|
// Defaults if $cursor is an invalid string.
|
|
$cursor_type = null;
|
|
$uri_id_cursor_map = [];
|
|
}
|
|
$next_cursor_map = [];
|
|
$prev_cursor_map = [];
|
|
$rows = [];
|
|
|
|
foreach ($board_uris as $uri) {
|
|
// Extract the cursor relative to the board.
|
|
$start_id = null;
|
|
if ($cursor_type !== null && isset($uri_id_cursor_map[$uri])) {
|
|
$value = $uri_id_cursor_map[$uri];
|
|
if (\is_numeric($value)) {
|
|
$start_id = (int)$value;
|
|
}
|
|
}
|
|
|
|
$posts = $callback($uri, $cursor_type, $start_id, $page_size);
|
|
|
|
$posts_count = \count($posts);
|
|
|
|
if ($posts_count === $page_size + 2) {
|
|
$has_extra_prev_post = true;
|
|
$has_extra_end_post = true;
|
|
} elseif ($posts_count === $page_size + 1) {
|
|
$has_extra_prev_post = $start_id !== null && $start_id === (int)$posts[0]['id'];
|
|
$has_extra_end_post = !$has_extra_prev_post;
|
|
} else {
|
|
$has_extra_prev_post = false;
|
|
$has_extra_end_post = false;
|
|
}
|
|
|
|
// Since we fetched one post bellow and/or above the limit, we always know if there are any posts after the current page.
|
|
|
|
// Get the previous cursor, if any.
|
|
if ($has_extra_prev_post) {
|
|
\array_shift($posts);
|
|
$posts_count--;
|
|
// Select the most recent post.
|
|
$prev_cursor_map[$uri] = $posts[0]['id'];
|
|
}
|
|
// Get the next cursor, if any.
|
|
if ($has_extra_end_post) {
|
|
\array_pop($posts);
|
|
// Select the oldest post.
|
|
$next_cursor_map[$uri] = $posts[$posts_count - 2]['id'];
|
|
}
|
|
|
|
$rows[$uri] = $posts;
|
|
}
|
|
|
|
$res = new PageFetchResult();
|
|
$res->by_uri = $rows;
|
|
$res->cursor_prev = !empty($prev_cursor_map) ? Net\encode_cursor(self::CURSOR_TYPE_PREV, $prev_cursor_map) : null;
|
|
$res->cursor_next = !empty($next_cursor_map) ? Net\encode_cursor(self::CURSOR_TYPE_NEXT, $next_cursor_map) : null;
|
|
|
|
return $res;
|
|
}
|
|
|
|
/**
|
|
* Fetch a page of user posts.
|
|
*
|
|
* @param array $board_uris The uris of the boards that should be included.
|
|
* @param string $ip The IP of the target user.
|
|
* @param integer $page_size The Number of posts that should be fetched.
|
|
* @param string|null $cursor The directional cursor to fetch the next or previous page. Null to start from the beginning.
|
|
* @return PageFetchResult
|
|
*/
|
|
public function fetchPaginatedByIp(array $board_uris, string $ip, int $page_size, ?string $cursor = null): PageFetchResult {
|
|
return $this->paginate($board_uris, $page_size, $cursor, function($uri, $cursor_type, $start_id, $page_size) use ($ip) {
|
|
if ($cursor_type === null) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `ip` = :ip ORDER BY `sticky` DESC, `id` DESC LIMIT :limit', $uri));
|
|
$query->bindValue(':ip', $ip);
|
|
$query->bindValue(':limit', $page_size + 1, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return $query->fetchAll(\PDO::FETCH_ASSOC);
|
|
} elseif ($cursor_type === self::CURSOR_TYPE_NEXT) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `ip` = :ip AND `id` <= :start_id ORDER BY `sticky` DESC, `id` DESC LIMIT :limit', $uri));
|
|
$query->bindValue(':ip', $ip);
|
|
$query->bindValue(':start_id', $start_id, \PDO::PARAM_INT);
|
|
$query->bindValue(':limit', $page_size + 2, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return $query->fetchAll(\PDO::FETCH_ASSOC);
|
|
} elseif ($cursor_type === self::CURSOR_TYPE_PREV) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `ip` = :ip AND `id` >= :start_id ORDER BY `sticky` ASC, `id` ASC LIMIT :limit', $uri));
|
|
$query->bindValue(':ip', $ip);
|
|
$query->bindValue(':start_id', $start_id, \PDO::PARAM_INT);
|
|
$query->bindValue(':limit', $page_size + 2, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return \array_reverse($query->fetchAll(\PDO::FETCH_ASSOC));
|
|
} else {
|
|
throw new \RuntimeException("Unknown cursor type '$cursor_type'");
|
|
}
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Fetch a page of user posts.
|
|
*
|
|
* @param array $board_uris The uris of the boards that should be included.
|
|
* @param string $password The password of the target user.
|
|
* @param integer $page_size The Number of posts that should be fetched.
|
|
* @param string|null $cursor The directional cursor to fetch the next or previous page. Null to start from the beginning.
|
|
* @return PageFetchResult
|
|
*/
|
|
public function fetchPaginateByPassword(array $board_uris, string $password, int $page_size, ?string $cursor = null): PageFetchResult {
|
|
return $this->paginate($board_uris, $page_size, $cursor, function($uri, $cursor_type, $start_id, $page_size) use ($password) {
|
|
if ($cursor_type === null) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `password` = :password ORDER BY `sticky` DESC, `id` DESC LIMIT :limit', $uri));
|
|
$query->bindValue(':password', $password);
|
|
$query->bindValue(':limit', $page_size + 1, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return $query->fetchAll(\PDO::FETCH_ASSOC);
|
|
} elseif ($cursor_type === self::CURSOR_TYPE_NEXT) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `password` = :password AND `id` <= :start_id ORDER BY `sticky` DESC, `id` DESC LIMIT :limit', $uri));
|
|
$query->bindValue(':password', $password);
|
|
$query->bindValue(':start_id', $start_id, \PDO::PARAM_INT);
|
|
$query->bindValue(':limit', $page_size + 2, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return $query->fetchAll(\PDO::FETCH_ASSOC);
|
|
} elseif ($cursor_type === self::CURSOR_TYPE_PREV) {
|
|
$query = $this->pdo->prepare(sprintf('SELECT * FROM `posts_%s` WHERE `password` = :ip AND `password` >= :start_id ORDER BY `sticky` ASC, `id` ASC LIMIT :limit', $uri));
|
|
$query->bindValue(':password', $password);
|
|
$query->bindValue(':start_id', $start_id, \PDO::PARAM_INT);
|
|
$query->bindValue(':limit', $page_size + 2, \PDO::PARAM_INT); // Always fetch more.
|
|
$query->execute();
|
|
return \array_reverse($query->fetchAll(\PDO::FETCH_ASSOC));
|
|
} else {
|
|
throw new \RuntimeException("Unknown cursor type '$cursor_type'");
|
|
}
|
|
});
|
|
}
|
|
}
|