book.module 40.9 KB
Newer Older
Dries's avatar
 
Dries committed
1
<?php
2
// $Id$
Dries's avatar
 
Dries committed
3

Dries's avatar
 
Dries committed
4 5
/**
 * @file
6
 * Allows users to create and organize related content in an outline.
Dries's avatar
 
Dries committed
7 8
 */

9
/**
10
 * Implement hook_theme().
11 12 13 14
 */
function book_theme() {
  return array(
    'book_navigation' => array(
Dries's avatar
Dries committed
15
      'arguments' => array('book_link' => NULL),
16
      'template' => 'book-navigation',
17 18
    ),
    'book_export_html' => array(
19 20
      'arguments' => array('title' => NULL, 'contents' => NULL, 'depth' => NULL),
      'template' => 'book-export-html',
21 22 23 24
    ),
    'book_admin_table' => array(
      'arguments' => array('form' => NULL),
    ),
Dries's avatar
Dries committed
25 26 27 28 29
    'book_title_link' => array(
      'arguments' => array('link' => NULL),
    ),
    'book_all_books_block' => array(
      'arguments' => array('book_menus' => array()),
30 31 32 33 34
      'template' => 'book-all-books-block',
    ),
    'book_node_export_html' => array(
      'arguments' => array('node' => NULL, 'children' => NULL),
      'template' => 'book-node-export-html',
Dries's avatar
Dries committed
35
    ),
36 37 38
  );
}

39
/**
40
 * Implement hook_permission().
41
 */
42
function book_permission() {
43
  return array(
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
    'administer book outlines' => array(
      'title' => t('Administer book outlines'),
      'description' => t('Manage books through the administration panel.'),
    ),
    'create new books' => array(
      'title' => t('Create new books'),
      'description' => t('Add new top-level books.'),
    ),
    'add content to books' => array(
      'title' => t('Add content to books'),
      'description' => t('Add new content and child pages to books.'),
    ),
    'access printer-friendly version' => array(
      'title' => t('Access printer-friendly version'),
      'description' => t('View a book page and all of its sub-pages as a single document for ease of printing. Can be performance heavy.'),
    ),
60
  );
Dries's avatar
 
Dries committed
61 62
}

Dries's avatar
 
Dries committed
63
/**
Dries's avatar
Dries committed
64
 * Inject links into $node as needed.
Dries's avatar
 
Dries committed
65
 */
66
function book_node_view_link($node, $build_mode) {
Dries's avatar
 
Dries committed
67
  $links = array();
Dries's avatar
Dries committed
68

69
  if (isset($node->book['depth'])) {
70
    if ($build_mode == 'full') {
Dries's avatar
Dries committed
71 72
      $child_type = variable_get('book_child_type', 'book');
      if ((user_access('add content to books') || user_access('administer book outlines')) && node_access('create', $child_type) && $node->status == 1 && $node->book['depth'] < MENU_MAX_DEPTH) {
73
        $links['book_add_child'] = array(
74
          'title' => t('Add child page'),
75 76
          'href' => 'node/add/' . str_replace('_', '-', $child_type),
          'query' => 'parent=' . $node->book['mlid'],
77
        );
Dries's avatar
 
Dries committed
78
      }
79

Dries's avatar
Dries committed
80
      if (user_access('access printer-friendly version')) {
81
        $links['book_printer'] = array(
82
          'title' => t('Printer-friendly version'),
83
          'href' => 'book/export/html/' . $node->nid,
84
          'attributes' => array('title' => t('Show a printer-friendly version of this book page and its sub-pages.'))
85
        );
86
      }
Dries's avatar
 
Dries committed
87
    }
Dries's avatar
 
Dries committed
88
  }
89

90 91
  if (!empty($links)) {
    $node->content['links']['book'] = array(
92 93
      '#theme' => 'links',
      '#links' => $links,
94
      '#attributes' => array('class' => array('links', 'inline')),
95 96
    );
  }
Dries's avatar
 
Dries committed
97 98
}

Dries's avatar
 
Dries committed
99
/**
100
 * Implement hook_menu().
Dries's avatar
 
Dries committed
101
 */
102 103
function book_menu() {
  $items['admin/content/book'] = array(
104
    'title' => 'Books',
Dries's avatar
Dries committed
105 106
    'page callback' => 'book_admin_overview',
    'access arguments' => array('administer book outlines'),
107
    'type' => MENU_LOCAL_TASK,
108
    'file' => 'book.admin.inc',
109 110
  );
  $items['admin/content/book/list'] = array(
111
    'title' => 'List',
112 113
    'type' => MENU_DEFAULT_LOCAL_TASK,
  );
Dries's avatar
Dries committed
114 115
  $items['admin/content/book/settings'] = array(
    'title' => 'Settings',
116
    'page callback' => 'drupal_get_form',
Dries's avatar
Dries committed
117 118
    'page arguments' => array('book_admin_settings'),
    'access arguments' => array('administer site configuration'),
119 120
    'type' => MENU_LOCAL_TASK,
    'weight' => 8,
121
    'file' => 'book.admin.inc',
122
  );
Dries's avatar
Dries committed
123 124 125 126 127 128 129
  $items['admin/content/book/%node'] = array(
    'title' => 'Re-order book pages and change titles',
    'page callback' => 'drupal_get_form',
    'page arguments' => array('book_admin_edit', 3),
    'access callback' => '_book_outline_access',
    'access arguments' => array(3),
    'type' => MENU_CALLBACK,
130
    'file' => 'book.admin.inc',
Dries's avatar
Dries committed
131
  );
132
  $items['book'] = array(
133
    'title' => 'Books',
134 135 136
    'page callback' => 'book_render',
    'access arguments' => array('access content'),
    'type' => MENU_SUGGESTED_ITEM,
137
    'file' => 'book.pages.inc',
138 139 140 141
  );
  $items['book/export/%/%'] = array(
    'page callback' => 'book_export',
    'page arguments' => array(2, 3),
Dries's avatar
Dries committed
142
    'access arguments' => array('access printer-friendly version'),
143
    'type' => MENU_CALLBACK,
144
    'file' => 'book.pages.inc',
145
  );
146
  $items['node/%node/outline'] = array(
147
    'title' => 'Outline',
Dries's avatar
Dries committed
148 149
    'page callback' => 'book_outline',
    'page arguments' => array(1),
150 151 152 153
    'access callback' => '_book_outline_access',
    'access arguments' => array(1),
    'type' => MENU_LOCAL_TASK,
    'weight' => 2,
154
    'file' => 'book.pages.inc',
155
  );
Dries's avatar
Dries committed
156 157 158 159 160 161 162
  $items['node/%node/outline/remove'] = array(
    'title' => 'Remove from outline',
    'page callback' => 'drupal_get_form',
    'page arguments' => array('book_remove_form', 1),
    'access callback' => '_book_outline_remove_access',
    'access arguments' => array(1),
    'type' => MENU_CALLBACK,
163
    'file' => 'book.pages.inc',
Dries's avatar
Dries committed
164
  );
165
  $items['book/js/form'] = array(
Dries's avatar
Dries committed
166 167 168
    'page callback' => 'book_form_update',
    'access arguments' => array('access content'),
    'type' => MENU_CALLBACK,
169
    'file' => 'book.pages.inc',
Dries's avatar
Dries committed
170
  );
171

Dries's avatar
 
Dries committed
172 173 174
  return $items;
}

Dries's avatar
Dries committed
175 176 177
/**
 * Menu item access callback - determine if the outline tab is accessible.
 */
178
function _book_outline_access($node) {
Dries's avatar
Dries committed
179 180 181 182 183 184 185 186
  return user_access('administer book outlines') && node_access('view', $node);
}

/**
 * Menu item access callback - determine if the user can remove nodes from the outline.
 */
function _book_outline_remove_access($node) {
  return isset($node->book) && ($node->book['bid'] != $node->nid) && _book_outline_access($node);
187 188
}

Dries's avatar
Dries committed
189
/**
190
 * Implement hook_init().
Dries's avatar
Dries committed
191
 */
192
function book_init() {
193
  drupal_add_css(drupal_get_path('module', 'book') . '/book.css');
194 195
}

196 197 198 199 200 201 202 203 204 205 206 207 208
/**
 * Implement hook_field_build_modes().
 */
function book_field_build_modes($obj_type) {
  $modes = array();
  if ($obj_type == 'node') {
    $modes = array(
      'print' => t('Print'),
    );
  }
  return $modes;
}

209
/**
210
 * Implement hook_block_info().
211
 */
212
function book_block_info() {
213 214
  $block = array();
  $block['navigation']['info'] = t('Book navigation');
215
  $block['navigation']['cache'] = DRUPAL_CACHE_PER_PAGE | DRUPAL_CACHE_PER_ROLE;
216 217 218 219 220

  return $block;
}

/**
221
 * Implement hook_block_view().
222
 *
Dries's avatar
 
Dries committed
223 224
 * Displays the book table of contents in a block when the current page is a
 * single-node view of a book node.
225
 */
226
function book_block_view($delta = '') {
Dries's avatar
 
Dries committed
227
  $block = array();
228 229 230 231
  $current_bid = 0;
  if ($node = menu_get_object()) {
    $current_bid = empty($node->book['bid']) ? 0 : $node->book['bid'];
  }
232

233 234 235 236 237 238 239 240 241
  if (variable_get('book_block_mode', 'all pages') == 'all pages') {
    $block['subject'] = t('Book navigation');
    $book_menus = array();
    $pseudo_tree = array(0 => array('below' => FALSE));
    foreach (book_get_books() as $book_id => $book) {
      if ($book['bid'] == $current_bid) {
        // If the current page is a node associated with a book, the menu
        // needs to be retrieved.
        $book_menus[$book_id] = menu_tree_output(menu_tree_all_data($node->book['menu_name'], $node->book));
Dries's avatar
 
Dries committed
242
      }
243 244 245 246 247 248
      else {
        // Since we know we will only display a link to the top node, there
        // is no reason to run an additional menu tree query for each book.
        $book['in_active_trail'] = FALSE;
        $pseudo_tree[0]['link'] = $book;
        $book_menus[$book_id] = menu_tree_output($pseudo_tree);
Dries's avatar
Dries committed
249
      }
250
    }
251 252
    $book_menus['#theme'] = 'book_all_books_block';
    $block['content'] = $book_menus;
253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
  }
  elseif ($current_bid) {
    // Only display this block when the user is browsing a book.
    $select = db_select('node');
    $select->addField('node', 'title');
    $select->condition('nid', $node->book['bid']);
    $select->addTag('node_access');
    $title = $select->execute()->fetchField();
    // Only show the block if the user has view access for the top-level node.
    if ($title) {
      $tree = menu_tree_all_data($node->book['menu_name'], $node->book);
      // There should only be one element at the top level.
      $data = array_shift($tree);
      $block['subject'] = theme('book_title_link', $data['link']);
      $block['content'] = ($data['below']) ? menu_tree_output($data['below']) : '';
    }
  }
270

271 272
  return $block;
}
273

274
/**
275
 * Implement hook_block_configure().
276 277 278 279 280 281 282 283 284 285 286 287 288 289
 */
function book_block_configure($delta = '') {
  $block = array();
  $options = array(
    'all pages' => t('Show block on all pages'),
    'book pages' => t('Show block only on book pages'),
  );
  $form['book_block_mode'] = array(
    '#type' => 'radios',
    '#title' => t('Book navigation block display'),
    '#options' => $options,
    '#default_value' => variable_get('book_block_mode', 'all pages'),
    '#description' => t("If <em>Show block on all pages</em> is selected, the block will contain the automatically generated menus for all of the site's books. If <em>Show block only on book pages</em> is selected, the block will contain only the one menu corresponding to the current page's book. In this case, if the current page is not in a book, no block will be displayed. The <em>Page specific visibility settings</em> or other visibility settings can be used in addition to selectively display this block."),
    );
290

291 292
  return $form;
}
293

294
/**
295
 * Implement hook_block_save().
296 297 298 299
 */
function book_block_save($delta = '', $edit = array()) {
  $block = array();
  variable_set('book_block_mode', $edit['book_block_mode']);
Dries's avatar
Dries committed
300
}
Dries's avatar
 
Dries committed
301

302
/**
Dries's avatar
Dries committed
303 304 305
 * Generate the HTML output for a link to a book title when used as a block title.
 *
 * @ingroup themeable
306
 */
Dries's avatar
Dries committed
307
function theme_book_title_link($link) {
308
  $link['options']['attributes']['class'] = array('book-title');
309

Dries's avatar
Dries committed
310
  return l($link['title'], $link['href'], $link['options']);
Dries's avatar
 
Dries committed
311
}
Dries's avatar
 
Dries committed
312

313
/**
Dries's avatar
Dries committed
314 315 316 317
 * Returns an array of all books.
 *
 * This list may be used for generating a list of all the books, or for building
 * the options for a form select.
318
 */
Dries's avatar
Dries committed
319
function book_get_books() {
320
  $all_books = &drupal_static(__FUNCTION__);
Dries's avatar
Dries committed
321 322 323

  if (!isset($all_books)) {
    $all_books = array();
324
    $nids = db_query("SELECT DISTINCT(bid) FROM {book}")->fetchCol();
325

Dries's avatar
Dries committed
326
    if ($nids) {
327 328 329 330 331 332 333 334 335 336 337 338 339 340
      $query = db_select('book', 'b', array('fetch' => PDO::FETCH_ASSOC));
      $node_alias = $query->join('node', 'n', 'b.nid = n.nid');
      $menu_links_alias = $query->join('menu_links', 'ml', 'b.mlid = ml.mlid');
      $query->addField('n', 'type', 'type');
      $query->addField('n', 'title', 'title');
      $query->fields('b');
      $query->fields($menu_links_alias);
      $query->condition('n.nid', $nids, 'IN');
      $query->condition('n.status', 1);
      $query->orderBy('ml.weight');
      $query->orderBy('ml.link_title');
      $query->addTag('node_access');
      $result2 = $query->execute();
      foreach ($result2 as $link) {
Dries's avatar
Dries committed
341 342
        $link['href'] = $link['link_path'];
        $link['options'] = unserialize($link['options']);
343
        $all_books[$link['bid']] = $link;
Dries's avatar
Dries committed
344 345
      }
    }
346
  }
347

Dries's avatar
Dries committed
348
  return $all_books;
349
}
350

Dries's avatar
Dries committed
351
/**
352
 * Implement hook_form_alter().
353 354
 *
 * Adds the book fieldset to the node form.
Dries's avatar
Dries committed
355 356 357 358 359 360
 *
 * @see book_pick_book_submit()
 * @see book_submit()
 */
function book_form_alter(&$form, $form_state, $form_id) {

361
  if (!empty($form['#node_edit_form'])) {
362
    // Add elements to the node form.
Dries's avatar
Dries committed
363 364 365 366 367
    $node = $form['#node'];

    $access = user_access('administer book outlines');
    if (!$access) {
      if (user_access('add content to books') && ((!empty($node->book['mlid']) && !empty($node->nid)) || book_type_is_allowed($node->type))) {
368
        // Already in the book hierarchy, or this node type is allowed.
Dries's avatar
Dries committed
369 370 371 372 373 374 375 376 377
        $access = TRUE;
      }
    }

    if ($access) {
      _book_add_form_elements($form, $node);
      $form['book']['pick-book'] = array(
        '#type' => 'submit',
        '#value' => t('Change book (update list of parents)'),
378
         // Submit the node form so the parent select options get updated.
379
         // This is typically only used when JS is disabled. Since the parent options
380 381
         // won't be changed via AJAX, a button is provided in the node form to submit
         // the form and generate options in the parent select corresponding to the
382
         // selected book. This is similar to what happens during a node preview.
383
        '#submit' => array('node_form_submit_build_node'),
Dries's avatar
Dries committed
384 385 386
        '#weight' => 20,
      );
    }
387
  }
Dries's avatar
Dries committed
388
}
389

Dries's avatar
Dries committed
390
/**
391
 * Build the parent selection form element for the node form or outline tab.
Dries's avatar
Dries committed
392 393 394 395 396 397
 *
 * This function is also called when generating a new set of options during the
 * AJAX callback, so an array is returned that can be used to replace an existing
 * form element.
 */
function _book_parent_select($book_link) {
398 399 400
  if (variable_get('menu_override_parent_selector', FALSE)) {
    return array();
  }
Dries's avatar
Dries committed
401 402 403 404 405 406 407
  // Offer a message or a drop-down to choose a different parent page.
  $form = array(
    '#type' => 'hidden',
    '#value' => -1,
    '#prefix' => '<div id="edit-book-plid-wrapper">',
    '#suffix' => '</div>',
  );
408

Dries's avatar
Dries committed
409 410 411
  if ($book_link['nid'] === $book_link['bid']) {
    // This is a book - at the top level.
    if ($book_link['original_bid'] === $book_link['bid']) {
412
      $form['#prefix'] .= '<em>' . t('This is the top-level page in this book.') . '</em>';
Dries's avatar
Dries committed
413 414
    }
    else {
415
      $form['#prefix'] .= '<em>' . t('This will be the top-level page in this book.') . '</em>';
Dries's avatar
Dries committed
416 417 418
    }
  }
  elseif (!$book_link['bid']) {
419
    $form['#prefix'] .= '<em>' . t('No book selected.') . '</em>';
420 421
  }
  else {
Dries's avatar
Dries committed
422 423 424 425
    $form = array(
      '#type' => 'select',
      '#title' => t('Parent item'),
      '#default_value' => $book_link['plid'],
426
      '#description' => t('The parent page in the book. The maximum depth for a book and all child pages is !maxdepth. Some pages in the selected book may not be available as parents if selecting them would exceed this limit.', array('!maxdepth' => MENU_MAX_DEPTH)),
427
      '#options' => book_toc($book_link['bid'], $book_link['parent_depth_limit'], array($book_link['mlid'])),
428
      '#attributes' => array('class' => array('book-title-select')),
429
    );
Dries's avatar
 
Dries committed
430
  }
431

432
  return $form;
Dries's avatar
 
Dries committed
433 434
}

435
/**
Dries's avatar
Dries committed
436
 * Build the common elements of the book form for the node and outline forms.
437
 */
Dries's avatar
Dries committed
438 439 440 441 442 443 444 445 446 447
function _book_add_form_elements(&$form, $node) {
  // Need this for AJAX.
  $form['#cache'] = TRUE;

  $form['book'] = array(
    '#type' => 'fieldset',
    '#title' => t('Book outline'),
    '#weight' => 10,
    '#collapsible' => TRUE,
    '#collapsed' => TRUE,
448
    '#group' => 'additional_settings',
449 450 451
    '#attached' => array(
      'js' => array(drupal_get_path('module', 'book') . '/book.js'),
    ),
Dries's avatar
Dries committed
452
    '#tree' => TRUE,
453
    '#attributes' => array('class' => array('book-outline-form')),
454
  );
455
  foreach (array('menu_name', 'mlid', 'nid', 'router_path', 'has_children', 'options', 'module', 'original_bid', 'parent_depth_limit') as $key) {
Dries's avatar
Dries committed
456 457 458 459 460 461 462 463 464 465
    $form['book'][$key] = array(
      '#type' => 'value',
      '#value' => $node->book[$key],
    );
  }

  $form['book']['plid'] = _book_parent_select($node->book);

  $form['book']['weight'] = array(
    '#type' => 'weight',
466
    '#title' => t('Weight'),
Dries's avatar
Dries committed
467
    '#default_value' => $node->book['weight'],
468
    '#delta' => 15,
Dries's avatar
Dries committed
469
    '#weight' => 5,
470 471
    '#description' => t('Pages at a given level are ordered first by weight and then by title.'),
  );
Dries's avatar
Dries committed
472 473
  $options = array();
  $nid = isset($node->nid) ? $node->nid : 'new';
474

475
  if (isset($node->nid) && ($nid == $node->book['original_bid']) && ($node->book['parent_depth_limit'] == 0)) {
476 477 478 479 480 481 482 483 484
    // This is the top level node in a maximum depth book and thus cannot be moved.
    $options[$node->nid] = $node->title;
  }
  else {
    foreach (book_get_books() as $book) {
      $options[$book['nid']] = $book['title'];
    }
  }

Dries's avatar
Dries committed
485 486
  if (user_access('create new books') && ($nid == 'new' || ($nid != $node->book['original_bid']))) {
    // The node can become a new book, if it is not one already.
487
    $options = array($nid => '<' . t('create a new book') . '>') + $options;
Dries's avatar
Dries committed
488 489
  }
  if (!$node->book['mlid']) {
490
    // The node is not currently in the hierarchy.
491
    $options = array(0 => '<' . t('none') . '>') + $options;
Dries's avatar
Dries committed
492 493 494 495 496 497 498 499 500 501 502
  }

  // Add a drop-down to select the destination book.
  $form['book']['bid'] = array(
    '#type' => 'select',
    '#title' => t('Book'),
    '#default_value' => $node->book['bid'],
    '#options' => $options,
    '#access' => (bool)$options,
    '#description' => t('Your page will be a part of the selected book.'),
    '#weight' => -5,
503
    '#attributes' => array('class' => array('book-title-select')),
504
    '#ajax' => array(
505 506
      'path' => 'book/js/form',
      'wrapper' => 'edit-book-plid-wrapper',
507 508
      'effect' => 'fade',
      'speed' => 'fast',
509
    ),
510
  );
Dries's avatar
Dries committed
511
}
Dries's avatar
 
Dries committed
512

513
/**
Dries's avatar
Dries committed
514
 * Common helper function to handles additions and updates to the book outline.
Dries's avatar
Dries committed
515
 *
Dries's avatar
Dries committed
516 517 518
 * Performs all additions and updates to the book outline through node addition,
 * node editing, node deletion, or the outline tab.
 */
519
function _book_update_outline($node) {
Dries's avatar
Dries committed
520 521 522 523 524
  if (empty($node->book['bid'])) {
    return FALSE;
  }
  $new = empty($node->book['mlid']);

525
  $node->book['link_path'] = 'node/' . $node->nid;
Dries's avatar
Dries committed
526 527 528 529 530 531 532 533 534 535
  $node->book['link_title'] = $node->title;
  $node->book['parent_mismatch'] = FALSE; // The normal case.

  if ($node->book['bid'] == $node->nid) {
    $node->book['plid'] = 0;
    $node->book['menu_name'] = book_menu_name($node->nid);
  }
  else {
    // Check in case the parent is not is this book; the book takes precedence.
    if (!empty($node->book['plid'])) {
536 537 538
      $parent = db_query("SELECT * FROM {book} WHERE mlid = :mlid", array(
        ':mlid' => $node->book['plid'],
      ))->fetchAssoc();
Dries's avatar
Dries committed
539 540
    }
    if (empty($node->book['plid']) || !$parent || $parent['bid'] != $node->book['bid']) {
541 542 543
      $node->book['plid'] = db_query("SELECT mlid FROM {book} WHERE nid = :nid", array(
        ':nid' => $node->book['bid'],
      ))->fetchField();
Dries's avatar
Dries committed
544 545 546
      $node->book['parent_mismatch'] = TRUE; // Likely when JS is disabled.
    }
  }
547

Dries's avatar
Dries committed
548 549 550
  if (menu_link_save($node->book)) {
    if ($new) {
      // Insert new.
551 552 553 554 555 556 557
      db_insert('book')
        ->fields(array(
          'nid' => $node->nid,
          'mlid' => $node->book['mlid'],
          'bid' => $node->book['bid'],
        ))
        ->execute();
Dries's avatar
Dries committed
558 559
    }
    else {
560 561 562
      if ($node->book['bid'] != db_query("SELECT bid FROM {book} WHERE nid = :nid", array(
          ':nid' => $node->nid,
        ))->fetchField()) {
Dries's avatar
Dries committed
563 564 565 566
        // Update the bid for this page and all children.
        book_update_bid($node->book);
      }
    }
567

Dries's avatar
Dries committed
568 569
    return TRUE;
  }
570 571

  // Failed to save the menu link.
Dries's avatar
Dries committed
572 573 574
  return FALSE;
}

575
/**
Dries's avatar
Dries committed
576
 * Update the bid for a page and its children when it is moved to a new book.
577
 *
Dries's avatar
Dries committed
578 579 580 581
 * @param $book_link
 *   A fully loaded menu link that is part of the book hierarchy.
 */
function book_update_bid($book_link) {
582 583
  $query = db_select('menu_links');
  $query->addField('menu_links', 'mlid');
Dries's avatar
Dries committed
584
  for ($i = 1; $i <= MENU_MAX_DEPTH && $book_link["p$i"]; $i++) {
585
    $query->condition("p$i", $book_link["p$i"]);
Dries's avatar
Dries committed
586
  }
587
  $mlids = $query->execute()->fetchCol();
588

Dries's avatar
Dries committed
589
  if ($mlids) {
590
    db_update('book')
591
      ->fields(array('bid' => $book_link['bid']))
592 593
      ->condition('mlid', $mlids, 'IN')
      ->execute();
Dries's avatar
Dries committed
594 595 596 597 598
  }
}

/**
 * Get the book menu tree for a page, and return it as a linear array.
599
 *
Dries's avatar
Dries committed
600 601
 * @param $book_link
 *   A fully loaded menu link that is part of the book hierarchy.
602
 * @return
Dries's avatar
Dries committed
603 604
 *   A linear array of menu links in the order that the links are shown in the
 *   menu, so the previous and next pages are the elements before and after the
605
 *   element corresponding to $node. The children of $node (if any) will come
606 607
 *   immediately after it in the array, and links will only be fetched as deep
 *   as one level deeper than $book_link.
608
 */
Dries's avatar
Dries committed
609
function book_get_flat_menu($book_link) {
610
  $flat = &drupal_static(__FUNCTION__, array());
Dries's avatar
Dries committed
611 612

  if (!isset($flat[$book_link['mlid']])) {
613
    // Call menu_tree_all_data() to take advantage of the menu system's caching.
614
    $tree = menu_tree_all_data($book_link['menu_name'], $book_link, $book_link['depth'] + 1);
Dries's avatar
Dries committed
615 616
    $flat[$book_link['mlid']] = array();
    _book_flatten_menu($tree, $flat[$book_link['mlid']]);
Dries's avatar
 
Dries committed
617
  }
618

Dries's avatar
Dries committed
619
  return $flat[$book_link['mlid']];
Dries's avatar
 
Dries committed
620 621
}

622
/**
Dries's avatar
Dries committed
623
 * Recursive helper function for book_get_flat_menu().
624
 */
Dries's avatar
Dries committed
625 626 627 628 629 630 631 632
function _book_flatten_menu($tree, &$flat) {
  foreach ($tree as $data) {
    if (!$data['link']['hidden']) {
      $flat[$data['link']['mlid']] = $data['link'];
      if ($data['below']) {
        _book_flatten_menu($data['below'], $flat);
      }
    }
Dries's avatar
 
Dries committed
633
  }
Dries's avatar
Dries committed
634
}
Dries's avatar
 
Dries committed
635

Dries's avatar
Dries committed
636 637 638 639 640 641 642
/**
 * Fetches the menu link for the previous page of the book.
 */
function book_prev($book_link) {
  // If the parent is zero, we are at the start of a book.
  if ($book_link['plid'] == 0) {
    return NULL;
Dries's avatar
 
Dries committed
643
  }
Dries's avatar
Dries committed
644 645 646 647 648 649 650 651 652 653 654 655
  $flat = book_get_flat_menu($book_link);
  // Assigning the array to $flat resets the array pointer for use with each().
  $curr = NULL;
  do {
    $prev = $curr;
    list($key, $curr) = each($flat);
  } while ($key && $key != $book_link['mlid']);

  if ($key == $book_link['mlid']) {
    // The previous page in the book may be a child of the previous visible link.
    if ($prev['depth'] == $book_link['depth'] && $prev['has_children']) {
      // The subtree will have only one link at the top level - get its data.
656 657
      $tree = book_menu_subtree_data($prev);
      $data = array_shift($tree);
Dries's avatar
Dries committed
658 659 660 661
      // The link of interest is the last child - iterate to find the deepest one.
      while ($data['below']) {
        $data = end($data['below']);
      }
662

Dries's avatar
Dries committed
663 664 665 666 667
      return $data['link'];
    }
    else {
      return $prev;
    }
Dries's avatar
 
Dries committed
668 669 670
  }
}

671
/**
Dries's avatar
Dries committed
672
 * Fetches the menu link for the next page of the book.
673
 */
Dries's avatar
Dries committed
674 675 676 677 678
function book_next($book_link) {
  $flat = book_get_flat_menu($book_link);
  // Assigning the array to $flat resets the array pointer for use with each().
  do {
    list($key, $curr) = each($flat);
679 680 681
  }
  while ($key && $key != $book_link['mlid']);

Dries's avatar
Dries committed
682 683
  if ($key == $book_link['mlid']) {
    return current($flat);
Dries's avatar
 
Dries committed
684
  }
Dries's avatar
Dries committed
685
}
Dries's avatar
 
Dries committed
686

Dries's avatar
Dries committed
687 688 689 690 691
/**
 * Format the menu links for the child pages of the current page.
 */
function book_children($book_link) {
  $flat = book_get_flat_menu($book_link);
692

Dries's avatar
Dries committed
693 694 695 696 697 698
  $children = array();

  if ($book_link['has_children']) {
    // Walk through the array until we find the current page.
    do {
      $link = array_shift($flat);
699 700
    }
    while ($link && ($link['mlid'] != $book_link['mlid']));
Dries's avatar
Dries committed
701 702 703 704 705
    // Continue though the array and collect the links whose parent is this page.
    while (($link = array_shift($flat)) && $link['plid'] == $book_link['mlid']) {
      $data['link'] = $link;
      $data['below'] = '';
      $children[] = $data;
Dries's avatar
 
Dries committed
706 707
    }
  }
708

709
  return $children ? drupal_render(menu_tree_output($children)) : '';
Dries's avatar
 
Dries committed
710 711
}

712
/**
Dries's avatar
Dries committed
713
 * Generate the corresponding menu name from a book ID.
714
 */
Dries's avatar
Dries committed
715
function book_menu_name($bid) {
716
  return 'book-toc-' . $bid;
Dries's avatar
Dries committed
717 718
}

719
/**
720
 * Implement hook_node_load().
721
 */
722
function book_node_load($nodes, $types) {
723
  $result = db_query("SELECT * FROM {book} b INNER JOIN {menu_links} ml ON b.mlid = ml.mlid WHERE b.nid IN (:nids)", array(':nids' =>  array_keys($nodes)), array('fetch' => PDO::FETCH_ASSOC));
724 725 726 727 728
  foreach ($result as $record) {
    $nodes[$record['nid']]->book = $record;
    $nodes[$record['nid']]->book['href'] = $record['link_path'];
    $nodes[$record['nid']]->book['title'] = $record['link_title'];
    $nodes[$record['nid']]->book['options'] = unserialize($record['options']);
729 730
  }
}
731

732
/**
733
 * Implement hook_node_view().
734
 */
735 736 737
function book_node_view($node, $build_mode) {
  if ($build_mode == 'full') {
    if (!empty($node->book['bid']) && empty($node->in_preview)) {
738 739 740 741 742 743
      $node->content['book_navigation'] = array(
        '#markup' => theme('book_navigation', $node->book),
        '#weight' => 100,
      );
    }
  }
Dries's avatar
Dries committed
744

745 746
  if ($build_mode != 'rss') {
    book_node_view_link($node, $build_mode);
747
  }
748 749 750
}

/**
751
 * Implement hook_page_alter().
752 753 754 755 756 757 758 759 760 761
 *
 * Add the book menu to the list of menus used to build the active trail when
 * viewing a book page.
 */
function book_page_alter(&$page) {
  if (($node = menu_get_object()) && !empty($node->book['bid'])) {
    $active_menus = menu_get_active_menu_names();
    $active_menus[] = $node->book['menu_name'];
    menu_set_active_menu_names($active_menus);
  }
762
}
763

764
/**
765
 * Implement hook_node_presave().
766
 */
767
function book_node_presave($node) {
768 769 770
  // Always save a revision for non-administrators.
  if (!empty($node->book['bid']) && !user_access('administer nodes')) {
    $node->revision = 1;
771 772 773 774
    // The database schema requires a log message for every revision.
    if (!isset($node->log)) {
      $node->log = '';
    }
775 776 777 778 779 780
  }
  // Make sure a new node gets a new menu link.
  if (empty($node->nid)) {
    $node->book['mlid'] = NULL;
  }
}
781

782
/**
783
 * Implement hook_node_insert().
784
 */
785
function book_node_insert($node) {
786 787 788 789 790 791 792 793 794 795
  if (!empty($node->book['bid'])) {
    if ($node->book['bid'] == 'new') {
      // New nodes that are their own book.
      $node->book['bid'] = $node->nid;
    }
    $node->book['nid'] = $node->nid;
    $node->book['menu_name'] = book_menu_name($node->book['bid']);
    _book_update_outline($node);
  }
}
796

797
/**
798
 * Implement hook_node_update().
799
 */
800
function book_node_update($node) {
801 802 803 804 805 806 807 808 809 810
  if (!empty($node->book['bid'])) {
    if ($node->book['bid'] == 'new') {
      // New nodes that are their own book.
      $node->book['bid'] = $node->nid;
    }
    $node->book['nid'] = $node->nid;
    $node->book['menu_name'] = book_menu_name($node->book['bid']);
    _book_update_outline($node);
  }
}
811

812
/**
813
 * Implement hook_node_delete().
814
 */
815
function book_node_delete($node) {
816 817 818
  if (!empty($node->book['bid'])) {
    if ($node->nid == $node->book['bid']) {
      // Handle deletion of a top-level post.
819 820 821 822 823
      $result = db_query("SELECT b.nid FROM {menu_links} ml INNER JOIN {book} b on b.mlid = ml.mlid WHERE ml.plid = :plid", array(
        ':plid' => $node->book['mlid']
      ));
      foreach ($result as $child) {
        $child_node = node_load($child->nid);
824 825
        $child_node->book['bid'] = $child_node->nid;
        _book_update_outline($child_node);
Dries's avatar
Dries committed
826
      }
827 828
    }
    menu_link_delete($node->book['mlid']);
829 830 831
    db_delete('book')
      ->condition('mlid', $node->book['mlid'])
      ->execute();
832 833 834 835
  }
}

/**
836
 * Implement hook_node_prepare().
837
 */
838
function book_node_prepare($node) {
839 840 841 842 843 844 845 846 847 848 849 850
  // Prepare defaults for the add/edit form.
  if (empty($node->book) && (user_access('add content to books') || user_access('administer book outlines'))) {
    $node->book = array();

    if (empty($node->nid) && isset($_GET['parent']) && is_numeric($_GET['parent'])) {
      // Handle "Add child page" links:
      $parent = book_link_load($_GET['parent']);

      if ($parent && $parent['access']) {
        $node->book['bid'] = $parent['bid'];
        $node->book['plid'] = $parent['mlid'];
        $node->book['menu_name'] = $parent['menu_name'];
851
      }
852 853 854 855 856 857 858 859 860 861 862 863
    }
    // Set defaults.
    $node->book += _book_link_defaults(!empty($node->nid) ? $node->nid : 'new');
  }
  else {
    if (isset($node->book['bid']) && !isset($node->book['original_bid'])) {
      $node->book['original_bid'] = $node->book['bid'];
    }
  }
  // Find the depth limit for the parent select.
  if (isset($node->book['bid']) && !isset($node->book['parent_depth_limit'])) {
    $node->book['parent_depth_limit'] = _book_parent_depth_limit($node->book);
Dries's avatar
Dries committed
864 865 866
  }
}

867 868 869 870 871 872 873
/**
 * Find the depth limit for items in the parent select.
 */
function _book_parent_depth_limit($book_link) {
  return MENU_MAX_DEPTH - 1 - (($book_link['mlid'] && $book_link['has_children']) ? menu_link_children_relative_depth($book_link) : 0);
}

Dries's avatar
Dries committed
874 875 876 877 878 879 880 881
/**
 * Form altering function for the confirm form for a single node deletion.
 */
function book_form_node_delete_confirm_alter(&$form, $form_state) {
  $node = node_load($form['nid']['#value']);

  if (isset($node->book) && $node->book['has_children']) {
    $form['book_warning'] = array(
882
      '#markup' => '<p>' . t('%title is part of a book outline, and has associated child pages. If you proceed with deletion, the child pages will be relocated automatically.', array('%title' => $node->title)) . '</p>',
Dries's avatar
Dries committed
883 884
      '#weight' => -10,
    );
Dries's avatar
 
Dries committed
885
  }
Dries's avatar
 
Dries committed
886
}
Dries's avatar
 
Dries committed
887

888
/**
Dries's avatar
Dries committed
889 890 891 892 893 894
 * Return an array with default values for a book link.
 */
function _book_link_defaults($nid) {
  return array('original_bid' => 0, 'menu_name' => '', 'nid' => $nid, 'bid' => 0, 'router_path' => 'node/%', 'plid' => 0, 'mlid' => 0, 'has_children' => 0, 'weight' => 0, 'module' => 'book', 'options' => array());
}

895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915
/**
 * Process variables for book-all-books-block.tpl.php.
 *
 * The $variables array contains the following arguments:
 * - $book_menus
 *
 * All non-renderable elements are removed so that the template has full
 * access to the structured data but can also simply iterate over all
 * elements and render them (as in the default template).
 *
 * @see book-navigation.tpl.php
 */
function template_preprocess_book_all_books_block(&$variables) {
  // Remove all non-renderable elements.
  $elements = $variables['book_menus'];
  $variables['book_menus'] = array();
  foreach (element_children($elements) as $index) {
    $variables['book_menus'][$index] = $elements[$index];
  }
}

Dries's avatar
Dries committed
916
/**
917
 * Process variables for book-navigation.tpl.php.
Dries's avatar
Dries committed
918
 *
919 920
 * The $variables array contains the following arguments:
 * - $book_link
921
 *
922
 * @see book-navigation.tpl.php
923
 */
924 925 926 927 928 929
function template_preprocess_book_navigation(&$variables) {
  $book_link = $variables['book_link'];

  // Provide extra variables for themers. Not needed by default.
  $variables['book_id'] = $book_link['bid'];
  $variables['book_title'] = check_plain($book_link['link_title']);
Dries's avatar