From 00aa5be55dada3036095f57ee9d6eb818bf31cab Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Thu, 28 Feb 2019 12:37:07 -0500
Subject: [PATCH] Use navigation bar for image editor buttons.

---
 .../Views/ImageEditor/ImageEditorView.swift      | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/SignalMessaging/Views/ImageEditor/ImageEditorView.swift b/SignalMessaging/Views/ImageEditor/ImageEditorView.swift
index 5792edc16..31bd3779b 100644
--- a/SignalMessaging/Views/ImageEditor/ImageEditorView.swift
+++ b/SignalMessaging/Views/ImageEditor/ImageEditorView.swift
@@ -40,6 +40,7 @@ public class ImageEditorView: UIView {
 
             updateButtons()
             updateGestureState()
+            delegate?.imageEditorUpdateNavigationBar()
         }
     }
 
@@ -264,10 +265,17 @@ public class ImageEditorView: UIView {
         let captionButton = navigationBarButton(imageName: "image_editor_caption",
                                              selector: #selector(didTapCaption(sender:)))
 
-        if model.canUndo() {
-            return [undoButton, newTextButton, brushButton, cropButton, captionButton]
-        } else {
-            return [newTextButton, brushButton, cropButton, captionButton]
+        switch editorMode {
+        case .text:
+            return []
+        case .brush:
+            return []
+        case .none:
+            if model.canUndo() {
+                return [undoButton, newTextButton, brushButton, cropButton, captionButton]
+            } else {
+                return [newTextButton, brushButton, cropButton, captionButton]
+            }
         }
     }