lghorizon 0.9.0.dev3__py3-none-any.whl → 0.9.1__py3-none-any.whl

lghorizon/lghorizon_device_state_processor.py

@@ -3,6 +3,9 @@
 import random
 import json
 import urllib.parse
+from datetime import datetime as dt, timezone
+
+import time
 
 from typing import cast, Dict, Optional
 
@@ -15,12 +18,10 @@ from .lghorizon_models import (
     LGHorizonReplaySource,
     LGHorizonNDVRSource,
     LGHorizonReviewBufferSource,
+    LGHorizonRecordingSource,
 )
 from .lghorizon_models import LGHorizonAuth
-from .lghorizon_models import (
-    LGHorizonReplayEvent,
-    LGHorizonVOD,
-)
+from .lghorizon_models import LGHorizonReplayEvent, LGHorizonVOD, LGHorizonVODType
 
 from .lghorizon_models import LGHorizonRecordingSingle
 from .lghorizon_models import LGHorizonChannel
@@ -99,6 +100,9 @@ class LGHorizonDeviceStateProcessor:
             return
         await device_state.reset()
         device_state.source_type = player_state.source_type
+        device_state.ui_state_type = LGHorizonUIStateType.MAINUI
+        device_state.speed = player_state.speed
+
         match player_state.source_type:
             case LGHorizonSourceType.LINEAR:
                 await self._process_linear_state(device_state, player_state)
@@ -116,9 +120,10 @@ class LGHorizonDeviceStateProcessor:
         device_state: LGHorizonDeviceState,
         apps_state: LGHorizonAppsState,
     ) -> None:
-        device_state.channel_id = apps_state.id
-        device_state.title = apps_state.app_name
+        device_state.id = apps_state.id
+        device_state.show_title = apps_state.app_name
         device_state.image = apps_state.logo_path
+        device_state.ui_state_type = LGHorizonUIStateType.APPS
 
     async def _process_linear_state(
         self,
@@ -140,12 +145,22 @@ class LGHorizonDeviceStateProcessor:
             service_path,
         )
         replay_event = LGHorizonReplayEvent(event_json)
+        device_state.id = replay_event.event_id
         channel = self._channels[replay_event.channel_id]
         device_state.source_type = source.source_type
-        device_state.channel_id = channel.channel_number
+        device_state.channel_id = channel.id
         device_state.channel_name = channel.title
-        device_state.title = replay_event.title
-        device_state.sub_title = replay_event.full_episode_title
+        device_state.episode_title = replay_event.episode_name
+        device_state.season_number = replay_event.season_number
+        device_state.episode_number = replay_event.episode_number
+        device_state.show_title = replay_event.title
+        now_in_ms = int(time.time() * 1000)
+
+        device_state.last_position_update = int(time.time() * 1000)
+        device_state.start_time = replay_event.start_time
+        device_state.end_time = replay_event.end_time
+        device_state.duration = replay_event.end_time - replay_event.start_time
+        device_state.position = now_in_ms - int(replay_event.start_time * 1000)
 
         # Add random number to url to force refresh
         join_param = "?"
@@ -155,7 +170,6 @@ class LGHorizonDeviceStateProcessor:
             f"{channel.stream_image}{join_param}{str(random.randrange(1000000))}"
         )
         device_state.image = image_url
-        await device_state.reset_progress()
 
     async def _process_reviewbuffer_state(
         self,
@@ -177,13 +191,20 @@ class LGHorizonDeviceStateProcessor:
             service_path,
         )
         replay_event = LGHorizonReplayEvent(event_json)
+        device_state.id = replay_event.event_id
         channel = self._channels[replay_event.channel_id]
         device_state.source_type = source.source_type
-        device_state.channel_id = channel.channel_number
+        device_state.channel_id = channel.id
         device_state.channel_name = channel.title
-        device_state.title = replay_event.title
-        device_state.sub_title = replay_event.full_episode_title
-
+        device_state.episode_title = replay_event.episode_name
+        device_state.season_number = replay_event.season_number
+        device_state.episode_number = replay_event.episode_number
+        device_state.show_title = replay_event.title
+        device_state.last_position_update = player_state.last_speed_change_time
+        device_state.position = player_state.relative_position
+        device_state.start_time = replay_event.start_time
+        device_state.end_time = replay_event.end_time
+        device_state.duration = replay_event.end_time - replay_event.start_time
         # Add random number to url to force refresh
         join_param = "?"
         if join_param in channel.stream_image:
@@ -192,7 +213,6 @@ class LGHorizonDeviceStateProcessor:
             f"{channel.stream_image}{join_param}{str(random.randrange(1000000))}"
         )
         device_state.image = image_url
-        await device_state.reset_progress()
 
     async def _process_replay_state(
         self,
@@ -214,15 +234,27 @@ class LGHorizonDeviceStateProcessor:
             service_path,
         )
         replay_event = LGHorizonReplayEvent(event_json)
+        device_state.id = replay_event.event_id
+        # Iets met buffer doen
+        channel = self._channels[replay_event.channel_id]
+        padding = channel.replay_pre_padding + channel.replay_post_padding
         device_state.source_type = source.source_type
-        device_state.channel_id = None
-        device_state.title = replay_event.title
-        if replay_event.full_episode_title:
-            device_state.sub_title = replay_event.full_episode_title
-
+        device_state.channel_id = channel.id
+        device_state.episode_title = replay_event.episode_name
+        device_state.season_number = replay_event.season_number
+        device_state.episode_number = replay_event.episode_number
+        device_state.show_title = replay_event.title
+        device_state.last_position_update = int(time.time() * 1000)
+        device_state.start_time = replay_event.start_time
+        device_state.end_time = replay_event.end_time
+        device_state.duration = (
+            replay_event.end_time - replay_event.start_time + padding
+        )
+        device_state.position = (
+            player_state.relative_position + channel.replay_pre_padding
+        )
         # Add random number to url to force refresh
         device_state.image = await self._get_intent_image_url(replay_event.event_id)
-        await device_state.reset_progress()
 
     async def _process_vod_state(
         self,
@@ -244,11 +276,20 @@ class LGHorizonDeviceStateProcessor:
             service_path,
         )
         vod = LGHorizonVOD(vod_json)
-        device_state.title = vod.title
-        device_state.sub_title = vod.full_episode_title
+        device_state.id = vod.id
+        if vod.vod_type == LGHorizonVODType.EPISODE:
+            device_state.show_title = vod.series_title
+            device_state.episode_title = vod.title
+            device_state.season_number = vod.season
+            device_state.episode_number = vod.episode
+        else:
+            device_state.show_title = vod.title
+
         device_state.duration = vod.duration
+        device_state.last_position_update = int(time.time() * 1000)
+        device_state.position = player_state.relative_position
+
         device_state.image = await self._get_intent_image_url(vod.id)
-        await device_state.reset_progress()
 
     async def _process_ndvr_state(
         self, device_state: LGHorizonDeviceState, player_state: LGHorizonPlayerState
@@ -267,13 +308,36 @@ class LGHorizonDeviceStateProcessor:
             service_path,
         )
         recording = LGHorizonRecordingSingle(recording_json)
-        device_state.title = recording.title
-        device_state.sub_title = recording.full_episode_title
+        device_state.id = recording.id
         device_state.channel_id = recording.channel_id
         if recording.channel_id:
             channel = self._channels[recording.channel_id]
             device_state.channel_name = channel.title
 
+        device_state.episode_title = recording.episode_title
+        device_state.season_number = recording.season_number
+        device_state.episode_number = recording.episode_number
+        device_state.last_position_update = player_state.last_speed_change_time
+        device_state.position = player_state.relative_position
+        if recording.start_time:
+            device_state.start_time = int(
+                dt.fromisoformat(
+                    recording.start_time.replace("Z", "+00:00")
+                ).timestamp()
+            )
+        if recording.end_time:
+            device_state.end_time = int(
+                dt.fromisoformat(recording.end_time.replace("Z", "+00:00")).timestamp()
+            )
+        if recording.start_time and recording.end_time:
+            device_state.duration = device_state.end_time - device_state.start_time
+        if recording.source == LGHorizonRecordingSource.SHOW:
+            device_state.show_title = recording.title
+        else:
+            device_state.show_title = recording.show_title
+
+        device_state.image = await self._get_intent_image_url(recording.id)
+
     async def _get_intent_image_url(self, intent_id: str) -> Optional[str]:
         """Get intent image url."""
         service_config = await self._auth.get_service_config()

lghorizon/lghorizon_message_factory.py

@@ -22,7 +22,6 @@ class LGHorizonMessageFactory:
             case LGHorizonMessageType.STATUS:
                 return LGHorizonStatusMessage(payload, topic)
             case LGHorizonMessageType.UI_STATUS:
-                # Placeholder for UI_STATUS message handling
                 return LGHorizonUIStatusMessage(payload, topic)
             case LGHorizonMessageType.UNKNOWN:
                 return LGHorizonUnknownMessage(payload, topic)