Add DRM support (widevine)

Author: rbuehlma

pvr.teleboy/addon.xml.in

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <addon id="pvr.teleboy"
-       version="19.8.2"
+       version="19.8.3"
        name="Teleboy PVR Client"
        provider-name="rbuehlma">
   <requires>
@@ -32,6 +32,8 @@
       <fanart>fanart.jpg</fanart>
     </assets>
     <news>
+v19.8.3
+ - Support drm
 v19.8.2
  - Fix session handling
 v19.8.1

src/TeleBoy.cpp

@@ -504,39 +504,53 @@ void TeleBoy::TransferChannel(kodi::addon::PVRChannelsResultSet& results, TeleBo
   results.Add(kodiChannel);
 }
 
-void TeleBoy::SetStreamProperties(std::vector<kodi::addon::PVRStreamProperty>& properties, const std::string& url, bool realtime)
+PVR_ERROR TeleBoy::SetStreamProperties(std::vector<kodi::addon::PVRStreamProperty>& properties, const Value& stream, bool realtime)
 {
+  PVR_ERROR ret = PVR_ERROR_FAILED;
+
+  string url = GetStringOrEmpty(stream, "url");
+  kodi::Log(ADDON_LOG_INFO, "Play URL: %s.", url.c_str());
+  url = FollowRedirect(url);
+
+  if (url.empty())
+  {
+    return PVR_ERROR_FAILED;
+  }
+
   properties.emplace_back(PVR_STREAM_PROPERTY_STREAMURL, url);
   properties.emplace_back(PVR_STREAM_PROPERTY_INPUTSTREAM, "inputstream.adaptive");
   properties.emplace_back("inputstream.adaptive.manifest_type", "mpd");
   properties.emplace_back("inputstream.adaptive.manifest_update_parameter", "full");
   properties.emplace_back(PVR_STREAM_PROPERTY_MIMETYPE, "application/xml+dash");
   properties.emplace_back(PVR_STREAM_PROPERTY_ISREALTIMESTREAM, realtime ? "true" : "false");
+      
+  if (stream.HasMember("drm")) {
+    string drmType = GetStringOrEmpty(stream["drm"], "type");
+    if (drmType == "widevine") {
+      string licenseUrl = GetStringOrEmpty(stream["drm"], "license_url");
+      properties.emplace_back("inputstream.adaptive.license_key", licenseUrl + "||A{SSM}|");
+      properties.emplace_back("inputstream.adaptive.license_type", "com.widevine.alpha"); 
+    } else {
+      kodi::Log(ADDON_LOG_ERROR, "Unsupported drm type: %s.", drmType.c_str());
+    }      
+  }
+  return PVR_ERROR_NO_ERROR;
 }
 
 PVR_ERROR TeleBoy::GetChannelStreamProperties(const kodi::addon::PVRChannel& channel, std::vector<kodi::addon::PVRStreamProperty>& properties)
 {
-  PVR_ERROR ret = PVR_ERROR_FAILED;
-
   Document json;
   if (!ApiGet(
       "/users/" + userId + "/stream/live/" + to_string(channel.GetUniqueId())
           + "?expand=primary_image,flags&https=1" + GetStreamParameters(), json, 0))
   {
     kodi::Log(ADDON_LOG_ERROR, "Error getting live stream url for channel %i.",
         channel.GetUniqueId());
-    return ret;
+    return PVR_ERROR_FAILED;
   }
-  string url = GetStringOrEmpty(json["data"]["stream"], "url");
-  kodi::Log(ADDON_LOG_INFO, "Play URL: %s.", url.c_str());
-  url = FollowRedirect(url);
+  const Value& stream = json["data"]["stream"];
+  return SetStreamProperties(properties, stream, true);
 
-  if (!url.empty())
-  {
-    SetStreamProperties(properties, url, true);
-    ret = PVR_ERROR_NO_ERROR;
-  }
-  return ret;
 }
 
 string TeleBoy::FollowRedirect(string url)
@@ -739,15 +753,8 @@ PVR_ERROR TeleBoy::GetRecordingStreamProperties(const kodi::addon::PVRRecording&
         recording.GetRecordingId().c_str());
     return ret;
   }
-  string url = GetStringOrEmpty(json["data"]["stream"], "url");
-  url = FollowRedirect(url);
-
-  if (!url.empty())
-  {
-    SetStreamProperties(properties, url, false);
-    ret = PVR_ERROR_NO_ERROR;
-  }
-  return ret;
+  const Value& stream = json["data"]["stream"];
+  return SetStreamProperties(properties, stream, false);
 }
 
 PVR_ERROR TeleBoy::GetRecordingEdl(const kodi::addon::PVRRecording& recording, std::vector<kodi::addon::PVREDLEntry>& edl)
@@ -904,14 +911,8 @@ PVR_ERROR TeleBoy::GetEPGTagStreamProperties(const kodi::addon::PVREPGTag& tag,
     kodi::Log(ADDON_LOG_ERROR, "Could not get URL for epg tag.");
     return ret;
   }
-  string url = GetStringOrEmpty(json["data"]["stream"], "url");
-  url = FollowRedirect(url);
-  if (!url.empty())
-  {
-    SetStreamProperties(properties, url, false);
-    ret = PVR_ERROR_NO_ERROR;
-  }
-  return ret;
+  const Value& stream = json["data"]["stream"];
+  return SetStreamProperties(properties, stream, false);
 }
 
 PVR_ERROR TeleBoy::GetEPGTagEdl(const kodi::addon::PVREPGTag& tag, std::vector<kodi::addon::PVREDLEntry>& edl)

src/TeleBoy.h

@@ -105,7 +105,7 @@ class ATTRIBUTE_HIDDEN TeleBoy : public kodi::addon::CAddonBase,
   std::string GetStreamParameters();
   void LoadGenres();
   bool LoadChannels();
-  void SetStreamProperties(std::vector<kodi::addon::PVRStreamProperty>& properties,
-        const std::string& url, bool realtime);
+  PVR_ERROR SetStreamProperties(std::vector<kodi::addon::PVRStreamProperty>& properties,
+        const Value& stream, bool realtime);
   void AddTimerType(std::vector<kodi::addon::PVRTimerType>& types, int idx, int attributes);
 };