Allow Discovery and GetStatus commands while updating firmware

The do not interfere with the firmware update, but can be relevant for
the client.

Author: krksgbr

src/dividat-driver/senso/websocket.go

@@ -265,12 +265,13 @@ func (handle *Handle) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 				return
 			}
 
-			if handle.isUpdatingFirmware() {
-				handle.log.Info("Firmware update in progress, ignoring websocket message.")
-				continue
-			}
-
 			if messageType == websocket.BinaryMessage {
+
+				if handle.isUpdatingFirmware() {
+					handle.log.Info("Firmware update in progress, ignoring Senso command.")
+					continue
+				}
+
 				handle.broker.TryPub(msg, "tx")
 
 			} else if messageType == websocket.TextMessage {
@@ -283,6 +284,11 @@ func (handle *Handle) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 				}
 				log.WithField("command", prettyPrintCommand(command)).Debug("Received command.")
 
+				if handle.isUpdatingFirmware() && (command.GetStatus == nil || command.Discover == nil) {
+					handle.log.Info("Firmware update in progress, ignoring UpdateFirmware command.")
+					continue
+				}
+
 				err := handle.dispatchCommand(ctx, log, command, sendMessage)
 				if err != nil {
 					return