more refinements from PR jessesquires#811

Author: jessesquires

JSQMessagesDemo/DemoMessagesViewController.m

@@ -70,9 +70,10 @@ - (void)viewDidLoad
                                                                              action:@selector(receiveMessagePressed:)];
 
     /**
-     *  Register custom menu actions for cells
+     *  Register custom menu actions for cells.
      */
-    [JSQMessagesCollectionViewCell registerMenuAction:@selector(delete:)];
+    [JSQMessagesCollectionViewCell registerMenuAction:@selector(customAction:)];
+    [UIMenuController sharedMenuController].menuItems = @[ [[UIMenuItem alloc] initWithTitle:@"Custom Action" action:@selector(customAction:)] ];
 
 
     /**
@@ -512,9 +513,11 @@ - (UICollectionViewCell *)collectionView:(JSQMessagesCollectionView *)collection
 
 #pragma mark - UICollectionView Delegate
 
+#pragma mark - Custom menu items
+
 - (BOOL)collectionView:(UICollectionView *)collectionView canPerformAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender
 {
-    if (action == @selector(delete:)) {
+    if (action == @selector(customAction:)) {
         return YES;
     }
 
@@ -523,14 +526,26 @@ - (BOOL)collectionView:(UICollectionView *)collectionView canPerformAction:(SEL)
 
 - (void)collectionView:(UICollectionView *)collectionView performAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender
 {
-    if (action == @selector(delete:)) {
-        NSLog(@"Did receive delegate action %@", NSStringFromSelector(action));
+    if (action == @selector(customAction:)) {
+        [self customAction:sender];
         return;
     }
 
     [super collectionView:collectionView performAction:action forItemAtIndexPath:indexPath withSender:sender];
 }
 
+- (void)customAction:(id)sender
+{
+    NSLog(@"Custom action received! Sender: %@", sender);
+
+    [[[UIAlertView alloc] initWithTitle:@"Custom Action"
+                               message:nil
+                              delegate:nil
+                     cancelButtonTitle:@"OK"
+                      otherButtonTitles:nil]
+     show];
+}
+
 
 
 #pragma mark - JSQMessages collection view flow layout delegate

JSQMessagesViewController/Views/JSQMessagesCollectionViewCell.h

@@ -196,7 +196,8 @@
  *
  *  @param action The selector to register with the cell.
  *
- *  @discussion Non-Standard actions still need to be added to the `UIMenuController` manually.
+ *  @discussion Non-standard or non-system actions must be added to the `UIMenuController` manually.
+ *  You can do this by creating a new `UIMenuItem` and adding it via the controller's `menuItems` property.
  *
  *  @warning Note that all message cells share the all actions registered here.
  */