<?php

/*
 * (c) Samuel De Backer <sdebacker@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Modules\Core\Support;

use App;
use Illuminate\Database\Eloquent\Collection;
use Illuminate\Support\Collection as BaseCollection;

class NestableCollection extends Collection
{
    private $total;
    private $parentColumn;

    public function __construct($items = array())
    {
        parent::__construct($items);
        $this->parentColumn = 'parent_id';
        $this->total = count($items);
    }

    /**
     * Nest items
     *
     * @return mixed boolean|NestableCollection
     */
    public function nest()
    {
        $parentColumn = $this->parentColumn;
        if (!$parentColumn) {
            return $this;
        }

        // Set id as keys
        $this->items = $this->getDictionary();

        $keysToDelete = array();

        // add empty children collection.
        $this->each(function ($item) {
            if (!$item->items) {
                $item->items = App::make('Illuminate\Support\Collection');
            }
        });

        // add items to children collection
        foreach ($this->items as $item) {
            if ($item->$parentColumn && isset($this->items[$item->$parentColumn])) {
                $this->items[$item->$parentColumn]->items->push($item);
                $keysToDelete[] = $item->id;
            }
        }

        // Delete moved items
        $this->items = array_values(array_except($this->items, $keysToDelete));

        return $this;
    }

    /**
     * Recursive function that flatten a nested Collection
     * with characters (default is four spaces)
     *
     * @param  BaseCollection|null $collection
     * @param  string $column
     * @param  integer $level
     * @param  array &$flattened
     * @param  string $indentChars
     * @return array
     */
    public function listsFlattened($column = 'title', BaseCollection $collection = null, $level = 0, array &$flattened = [], $indentChars = '&nbsp;&nbsp;&nbsp;&nbsp;')
    {
        $collection = $collection ?: $this;
        foreach ($collection as $item) {
            $flattened[$item->id] = str_repeat($indentChars, $level) . $item->$column;
            if ($item->items) {
                $this->listsFlattened($column, $item->items, $level + 1, $flattened, $indentChars);
            }
        }

        return $flattened;
    }

    /**
     * Get total items for laravel 4 compatibility
     *
     * @return int
     */
    public function getTotal()
    {
        return $this->total();
    }

    /**
     * Get total items in nested collection
     *
     * @return int
     */
    public function total()
    {
        return $this->total;
    }
}