<?php


namespace ColibriWP\Theme\Core;


class Tree {

    const SEPARATOR = ".";

    private $data;

    public function __construct( $data = array() ) {
        $this->data = $data;
    }

    public function walkFirstLevel( $callback ) {
        $this->walkElementsAt( "", $callback );
    }

    public function walkElementsAt( $path, $callback ) {

        $data = $this->getData();

        if ( ! empty( $path ) ) {
            $data = $this->findAt( $path, array() );
        }

        if ( is_array( $data ) ) {
            foreach ( $data as $key => $item ) {
                call_user_func( $callback, $key, $item );
            }
        }
    }

    /**
     * @return array
     */
    public function getData() {
        return $this->data;
    }

    /**
     * @param array $data
     *
     * @return Tree
     */
    public function setData( $data ) {
        $this->data = $data;

        return $this;
    }

    public function findAt( $path, $default = null ) {
        $path_parts = explode( Tree::SEPARATOR, $path );
        $result     = $this->data;

        if ( $path === '' ) {
            return $result;
        }

        while ( $path_parts ) {
            $part = array_shift( $path_parts );

            if ( $this->entityHasKey( $result, $part ) ) {
                if ( is_array( $result ) ) {
                    $result = $result[ $part ];
                } else {
                    $result = $result->$part;
                }
            } else {
                $result = $default;
                break;
            }
        }

        return $result;
    }

    /**
     * @param $entity
     * @param $key
     *
     * @return bool
     */
    private function entityHasKey( $entity, $key ) {
        if ( is_array( $entity ) && array_key_exists( $key, $entity ) ) {
            return true;
        }

        if ( is_object( $entity ) && property_exists( $entity, $key ) ) {
            return true;
        }

        return false;
    }

}

