This is getting nuts, I need an internal branch.

[cydia.git] / UICaboodle / RVBook.mm

diff --git a/UICaboodle/RVBook.mm b/UICaboodle/RVBook.mm
index d7a39490cdc1eebb45a33a8a9e872bb321345d93..4b5a3999303e56812adf5181460851e8afef8a3c 100644 (file)
--- a/UICaboodle/RVBook.mm
+++ b/UICaboodle/RVBook.mm
@@ -121,6 +121,10 @@
     delegate_ = delegate;
 }
 
+- (void) setHook:(id)hook {
+    hook_ = hook;
+}
+
 - (void) setPage:(RVPage *)page {
     if ([pages_ count] != 0)
         [[pages_ lastObject] setPageActive:NO];
@@ -266,10 +270,16 @@
 }
 
 - (void) reloadData {
-    for (int i(0), e([pages_ count]); i != e; ++i) {
-        RVPage *page([pages_ objectAtIndex:(e - i - 1)]);
+    size_t count([pages_ count]);
+    for (size_t i(0); i != count; ++i) {
+        RVPage *page([pages_ objectAtIndex:(count - i - 1)]);
         [page reloadData];
     }
+
+    if (count != 0) {
+        RVPage *page([pages_ lastObject]);
+        [self reloadButtonsForPage:page];
+    }
 }
 
 - (CGRect) pageBounds {
@@ -277,6 +287,8 @@
 }
 
 - (void) close {
+    if (hook_ != nil)
+        [hook_ didCloseBook:self];
 }
 
 @end
@@ -300,6 +312,7 @@
 
 - (void) close {
     [self popFromSuperviewAnimated:YES];
+    [super close];
 }
 
 @end