io.appium.settings: recording: Add timer limitor

Signed-off-by: sirmordred <[email protected]>

Author: sirmordred

app/src/main/java/io/appium/settings/Settings.java

@@ -48,12 +48,14 @@
 
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_BASE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_FILENAME;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_MAX_DURATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_RESULT_CODE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_ROTATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_START;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_STOP;
 import static io.appium.settings.recorder.RecorderConstant.NO_ROTATION_SET;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_MAX_DURATION_DEFAULT_MS;
 import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MAX;
 import static io.appium.settings.recorder.RecorderConstant.REQUEST_CODE_SCREEN_CAPTURE;
 
@@ -63,6 +65,7 @@ public class Settings extends Activity {
     private String recordingOutputPath = "";
     private int recordingRotation = NO_ROTATION_SET;
     private int recordingPriority = RECORDING_PRIORITY_MAX;
+    private int recordingMaxDuration = RECORDING_MAX_DURATION_DEFAULT_MS;
 
     @Override
     public void onCreate(Bundle savedInstanceState) {
@@ -159,6 +162,8 @@ so we need to call getExternalFilesDir() method twice
 
             recordingPriority = RecorderUtil.getRecordingPriority(intent);
 
+            recordingMaxDuration = RecorderUtil.getRecordingMaxDuration(intent);
+
             // start record
             final MediaProjectionManager manager
                     = (MediaProjectionManager) getSystemService(
@@ -219,6 +224,7 @@ protected void onActivityResult(final int requestCode, final int resultCode, fin
         intent.putExtra(ACTION_RECORDING_FILENAME, recordingOutputPath);
         intent.putExtra(ACTION_RECORDING_ROTATION, recordingRotation);
         intent.putExtra(ACTION_RECORDING_PRIORITY, recordingPriority);
+        intent.putExtra(ACTION_RECORDING_MAX_DURATION, recordingMaxDuration);
         intent.putExtras(data);
 
         startService(intent);

app/src/main/java/io/appium/settings/recorder/RecorderConstant.java

@@ -27,6 +27,7 @@ public class RecorderConstant {
     public static final String ACTION_RECORDING_FILENAME = "recording_filename";
     public static final String ACTION_RECORDING_ROTATION = "recording_rotation";
     public static final String ACTION_RECORDING_PRIORITY = "recording_priority";
+    public static final String ACTION_RECORDING_MAX_DURATION = "recording_max_duration";
     public static final float BITRATE_MULTIPLIER = 0.25f;
     public static final int AUDIO_CODEC_SAMPLE_RATE_HZ = 44100;
     public static final int AUDIO_CODEC_CHANNEL_COUNT = 1;
@@ -43,4 +44,5 @@ public class RecorderConstant {
     public static final int RECORDING_PRIORITY_MIN = 1;
     public static final int RECORDING_PRIORITY_NORM = 2;
     public static final int RECORDING_PRIORITY_DEFAULT = Thread.MAX_PRIORITY;
+    public static final int RECORDING_MAX_DURATION_DEFAULT_MS = 15 * 60 * 1000; // 15 Minute, in miliseconds
 }

app/src/main/java/io/appium/settings/recorder/RecorderService.java

@@ -31,11 +31,13 @@
 import io.appium.settings.helpers.NotificationHelpers;
 
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_FILENAME;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_MAX_DURATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_RESULT_CODE;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_ROTATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_START;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_STOP;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_MAX_DURATION_DEFAULT_MS;
 import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_DEFAULT;
 
 public class RecorderService extends Service {
@@ -162,8 +164,12 @@ private void startScreenRecord(MediaProjectionManager mediaProjectionManager,
         int recordingPriority = intent.getIntExtra(ACTION_RECORDING_PRIORITY,
                 RECORDING_PRIORITY_DEFAULT);
 
+        int recordingMaxDuration = intent.getIntExtra(ACTION_RECORDING_MAX_DURATION,
+                RECORDING_MAX_DURATION_DEFAULT_MS);
+
         recorderThread = new RecorderThread(projection, outputFilePath,
-                rawWidth, rawHeight, recordingRotation, recordingPriority);
+                rawWidth, rawHeight, recordingRotation, recordingPriority,
+                recordingMaxDuration);
         recorderThread.startRecording();
     }
 

app/src/main/java/io/appium/settings/recorder/RecorderThread.java

@@ -53,7 +53,8 @@ public class RecorderThread implements Runnable {
     private final int videoWidth;
     private final int videoHeight;
     private final int recordingRotation;
-    private int recordingPriority = RECORDING_PRIORITY_DEFAULT;
+    private final int recordingPriority;
+    private final int recordingMaxDuration;
 
     private boolean muxerStarted = false;
     private boolean isStartTimestampInitialized = false;
@@ -86,13 +87,14 @@ public void onStopped() {
 
     public RecorderThread(MediaProjection mediaProjection, String outputFilePath,
                           int videoWidth, int videoHeight, int recordingRotation,
-                          int recordingPriority) {
+                          int recordingPriority, int recordingMaxDuration) {
         this.mediaProjection = mediaProjection;
         this.outputFilePath = outputFilePath;
         this.videoWidth = videoWidth;
         this.videoHeight = videoHeight;
         this.recordingRotation = recordingRotation;
         this.recordingPriority = recordingPriority;
+        this.recordingMaxDuration = recordingMaxDuration;
     }
 
     public void startRecording() {
@@ -399,6 +401,8 @@ public void run() {
             MediaCodec.BufferInfo bufferInfo = new MediaCodec.BufferInfo();
             lastAudioTimestampUs = NO_TIMESTAMP_SET;
 
+            long recordingStartTime = System.currentTimeMillis();
+
             while (!stopped && !hasAsyncError) {
                 if (!writeAudioBufferToFile(audioEncoder, muxer, bufferInfo)) {
                     break;
@@ -411,6 +415,11 @@ public void run() {
                 if (!writeVideoBufferToFile(videoEncoder, muxer, bufferInfo)) {
                     break;
                 }
+
+                if ((System.currentTimeMillis() - recordingStartTime) < this.recordingMaxDuration) {
+                    Log.v(TAG, "Recording stopped, reached maximum duration");
+                    stopped = true;
+                }
             }
         } catch (Exception mainException) {
             mainException.printStackTrace();

app/src/main/java/io/appium/settings/recorder/RecorderUtil.java

@@ -29,8 +29,10 @@
 import androidx.core.app.ActivityCompat;
 
 import static android.content.Context.WINDOW_SERVICE;
+import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_MAX_DURATION;
 import static io.appium.settings.recorder.RecorderConstant.ACTION_RECORDING_PRIORITY;
 import static io.appium.settings.recorder.RecorderConstant.NO_ROTATION_SET;
+import static io.appium.settings.recorder.RecorderConstant.RECORDING_MAX_DURATION_DEFAULT_MS;
 import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_DEFAULT;
 import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MAX;
 import static io.appium.settings.recorder.RecorderConstant.RECORDING_PRIORITY_MIN;
@@ -141,4 +143,24 @@ public static int getRecordingPriority(Intent intent) {
         }
         return RECORDING_PRIORITY_DEFAULT;
     }
+
+    public static int getRecordingMaxDuration(Intent intent) {
+        if (intent.hasExtra(ACTION_RECORDING_MAX_DURATION)) {
+            try {
+                int userRequestedMaxDurationInSecond =
+                        Integer.parseInt(intent.getStringExtra(ACTION_RECORDING_MAX_DURATION));
+                if (userRequestedMaxDurationInSecond <= 0) {
+                    Log.e(TAG, "Maximum recording duration must be greater than 0 second");
+                    return RECORDING_MAX_DURATION_DEFAULT_MS;
+                }
+                // Convert it to millisecond and return
+                return userRequestedMaxDurationInSecond * 1000;
+            } catch (NumberFormatException e) {
+                Log.e(TAG, "Exception while retrieving recording max duration", e);
+            }
+        } else {
+            Log.e(TAG, "Unable to retrieve recording max duration");
+        }
+        return RECORDING_MAX_DURATION_DEFAULT_MS;
+    }
 }