Merge pull request #467 from GoSecure/feature-player-show-time

Player GUI: Added a label that shows time / total_time
GoSecure · Jan 19, 2024 · 3104808 · 3104808
2 parents a664c43 + 8ed6294
commit 3104808
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/pyrdp/player/ReplayBar.py b/pyrdp/player/ReplayBar.py
@@ -28,6 +28,18 @@ def __init__(self, duration: float, parent: QWidget = None):
 
         self.log = logging.getLogger(LOGGER_NAMES.PLAYER)
 
+        # pretty display capture duration
+        self.duration = duration
+        self.duration_hours = int(self.duration // 3600)
+        remaining_seconds = int(self.duration % 3600)
+        minutes = int(remaining_seconds // 60)
+        seconds = int(remaining_seconds % 60)
+        if self.duration_hours > 0:
+            self.duration_pretty = "{:d}:{:02d}:{:02d}".format(
+                                    self.duration_hours, minutes, seconds)
+        else:
+            self.duration_pretty = "{:02d}:{:02d}".format(minutes, seconds)
+
         self.button = PlayPauseButton()
         self.button.setMaximumWidth(100)
         self.button.clicked.connect(self.onButtonClicked)
@@ -36,9 +48,11 @@ def __init__(self, duration: float, parent: QWidget = None):
 
         self.timeSlider = SeekBar()
         self.timeSlider.setMinimum(0)
-        self.timeSlider.setMaximum(int(duration * 1000))
+        self.timeSlider.setMaximum(int(self.duration * 1000))
         self.timeSlider.valueChanged.connect(self.onSeek)
 
+        self.timeLabel = QLabel(self.formatTimeLabel(0))
+
         self.speedLabel = QLabel("Speed: 1x")
 
         self.speedSlider = QSlider(Qt.Horizontal)
@@ -59,6 +73,7 @@ def __init__(self, duration: float, parent: QWidget = None):
         horizontal = QHBoxLayout()
         horizontal.addWidget(self.button)
         horizontal.addWidget(self.timeSlider)
+        horizontal.addWidget(self.timeLabel)
         vertical.addLayout(horizontal)
 
         self.setLayout(vertical)
@@ -82,3 +97,20 @@ def onSpeedChanged(self):
         self.log.debug("Slider changed value: %(arg1)d", {"arg1": speed})
         self.speedLabel.setText("Speed: {}x".format(speed))
         self.speedChanged.emit(speed)
+
+    def onTimeChanged(self, currentTime: float):
+        if currentTime >= self.duration:
+            currentTime = self.duration
+
+        self.timeLabel.setText(self.formatTimeLabel(currentTime))
+
+    def formatTimeLabel(self, current: float) -> str:
+        hours = int(current // 3600)
+        remaining_seconds = int(current % 3600)
+        minutes = int(remaining_seconds // 60)
+        seconds = int(remaining_seconds % 60)
+        if self.duration_hours > 0:
+            return "{:d}:{:02d}:{:02d} / {}".format(
+                hours, minutes, seconds, self.duration_pretty)
+        else:
+            return "{:02d}:{:02d} / {}".format(minutes, seconds, self.duration_pretty)
diff --git a/pyrdp/player/ReplayTab.py b/pyrdp/player/ReplayTab.py
@@ -70,6 +70,7 @@ def onTimeUpdated(self, currentTime: float):
         self.controlBar.timeSlider.blockSignals(True)
         self.controlBar.timeSlider.setValue(int(currentTime * 1000))
         self.controlBar.timeSlider.blockSignals(False)
+        self.controlBar.onTimeChanged(currentTime)
 
     def clear(self):
         """