<?php
/**
 * @version   $Id: CategoryPopulator.php 29018 2015-10-15 21:04:25Z jakub $
 * @author    RocketTheme http://www.rockettheme.com
 * @copyright Copyright (C) 2007 - 2017 RocketTheme, LLC
 * @license   http://www.gnu.org/licenses/gpl-2.0.html GNU/GPLv2 only
 */

class RokSprocket_Provider_Wordpress_CategoryPopulator implements RokCommon_Filter_IPicklistPopulator
{
    /**
     *
     * @return array;
     */
    public function getPicklistOptions()
    {
        $tax_args = array(
            'public'   => true,
            '_builtin' => false
        );

        $taxonomies = get_taxonomies($tax_args);
        $taxonomies[] = 'category';

        $args = array(
        	'type'                     => 'post',
        	'child_of'                 => 0,
        	'orderby'                  => 'name',
        	'order'                    => 'ASC',
        	'hide_empty'               => 1,
        	'hierarchical'             => 1,
        	'taxonomy'                 => $taxonomies,
        	'pad_counts'               => true );

        $categories = get_categories( $args );

        $children = array ();
        if ($categories) {
            foreach ($categories as $v) {
                $v->title = $v->cat_name;
                $v->parent_id = $v->category_parent;
                $v->id = $v->cat_ID;
                $pt = $v->category_parent;
                $list = @$children[$pt]?$children[$pt]: array ();
                array_push($list, $v);
                $children[$pt] = $list;
            }
        }

        //treecurse function from functions.php
        $list = treerecurse($children);

        $mitems = array();
        foreach ($list as $item) {
            $mitems[$item->id] = $item->treename;
        }
        return $mitems;
    }
}