differentiate between removing listener entries

squirro · Jul 26, 2010 · 55656bc · 55656bc
1 parent 5404f3b
commit 55656bc
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/iphone/Classes/AppModule.m b/iphone/Classes/AppModule.m
@@ -237,8 +237,17 @@ -(void)willShutdownContext:(NSNotification*)note
 			{
 				if ([entry context] == context)
 				{
-					[found addObject:[NSArray
-									   arrayWithObjects:[entry type],[entry listener],nil]];
+					id listener = [entry listener];
+					if ([listener isKindOfClass:[KrollCallback class]])
+					{
+						[found addObject:[NSArray
+										  arrayWithObjects:((KrollCallback*)listener).type,listener,nil]];
+					}
+					else 
+					{
+						[found addObject:[NSArray
+									   arrayWithObjects:[entry type],listener,nil]];
+					}
 				}
 			}
 		}