silverstripe-dms/code/DMSShortcodeHandler.php

<?php

/**
 * Handles replacing `dms_document_link` shortcodes with links to the actual
 * document.
 *
 * @package dms
 */
class DMSShortcodeHandler
{

    public static function handle(
        $arguments,
        $content,
        ShortcodeParser $parser,
        $tag,
        array $extra = array()
    ) {
        if (!empty($arguments['id'])) {
            $document = DMSDocument::get()->byID($arguments['id']);

            if ($document && !$document->isHidden()) {
                if ($content) {
                    return sprintf(
                        '<a href="%s">%s</a>',
                        $document->Link(),
                        $parser->parse($content)
                    );
                } else {
                    if (isset($extra['element'])) {
                        $extra['element']->setAttribute('data-ext', $document->getExtension());
                        $extra['element']->setAttribute('data-size', $document->getFileSizeFormatted());
                    }

                    return $document->Link();
                }
            }
        }

        $error = ErrorPage::get()->filter('ErrorCode', '404')->First();

        if ($error) {
            return $error->Link();
        }

        return '';
    }
}