Shortcode WordPress : afficher les documents liés aux billets

<?php
/**
 * Add the new shortcode `attachments`.
 */
$attachments_list = new AttachmentsListShortCode('attachments');
$attachments_list->register();
 
/**
 * Shortcode to display a list of attachments of the current post.
 *
 * Attributes to use as follows:
 *     'exclude' - Default is none. List of IDs of the attachments to exclude of search.
 *     'orderby' - Default is 'title ASC'. How to order the attachments.
 *     'style'   - Default is 'large'. Presentation style to use.
 *
 * @author Mehdi Kabab <http://pioupioum.fr/>
 * @copyright Copyright (C) 2010 Mehdi Kabab
 * @license http://www.gnu.org/licenses/gpl.html  GNU GPL version 3 or later
 * @version 1.0.0
 * @link http://pioupioum.fr/outils-astuces/wordpress-shortcode-afficher-fichiers-joints.html
 */
class AttachmentsListShortCode extends PPM_Shortcode
{
    /**
     * Default options.
     *
     * @var array
     **/
    protected $_default_options = array(
        'exclude' => array(),
        'orderby' => 'title ASC',
        'style'   => null,
    );
 
    /**
     * Keys allowed in the query ORDER BY clause.
     *
     * @param array
     */
    protected $_allowed_orderby = array(
        'mime_type'
    );
 
    /**
     * Directory of MIME icons.
     * The path is relative to the current theme.
     *
     * @var string
     */
    private $_icons_dir;
 
    /**
     * Clause ORDER BY to be applied to sorting.
     *
     * @var string
     */
    private $_orderby = null;
 
    /**
     * Stack of the presentation styles for the attachments.
     *
     * @var ArrayObject
     **/
    private $_styles;
 
    /**
     * Constructor.
     *
     * @param string $tag Shortcode tag to be searched in post content.
     * @param string $icons_dir Directory of MIME icons.
     */
    public function __construct($tag, $icons_dir = 'images/mime')
    {
        parent::__construct($tag);
        $this->setIconsDirectory($icons_dir);
 
        $this->_styles = new ArrayObject(array(),
                                         ArrayObject::ARRAY_AS_PROPS);
        foreach ($this->_getDefaultStyles() as $tpl => $data) {
            if (isset($data['default'])) {
                $default = (boolean) $data['default'];
                unset($data['default']);
            } else {
                $default = false;
            }
            $this->setStyle($tpl, $data, $default);
        }
    }
 
    /**
     * Retrieve directory of MIME icons.
     *
     * @return string
     */
    public function getIconsDirectory()
    {
        return $this->_icons_dir;
    }
 
    /**
     * Set the directory of MIME icons.
     *
     * @return PPM_Shortcode
     */
    public function setIconsDirectory($dir)
    {
        $this->_icons_dir = trim($dir, '/') . '/';
 
        return $this;
    }
 
    /**
     * Retrieve the stack of the presentation styles.
     *
     * @return ArrayObject
     */
    public function getStyles()
    {
        return $this->_styles;
    }
 
    /**
     * Set a presentation style of the attachments.
     *
     * @param string $name Presentation style name.
     * @param array $data Style data.
     * @param boolean $default True for set this style as default style.
     * @return PPM_Shortcode
     * @throws Exception If the data are not valid.
     */
    public function setStyle($name, $data, $default = false)
    {
        if (!$this->_isValidStyle($data)) {
            throw new Exception(sprintf('Data for the style `%s` are not valid!',
                                        $name));
        }
        $this->_styles[$name] = new ArrayObject($data,
                                                   ArrayObject::ARRAY_AS_PROPS);
        if (true === $default) {
            $this->setStyleByDefault($name);
        }
 
        return $this;
    }
 
    /**
     * Retrieve name of the presentation style used by default.
     *
     * @return string
     */
    public function getStyleByDefault()
    {
        return $this->_default_options['style'];
    }
 
    /**
     * Set name of the presentation style to use by default.
     *
     * @param string $name
     * @return PPM_Shortcode
     */
    public function setStyleByDefault($name)
    {
        $this->_default_options['style'] = $name;
 
        return $this;
    }
 
    /**
     * [filter] Edit the clause ORDER BY to allow sorting by mime_type.
     *
     * @param string $orderby The ORDER BY clause.
     * @return string The new ORDER BY clause.
     */
    public function filterOrderby($orderby)
    {
        if (null !== $this->_orderby && in_array($this->_orderby, $this->_allowed_orderby)) {
            $orderby = preg_replace('#^([^\.]+\.)(\w+)(\s+ASC|DESC)$#i',
                                    '$1post_' . $this->_orderby . '$3',
                                    $orderby);
        }
 
        return $orderby;
    }
 
    /**
     * @see PPM_Shortcode::_process()
     */
    protected function _process($attributes, $content = null)
    {
        if ($this->getStyles()->offsetExists($this->getOption('style'))) {
            $style = $this->getOption('style');
        } else {
            $style = $this->getStyleByDefault();
        }
 
        // Order and Order by clauses
        list($orderby, $order) = explode(' ', $this->_default_options['orderby']);
        if (false !== strpos($this->getOption('orderby'), ' ')) {
            list($orderby, $order) = explode(' ', $this->getOption('orderby'));
        } else {
            $orderby = $this->getOption('orderby');
        }
        $this->_orderby = $orderby;
 
        // Retrieve the attachments
        $attachments = get_children(array(
            'post_parent' => get_the_ID(),
            'post_type'   => 'attachment',
            'numberposts' => -1,
            'exclude'     => $this->getOption('exclude'),
            'orderby'     => $this->_orderby,
            'order'       => ('ASC' === strtoupper($order)) ? 'ASC' : 'DESC',
            'exclude'     => $this->getOption('exclude'),
            'suppress_filters' => false // Allows to edit the query
        ));
        if (false === $attachments) {
            return '';
        }
 
        // Edit the query
        add_filter('posts_orderby', array($this, 'filterOrderby'));
 
        $result = '';
        $tpl = $this->getStyles()->$style;
        foreach ($attachments as $attachment) {
            // Do not display image attachments
            if (0 === strpos($attachment->post_mime_type, 'image/')) {
                continue;
            }
 
            // Determine the filename
            $file_type = strtr($attachment->post_mime_type, '/-', '__');
            $file_name = $this->getIconsDirectory() . $file_type . '.png';
            if (file_exists(get_stylesheet_directory() . '/' . $file_name))
            {
                $icon_file = get_bloginfo('stylesheet_directory', false) .
                             '/' . $file_name;
            }
            else
            {
                $icon_file = wp_mime_type_icon($attachment->ID);
            }
 
            if (function_exists('getimagesize')) {
                $size_attr =  getimagesize($icon_file);
                $size_attr = $size_attr[3];
            } else {
                $size_attr = '';
            }
 
            $file_path = get_attached_file($attachment->ID);
            $context = array(
                'ID'        => $attachment->ID,
                'filename'  => basename($attachment->guid),
                'file_date' => mysql2date(get_option('date_format'),
                                          $attachment->post_date),
                'file_size' => esc_attr(size_format(filesize($file_path))),
                'file_type' => $attachment->post_mime_type,
                'file_url'  => esc_attr($attachment->guid),
                'icon_atts' => $size_attr,
                'icon_path' => esc_attr($icon_file),
                'title'     => esc_attr(get_the_title($attachment->ID)),
                'label_filename'  => __('File name:'),
                'label_file_date' => __('Upload date:'),
                'label_file_size' => __('Size:')
            );
 
            // Hook for customize the context
            $context = apply_filters('attachments_list_context',
                                     $context,
                                     $attachment,
                                     $this->getStyles()->$style);
 
            $result .= $this->_compileMarkup($tpl, $context);
        }
 
        return $tpl->before . do_shortcode($content) . $result . $tpl->after;
    }
 
    /**
     * Performs substitutions in the markup of the selected presentation style.
     *
     * @param ArrayObject $tpl
     * @param array $context
     * @return string The markup after substitutions.
     */
    protected function _compileMarkup($tpl, $context)
    {
        $markup = $tpl->markup;
        foreach ($context as $key => $value) {
            $markup = str_replace('%' . $key . '%', $value, $markup);
        }
 
        return $markup;
    }
 
    /**
     * Retrieve the default stack of the presentation styles.
     *
     * Declare the `medium`, `small` and `large` presentation styles.
     *
     * @return array
     */
    protected function _getDefaultStyles()
    {
        return array(
            'medium' => array(
                'before' => '<div class="attachments-list-medium">',
                'after'  => '</div>',
                'markup' => '<div class="attachment-item attachment-%ID%">' .
                            '<a href="%file_url%" title="%title% (%file_size%)" rel="attachment">' .
                            '<img src="%icon_path%"%icon_atts% alt=""/><small>%title%</small></a>' . 
                            '</div>'
            ),
            'small'  => array(
                'before' => '<div class="attachments-list-small">',
                'after'  => '</div>',
                'markup' => '<div class="attachment-small-icon attachment-%ID%">' .
                            '<img src="%icon_path%"%icon_atts% alt=""/>' .
                            '<a href="%file_url%" title="%title%" rel="attachment">' .
                            '%title%</a>&nbsp;(%file_size%)' . 
                            '</div>'
            ),
            'large'  => array(
                'before' => '<div class="attachments-list-large">',
                'after'  => '</div>',
                'markup' => '<dl class="attachment-item attachment-%ID%">' .
                            '<dt><a href="%file_url%" title="%title%" rel="attachment">' .
                            '<img class="alignleft" src="%icon_path%"%icon_atts% alt=""/>' .
                            '%title%</a></dt>' .
                            '<dd><span>%label_filename%</span> %filename%</dd>' .
                            '<dd><span>%label_file_date%</span> %file_date%</dd>' .
                            '<dd><span>%label_file_size%</span> %file_size%</dd>' .
                            '</dl>',
                'default' => true
            )
        );
    }
 
    /**
     * Test if the data of a presentation style are well formed.
     *
     * @param array $data
     * @return boolean
     */
    protected function _isValidStyle($data)
    {
        $ok = 3 === count($data) &&
              isset($data['before']) &&
              isset($data['after']) &&
              isset($data['markup']);
 
        return $ok;
    }
 
}
?>

.attachments-list-large, .attachments-list-medium, .attachments-list-small {
  margin-bottom: 1em;
}
.attachments-list-large .attachment-item img, .attachments-list-medium .attachment-item img, .attachments-list-small .attachment-item img {
    border: none;
}
.attachments-list-large .attachment-item dl {
    margin: 0 0 1.5em 1em;
}
.attachments-list-large .attachment-item dt {
    font-weight: bold;
}
.attachments-list-large .attachment-item dd {
    font-size: 0.85em;
    margin-left: 1.5em;
}
.attachments-list-large .attachment-item img {
    display: inline;
    float: left;
}
.attachments-list-large .attachment-item img {
    margin-right: 1em;
}
.attachments-list-medium .attachment-item {
    display: inline-table;
    margin-bottom: 1.5em;
    padding: 0 0.15em;
    text-align: center;
    vertical-align: top;
    width: 10em;
}
.attachments-list-medium .attachment-item a {
    text-decoration: none;
}
.attachments-list-medium .attachment-item small {
    display: block;
    font-size: 0.85em;
}
.attachments-list-small img {
    margin-right: 5px;
    vertical-align: text-bottom;
    height: 16px;
    width: 16px;
}

.attachments-list-large, .attachments-list-medium, .attachments-list-small {
    margin-bottom: 1em;
    .attachment-item {
        img {
            border: none;
        }
    }
}
.attachments-list-large {
    .attachment-item {
        $space: 1em;
        dl {
            margin: 0 0 ($space + 0.5em) $space;
        }
        dt {
            font-weight: bold;
        }
        dd {
            font-size: 0.85em;
            margin-left: ($space + 0.5em);
        }
        img {
            display: inline;
            float: left;
            margin: {
                right: $space;
            }
        }
    }
}
.attachments-list-medium {
    .attachment-item {
        display: inline-table;
        margin-bottom: 1.5em;
        padding: 0 0.15em;
        text-align: center;
        width: 10em;
        a {
            text-decoration: none;
        }
        small {
            display: block;
            font-size: 0.85em;
        }
    }
}
.attachments-list-small {
    img {
        margin-right: 5px;
        vertical-align: text-bottom;
        height: 16px;
        width: 16px;
    }
}

<?php
$attachments_list = new AttachmentsListShortCode('attachments');
$ol_style = array(
    'before' => '<ol class="attachments-list-ol">',
    'after'  => '</ol>',
    'markup' => '<li><a href="%file_url%" rel="attachment">%title%</a></li>'
);
$attachments_list->setStyle('ol', $ol_style)
                 ->register();
?>

<?php
$attachments_list = new AttachmentsListShortCode('pièces jointes');
$attachments_list->setIconsDirectory('img/icons')
                 ->register();
?>

<?php
function filter_attachments_list_context($context, $attachment, $size)
{
    // WordPress ne comportant pas de chaîne 'Size:', cette
    // dernière ne peut être localisée automatiquement.
    // Travaillant sur un site exclusivement localisé en français
    // on peut se permettre les deux traitements qui suivent.
    $context['label_file_size'] = 'Taille&nbsp;:';
    $context['file_size'] = strtr($context['file_size'], 'B', 'o');
 
    return $context;
}
 
add_filter('attachments_list_context', 'filter_attachments_list_context', 10, 3);
?>