Move setActive(true) from the trigger 'PUSH_REWRITE' to 'STREAM_BUFFER'

mapic/mistapiconnector_app.go

@@ -189,27 +189,27 @@
 }
 
 func (mc *mac) handleStreamBuffer(ctx context.Context, payload *misttriggers.StreamBufferPayload) error {
-	// We only care about connections ending
-	if !payload.IsEmpty() {
-		return nil
-	}
-
+	isActive := !payload.IsEmpty()
 	playbackID := payload.StreamName
 	if mc.baseStreamName != "" && strings.Contains(playbackID, "+") {
 		playbackID = strings.Split(playbackID, "+")[1]
 	}
 	if info, ok := mc.getStreamInfoLogged(playbackID); ok {
-		glog.Infof("Setting stream's manifestID=%s playbackID=%s active status to false", info.id, playbackID)
-		_, err := mc.lapi.SetActive(info.id, false, info.startedAt)
+		glog.Infof("Setting stream's manifestID=%s playbackID=%s active status to %v", info.id, playbackID, isActive)
+		_, err := mc.lapi.SetActive(info.id, isActive, info.startedAt)
 		if err != nil {
 			glog.Error(err)
 		}
-		mc.emitStreamStateEvent(info.stream, data.StreamState{Active: false})
-		info.mu.Lock()
-		info.stopped = true
-		info.mu.Unlock()
-		mc.removeInfoDelayed(playbackID, info.done)
-		metrics.StopStream(true)
+		mc.emitStreamStateEvent(info.stream, data.StreamState{Active: isActive})
+		if isActive {
+			metrics.StartStream()
+		} else {
+			info.mu.Lock()
+			info.stopped = true
+			info.mu.Unlock()
+			mc.removeInfoDelayed(playbackID, info.done)
+			metrics.StopStream(true)
+		}
 	}
 
 	return nil
@@ -263,19 +263,9 @@
 		} else {
 			responseName = mc.wildcardPlaybackID(stream)
 		}
-		ok, err := mc.lapi.SetActive(stream.ID, true, info.startedAt)
-		if err != nil {
-			return "", fmt.Errorf("Error calling SetActive err=%s", err)
-		} else if !ok {
-			glog.Infof("Stream id=%s streamKey=%s playbackId=%s forbidden by webhook, rejecting", stream.ID, stream.StreamKey, stream.PlaybackID)
-			mc.removeInfo(stream.PlaybackID)
-			return "", nil
-		}
 	} else {
 		glog.Errorf("Shouldn't happen streamID=%s", stream.ID)
 	}
-	go mc.emitStreamStateEvent(stream, data.StreamState{Active: true})
-	metrics.StartStream()
 	glog.Infof("Responded with '%s'", responseName)
 	return responseName, nil
 }