ui refresh after mode change and activate display for DocumentMode

Author: krVatsal

editor/src/messages/portfolio/document/document_message_handler.rs

@@ -1124,15 +1124,25 @@ impl MessageHandler<DocumentMessage, DocumentMessageContext<'_>> for DocumentMes
 					DocumentMode::MaskMode => DocumentMode::DesignMode,
 					_ => DocumentMode::MaskMode,
 				};
+				responses.add(PortfolioMessage::UpdateDocumentWidgets);
 			}
 			DocumentMessage::SetDocumentMode { document_mode } => {
-				self.document_mode = document_mode;
+				if self.document_mode != document_mode {
+					self.document_mode = document_mode;
+					responses.add(PortfolioMessage::UpdateDocumentWidgets);
+				}
 			}
 			DocumentMessage::EnterMaskMode => {
-				self.document_mode = DocumentMode::MaskMode;
+				if self.document_mode != DocumentMode::MaskMode {
+					self.document_mode = DocumentMode::MaskMode;
+					responses.add(PortfolioMessage::UpdateDocumentWidgets);
+				}
 			}
 			DocumentMessage::ExitMaskMode { discard: _ } => {
-				self.document_mode = DocumentMode::DesignMode;
+				if self.document_mode != DocumentMode::DesignMode {
+					self.document_mode = DocumentMode::DesignMode;
+					responses.add(PortfolioMessage::UpdateDocumentWidgets);
+				}
 			}
 			DocumentMessage::DrawMarchingAntsOverlay { context: _ } => {}
 			DocumentMessage::AddTransaction => {

editor/src/messages/portfolio/document/utility_types/misc.rs

@@ -39,16 +39,16 @@ pub enum DocumentMode {
 	MaskMode,
 }
 
-// impl fmt::Display for DocumentMode {
-// 	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-// 		match self {
-// 			DocumentMode::DesignMode => write!(f, "Design Mode"),
-// 			DocumentMode::SelectMode => write!(f, "Select Mode"),
-// 			DocumentMode::GuideMode => write!(f, "Guide Mode"),
-// 			DocumentMode::MaskMode => write!(f, "Mask Mode"),
-// 		}
-// 	}
-// }
+impl fmt::Display for DocumentMode {
+	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+		match self {
+			DocumentMode::DesignMode => write!(f, "Design Mode"),
+			DocumentMode::SelectMode => write!(f, "Select Mode"),
+			DocumentMode::GuideMode => write!(f, "Guide Mode"),
+			DocumentMode::MaskMode => write!(f, "Mask Mode"),
+		}
+	}
+}
 
 // impl DocumentMode {
 // 	pub fn icon_name(&self) -> String {