88 lines
3 KiB
PHP
88 lines
3 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Shimmie2;
|
|
|
|
require_once "mime_map.php";
|
|
require_once "file_extension.php";
|
|
require_once "mime_type.php";
|
|
|
|
class MimeSystem extends Extension
|
|
{
|
|
/** @var MimeSystemTheme */
|
|
protected Themelet $theme;
|
|
|
|
public const VERSION = "ext_mime_version";
|
|
|
|
public function onParseLinkTemplate(ParseLinkTemplateEvent $event): void
|
|
{
|
|
$event->replace('$ext', $event->image->get_ext());
|
|
$event->replace('$mime', $event->image->get_mime());
|
|
}
|
|
|
|
|
|
public function onDatabaseUpgrade(DatabaseUpgradeEvent $event): void
|
|
{
|
|
global $database;
|
|
|
|
// These upgrades are primarily for initializing mime types on upgrade, and for adjusting mime types whenever an
|
|
// adjustment needs to be made to the mime types.
|
|
|
|
if ($this->get_version(self::VERSION) < 1) {
|
|
if ($database->is_transaction_open()) {
|
|
// Each of these commands could hit a lot of data, combining
|
|
// them into one big transaction would not be a good idea.
|
|
$database->commit();
|
|
}
|
|
$database->set_timeout(null); // These updates can take a little bit
|
|
|
|
$extensions = $database->get_col_iterable("SELECT DISTINCT ext FROM images");
|
|
|
|
foreach ($extensions as $ext) {
|
|
$mime = MimeType::get_for_extension($ext);
|
|
|
|
if (empty($mime) || $mime === MimeType::OCTET_STREAM) {
|
|
throw new UserError("Unknown extension: $ext");
|
|
}
|
|
|
|
$normalized_extension = FileExtension::get_for_mime($mime);
|
|
|
|
$database->execute(
|
|
"UPDATE images SET mime = :mime, ext = :new_ext WHERE ext = :ext AND (mime IS NULL OR mime != :mime OR ext != :new_ext)",
|
|
["mime" => $mime, "new_ext" => $normalized_extension, "ext" => $ext]
|
|
);
|
|
}
|
|
|
|
$this->set_version(self::VERSION, 1);
|
|
$database->begin_transaction();
|
|
}
|
|
}
|
|
|
|
public function onHelpPageBuilding(HelpPageBuildingEvent $event): void
|
|
{
|
|
if ($event->key === HelpPages::SEARCH) {
|
|
$block = new Block();
|
|
$block->header = "File Types";
|
|
$block->body = $this->theme->get_help_html();
|
|
$event->add_block($block);
|
|
}
|
|
}
|
|
|
|
public function onSearchTermParse(SearchTermParseEvent $event): void
|
|
{
|
|
if (is_null($event->term)) {
|
|
return;
|
|
}
|
|
|
|
$matches = [];
|
|
// check for tags first as tag based searches are more common.
|
|
if (preg_match("/^ext[=|:]([a-zA-Z0-9]+)$/i", $event->term, $matches)) {
|
|
$ext = strtolower($matches[1]);
|
|
$event->add_querylet(new Querylet('images.ext = :ext', ["ext" => $ext]));
|
|
} elseif (preg_match("/^mime[=|:](.+)$/i", $event->term, $matches)) {
|
|
$mime = strtolower($matches[1]);
|
|
$event->add_querylet(new Querylet("images.mime = :mime", ["mime" => $mime]));
|
|
}
|
|
}
|
|
}
|