book.module 40.2 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_list().
211 212 213 214 215 216 217 218 219 220
 */
function book_block_list() {
  $block = array();
  $block['navigation']['info'] = t('Book navigation');
  $block['navigation']['cache'] = BLOCK_CACHE_PER_PAGE | BLOCK_CACHE_PER_ROLE;

  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 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
    }
    $block['content'] = theme('book_all_books_block', $book_menus);
  }
  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']) : '';
    }
  }
269

270 271
  return $block;
}
272

273
/**
274
 * Implement hook_block_configure().
275 276 277 278 279 280 281 282 283 284 285 286 287 288
 */
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."),
    );
289

290 291
  return $form;
}
292

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

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

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

312
/**
Dries's avatar
Dries committed
313 314 315 316
 * 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.
317
 */
Dries's avatar
Dries committed
318
function book_get_books() {
319
  $all_books = &drupal_static(__FUNCTION__);
Dries's avatar
Dries committed
320 321 322

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

Dries's avatar
Dries committed
325
    if ($nids) {
326 327 328 329 330 331 332 333 334 335 336 337 338 339
      $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
340 341
        $link['href'] = $link['link_path'];
        $link['options'] = unserialize($link['options']);
342
        $all_books[$link['bid']] = $link;
Dries's avatar
Dries committed
343 344
      }
    }
345
  }
346

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

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

360
  if (!empty($form['#node_edit_form'])) {
361
    // Add elements to the node form.
Dries's avatar
Dries committed
362 363 364 365 366
    $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))) {
367
        // Already in the book hierarchy, or this node type is allowed.
Dries's avatar
Dries committed
368 369 370 371 372 373 374 375 376
        $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)'),
377
         // Submit the node form so the parent select options get updated.
378
         // This is typically only used when JS is disabled. Since the parent options
379 380
         // 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
381
         // selected book. This is similar to what happens during a node preview.
382
        '#submit' => array('node_form_submit_build_node'),
Dries's avatar
Dries committed
383 384 385
        '#weight' => 20,
      );
    }
386
  }
Dries's avatar
Dries committed
387
}
388

Dries's avatar
Dries committed
389
/**
390
 * Build the parent selection form element for the node form or outline tab.
Dries's avatar
Dries committed
391 392 393 394 395 396
 *
 * 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) {
397 398 399
  if (variable_get('menu_override_parent_selector', FALSE)) {
    return array();
  }
Dries's avatar
Dries committed
400 401 402 403 404 405 406
  // 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>',
  );
407

Dries's avatar
Dries committed
408 409 410
  if ($book_link['nid'] === $book_link['bid']) {
    // This is a book - at the top level.
    if ($book_link['original_bid'] === $book_link['bid']) {
411
      $form['#prefix'] .= '<em>' . t('This is the top-level page in this book.') . '</em>';
Dries's avatar
Dries committed
412 413
    }
    else {
414
      $form['#prefix'] .= '<em>' . t('This will be the top-level page in this book.') . '</em>';
Dries's avatar
Dries committed
415 416 417
    }
  }
  elseif (!$book_link['bid']) {
418
    $form['#prefix'] .= '<em>' . t('No book selected.') . '</em>';
419 420
  }
  else {
Dries's avatar
Dries committed
421 422 423 424
    $form = array(
      '#type' => 'select',
      '#title' => t('Parent item'),
      '#default_value' => $book_link['plid'],
425
      '#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)),
426
      '#options' => book_toc($book_link['bid'], $book_link['parent_depth_limit'], array($book_link['mlid'])),
427
      '#attributes' => array('class' => array('book-title-select')),
428
    );
Dries's avatar
 
Dries committed
429
  }
430

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

434
/**
Dries's avatar
Dries committed
435
 * Build the common elements of the book form for the node and outline forms.
436
 */
Dries's avatar
Dries committed
437 438 439 440 441 442 443 444 445 446
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,
447 448
    '#group' => 'additional_settings',
    '#attached_js' => array(drupal_get_path('module', 'book') . '/book.js'),
Dries's avatar
Dries committed
449
    '#tree' => TRUE,
450
    '#attributes' => array('class' => array('book-outline-form')),
451
  );
452
  foreach (array('menu_name', 'mlid', 'nid', 'router_path', 'has_children', 'options', 'module', 'original_bid', 'parent_depth_limit') as $key) {
Dries's avatar
Dries committed
453 454 455 456 457 458 459 460 461 462
    $form['book'][$key] = array(
      '#type' => 'value',
      '#value' => $node->book[$key],
    );
  }

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

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

472
  if (isset($node->nid) && ($nid == $node->book['original_bid']) && ($node->book['parent_depth_limit'] == 0)) {
473 474 475 476 477 478 479 480 481
    // 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
482 483
  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.
484
    $options = array($nid => '<' . t('create a new book') . '>') + $options;
Dries's avatar
Dries committed
485 486
  }
  if (!$node->book['mlid']) {
487
    // The node is not currently in the hierarchy.
488
    $options = array(0 => '<' . t('none') . '>') + $options;
Dries's avatar
Dries committed
489 490 491 492 493 494 495 496 497 498 499
  }

  // 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,
500
    '#attributes' => array('class' => array('book-title-select')),
501
    '#ajax' => array(
502 503
      'path' => 'book/js/form',
      'wrapper' => 'edit-book-plid-wrapper',
504 505
      'effect' => 'fade',
      'speed' => 'fast',
506
    ),
507
  );
Dries's avatar
Dries committed
508
}
Dries's avatar
 
Dries committed
509

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

522
  $node->book['link_path'] = 'node/' . $node->nid;
Dries's avatar
Dries committed
523 524 525 526 527 528 529 530 531 532
  $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'])) {
533 534 535
      $parent = db_query("SELECT * FROM {book} WHERE mlid = :mlid", array(
        ':mlid' => $node->book['plid'],
      ))->fetchAssoc();
Dries's avatar
Dries committed
536 537
    }
    if (empty($node->book['plid']) || !$parent || $parent['bid'] != $node->book['bid']) {
538 539 540
      $node->book['plid'] = db_query("SELECT mlid FROM {book} WHERE nid = :nid", array(
        ':nid' => $node->book['bid'],
      ))->fetchField();
Dries's avatar
Dries committed
541 542 543
      $node->book['parent_mismatch'] = TRUE; // Likely when JS is disabled.
    }
  }
544

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

Dries's avatar
Dries committed
565 566
    return TRUE;
  }
567 568

  // Failed to save the menu link.
Dries's avatar
Dries committed
569 570 571
  return FALSE;
}

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

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

/**
 * Get the book menu tree for a page, and return it as a linear array.
596
 *
Dries's avatar
Dries committed
597 598
 * @param $book_link
 *   A fully loaded menu link that is part of the book hierarchy.
599
 * @return
Dries's avatar
Dries committed
600 601
 *   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
602
 *   element corresponding to $node. The children of $node (if any) will come
603 604
 *   immediately after it in the array, and links will only be fetched as deep
 *   as one level deeper than $book_link.
605
 */
Dries's avatar
Dries committed
606
function book_get_flat_menu($book_link) {
607
  $flat = &drupal_static(__FUNCTION__, array());
Dries's avatar
Dries committed
608 609

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

Dries's avatar
Dries committed
616
  return $flat[$book_link['mlid']];
Dries's avatar
 
Dries committed
617 618
}

619
/**
Dries's avatar
Dries committed
620
 * Recursive helper function for book_get_flat_menu().
621
 */
Dries's avatar
Dries committed
622 623 624 625 626 627 628 629
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
630
  }
Dries's avatar
Dries committed
631
}
Dries's avatar
 
Dries committed
632

Dries's avatar
Dries committed
633 634 635 636 637 638 639
/**
 * 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
640
  }
Dries's avatar
Dries committed
641 642 643 644 645 646 647 648 649 650 651 652
  $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.
653 654
      $tree = book_menu_subtree_data($prev);
      $data = array_shift($tree);
Dries's avatar
Dries committed
655 656 657 658
      // The link of interest is the last child - iterate to find the deepest one.
      while ($data['below']) {
        $data = end($data['below']);
      }
659

Dries's avatar
Dries committed
660 661 662 663 664
      return $data['link'];
    }
    else {
      return $prev;
    }
Dries's avatar
 
Dries committed
665 666 667
  }
}

668
/**
Dries's avatar
Dries committed
669
 * Fetches the menu link for the next page of the book.
670
 */
Dries's avatar
Dries committed
671 672 673 674 675
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);
676 677 678
  }
  while ($key && $key != $book_link['mlid']);

Dries's avatar
Dries committed
679 680
  if ($key == $book_link['mlid']) {
    return current($flat);
Dries's avatar
 
Dries committed
681
  }
Dries's avatar
Dries committed
682
}
Dries's avatar
 
Dries committed
683

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

Dries's avatar
Dries committed
690 691 692 693 694 695
  $children = array();

  if ($book_link['has_children']) {
    // Walk through the array until we find the current page.
    do {
      $link = array_shift($flat);
696 697
    }
    while ($link && ($link['mlid'] != $book_link['mlid']));
Dries's avatar
Dries committed
698 699 700 701 702
    // 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
703 704
    }
  }
705

Dries's avatar
Dries committed
706
  return $children ? menu_tree_output($children) : '';
Dries's avatar
 
Dries committed
707 708
}

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

716
/**
717
 * Implement hook_node_load().
718
 */
719
function book_node_load($nodes, $types) {
720
  $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));
721 722 723 724 725
  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']);
726 727
  }
}
728

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

742 743
  if ($build_mode != 'rss') {
    book_node_view_link($node, $build_mode);
744
  }
745 746 747
}

/**
748
 * Implement hook_page_alter().
749 750 751 752 753 754 755 756 757 758
 *
 * 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);
  }
759
}
760

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

779
/**
780
 * Implement hook_node_insert().
781
 */
782
function book_node_insert($node) {
783 784 785 786 787 788 789 790 791 792
  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);
  }
}
793

794
/**
795
 * Implement hook_node_update().
796
 */
797
function book_node_update($node) {
798 799 800 801 802 803 804 805 806 807
  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);
  }
}
808

809
/**
810
 * Implement hook_node_delete().
811
 */
812
function book_node_delete($node) {
813 814 815
  if (!empty($node->book['bid'])) {
    if ($node->nid == $node->book['bid']) {
      // Handle deletion of a top-level post.
816 817 818 819 820
      $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);
821 822
        $child_node->book['bid'] = $child_node->nid;
        _book_update_outline($child_node);
Dries's avatar
Dries committed
823
      }
824 825
    }
    menu_link_delete($node->book['mlid']);
826 827 828
    db_delete('book')
      ->condition('mlid', $node->book['mlid'])
      ->execute();
829 830 831 832
  }
}

/**
833
 * Implement hook_node_prepare().
834
 */
835
function book_node_prepare($node) {
836 837 838 839 840 841 842 843 844 845 846 847
  // 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'];
848
      }
849 850 851 852 853 854 855 856 857 858 859 860
    }
    // 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
861 862 863
  }
}

864 865 866 867 868 869 870
/**
 * 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
871 872 873 874 875 876 877 878
/**
 * 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(
879
      '#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
880 881
      '#weight' => -10,
    );
Dries's avatar
 
Dries committed
882
  }
Dries's avatar
 
Dries committed
883
}
Dries's avatar
 
Dries committed
884

885
/**
Dries's avatar
Dries committed
886 887 888 889 890 891 892
 * 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());
}

/**
893
 * Process variables for book-navigation.tpl.php.
Dries's avatar
Dries committed
894
 *
895 896
 * The $variables array contains the following arguments:
 * - $book_link
897
 *
898
 * @see book-navigation.tpl.php
899
 */
900 901 902 903 904 905
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']);
906
  $variables['book_url'] = 'node/' . $book_link['bid'];
907 908
  $variables['current_depth'] = $book_link['depth'];
  $variables['tree'] = '';
909

Dries's avatar
Dries committed
910
  if ($book_link['mlid']) {
911
    $variables['tree'] = book_children($book_link);
Dries's avatar
 
Dries committed
912

Dries's avatar
Dries committed
913
    if ($prev = book_prev($book_link)) {
914 915 916 917
      $prev_href = url($prev['href']);
      drupal_add_link(array('rel' => 'prev', 'href' => $prev_href));
      $variables['prev_url'] = $prev_href;
      $variables['prev_title'] = check_plain($prev['title']);
Dries's avatar
 
Dries committed
918
    }
919

Dries's avatar
Dries committed
920
    if ($book_link['plid'] && $parent = book_link_load($book_link['plid'