From f8a709fe985bf81f67e99accc426f870e7e59e03 Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <blacklight86@gmail.com>
Date: Mon, 9 Dec 2019 11:00:31 +0000
Subject: [PATCH] Improved Pushbullet plugin

---
 platypush/plugins/pushbullet.py | 94 +++++++++++++++++++++++++++++----
 1 file changed, 83 insertions(+), 11 deletions(-)

diff --git a/platypush/plugins/pushbullet.py b/platypush/plugins/pushbullet.py
index 70900c141..9763e76c9 100644
--- a/platypush/plugins/pushbullet.py
+++ b/platypush/plugins/pushbullet.py
@@ -18,45 +18,116 @@ class PushbulletPlugin(Plugin):
         * The :class:`platypush.backend.pushbullet.Pushbullet` backend enabled
     """
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, token: str = None, *args, **kwargs):
+        """
+        :param token: Pushbullet API token. If not set the plugin will try to retrieve it from
+            the Pushbullet backend configuration, if available
+        """
         super().__init__(*args, **kwargs)
 
+        if not token:
+            backend = get_backend('pushbullet')
+            if not backend or not backend.token:
+                raise AttributeError('No Pushbullet token specified')
+
+            self.token = backend.token
+        else:
+            self.token = token
+
+        self._devices = []
+        self._devices_by_id = {}
+        self._devices_by_name = {}
+
     @action
-    def send_note(self, body=None, title=None, **kwargs):
+    def get_devices(self):
+        """
+        Get the list of available devices
+        """
+        resp = requests.get('https://api.pushbullet.com/v2/devices',
+                headers={'Authorization': 'Bearer ' + self.token,
+                    'Content-Type': 'application/json'})
+
+        self._devices = resp.json().get('devices', [])
+        self._devices_by_id = {
+                dev['iden']: dev
+                for dev in self._devices
+        }
+
+        self._devices_by_name = {
+                dev['nickname']: dev
+                for dev in self._devices if 'nickname' in dev
+        }
+
+    @action
+    def get_device(self, device):
+        """
+        :param device: Device ID or name
+        """
+        output = None
+        refreshed = False
+
+        while not output:
+            if device in self._devices_by_id:
+                return self._devices_by_id[device]
+            if device in self._devices_by_name:
+                return self._devices_by_name[device]
+            if refreshed:
+                return None
+
+            self.get_devices()
+            refreshed = True
+
+    @action
+    def send_note(self, device: str = None, body: str = None, title: str = None, url: str = None, **kwargs):
         """
         Send a note push.
 
+        :param device: Device ID or name (default: None, all devices)
+        :param body: Note body
+        :param title: Note title
+        :param url: URL attached to the note
         :param kwargs: Push arguments, see https://docs.pushbullet.com/#create-push
-        :type kwargs: dict
         """
 
-        pushbullet = get_backend('pushbullet')
+        if device:
+            device = self.get_device(device).output
+            if not device:
+                raise RuntimeError('No such device')
+
         kwargs['body'] = body
         kwargs['title'] = title
-        kwargs['type'] = 'note'
+        kwargs['type'] = 'link' if url else 'note'
+
+        if device:
+            kwargs['device_iden'] = device['iden']
+
         resp = requests.post('https://api.pushbullet.com/v2/pushes',
                              data=json.dumps(kwargs),
-                             headers={'Authorization': 'Bearer ' + pushbullet.token,
+                             headers={'Authorization': 'Bearer ' + self.token,
                                       'Content-Type': 'application/json'})
 
         if resp.status_code >= 400:
             raise Exception('Pushbullet push failed with status {}: {}'.
                             format(resp.status_code, resp.json()))
 
-
     @action
-    def send_file(self, filename):
+    def send_file(self, filename: str, device: str = None):
         """
         Send a file.
 
+        :param device: Device ID or name (default: None, all devices)
         :param filename: Path to the local file
-        :type filename: str
         """
 
+        if device:
+            device = self.get_device(device).output
+            if not device:
+                raise RuntimeError('No such device')
+
         pushbullet = get_backend('pushbullet')
         resp = requests.post('https://api.pushbullet.com/v2/upload-request',
                              data=json.dumps({'file_name': os.path.basename(filename)}),
-                             headers={'Authorization': 'Bearer ' + pushbullet.token,
+                             headers={'Authorization': 'Bearer ' + self.token,
                                       'Content-Type': 'application/json'})
 
         if resp.status_code != 200:
@@ -72,11 +143,12 @@ class PushbulletPlugin(Plugin):
                             format(resp.status_code))
 
         resp = requests.post('https://api.pushbullet.com/v2/pushes',
-                             headers={'Authorization': 'Bearer ' + pushbullet.token,
+                             headers={'Authorization': 'Bearer ' + self.token,
                                       'Content-Type': 'application/json'},
 
                              data=json.dumps({
                                  'type': 'file',
+                                 'device_iden': device['iden'] if device else None,
                                  'file_name': r['file_name'],
                                  'file_type': r['file_type'],
                                  'file_url': r['file_url'] }))