rendered paste body<?php// $Id: book.install,v 1.7 2007/05/25 12:46:43 dries Exp $/** * Implementation of hook_install(). */function book_install() { // Create tables. drupal_install_schema('book');}/** * Implementation of hook_uninstall(). */function book_uninstall() { // Delete menu links db_query("DELETE FROM {menu_links} WHERE module = 'book'"); menu_cache_clear_all(); // Remove tables. drupal_uninstall_schema('book');}function book_update_6000() { $ret = array(); // Multi-part update if (!isset($_SESSION['book_update_6000'])) { $book_node_type = array( 'type' => 'book', 'name' => t('Book page'), 'module' => 'node', // This is cheating, but it works 'description' => t("A simple node type like page. Maybe be used "), 'custom' => TRUE, 'modified' => TRUE, 'locked' => FALSE, ); $book_node_type = (object)_node_type_set_defaults($book_node_type); node_type_save($book_node_type); if (db_result(db_query("SELECT COUNT(*) FROM {book}"))) { // Do the update $schema['book_temp'] = array( 'fields' => array( 'nid' => array('type' => 'int', 'unsigned' => TRUE, 'not null' => TRUE, 'default' => 0), 'parent' => array('type' => 'int', 'not null' => TRUE, 'default' => 0), 'weight' => array('type' => 'int', 'not null' => TRUE, 'default' => 0, 'size' => 'tiny') ), 'indexes' => array( 'parent' => array('parent') ), 'primary key' => array('nid'), ); $schema['book'] = drupal_get_schema_unprocessed('book', 'book'); _drupal_initialize_schema('book', $schema); db_create_table($ret, $schema['book_temp']); $ret[] = update_sql("INSERT INTO {book_temp} (nid, parent, weight) SELECT b.nid, b.parent, b.weight FROM {book} b INNER JOIN {node} n on b.vid = n.vid"); $ret[] = update_sql("DROP TABLE {book}"); db_create_table($ret, $schema['book']); $_SESSION['book_update_6000'] = array(); $r = db_query("SELECT * from {book_temp} WHERE parent = 0"); // Collect all books - top-level nodes while ($a = db_fetch_array($r)) { $_SESSION['book_update_6000'][] = $a; } $ret['#finished'] = FALSE; return $ret; } else { // no exising nodes in the hierarchy $ret[] = update_sql("DROP TABLE {book}"); $schema['book'] = drupal_get_schema_unprocessed('book', 'book'); _drupal_initialize_schema('book', $schema); db_create_table($ret, $schema['book']); return $ret; } } else { // Do the batched part of the update $i = 20; // Update this many at a time print_r($_SESSION['book_update_6000']); while ($i && $_SESSION['book_update_6000']) { $book = array_pop($_SESSION['book_update_6000']); $r = db_query("SELECT * FROM {book_temp} WHERE parent = %d", $book['nid']); while ($a = db_fetch_array($r)) { $_SESSION['book_update_6000'][] = $a; } if ($book['parent']) { $book = array_merge($book, db_fetch_array(db_query("SELECT b.mlid AS plid, b.book_id FROM {book} b WHERE b.nid = %d", $book['parent']))); } else { $book['plid'] = 0; $book['book_id'] = $book['nid']; } $book += array( 'module' => 'book', 'link_path' => 'node/'. $book['nid'], 'router_path' => 'node/%', 'menu_name' => book_menu_name($book['book_id']), ); $book = array_merge($book, db_fetch_array(db_query("SELECT title AS link_title, type AS child_type FROM {node} WHERE nid = %d", $book['nid']))); if (menu_link_save($book)) { // Note - items with depth > 6 drop out of the hierarchy db_query("INSERT INTO {book} (mlid, nid, book_id, child_type) VALUES (%d, %d, %d, '%s')", $book['mlid'], $book['nid'], $book['book_id'], $book['child_type']); } else { // Too deep $book['plid'] = db_result(db_query("SELECT plid FROM {menu_links} WHERE mlid = %d", $book['plid'])); if (menu_link_save($book)) { db_query("INSERT INTO {book} (mlid, nid, book_id, child_type) VALUES (%d, %d, %d, '%s')", $book['mlid'], $book['nid'], $book['book_id'], $book['child_type']); } } $i--; } $ret['#finished'] = FALSE; } $ret['#finished'] = empty($_SESSION['book_update_6000']); if ($ret['#finished']) { $ret[] = update_sql("DROP TABLE {book_temp}"); unset($_SESSION['book_update_6000']); } return $ret;}