shimmie2/ext/autocomplete/main.php

<?php

declare(strict_types=1);

namespace Shimmie2;

class AutoComplete extends Extension
{
    public function get_priority(): int
    {
        return 30;
    } // before Home

    public function onPageRequest(PageRequestEvent $event): void
    {
        global $page;

        if ($event->page_matches("api/internal/autocomplete")) {
            $limit = (int)($_GET["limit"] ?? 1000);
            $s = $_GET["s"] ?? "";

            $res = $this->complete($s, $limit);

            $page->set_mode(PageMode::DATA);
            $page->set_mime(MimeType::JSON);
            $page->set_data(json_encode($res));
        }
    }

    private function complete(string $search, int $limit): array
    {
        global $cache, $database;

        $search = strtolower($search);
        if (
            $search == '' ||
            $search[0] == '_' ||
            $search[0] == '%' ||
            strlen($search) > 32
        ) {
            return [];
        }

        # memcache keys can't contain spaces
        $cache_key = "autocomplete:$limit:" . md5($search);
        $limitSQL = "";
        $search = str_replace('_', '\_', $search);
        $search = str_replace('%', '\%', $search);
        $SQLarr = [
            "search" => "$search%",
            "cat_search" => Extension::is_enabled(TagCategoriesInfo::KEY) ? "%:$search%" : "",
        ];
        if ($limit !== 0) {
            $limitSQL = "LIMIT :limit";
            $SQLarr['limit'] = $limit;
            $cache_key .= "-" . $limit;
        }

        return cache_get_or_set($cache_key, fn () => $database->get_pairs(
            "
                SELECT tag, count
                FROM tags
                WHERE (
                    LOWER(tag) LIKE LOWER(:search)
                    OR LOWER(tag) LIKE LOWER(:cat_search)
                )
                AND count > 0
                ORDER BY count DESC, tag ASC
                $limitSQL
                ",
            $SQLarr
        ), 600);
    }
}