From 81832d0b5afcf6187e49850fe5cff6006fbb7022 Mon Sep 17 00:00:00 2001
From: John Bintz <john@coswellproductions.com>
Date: Sat, 28 Nov 2009 16:55:09 -0500
Subject: [PATCH] make first link work with multipage

---
 test/widgets/GraphicalNavigationWidgetTest.php | 18 ++++++++++++------
 widgets/GraphicalNavigationWidget.inc          | 12 +++++++++++-
 2 files changed, 23 insertions(+), 7 deletions(-)

diff --git a/test/widgets/GraphicalNavigationWidgetTest.php b/test/widgets/GraphicalNavigationWidgetTest.php
index 5bb2edf..faa84f7 100644
--- a/test/widgets/GraphicalNavigationWidgetTest.php
+++ b/test/widgets/GraphicalNavigationWidgetTest.php
@@ -28,18 +28,24 @@ class GraphicalNavigationWidgetTest extends PHPUnit_Framework_TestCase {
 
 	function providerTestIsNavLinkVisible() {
 		return array(
-			array('first', 1, 2, true),
-			array('first', 1, 1, false),
-			array('last', 1, 2, true),
-			array('last', 1, 1, false),
-			array('prev', 1, 2, true),
+			array('first', 1, 2, 1, 1, true),
+			array('first', 1, 1, 1, 1, false),
+			array('last', 1, 2, 1, 1, true),
+			array('last', 1, 1, 1, 1, false),
+			array('prev', 1, 2, 1, 1, true),
+			array('first', 1, 1, 2, 2, true),
+			array('first', 1, 1, 1, 2, false),
 		);
 	}
 
 	/**
 	 * @dataProvider providerTestIsNavLinkVisible
 	 */
-	function testIsNavLinkVisible($which, $current_id, $target_id, $expected_result) {
+	function testIsNavLinkVisible($which, $current_id, $target_id, $_page, $_numpages, $expected_result) {
+		global $page, $numpages;
+		$page = $_page;
+		$numpages = $_numpages;
+
 		$current = (object)array('ID' => $current_id);
 		$target  = (object)array('ID' => $target_id);
 
diff --git a/widgets/GraphicalNavigationWidget.inc b/widgets/GraphicalNavigationWidget.inc
index e081feb..c6112eb 100644
--- a/widgets/GraphicalNavigationWidget.inc
+++ b/widgets/GraphicalNavigationWidget.inc
@@ -170,10 +170,16 @@ class GraphicalNavigationWidget extends WP_Widget {
 	 * @return boolean True if this link should be visible.
 	 */
 	function _will_display_nav_link($which, $current, $target) {
+		global $page, $numpages;
+
 		$return = true;
 
 		switch ($which) {
 			case 'first':
+				if ($page != 1 && $numpages > 1) {
+					$return = true;
+					break;
+				}
 			case 'last':
 				$return = ($target->ID != $current->ID);
 				break;
@@ -335,6 +341,10 @@ class GraphicalNavigationWidget extends WP_Widget {
 		return array($post_nav, $post, $instance);
 	}
 
+	function _get_page_count($post) {
+		return preg_match_all('#<!--nextpage-->#', $post->post_content, $matches);
+	}
+
 	function _comicpress_set_up_post_nav_multi_page_support($info) {
 		global $page, $numpages;
 		list($post_nav, $post, $instance) = $info;
@@ -349,7 +359,7 @@ class GraphicalNavigationWidget extends WP_Widget {
 			if ($page == 1) {
 				foreach (array('previous', 'previous_storyline') as $which) {
 					if (isset($post_nav[$which])) {
-						$count = preg_match_all('#<!--nextpage-->#', $post_nav[$which]->post_content, $matches);
+						$count = $this->_get_page_count($post_nav[$which]);
 						if (!empty($count)) {
 							$post_nav[$which] = $this->_build_in_post_page_link($post_nav[$which], $count + 1);
 						}