Add method for probing for torch presence

LinusU · LinusU · commit 66f91b43f541 · 2019-10-26T22:58:17.000+02:00
diff --git a/ios/Classes/FlutterRTCMediaStream.h b/ios/Classes/FlutterRTCMediaStream.h
@@ -11,6 +11,9 @@
 
 -(void)getSources:(FlutterResult)result;
 
+-(void)mediaStreamTrackHasTorch:(RTCMediaStreamTrack *)track
+                         result:(FlutterResult) result;
+
 -(void)mediaStreamTrackSetTorch:(RTCMediaStreamTrack *)track
                           torch:(BOOL) torch
                          result:(FlutterResult) result;
diff --git a/ios/Classes/FlutterRTCMediaStream.m b/ios/Classes/FlutterRTCMediaStream.m
@@ -474,6 +474,23 @@ -(void)mediaStreamTrackSetEnabled:(RTCMediaStreamTrack *)track : (BOOL)enabled
   }
 }
 
+-(void)mediaStreamTrackHasTorch:(RTCMediaStreamTrack *)track result:(FlutterResult) result
+{
+    if (!self.videoCapturer) {
+        result(@NO);
+        return;
+    }
+    if (self.videoCapturer.captureSession.inputs.count == 0) {
+        result(@NO);
+        return;
+    }
+
+    AVCaptureDeviceInput *deviceInput = [self.videoCapturer.captureSession.inputs objectAtIndex:0];
+    AVCaptureDevice *device = deviceInput.device;
+
+    result(@([device isTorchModeSupported:AVCaptureTorchModeOn]));
+}
+
 -(void)mediaStreamTrackSetTorch:(RTCMediaStreamTrack *)track torch:(BOOL)torch result:(FlutterResult)result
 {
     if (!self.videoCapturer) {
diff --git a/ios/Classes/FlutterWebRTCPlugin.m b/ios/Classes/FlutterWebRTCPlugin.m
@@ -353,6 +353,20 @@ - (void)handleMethodCall:(FlutterMethodCall*)call result:(FlutterResult) result
             [self setStreamId:streamId view:render];
         }
         result(nil);
+    }else if ([@"mediaStreamTrackHasTorch" isEqualToString:call.method]) {
+        NSDictionary* argsMap = call.arguments;
+        NSString* trackId = argsMap[@"trackId"];
+        RTCMediaStreamTrack *track = self.localTracks[trackId];
+        if (track != nil && [track isKindOfClass:[RTCVideoTrack class]]) {
+            RTCVideoTrack *videoTrack = (RTCVideoTrack *)track;
+            [self mediaStreamTrackHasTorch:videoTrack result:result];
+        } else {
+            if (track == nil) {
+                result([FlutterError errorWithCode:@"Track is nil" message:nil details:nil]);
+            } else {
+                result([FlutterError errorWithCode:[@"Track is class of " stringByAppendingString:[[track class] description]] message:nil details:nil]);
+            }
+        }
     }else if ([@"mediaStreamTrackSetTorch" isEqualToString:call.method]) {
         NSDictionary* argsMap = call.arguments;
         NSString* trackId = argsMap[@"trackId"];
diff --git a/lib/media_stream_track.dart b/lib/media_stream_track.dart
@@ -23,6 +23,12 @@ class MediaStreamTrack {
   String get kind => _kind;
   String get id => _trackId;
 
+  Future<bool> hasTorch() =>
+    _channel.invokeMethod(
+      'mediaStreamTrackHasTorch',
+      <String, dynamic>{'trackId': _trackId},
+    );
+
   Future<void> setTorch(bool torch) =>
     _channel.invokeMethod(
       'mediaStreamTrackSetTorch',
