tp_link_smartplug 0.0.3 → 0.1.0.alpha1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f53d7ea800a82c44bc660a56066f5e7ee81d2ff2f786ddcb668c8d3267b8960c
-  data.tar.gz: af92b7cd428a8a3e1ceafb1868b3651153b5b192d051f0266174b14d4bbfba67
+  metadata.gz: 23a1848523acc948dd653d39fa3b9cec937bdba9d2a0ad6970308bac4af64d8d
+  data.tar.gz: dd4d4a0fda92aeabfdc7adfb5f751f1b957438b168e5c96efaf4950d69e1bb3d
 SHA512:
-  metadata.gz: f406f2bc6b7a7ebb0d0e8434146593a6080f4a2840d4420f96b3fd9454f31645adbfb8e248c5dab75e35c54ed9e72129b0e2355b508949271f1aa75371eb7269
-  data.tar.gz: 1b21d836bac38def6558452bd6fa6bbb9a4c0fe5023f70abb07c91bdd6921d5f4bb6db7db44d52ecf47d07d7fd417c0a5d8b71e10ccf5832becf864abb90844d
+  metadata.gz: e2259524ac344b200b19d4824178dfe18b5b7d92c8fbead1b3a4c40f3c073a3078b529f73bf035c4e1505f9827c9323f2f5c8e98f8076270783ed701d1ce6058
+  data.tar.gz: 24ae9be641e069804c884abd18594ee8a2c355370033c06c36091dff990c0846325cc9878c1e08550658e88fabebe42e98ce60b0edf82cc57f71006b0b5d1753

data/lib/tp_link_smartplug/command.rb

@@ -1,18 +1,31 @@
 # frozen_string_literal: true
 
 module TpLinkSmartplug
+  # Module containing the static commands for the plug
   module Command
+    # Plug information command
     INFO = '{"system":{"get_sysinfo":{}}}'
+    # Plug output on command
     ON = '{"system":{"set_relay_state":{"state":1}}}'
+    # Plug output off command
     OFF = '{"system":{"set_relay_state":{"state":0}}}'
+    # Plug cloud info command
     CLOUDINFO = '{"cnCloud":{"get_info":{}}}'
+    # Plug WLAN SSID scan command
     WLANSCAN = '{"netif":{"get_scaninfo":{"refresh":0}}}'
+    # Plug time command
     TIME = '{"time":{"get_time":{}}}'
+    # Plug schedule command
     SCHEDULE = '{"schedule":{"get_rules":{}}}'
+    # Plug countdown command
     COUNTDOWN = '{"count_down":{"get_rules":{}}}'
+    # Plug antitheft command
     ANTITHEFT = '{"anti_theft":{"get_rules":{}}}'
+    # Plug reboot command
     REBOOT = '{"system":{"reboot":{"delay":1}}}'
+    # Plug reset command
     RESET = '{"system":{"reset":{"delay":1}}}'
+    # Plug energy command
     ENERGY = '{"emeter":{"get_realtime":{}}}'
   end
 end

data/lib/tp_link_smartplug/device.rb

@@ -3,16 +3,24 @@
 require 'socket'
 require 'ipaddr'
 require 'json'
+require 'tp_link_smartplug/helpers'
 require 'tp_link_smartplug/message'
 
 module TpLinkSmartplug
-  # Provides an interface to a plug
+  # Provides an object to represent to a plug
+  #
+  # @author Ben Hughes
+  # @attr [IPAddr] address IP address of the plug
+  # @attr [Integer] port Port to connect to on the plug
+  # @attr [Integer] timeout Timeout value for connecting and sending commands to the plug
+  # @attr [true, false] debug Control debug logging
   class Device
+    include TpLinkSmartplug::Helpers
     include TpLinkSmartplug::Message
 
     attr_accessor :address
-    attr_accessor :timeout
     attr_accessor :port
+    attr_accessor :timeout
     attr_accessor :debug
 
     def initialize(address:, port: 9999)
@@ -20,8 +28,130 @@ module TpLinkSmartplug
       @port = port
       @timeout = 3
       @debug = false
+      @sockaddr = Addrinfo.getaddrinfo(address.to_s, port, Socket::PF_INET, :STREAM, 6).first.to_sockaddr
+      @socket = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM)
+    end
+
+    # Open connection to plug
+    #
+    def connect
+      debug_message("Connecting to #{@address} port #{@port}") if @debug
+      debug_message("Connecting, socket state: #{@socket.closed? ? 'closed' : 'open'}") if @debug
+
+      @socket = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM) if closed?
+
+      begin
+        @socket.connect_nonblock(@sockaddr)
+      rescue IO::WaitWritable
+        if IO.select(nil, [@socket], nil, timeout)
+          begin
+            @socket.connect_nonblock(@sockaddr)
+          rescue Errno::EISCONN
+            debug_message('Connected') if debug
+          rescue StandardError => e
+            disconnect
+            debug_message('Unexpected exception encountered.') if debug
+            raise e
+          end
+        else
+          @socket.close
+          raise "Connection timeout connecting to address #{@address}, port #{@port}."
+        end
+      rescue Errno::EISCONN
+        debug_message('Connected') if debug
+      end
+    end
+
+    # Close connection to plug
+    #
+    def disconnect
+      @socket.close unless @socket.closed?
+    end
+
+    # Return connection state
+    #
+    # @return [True, False]
+    def open?
+      !@socket.closed?
+    end
+
+    # Return connection state
+    #
+    # @return [True, False]
+    def closed?
+      @socket.closed?
+    end
+
+    # Polls a plug with a command
+    #
+    # @param command [String] the command to send to the plug
+    # @return [Hash] the output from the plug command
+    def poll(command:)
+      connect
+
+      begin
+        debug_message("Sending: #{decrypt(encrypt(command)[4..(command.length + 4)])}") if @debug
+        @socket.write_nonblock(encrypt(command))
+      rescue IO::WaitWritable
+        IO.select(nil, [@socket])
+        retry
+      end
+
+      begin
+        data = @socket.recv_nonblock(2048)
+      rescue IO::WaitReadable
+        IO.select([@socket])
+        retry
+      ensure
+        disconnect unless closed?
+      end
+
+      raise 'Error occured during disconnect' unless closed?
+      raise 'No data received' if data.nil? || data.empty?
+
+      debug_message("Received Raw: #{data}") if debug
+      data = decrypt(data[4..data.length])
+      debug_message("Received Decrypted: #{data}") if debug
+
+      data
     end
 
+    # @!method info
+    #   Return plug information
+    #   @return [Hash]
+    # @!method on
+    #   Turn plug output on
+    #   @return [Hash]
+    # @!method off
+    #   Turn plug output off
+    #   @return [Hash]
+    # @!method cloudinfo
+    #   Return plug cloud account configuration
+    #   @return [Hash]
+    # @!method wlanscan
+    #   Perform a scan for wireless SSIDs
+    #   @return [Hash]
+    # @!method time
+    #   Return system time from the plug
+    #   @return [Hash]
+    # @!method schedule
+    #   Return schedule configured on the plug
+    #   @return [Hash]
+    # @!method countdown
+    #   Return countdown configured on the plug
+    #   @return [Hash]
+    # @!method antitheft
+    #   Unsure
+    #   @return [Hash]
+    # @!method reboot
+    #   Reboot plug
+    #   @return [Hash]
+    # @!method resry
+    #   Reset plug
+    #   @return [Hash]
+    # @!method energy
+    #   Return plug energy data
+    #   @return [Hash]
     [
       :info,
       :on,
@@ -37,7 +167,7 @@ module TpLinkSmartplug
       :energy
     ].each do |method|
       define_method method do
-        JSON.parse(poll(address: @address, port: @port, command: TpLinkSmartplug::Command.const_get(method.upcase), timeout: @timeout, debug: @debug))
+        JSON.parse(poll(command: TpLinkSmartplug::Command.const_get(method.upcase)))
       end
     end
   end

data/lib/tp_link_smartplug/helpers.rb

@@ -5,9 +5,12 @@ require 'time'
 module TpLinkSmartplug
   # Generic helper methods
   module Helpers
+    # Formats a message for output as a debug log
+    #
+    # @param string [String] the message to be formatted for debug output
     def debug_message(string)
-      caller_method = caller_locations(1..1).label
-      Time.now.strftime('%Y-%m-%d %H:%M:%S: ').concat("#{caller_method}: ").concat(string)
+      caller_method = caller_locations(1..1).first.label
+      STDOUT.puts(Time.now.strftime('%Y-%m-%d %H:%M:%S: ').concat("#{caller_method}: ").concat(string))
     end
   end
 end

data/lib/tp_link_smartplug/message.rb

@@ -1,68 +1,33 @@
-# frozen_string_literal: true
-
-require 'socket'
-require 'tp_link_smartplug/helpers'
-require 'tp_link_smartplug/message_helpers'
+# frozen_string_literal: false
 
 module TpLinkSmartplug
-  # Provides methods to send and receive messages from plugs
+  # Helper methods for plug communication messages
   module Message
-    include TpLinkSmartplug::Helpers
-    include TpLinkSmartplug::MessageHelpers
-
-    def poll(address:, port:, command:, timeout: 3, debug: false)
-      socket = connect(address: address, port: port, timeout: timeout, debug: debug)
-      STDOUT.puts(debug_message("Sending: #{decrypt(encrypt(command)[4..command.length])}")) if debug
-
-      begin
-        socket.write_nonblock(encrypt(command))
-        data = socket.recv_nonblock(2048)
-      rescue IO::WaitReadable, IO::WaitWritable
-        IO.select([socket], [socket])
-        retry
-      ensure
-        disconnect(socket: socket)
+    # Encrypts a message to send to the smart plug
+    #
+    # @param string [String] the message to be encrypted
+    def encrypt(string)
+      key = 171
+      result = [string.length].pack('N')
+      string.each_char do |char|
+        key = a = key ^ char.ord
+        result.concat(a.chr)
       end
-
-      raise 'Error occured during disconnect' unless socket.closed?
-      raise 'No data received' if data.nil? || data.empty?
-
-      STDOUT.puts(debug_message("Received Raw: #{data}")) if debug
-      data = decrypt(data[4..data.length])
-      STDOUT.puts(debug_message("Received: #{data}")) if debug
-
-      data
+      result
     end
 
-    private
-
-    def connect(address:, port:, timeout: 3, debug: false)
-      STDOUT.puts(debug_message("Connecting to #{address} port #{port}")) if debug
-
-      Socket.new(Socket::AF_INET, Socket::SOCK_STREAM).tap do |socket|
-        sockaddr = Addrinfo.getaddrinfo(address.to_s, port, Socket::PF_INET, :STREAM, 6).first.to_sockaddr
-        STDOUT.puts(debug_message("Connecting, socket closed: #{socket.closed?}")) if debug
-        socket.connect_nonblock(sockaddr)
-      rescue IO::WaitWritable
-        if IO.select(nil, [socket], nil, timeout)
-          begin
-            socket.connect_nonblock(sockaddr)
-          rescue Errno::EISCONN
-            STDOUT.puts(debug_message('Connected')) if debug
-          rescue StandardError => e
-            socket.close
-            STDOUT.puts(debug_message('Unexpected exception encountered.')) if debug
-            raise e
-          end
-        else
-          socket.close
-          raise "Connection timeout connecting to address #{address}, port #{port}."
-        end
+    # Decrypts a message received from the smart plug
+    #
+    # @param string [String] the message to be decrypted
+    def decrypt(string)
+      key = 171
+      result = ''
+      string.each_char do |char|
+        a = key ^ char.ord
+        key = char.ord
+        result.concat(a.chr)
       end
-    end
-
-    def disconnect(socket:)
-      socket.close unless socket.closed? || socket.nil?
+      result
     end
   end
 end

data/lib/tp_link_smartplug/version.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+# @author Ben Hughes
 module TpLinkSmartplug
+  # Gem version
   VERSION = '0.0.1'
 end

data/lib/tp_link_smartplug.rb

@@ -3,6 +3,5 @@
 require 'tp_link_smartplug/command'
 require 'tp_link_smartplug/device'
 require 'tp_link_smartplug/helpers'
-require 'tp_link_smartplug/message_helpers'
 require 'tp_link_smartplug/message'
 require 'tp_link_smartplug/version'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tp_link_smartplug
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0.alpha1
 platform: ruby
 authors:
 - Ben Hughes
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-06-07 00:00:00.000000000 Z
+date: 2019-06-08 00:00:00.000000000 Z
 dependencies: []
 description: Control and retrieve data from a TP-Link HS100/110 (Metered) Smartplug
 email: bmhughes@bmhughes.co.uk
@@ -21,7 +21,6 @@ files:
 - "./lib/tp_link_smartplug/device.rb"
 - "./lib/tp_link_smartplug/helpers.rb"
 - "./lib/tp_link_smartplug/message.rb"
-- "./lib/tp_link_smartplug/message_helpers.rb"
 - "./lib/tp_link_smartplug/version.rb"
 homepage: https://github.com/bmhughes/tp_link_smartplug
 licenses:
@@ -38,9 +37,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '2.5'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - ">"
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.3.1
 requirements: []
 rubyforge_project: 
 rubygems_version: 2.7.7

data/lib/tp_link_smartplug/message_helpers.rb

@@ -1,31 +0,0 @@
-module TpLinkSmartplug
-  # Helper methods for plug communication messages
-  module MessageHelpers
-    # Encrypts a message to send to the smart plug
-    #
-    # @string format [String] the message to be encrypted
-    def encrypt(string)
-      key = 171
-      result = [string.length].pack('N')
-      string.each_char do |char|
-        key = a = key ^ char.ord
-        result.concat(a.chr)
-      end
-      result
-    end
-
-    # Decrypts a message received from the smart plug
-    #
-    # @string format [String] the message to be decrypted
-    def decrypt(string)
-      key = 171
-      result = ''
-      string.each_char do |char|
-        a = key ^ char.ord
-        key = char.ord
-        result.concat(a.chr)
-      end
-      result
-    end
-  end
-end