Fix navbar on edit screen

lockbox-ios/Presenter/ItemDetailPresenter.swift

@@ -19,6 +19,7 @@ protocol ItemDetailViewProtocol: class, StatusAlertView, AlertControllerView {
     var titleText: AnyObserver<String?> { get }
     var rightButtonText: AnyObserver<String?> { get }
     var leftButtonText: AnyObserver<String?> { get }
+    var leftButtonIcon: AnyObserver<UIImage?> { get }
     var deleteHidden: AnyObserver<Bool> { get }
 }
 
@@ -211,6 +212,16 @@ class ItemDetailPresenter {
                 .subscribe(self.view!.leftButtonText)
                 .disposed(by: self.disposeBag)
 
+        self.itemDetailStore.isEditing
+            .withLatestFrom(self.sizeClassStore.shouldDisplaySidebar) { (editing: Bool, sidebar: Bool) -> UIImage? in
+                if !editing && !sidebar {
+                    return UIImage(named: "back")
+                }
+                return nil
+            }
+            .subscribe(self.view!.leftButtonIcon)
+            .disposed(by: self.disposeBag)
+
         self.sizeClassStore.shouldDisplaySidebar
                 .subscribe(onNext: { (enableSidebar) in
                     self.view?.enableSwipeNavigation(enabled: !enableSidebar)

lockbox-ios/View/ItemDetailView.swift

@@ -79,15 +79,25 @@ extension ItemDetailView: ItemDetailViewProtocol {
     }
 
     var rightButtonText: AnyObserver<String?> {
-        // swiftlint:disable:next force_cast
-        return (self.navigationItem.rightBarButtonItem!.customView as! UIButton).rx.title().asObserver()
+        return AnyObserver<String?> { evt in
+            // swiftlint:disable:next force_cast
+            let button = (self.navigationItem.rightBarButtonItem!.customView as! UIButton)
+            button.setTitle(evt.element ?? "", for: UIControl.State.normal)
+            button.sizeToFit()
+
+        }
     }
 
     var leftButtonText: AnyObserver<String?> {
         // swiftlint:disable:next force_cast
         return (self.navigationItem.leftBarButtonItem!.customView as! UIButton).rx.title().asObserver()
     }
 
+    var leftButtonIcon: AnyObserver<UIImage?> {
+        // swiftlint:disable:next force_cast
+        return (self.navigationItem.leftBarButtonItem!.customView as! UIButton).rx.image().asObserver()
+    }
+
     var deleteHidden: AnyObserver<Bool> {
         return self.deleteButton.rx.isHidden.asObserver()
     }

lockbox-iosTests/ItemDetailPresenterSpec.swift

@@ -19,6 +19,7 @@ class ItemDetailPresenterSpec: QuickSpec {
         var rightButtonTextObserverStub: TestableObserver<String?>!
         var leftButtonTextObserverStub: TestableObserver<String?>!
         var deleteHiddenObserverStub: TestableObserver<Bool>!
+        var leftButtonIconObserverStub: TestableObserver<UIImage?>!
         let editStub = PublishSubject<Void>()
         let cellTappedStub = PublishSubject<String?>()
         let deleteTappedStub = PublishSubject<Void>()
@@ -78,6 +79,10 @@ class ItemDetailPresenterSpec: QuickSpec {
             return self.deleteHiddenObserverStub.asObserver()
         }
 
+        var leftButtonIcon: AnyObserver<UIImage?> {
+            return self.leftButtonIconObserverStub.asObserver()
+        }
+
         func displayTemporaryAlert(_ message: String, timeout: TimeInterval, icon: UIImage?) {
             self.tempAlertMessage = message
             self.tempAlertTimeout = timeout
@@ -175,6 +180,7 @@ class ItemDetailPresenterSpec: QuickSpec {
                 self.view.rightButtonTextObserverStub = self.scheduler.createObserver(String?.self)
                 self.view.leftButtonTextObserverStub = self.scheduler.createObserver(String?.self)
                 self.view.deleteHiddenObserverStub = self.scheduler.createObserver(Bool.self)
+                self.view.leftButtonIconObserverStub = self.scheduler.createObserver(UIImage?.self)
 
                 self.dispatcher = FakeDispatcher()
                 self.dataStore = FakeDataStore()