<?php
// $Id$

/**
 * @file book-upgrade.php
 *   File description.
 */

// Setup variables.
$book_nid = 1; // The nid of the parent book.
$book_mlid = 128; // The mlid of the parent book.
$mlid_start = 133; // The starting mlid to insert for this process.

$menu_name = 'book-toc-' . $book_nid;
$parents = array(128);
mysql_connect($argv[1], $argv[2], $argv[3]);
mysql_select_db($argv[4]);
$book_structure = book_structure($book_nid);
mysql_close();
$file = fopen('./book-upgrade.sql', 'w');

print_r($book_structure);

// Delete all the existing child entries in the book.
fwrite($file, "DELETE FROM menu_links WHERE menu_name = '$menu_name' AND plid != 0;\n");

book_links_save($file, $menu_name, $book_mlid, $book_structure, $parents, $mlid_start);

// Get the structure of a book.
function book_structure($parent) {
  $result = mysql_query("SELECT b.*, n.title FROM book b INNER JOIN node n ON b.nid = n.nid WHERE parent = ". (int) $parent ." GROUP BY b.nid ORDER BY b.vid DESC");
  $children = array();
  while ($child = mysql_fetch_object($result)) {
    $children[$child->nid] = $child;
    $children[$child->nid]->children = book_structure($child->nid);
  }
  return $children;
}

// Create SQL queries for the structure of a book.
function book_links_save($file, $menu_name, $book_mlid, $book_structure, $parents = array(), $mlid_start = 0) {
  static $mlid;

  if (!isset($mlid)) {
    $mlid = $mlid_start;
  }

  array_push($parents, $mlid);

  foreach ($book_structure as $page) {
    $mlid++;
    $plid = end($parents);
    $nid = $page->nid;
    $title = $page->title;
    $has_children = (int) !empty($page->children);
    $weight = $page->weight;
    $depth = count($parents);
    for ($index = 0; $index < 9; $index++) {
      ${'p' . ($index + 1)} = isset($parents[$index]) ? $parents[$index] : 0;
    }
    fwrite($file, "INSERT INTO menu_links (menu_name, mlid, plid, link_path, router_path, link_title, options, module, has_children, weight, depth, p1, p2, p3, p4, p5, p6, p7, p8, p9) VALUES ('$menu_name', $mlid, $plid, 'node/$nid', 'node/%', '$title', 'a:0:{}', 'book', $has_children, $weight, $depth, $p1, $p2, $p3, $p4, $p5, $p6, $p7, $p8, $p9);\n");

    if ($has_children) {
      book_links_save($file, $menu_name, $book_mlid, $page->children, $parents);
    }
  }
}

fclose($file);