AL+MP one-to-many

<?php

class Tree {

    /** Колонка по которой строится путь */
    static $col_pid = 'parent_id';

    /** Суффикс таблицы где хранятся пути */
    static $table_suffix = '_paths';

    /** @var PDO */
    static $pdo;

    /** Хендлер подключения PDO */
    public static function SetConnection ( PDO $pdo ) {
        self::$pdo = $pdo;
    }

    /** Запрос в БД и фетч массива */
    public static function Query ( $sql ) {
        $stmt = self::$pdo->query($sql);
        if ( $stmt )
            return $stmt->fetchAll(PDO::FETCH_ASSOC);
        $i = self::$pdo->errorInfo();
        return $i[2];
    }

    /** Путь к элементу */
    public static function GetPath ( $table, $id ) {
        $sql = 'SELECT t.*, p.level as p_level, p.order as p_order '
              .'FROM `'.$table.self::$table_suffix.'` p '
              .'JOIN `'.$table.'` t ON `id`=p.`parent_id` '
              .'WHERE item_id='.$id.' ORDER BY `order`';
        return self::Query($sql);
    }

    /** Все дети */
    public static function GetChildren ( $table, $id ) {
        $sql = 'SELECT t.*, p.level as p_level, p.order as p_order '
              .'FROM `'.$table.self::$table_suffix.'` p '
              .'JOIN `'.$table.'` t ON `id`=`item_id` '
              .'WHERE p.`parent_id`='.$id.' ORDER BY p.`level`, p.`order`';
        return self::Query($sql);
    }

    /** Количество детей */
    public static function GetChildrenCount ( $table, $id ) {
        $sql = 'SELECT count(*) FROM `'.$table.self::$table_suffix.'` WHERE parent_id='.$id.'';
        $stmt = self::$pdo->query($sql);
        if ( $stmt )
            return $stmt->fetchColumn(0);
        else
            return false;
    }

    /** Построение дерева из родительской таблицы */
    public static function GetTree ( $table ) {
        $sql = 'SELECT `id`, `'.self::$col_pid.'` FROM `'.$table.'`';
        $stmt = self::$pdo->query($sql);
        if ( !$stmt ) return false;
        $res = $stmt->fetchAll(PDO::FETCH_ASSOC);
        return self::BuildTree( $res );
    }

    /**
     * Создание путей в доп.таблице
     * @param $table имя основной таблицы
     * @param bool $ids если нужно достроить путь - указываются ID нужных веток
     * @return bool
     */
    public static function GeneratePaths ( $table, $ids = true ) {

        if ( !$tree = self::GetTree( $table ) ) return false;

        self::$pdo->beginTransaction();
        if ( $ids === true ) self::$pdo->query( 'TRUNCATE '.$table.self::$table_suffix );

        $sql = 'INSERT INTO '.$table.self::$table_suffix.' (`item_id`, `parent_id`, `level`, `order`) '
              .'VALUES (:item_id, :parent_id, :level, :order)';
        $stmt = self::$pdo->prepare($sql);

        try {
            self::WalkthruTree( $tree, $stmt, $ids );
            self::$pdo->commit();
        } catch ( Exception $e ) {
            echo $e->getMessage();
            self::$pdo->rollBack();
        }
    }

    /** Создание в таблице $table количества $total записей на $lvls уровней вложенности */
    public static function Fixture ( $table, $total, $lvls = 3 ) {

        //TRUNCATE не работает из-за внешнего ключа
        self::$pdo->query( 'DELETE FROM '.$table );
        self::$pdo->query( 'ALTER TABLE '.$table.' AUTO_INCREMENT=1' );

        //Распределение элементов по уровням вложенности
        $num = $total;
        $lvls_arr = array();
        for ( $i=$lvls; $i>0; $i-- ) {
            if ( $i == 1 ) $limit = $num;
            else {
                $limit = ceil($num*0.6);
                $num -= $limit;
            }
            $lvls_arr[$i] = $limit;
        }

        //Пихаем всё в БД
        $stmt = self::$pdo->prepare('INSERT INTO `'.$table.'` (parent_id, title) VALUES (:parent_id, :title)');
        $p_arr = array();
        for ( $i = 1; $i<=$lvls; $i++ ) {
            $limit = $lvls_arr[$i];
            echo $i.' уровень: '.$limit.' штук.<br>';
            while ( $limit-- ) {
                $prev = NULL;
                if ( $i>1 ) {
                    $c = count ( $p_arr[($i-1)] )-1;
                    $prev = $p_arr[($i-1)][mt_rand(0,$c)];
                }
                $stmt->bindValue(':parent_id', $prev);
                $stmt->bindValue(':title', md5(uniqid(NULL, true)));
                if ( $stmt->execute() )
                    $p_arr[$i][] = self::$pdo->lastInsertId();
                else throw new Exception ( 'Ошибка при добавлении страницы: '.$stmt->errorInfo() );
            }
        }
    }

    /** Проход по дереву и создание соотв. путей в БД */
    private static function WalkthruTree ( $arr, &$pdo_stmt, $ids = true, $prev = NULL, $lvl = 0 ) {

        if ( is_numeric( $ids ) ) $ids = array( $ids );

        if ( is_array( $arr ) ) foreach ( $arr as $id=>$a ) {
            if ( is_array( $prev ) ) foreach ( $prev as $pid=>$order ) {
                try {
                    if ( $ids === true )
                        self::PutPathItem( $pdo_stmt, $id, $pid, $order, $lvl );
                    elseif ( is_array( $ids ) && in_array( $id, $ids ) ) {
                        self::PutPathItem( $pdo_stmt, $id, $pid, $order, $lvl );
                        if ( is_array( $a ) ) $ids += array_merge($ids, array_keys($a) );
                    }
                } catch ( Exception $e ) {
                    echo '<p>'.$e->getMessage().'</p>';
                }
            }
            if ( is_array( $a ) ) {
                $prev_new = $prev;
                $prev_new[$id] = $lvl;
                self::WalkthruTree( $a, $pdo_stmt, $ids, $prev_new, ($lvl+1) );
            }
        }
    }

    /** Пихалово пути в БД */
    private static function PutPathItem ( &$pdo_stmt, $id, $pid, $order, $level ) {
        $str = 'item_id: '.$id.', parent_id: '.$pid.', order: '.$order.', level: '.$level;
        if ( $pdo_stmt instanceof PDOStatement ) {
            $pdo_stmt->bindValue(':item_id',     $id);
            $pdo_stmt->bindValue(':parent_id',   $pid);
            $pdo_stmt->bindValue(':order',       $order);
            $pdo_stmt->bindValue(':level',       $level);
            if ( !$pdo_stmt->execute() )
                throw new Exception ( 'Ошибка при добавлении пути: '.$str );
        } else //{}
            echo $str.'<br>';
    }

    /** Построение массива по таблице id => parent_id */
    private static function BuildTree ( $arr, $pid = NULL ) {
        $tmp = array();
        foreach ( $arr as $row ) if ( $row['parent_id'] == $pid )
            $tmp[$row['id']] = self::BuildTree( $arr, $row['id'] );
        return count($tmp) ? $tmp : true;
    }
}