- Implement result view saving in a more general way within CustomQuery; now when editing custom query results *and* sorting custom query results, scroll position and selection index are preserved

1 files changed, 13 insertions, 1 deletions

diff --git a/Source/CustomQuery.h b/Source/CustomQuery.h
index 15f0dba6..6650ee79 100644
--- a/Source/CustomQuery.h
+++ b/Source/CustomQuery.h
@@ -139,11 +139,15 @@
 	NSRect editedScrollViewRect;
 
 	BOOL isWorking;
-	BOOL tableReloadAfterEditing;
+	BOOL tableRowsSelectable;
+	BOOL reloadingExistingResult;
 	BOOL queryIsTableSorter;
 	BOOL isDesc;
 	NSNumber *sortField;
 
+	NSIndexSet *selectionIndexToRestore;
+	NSRect selectionViewportToRestore;
+
 	NSString *fieldIDQueryString;
 
 	NSUInteger numberOfQueries;
@@ -187,6 +191,14 @@
 - (NSArray *)currentResult;
 - (void)processResultIntoDataStorage:(MCPStreamingResult *)theResult;
 
+// Retrieving and setting table state
+- (NSIndexSet *) resultSelectedRowIndexes;
+- (NSRect) resultViewport;
+- (void) setResultSelectedRowIndexesToRestore:(NSIndexSet *)theIndexSet;
+- (void) setResultViewportToRestore:(NSRect)theViewport;
+- (void) storeCurrentResultViewForRestoration;
+- (void) clearResultViewDetailsToRestore;
+
 // MySQL Help
 - (void)showAutoHelpForCurrentWord:(id)sender;
 - (NSString *)getHTMLformattedMySQLHelpFor:(NSString *)searchString calledByAutoHelp:(BOOL)autoHelp;