BookOutlineStorage.php

<?php

namespace Drupal\book;

use Drupal\Core\Database\Connection;

/**
 * Defines a storage class for books outline.
 */
class BookOutlineStorage implements BookOutlineStorageInterface {
    
    /**
     * Database Service Object.
     *
     * @var \Drupal\Core\Database\Connection
     */
    protected $connection;
    
    /**
     * Constructs a BookOutlineStorage object.
     */
    public function __construct(Connection $connection) {
        $this->connection = $connection;
    }
    
    /**
     * {@inheritdoc}
     */
    public function getBooks() {
        return $this->connection
            ->query("SELECT DISTINCT([bid]) FROM {book}")
            ->fetchCol();
    }
    
    /**
     * {@inheritdoc}
     */
    public function hasBooks() {
        return (bool) $this->connection
            ->query('SELECT count([bid]) FROM {book}')
            ->fetchField();
    }
    
    /**
     * {@inheritdoc}
     */
    public function loadMultiple($nids, $access = TRUE) {
        $query = $this->connection
            ->select('book', 'b', [
            'fetch' => \PDO::FETCH_ASSOC,
        ]);
        $query->fields('b');
        $query->condition('b.nid', $nids, 'IN');
        if ($access) {
            $query->addTag('node_access');
            $query->addMetaData('base_table', 'book');
        }
        return $query->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function getChildRelativeDepth($book_link, $max_depth) {
        $query = $this->connection
            ->select('book');
        $query->addField('book', 'depth');
        $query->condition('bid', $book_link['bid']);
        $query->orderBy('depth', 'DESC');
        $query->range(0, 1);
        $i = 1;
        $p = 'p1';
        while ($i <= $max_depth && $book_link[$p]) {
            $query->condition($p, $book_link[$p]);
            $p = 'p' . ++$i;
        }
        return $query->execute()
            ->fetchField();
    }
    
    /**
     * {@inheritdoc}
     */
    public function delete($nid) {
        return $this->connection
            ->delete('book')
            ->condition('nid', $nid)
            ->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function loadBookChildren($pid) {
        return $this->connection
            ->query("SELECT * FROM {book} WHERE [pid] = :pid", [
            ':pid' => $pid,
        ])
            ->fetchAllAssoc('nid', \PDO::FETCH_ASSOC);
    }
    
    /**
     * {@inheritdoc}
     */
    public function getBookMenuTree($bid, $parameters, $min_depth, $max_depth) {
        $query = $this->connection
            ->select('book');
        $query->fields('book');
        for ($i = 1; $i <= $max_depth; $i++) {
            $query->orderBy('p' . $i, 'ASC');
        }
        $query->condition('bid', $bid);
        if (!empty($parameters['expanded'])) {
            $query->condition('pid', $parameters['expanded'], 'IN');
        }
        if ($min_depth != 1) {
            $query->condition('depth', $min_depth, '>=');
        }
        if (isset($parameters['max_depth'])) {
            $query->condition('depth', $parameters['max_depth'], '<=');
        }
        // Add custom query conditions, if any were passed.
        if (isset($parameters['conditions'])) {
            foreach ($parameters['conditions'] as $column => $value) {
                $query->condition($column, $value);
            }
        }
        return $query->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function insert($link, $parents) {
        return $this->connection
            ->insert('book')
            ->fields([
            'nid' => $link['nid'],
            'bid' => $link['bid'],
            'pid' => $link['pid'],
            'weight' => $link['weight'],
        ] + $parents)
            ->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function update($nid, $fields) {
        return $this->connection
            ->update('book')
            ->fields($fields)
            ->condition('nid', $nid)
            ->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function updateMovedChildren($bid, $original, $expressions, $shift) {
        $query = $this->connection
            ->update('book');
        $query->fields([
            'bid' => $bid,
        ]);
        foreach ($expressions as $expression) {
            $query->expression($expression[0], $expression[1], $expression[2]);
        }
        $query->expression('depth', '[depth] + :depth', [
            ':depth' => $shift,
        ]);
        $query->condition('bid', $original['bid']);
        $p = 'p1';
        for ($i = 1; !empty($original[$p]); $p = 'p' . ++$i) {
            $query->condition($p, $original[$p]);
        }
        return $query->execute();
    }
    
    /**
     * {@inheritdoc}
     */
    public function countOriginalLinkChildren($original) {
        return $this->connection
            ->select('book', 'b')
            ->condition('bid', $original['bid'])
            ->condition('pid', $original['pid'])
            ->condition('nid', $original['nid'], '<>')
            ->countQuery()
            ->execute()
            ->fetchField();
    }
    
    /**
     * {@inheritdoc}
     */
    public function getBookSubtree($link, $max_depth) {
        $query = $this->connection
            ->select('book', 'b', [
            'fetch' => \PDO::FETCH_ASSOC,
        ]);
        $query->fields('b');
        $query->condition('b.bid', $link['bid']);
        for ($i = 1; $i <= $max_depth && $link["p{$i}"]; ++$i) {
            $query->condition("p{$i}", $link["p{$i}"]);
        }
        for ($i = 1; $i <= $max_depth; ++$i) {
            $query->orderBy("p{$i}");
        }
        return $query->execute();
    }

}