From 5d60b7caad540008bb2afa7124a4e9710f1e6343 Mon Sep 17 00:00:00 2001
From: Matthew Chen <charlesmchen@gmail.com>
Date: Mon, 27 Feb 2017 10:47:19 -0500
Subject: [PATCH] =?UTF-8?q?Fix=20back=20button=20in=20=E2=80=9Cnew=20conve?=
 =?UTF-8?q?rsation=E2=80=9D=20view.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

// FREEBIE
---
 Signal/src/util/UIViewController+OWS.h             |  2 ++
 Signal/src/util/UIViewController+OWS.m             | 14 ++++++++++++--
 .../MessageComposeTableViewController.m            |  7 ++++++-
 3 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/Signal/src/util/UIViewController+OWS.h b/Signal/src/util/UIViewController+OWS.h
index b95a4b48f..3d5bdef3b 100644
--- a/Signal/src/util/UIViewController+OWS.h
+++ b/Signal/src/util/UIViewController+OWS.h
@@ -8,7 +8,9 @@ NS_ASSUME_NONNULL_BEGIN
 
 @interface UIViewController (OWS)
 
+- (UIBarButtonItem *)createOWSBackButtonWithSelector:(SEL)selector;
 - (UIBarButtonItem *)createOWSBackButton;
+- (void)useOWSBackButtonWithSelector:(SEL)selector;
 - (void)useOWSBackButton;
 
 @end
diff --git a/Signal/src/util/UIViewController+OWS.m b/Signal/src/util/UIViewController+OWS.m
index 238e38afa..7789ecfa0 100644
--- a/Signal/src/util/UIViewController+OWS.m
+++ b/Signal/src/util/UIViewController+OWS.m
@@ -8,17 +8,27 @@ NS_ASSUME_NONNULL_BEGIN
 
 @implementation UIViewController (OWS)
 
-- (UIBarButtonItem *)createOWSBackButton
+- (UIBarButtonItem *)createOWSBackButtonWithSelector:(SEL)selector
 {
     UIImage *backImage = [UIImage imageNamed:@"NavBarBack"];
     OWSAssert(backImage);
     UIBarButtonItem *backItem = [[UIBarButtonItem alloc] initWithImage:backImage
                                                                  style:UIBarButtonItemStylePlain
                                                                 target:self
-                                                                action:@selector(backButtonPressed:)];
+                                                                action:selector];
     return backItem;
 }
 
+- (UIBarButtonItem *)createOWSBackButton
+{
+    return [self createOWSBackButtonWithSelector:@selector(backButtonPressed:)];
+}
+
+- (void)useOWSBackButtonWithSelector:(SEL)selector
+{
+    self.navigationItem.leftBarButtonItem = [self createOWSBackButtonWithSelector:selector];
+}
+
 - (void)useOWSBackButton
 {
     self.navigationItem.leftBarButtonItem = [self createOWSBackButton];
diff --git a/Signal/src/view controllers/MessageComposeTableViewController.m b/Signal/src/view controllers/MessageComposeTableViewController.m
index ed8d5bf59..ba5a5b1b1 100644
--- a/Signal/src/view controllers/MessageComposeTableViewController.m	
+++ b/Signal/src/view controllers/MessageComposeTableViewController.m	
@@ -117,7 +117,7 @@ NSString *const MessageComposeTableViewControllerCellContact = @"ContactTableVie
 - (void)viewDidLoad {
     [super viewDidLoad];
     [self.navigationController.navigationBar setTranslucent:NO];
-    [self useOWSBackButton];
+    [self useOWSBackButtonWithSelector:@selector(closeButton:)];
     
     self.navigationItem.rightBarButtonItem.accessibilityLabel = NSLocalizedString(
         @"CREATE_NEW_GROUP", @"Accessibility label for the create group new group button");
@@ -150,6 +150,11 @@ NSString *const MessageComposeTableViewControllerCellContact = @"ContactTableVie
     }
 }
 
+- (void)closeButton:(id)sender
+{
+    [self dismissViewControllerAnimated:YES completion:nil];
+}
+
 - (UILabel *)createLabelWithFirstLine:(NSString *)firstLine andSecondLine:(NSString *)secondLine {
     UILabel *label      = [[UILabel alloc] init];
     label.textColor     = [UIColor grayColor];