IsekaiWidgets/includes/TileWidget.php

<?php

namespace Isekai\Widgets;

use MediaWiki\Title\Title;
use MediaWiki\Html\Html;
use MediaWiki\Parser\Parser;
use PPFrame;
use MediaWiki\MediaWikiServices;

class TileWidget {
    private $size = 'medium';
    private $icon = false;
    private $title = '';
    private $content = '';
    private $href = '';
    private $badge = false;
    private $color = false;
    private $cover = false;
    private $images = [];
    private $grid = false;

    private $attributes = [];

    public function __construct($args, $content) {
        $this->content = $content;
        $this->parseArgs($args);
    }

    public static function create(string $text, array $args, Parser $parser, PPFrame $frame) {
        $parser->getOutput()->addModules(['ext.isekai.tile']);

        $content = '';
        if ($text) {
            $content = $frame->expand($text);

            $title = preg_replace('/\[\[.*?\]\]/', '', $content);
            $title = preg_replace('/<img .*?src="(?<src>.*?)".*?srcset="(?<srcset>.*?)"[^\>]+>/', '', $title);
            $title = strip_tags(trim($title));
            $args['title'] = $title;
        }

        $tile = new TileWidget($args, $content);
        return [$tile->toHtml(), 'markerType' => 'nowiki'];
    }

    private function parseArgs($args) {
        $allowedArgs = ['size', 'icon', 'title', 'cover', 'badge', 'color', 'href', 'grid'];

        foreach ($args as $name => $arg) {
            if (in_array($name, $allowedArgs)) {
                $this->$name = $arg;
            } elseif (substr($name, 0, 2) !== 'on') {
                $this->attributes[$name] = $arg;
            }
        }
    }

    private function getSizeArgs(array &$element, array &$content) {
        $element['data-size'] = $this->size;
        $element['class'][] = 'tile-' . $this->size;
    }

    private function getColorArgs(array &$element, array &$content) {
        if ($this->color) {
            if (substr($this->color, 0, 1) == '#' || substr($this->color, 0, 3) == 'rgb') {
                $element['style'][] = 'background-color: ' . $this->color;
            } else {
                $color = str_replace($this->color, 'bg-', '');
                $element['class'][] = 'bg-' . $color;
            }
        }
    }

    private function getTitleArgs(array &$element, array &$content) {
        if (!empty($this->title)) {
            $content[] = Html::element('span', [
                'class' => ['branding-bar'],
            ], $this->title);
            $element['data-title'] = $this->title;
        }
    }

    private function getCoverArgs(array &$element, array &$content) {
        $element['data-cover'] = $this->cover;
    }

    private function getHrefArgs(array &$element, array &$content) {
        if (substr($this->href, 0, 2) == '[[' && substr($this->href, -2, 2) == ']]') { //内部链接
            $titleText = substr($this->href, 2, strlen($this->href) - 4);
            $title = Title::newFromText($titleText);
            $href = $title->getLocalURL();
        } else {
            $href = $this->href;
        }
        $element['href'] = $href;
    }

    private function getIconArgs(array &$element, array &$content) {
        if ($this->icon) {
            if (is_string($this->icon)) {
                if (preg_match('/\.[a-zA-Z0-9]{3,4}$/', $this->icon)) {
                    //图片图标
                    $iconSrc = $this->icon;
                    $type = 'image';
                } else {
                    $iconSrc = explode(' ', $this->icon);
                    $type = 'class';
                }
            } else {
                $type = 'class';
                $iconSrc = $this->icon;
            }

            if ($type == 'class') {
                $content[] = Html::element('span', [
                    'class' => array_merge($iconSrc, ['icon']),
                ]);
            } elseif ($type == 'image') {
                $content[] = Html::element('img', [
                    'src' => $iconSrc,
                    'class' => ['icon'],
                ]);
            }
        }
    }

    private function getBadgeArgs(array &$element, array &$content) {
        if ($this->badge) {
            $content[] = Html::element('span', [
                'class' => ['badge-bottom'],
            ], strval($this->badge));
        }
    }

    private function getImagesArgs(array &$element, array &$content) {
        $service = MediaWikiServices::getInstance();
        $this->images = [];
        // 提取wikitext图片
        preg_match_all('/\[\[(?<title>.+?:.+?)(\|.*?)?\]\]/', $this->content, $matches);
        if (isset($matches['title']) && !empty($matches['title'])) {
            foreach ($matches['title'] as $titleText) {
                $title = Title::newFromText($titleText);
                if ($title->inNamespace(NS_FILE)) {
                    $file = $service->getRepoGroup()->findFile($title);
                    $thumb = $file->getUrl();
                    $this->images[] = $thumb;
                }
            }
        }

        // 提取html图片
        preg_match_all('/<img .*?src="(?<src>.*?)".*?srcset="(?<srcset>.*?)"[^\>]+>/', $this->content, $matches);
        if (isset($matches['src']) && !empty($matches['src'])) {
            $this->images = array_merge($this->images, $matches['src']);
        }

        if (!empty($this->images)) {
            $element['data-effect'] = 'image-set';
            foreach ($this->images as $image) {
                $content[] = Html::element('img', [
                    'src' => $image,
                    'style' => 'display: none'
                ]);
            }
        }
    }

    private function getGridArgs(array &$element, array &$content) {
        if ($this->grid) {
            $grid = explode(' ', $this->grid);
            $element['class'][] = 'col-' . $grid[0];
            if (count($grid) > 1) {
                $element['class'][] = 'row-' . $grid[1];
            }
        }
    }

    public function toHtml() {
        $element = array_merge($this->attributes, [
            'data-role' => 'tile',
        ]);
        $content = [];

        if (isset($element['class'])) {
            $element['class'] = explode(' ', $element['class']);
        } else {
            $element['class'] = [];
        }
        if (isset($element['style'])) {
            $element['style'] = explode(' ', $element['style']);
        } else {
            $element['style'] = [];
        }

        $this->getSizeArgs($element, $content);
        $this->getColorArgs($element, $content);
        $this->getIconArgs($element, $content);
        $this->getTitleArgs($element, $content);
        $this->getCoverArgs($element, $content);
        $this->getHrefArgs($element, $content);
        $this->getBadgeArgs($element, $content);
        $this->getImagesArgs($element, $content);
        $this->getGridArgs($element, $content);

        $content = implode('', $content);

        if (!empty($element['class'])) {
            $element['class'] = implode(' ', $element['class']);
        } else {
            unset($element['class']);
        }
        if (!empty($element['style'])) {
            $element['style'] = implode('; ', $element['style']) . ';';
        } else {
            unset($element['style']);
        }

        return Html::rawElement('a', $element, $content);
    }
}