Implement automatic Widevine download methods

Author: emilsvennesson

addon.xml

@@ -5,6 +5,7 @@
        provider-name="emilsvennesson">
   <requires>
     <import addon="xbmc.python" version="2.25.0"/>
+    <import addon="script.module.requests" version="2.9.1"/>
   </requires>
   <extension library="lib" point="xbmc.python.module"/>
   <extension point="xbmc.addon.metadata">

lib/config.py

@@ -32,7 +32,7 @@
     }
 }
 
-WIDEVINE_CDM_EXTENSIONS = ['.so', '.dll', '.dylib']
+WIDEVINE_CDM_EXTENSIONS = ('.so', '.dll', '.dylib')
 
 WIDEVINE_SUPPORTED_ARCHS = ['x86_64', 'x86', 'aarch64', 'aarch64_be', 'armv7', 'armv8', 'arm64']
 

lib/inputstreamhelper.py

@@ -4,6 +4,8 @@
 import json
 from distutils.version import LooseVersion
 
+import requests
+
 import config
 import xbmc
 import xbmcaddon
@@ -63,7 +65,7 @@ def _has_widevine_cdm(self):
                     self._log('Found Widevine binary at {0}'.format(os.path.join(self._cdm_path(), filename)))
                     return True
 
-            self.log('Widevine is not installed.')
+            self._log('Widevine is not installed.')
             return False
 
     def _json_rpc_request(self, payload):
@@ -73,6 +75,46 @@ def _json_rpc_request(self, payload):
 
         return json.loads(response)
 
+    def _http_request(self, url, download=False, download_path=None):
+        busy_dialog = xbmcgui.DialogBusy()
+        dialog = xbmcgui.Dialog()
+        filename = url.split('/')[-1]
+        self._log('Request URL: {0}'.format(url))
+        try:
+            busy_dialog.create()
+            req = requests.get(url, stream=download, verify=False)
+            self._log('Response code: {0}'.format(req.status_code))
+            if not download:
+                self._log('Response: {0}'.format(req.content))
+            req.raise_for_status()
+        except requests.exceptions.HTTPError:
+            busy_dialog.close()
+            dialog.ok(self._language(30004), self._language(30013).format(filename))
+            return False
+
+        busy_dialog.close()
+        if download:
+            total_length = float(req.headers.get('content-length'))
+            progress_dialog = xbmcgui.DialogProgress()
+            progress_dialog.create(self._language(30014), self._language(30015).format(filename))
+
+            with open(download_path, 'wb') as f:
+                dl = 0
+                for chunk in req.iter_content(chunk_size=1024):
+                    f.write(chunk)
+                    dl += len(chunk)
+                    percent = int(dl * 100/ total_length)
+                    if progress_dialog.iscanceled():
+                        progress_dialog.close()
+                        req.close()
+                        return False
+                    progress_dialog.update(percent)
+                progress_dialog.close()
+                return True
+        else:
+            return req.content
+
+
     def _has_inputstream(self):
         """Checks if selected InputStream add-on is installed."""
         payload = {
@@ -145,6 +187,50 @@ def _supports_widevine(self):
 
         return True
 
+    def _install_widevine_cdm(self):
+        dialog = xbmcgui.Dialog()
+        download_path = os.path.join(xbmc.translatePath('special://temp'), 'widevine_cdm.zip')
+        cdm_platform = config.WIDEVINE_DOWNLOAD_MAP[self._arch][self._os]
+        cdm_source = json.loads(self._http_request(config.WIDEVINE_CDM_SOURCE))['vendors']['gmp-widevinecdm']['platforms']
+        cdm_zip_url = cdm_source[cdm_platform]['fileUrl']
+
+        downloaded = self._http_request(cdm_zip_url, download=True, download_path=download_path)
+        if downloaded:
+            if self._unzip_widevine_cdm(download_path):
+                dialog.ok(self._language(30001), self._language(30003))
+            else:
+                return False
+        else:
+            return False
+
+    def _unzip_widevine_cdm(self, zip_path):
+        busy_dialog = xbmcgui.DialogBusy()
+        zip_obj = zipfile.ZipFile(zip_path)
+        busy_dialog.create()
+        for filename in zip_obj.namelist():
+            if filename.endswith(config.WIDEVINE_CDM_EXTENSIONS):
+                zip_obj.extract(filename, self._cdm_path())
+                busy_dialog.close()
+                return True
+
+        busy_dialog.close()
+        dialog = xbmcgui.Dialog()
+        dialog.ok(self._language(30004), self._language(30016))
+        return False
+
+    def check_for_widevine(self):
+        if not self._supports_widevine():
+            return False
+        if not self._has_widevine_cdm():
+            dialog = xbmcgui.Dialog()
+            ok = dialog.yesno(self._language(30001), self._language(30002))
+            if ok:
+                return self._install_widevine_cdm()
+            else:
+                return False
+        else:
+            return True
+
     def check_for_inputstream(self):
         """Ensures that selected InputStream add-on is installed and enabled.
         Displays a select dialog if add-on is installed but not enabled."""

resources/language/resource.language.en_gb/strings.po

@@ -22,15 +22,15 @@ msgid "You do not have Widevine DRM installed. This is required for playback of
 msgstr ""
 
 msgctxt "#30003"
-msgid "Widevine DRM was successfully installed!"
+msgid "[B]Widevine DRM[/B] was successfully installed!"
 msgstr ""
 
 msgctxt "#30004"
 msgid "Error"
 msgstr ""
 
 msgctxt "#30005"
-msgid "An error occurred while installing Widevine DRM. Please enable debug logging and file a bug report on [B]github.com/emilsvennesson/script.module.inputstreamhelper[/B]."
+msgid "An error occurred while installing Widevine DRM. Please enable debug logging and file a bug report at:[CR][CR]github.com/emilsvennesson/script.module.inputstreamhelper"
 msgstr ""
 
 msgctxt "#30006"
@@ -60,3 +60,19 @@ msgstr ""
 msgctxt "#30012"
 msgid "The Windows Store version of Kodi does not support Widevine DRM. Please use the installer from kodi.tv instead."
 msgstr ""
+
+msgctxt "#30013"
+msgid "Failed to retrieve [B]{0}[/B]."
+msgstr ""
+
+msgctxt "#30014"
+msgid "Download in progress..."
+msgstr ""
+
+msgctxt "#30015"
+msgid "Downloading [B]{0}[/B]..."
+msgstr ""
+
+msgctxt "#30016"
+msgid "Failed to extract Widevine CDM from zip file."
+msgstr ""