presbeus 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5eccbc0625356b786735b43c6a71237c12f761dc
-  data.tar.gz: 957e3c0d6604fa0875ac5e6662fcbaa30fe55663
+  metadata.gz: 6d6e9f092c95ac7a42ab4a3d734fe240169b0995
+  data.tar.gz: 77c0037c697c764b6e4da23cc2eccfdb14e69dfb
 SHA512:
-  metadata.gz: 644f5372cab7a69065eb71e35b854efb0154c024ddcee54cbd5cd2db3adf4cad1d73c9fed53c3fb0da994c26dec0196d53705fb2c7ccf313c90f85a3d871abf1
-  data.tar.gz: ac7c6a6dafbd0d1d40847e0001b0bfaa4700b87c12f8fbb0d078910b7e13c9079e10d546a7076f27c73c685a3cfdecd0ca39110a79089dde1abcc221be42b5d6
+  metadata.gz: 006affe9abb7bd67066afa54acde980344f216a2f6bf87f52c695a9ba1bfbef80c86376120472701592ec51a7a1fb1b4417951a0c62532e3bdbbe30b07575089
+  data.tar.gz: 829fd5d5c602b3fb7a739a8bfdfa60ea30ba4d01a6e3cd9739b54f8a08315e5a82d93b6b11137b8d084df4b2b7a937572fd712f395fa01e5e94d41fdcb0c0d03

data/bin/presbeus

@@ -6,7 +6,10 @@ require 'time_ago_in_words'
 require 'terminal-table'
 require 'highline'
 require 'colorize'
-require "readline"
+require 'readline'
+#require 'websocket-eventmachine-client'
+require 'kontena-websocket-client'
+require 'open-uri'
 
 class Time
 
@@ -18,6 +21,7 @@ class Presbeus
 
   def initialize_arguments
     argument(:help, [], "show this help")  { help }
+    argument(:realtime, [], "handle realtime event stream")  { realtime }
     argument(:shell, [], "run in an interactive shell") { shell }
     argument(:devices, [], "list devices") { puts table devices }
     argument(:pushes, [], "list pushes") { puts table pushes_with_color }
@@ -42,18 +46,52 @@ class Presbeus
     initialize_arguments
     configuration_path = "#{ENV['HOME']}/.config/presbeus.yml"
     configuration = YAML.load_file configuration_path
-    token = `#{configuration['password_command']}`
-    @headers = {"Access-Token" => token, "Content-Type" => "json"}
+    @token = `#{configuration['password_command']}`.chomp
+    @notify_command = configuration['notify_command']
+    @headers = {"Access-Token" => @token, "Content-Type" => "json"}
+    @api_prefix = 'https://api.pushbullet.com/v2/'
+  end
+
+  def realtime_no_reconnect
+    uri = "wss://stream.pushbullet.com/websocket/#{@token}"
+    Kontena::Websocket::Client.connect(uri, ping_timeout: 60) do |client|
+      client.on_pong do |time, delay|
+        p time, delay
+      end
+      client.read do |message|
+        json = JSON.parse(message)
+        type = json["type"]
+        if type != "nop"
+          if type == "tickle"
+            puts(table(pushes_with_color(Time.now.to_i) do |push|
+              `#{@notify_command} "#{push[1]}"` if @notify_command
+            end))
+          end
+        else
+          client.send message # pong
+        end
+      end
+    end
+  end
+
+  def realtime
+    while true
+      begin
+        realtime_no_reconnect
+      rescue Kontena::Websocket::TimeoutError => e
+        puts "timeout #{e}"
+      end
+    end
   end
 
   def post_v2 what, payload
     RestClient.post(
-      'https://api.pushbullet.com/v2/' + what, payload.to_json, @headers)
+      @api_prefix + what, payload.to_json, @headers)
   end
 
   def get_v2 what
     response = RestClient.get(
-      'https://api.pushbullet.com/v2/' + what, @headers)
+      @api_prefix + what, @headers)
     JSON.parse response.body
   end
 
@@ -74,8 +112,10 @@ class Presbeus
     get_v2("permanents/#{device}_thread_#{id}")["thread"]
   end
 
-  def pushes
-    get_v2("pushes")["pushes"].map do |push|
+  def pushes modified_after = nil
+    path = "pushes"
+    path += "?modified_after=#{modified_after}" if modified_after
+    get_v2(path)["pushes"].map do |push|
       [Time.at(push["created"].to_i).ago_in_words, push["body"]]
     end
   end
@@ -151,8 +191,9 @@ class Presbeus
     res 
   end
 
-  def pushes_with_color
-    pushes.reverse.map do |push|
+  def pushes_with_color modified_after = nil
+    pushes(modified_after).reverse.map do |push|
+      yield(push) if block_given?
       ["\n" + push[0] + "\n" + with_color(wrap(push[1], width_2), width_2, :green)]
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: presbeus
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Olivier Abdesselam
@@ -80,6 +80,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1'
+- !ruby/object:Gem::Dependency
+  name: kontena-websocket-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
 description: Allows to view/send pushbullet SMS
 email: 
 executables: