<?php
if (!defined('__TYPECHO_ROOT_DIR__')) exit;

/**
 * 根据内容标题关系自动生成目录树 【<a href="https://github.com/typecho-fans/plugins" target="_blank">TF</a>社区维护版】
 *
 * @package Menu Tree
 * @author BeaconFire,Melon
 * @version 0.1.2
 * @link https://github.com/typecho-fans/plugins/tree/master/MenuTree
 */

// 使用方法：
// 在文章某处地方加上<!-- index-menu -->，程序会把这个注释替换成目录树

// 样式：
// .index-menu			整个目录
// .index-menu-list	列表 ul
// .index-menu-item	每个目录项 li
// .index-menu-link	目录项连接 a

// 独立模式下需要在主题模板中调用：$this->treeMenu();

class MenuTree_Plugin implements Typecho_Plugin_Interface
{

    /**
     * 索引ID
     */
    public static $id = 1;

    public static $pattern = '/(&lt;|<)!--\s*index-menu\s*--(&gt;|>)/i';

    /**
     * 标题匹配模式
     */
    public static $patternTitle = '/<h([1-6])[^>]*>.*?<\/h\1>/s';

    /**
     * 目录树
     */
    public static $tree = array();

    /**
     * 激活插件方法,如果激活失败,直接抛出异常
     *
     * @access public
     * @return void
     */
    public static function activate()
    {
        Typecho_Plugin::factory('admin/write-post.php')->bottom = array('MenuTree_Plugin', 'render');
        Typecho_Plugin::factory('admin/write-page.php')->bottom = array('MenuTree_Plugin', 'render');
        Typecho_Plugin::factory('Widget_Abstract_Contents')->contentEx = array('MenuTree_Plugin', 'contentEx');
        Typecho_Plugin::factory('Widget_Abstract_Contents')->excerptEx = array('MenuTree_Plugin', 'excerptEx');
        Typecho_Plugin::factory('Widget_Archive')->___treeMenu = array('MenuTree_Plugin', 'treeMenu');
    }


    /**
     * 禁用插件方法,如果禁用失败,直接抛出异常
     *
     * @static
     * @access public
     * @return void
     */
    public static function deactivate()
    {
        //do nothing
    }

    /**
     * 获取插件配置面板
     *
     * @access public
     * @param Typecho_Widget_Helper_Form $form 配置面板
     * @return void
     */
    public static function config(Typecho_Widget_Helper_Form $form)
    {
        $type = new Typecho_Widget_Helper_Form_Element_Checkbox('switch',
            array('normal' => _t('嵌入模式'), 'single' => _t('独立模式')),
            array('normal'),
            _t('显示模式'));

        $form->addInput($type);
    }

    /**
     * 个人用户的配置面板
     *
     * @access public
     * @param Typecho_Widget_Helper_Form $form
     * @return void
     */
    public static function personalConfig(Typecho_Widget_Helper_Form $form)
    {
        //do nothing
    }

    /**
     * 列表页忽略目录生成标记
     *
     * @access public
     * @return string
     */
    public static function excerptEx($html, $widget, $lastResult)
    {
        return self::rmMenuTag($html);
    }

    /**
     * 内容页构造索引目录
     *
     * @access public
     * @return string
     * @throws Typecho_Plugin_Exception
     */
    public static function contentEx($html, $widget, $lastResult)
    {
        $html = empty($lastResult) ? $html : $lastResult;

        $options = Helper::options()->plugin('MenuTree');

        /**开关判断*/
        if (!is_null($options->switch)
            && (in_array('single', $options->switch) or in_array('normal', $options->switch))) {

            $html = preg_replace_callback(self::$patternTitle, array('MenuTree_Plugin', 'parseCallback'), $html);

            if (in_array('normal', $options->switch)) {
                $html = preg_replace(self::$pattern, '<div class="index-menu">' . self::buildMenuHtml(self::$tree) . '</div>', $html);
            } else {
                $html = self::rmMenuTag($html);
            }

            self::$id = 1;
            self::$tree = array();
            return $html;
        }

        $html = self::rmMenuTag($html);
        return $html;
    }

    /**
     * 构造独立的索引目录
     *
     * @param $archive
     * @return string
     * @throws Typecho_Plugin_Exception
     */
    public static function treeMenu($archive)
    {
        $options = Helper::options()->plugin('MenuTree');

        if (is_null($options->switch) || !in_array('single', $options->switch)) {
            return '';
        }

        preg_replace_callback(self::$patternTitle, array('MenuTree_Plugin', 'parseCallback'), $archive->content);
        $result = '<div class="index-menu">' . self::buildMenuHtml(self::$tree) . '</div>';
        self::$id = 1;
        self::$tree = array();
        return $result;
    }

    /**
     * 解析
     *
     * @access public
     * @param array $match 解析值
     * @return string
     */
    public static function parseCallback($match)
    {
        $parent = &self::$tree;

        $html = $match[0];
        $n = $match[1];
        $menu = array(
            'num' => $n,
            'title' => trim(strip_tags($html)),
            'id' => 'menu_index_' . self::$id,
            'sub' => array()
        );
        $current = array();
        if ($parent) {
            $current = &$parent[count($parent) - 1];
        }
        // 根
        if (!$parent || (isset($current['num']) && $n <= $current['num'])) {
            $parent[] = $menu;
        } else {
            while (is_array($current['sub'])) {
                // 父子关系
                if ($current['num'] == $n - 1) {
                    $current['sub'][] = $menu;
                    break;
                } // 后代关系，并存在子菜单
                elseif ($current['num'] < $n && $current['sub']) {
                    $current = &$current['sub'][count($current['sub']) - 1];
                } // 后代关系，不存在子菜单
                else {
                    for ($i = 0; $i < $n - $current['num']; $i++) {
                        $current['sub'][] = array(
                            'num' => $current['num'] + 1,
                            'sub' => array()
                        );
                        $current = &$current['sub'][0];
                    }
                    $current['sub'][] = $menu;
                    break;
                }
            }
        }
        self::$id++;
        return "<span class=\"menu-target-fix\" id=\"{$menu['id']}\" name=\"{$menu['id']}\"></span>" . $html;
    }

    /**
     * 构建目录树，生成索引
     *
     * @access public
     * @param $tree
     * @param bool $include
     * @return string
     */
    public static function buildMenuHtml($tree, $include = true)
    {

        $menuHtml = '';
        foreach ($tree as $menu) {
            /**默认给第一个链接添加class: .current */
            $current = ($menu['id'] == 'menu_index_1') ? 'current' : '';

            if (!isset($menu['id']) && $menu['sub']) {
                $menuHtml .= self::buildMenuHtml($menu['sub'], false);
            } elseif ($menu['sub']) {
                $menuHtml .= "<li class=\"index-menu-item\"><a data-scroll class=\"index-menu-link {$current}\" href=\"#{$menu['id']}\" title=\"{$menu['title']}\">{$menu['title']}</a>" . self::buildMenuHtml($menu['sub']) . "</li>";
            } else {
                $menuHtml .= "<li class=\"index-menu-item\"><a data-scroll class=\"index-menu-link {$current}\" href=\"#{$menu['id']}\" title=\"{$menu['title']}\">{$menu['title']}</a></li>";
            }
        }
        if ($include) {
            $menuHtml = '<ul class="index-menu-list">' . $menuHtml . '</ul>';
        }
        return $menuHtml;
    }

    /**
     * 删除文章中的菜单标记注释
     *
     * @param $html
     * @return string|string[]|null
     */
    public static function rmMenuTag($html)
    {
        $html = preg_replace(self::$pattern, '', $html);
        return $html;
    }

    /**
     * 编辑器插入短代码功能。
     * @access public
     * @return void
     */
    public static function render(){
?>
<script>
$(function(){
        var wmd = $('#wmd-image-button');
        if (wmd.length>0) {
            wmd.after(
        '<li class="wmd-button" id="wmd-mn-button" style="padding-top:5px;" title="<?php _e("插入目录树"); ?>"><img src="data:image/svg+xml,%3csvg xmlns=\'http://www.w3.org/2000/svg\' width=\'16\' height=\'16\' viewBox=\'0 0 24 24\'%3e%3cpath fill=\'%23999\' d=\'M22 18v-7h-9v-5h3v-6h-8v6h3v5h-9v7h-2v6h6v-6h-2v-5h7v5h-2v6h6v-6h-2v-5h7v5h-2v6h6v-6z\'/%3e%3c/svg%3e"/></li>');
        } else {
            $('.url-slug').after('<button type="button" id="wmd-mn-button" class="btn btn-xs" style="margin-right:5px;"><?php _e("插入目录树"); ?></button>');
        }
        $('#wmd-mn-button').click(function(){
            var textarea = $('#text'),
            mninput = '<!-- index-menu -->',
            sel = textarea.getSelection(),
            offset = (sel ? sel.start : 0)+mninput.length;
            textarea.replaceSelection(mninput);
            textarea.setSelection(offset,offset);
        });
});
</script>
<?php
    }

}