autodesk_backburner_client 0.1.0 → 0.2.2

data/lib/backburner.rb

@@ -1,4 +1,3 @@
-require 'active_support/core_ext/hash'
 require "backburner/version"
 require "backburner/connection"
 require "backburner/job"
@@ -6,4 +5,5 @@ require "backburner/server"
 require "backburner/client"
 require "backburner/string/string_operations"
 require "backburner/data_object"
-require "backburner/backburner_object"
+require "backburner/backburner_object"
+require "backburner/remote_command"

data/lib/backburner/client.rb

@@ -2,10 +2,13 @@ module Backburner
 
   class Client
 
+    attr_reader :data
+
     def initialize host, port=3234
       @host = host
       @port = port
       @connection = Connection.new @host, @port
+      send_client_info
       self
     end
 
@@ -53,25 +56,54 @@ module Backburner
 
     private
 
+    def generate_client_info
+      @data = DataObject.new
+      @data.client_info = {
+          version: 300,
+          udp_port: 0,
+          controler: 0,
+          system_info: {
+              total_memory: total_memory,
+              total_memory_f: total_memory_f,
+              num_cpus: num_cpus,
+              platform: platform,
+              user: user,
+              computer_name: computer_name,
+              mac: '00E052B9A2F20000',
+              work_disk_space: 0,
+              i_p_address: '172.50.0.170'
+          }
+      }
+    end
+
     def send_client_info
-#      <?xml version="1.0" encoding="UTF-8" standalone="yes" ?>
-#          <ClientInfo>
-#      <Version>300</Version>
-#    <UdpPort>0</UdpPort>
-#      <Controler>0</Controler>
-#    <SystemInfo>
-#        <TotalMemory>4294967295</TotalMemory>
-#      <TotalMemoryF>34313162752.000000</TotalMemoryF>
-#        <NumCpus>8</NumCpus>
-#      <Platform>Windows XP 6.1 Build 7601 - Service Pack 1</Platform>
-#        <User> </User>
-#      <ComputerName>rendermama</ComputerName>
-#        <Mac>00E052B9A2F20000</Mac>
-#      <WorkDiskSPace>0</WorkDiskSPace>
-#        <IPAddress>10.128.23.74</IPAddress>
-#      </SystemInfo>
-#</ClientInfo>
+      generate_client_info unless @data
+      xml = @data.to_xml
+      @connection.set_clientinfo xml.bytesize
+    end
+
+    def total_memory
+      `free -b`.match(/\d+/).to_s.to_i
+    end
+
+    def total_memory_f
+      total_memory.to_f
+    end
+
+    def num_cpus
+      `cat /proc/cpuinfo`.split("\n\n" ).count
+    end
+
+    def platform
+      `uname`.gsub(/\n/, '')
+    end
+
+    def user
+      `whoami`.gsub(/\n/, '')
+    end
 
+    def computer_name
+      `hostname`.gsub(/\n/, '')
     end
 
   end

data/lib/backburner/connection.rb

@@ -1,25 +1,23 @@
 require 'net/telnet'
+require 'nori'
 
 module Backburner
 
   class Connection
 
-    def initialize host, port=3234
-      @session = Net::Telnet.new('Host' => host, 'Port' => port, 'Prompt' => /backburner>/i)
-      @session.waitfor 'Match' => /backburner>/i
-    end
+    STATUS_REGEXP = /(\d+) ([0-9A-z ]+)$/
+    PROMPT_REGEXP = /\nbackburner(\(controller\))?>/i
+    XML_COMMENT_REGEXP = /<!--.+-->/
 
-    def method_missing name, *args
-      if name =~ /^get_/
-        object = name.to_s.gsub /^get_/, ''
-        return get(object, args)
-      end
-      super
-    end
+    attr_reader :last_message, :last_size, :last_status, :last_body, :last_response
 
-    def get object, *params
-      exec 'String' => "get #{object} #{params.join ' ' if params.kind_of?(Array)}"
+    def initialize host='127.0.0.1', port=3234
+      @session = Net::Telnet.new('Host' => host, 'Port' => port, 'Prompt' => PROMPT_REGEXP)
+      @session.waitfor 'Match' => PROMPT_REGEXP
+    end
 
+    def method_missing method_name
+      RemoteCommand.new method_name, self
     end
 
     def close
@@ -32,31 +30,68 @@ module Backburner
       raise
     end
 
+    def exec arg_hash
+      xml_string = ''
+      first = true
+      @session.cmd arg_hash do |response|
+        if first
+          response =~ /(\d+) (\d*)(.*)/
+          @last_status = $1.to_i
+          @last_size = $2.to_i
+          @last_message = $3
+          first = false
+          next
+        end
+        xml_string += response
+      end
+
+      xml_string = xml_string.sub(/backburner>$/, '')
+      DataObject.new xml_string
+    end
+
+    def send_data data
+      @session.cmd 'String' => data
+    end
+
+    def send_text data
+      @last_response = @session.cmd data
+      parse_response
+    end
+
+    def proceed?
+      @last_status < 200 || @last_status > 299
+    end
+
+    def ready?
+      @last_status == 250
+    end
+
     private
 
-      def exec arg_hash
-        xml_string = ""
-        first = true
-        @session.cmd arg_hash do |response|
-          if first
-            response =~ /(\d+) (\d*)(.*)/
-            @last_status = $1.to_i
-            @last_size = $2.to_i
-            @last_message = $3
-            first = false
-            next
-          end
-          xml_string += response
-        end
+      def parse_response
+        parse_status
+        parse_body
+        parse_object
+      end
 
-        xml_string = xml_string.sub(/backburner>$/, '')
-        DataObject.new xml_string
+      def parse_status
+        @last_response.scan STATUS_REGEXP
+        @last_status = $1.to_i
+        @last_size = $2.to_i
+        @last_message = $2
       end
 
-      def send hash
+      def parse_body
+        @last_body = @last_response.gsub PROMPT_REGEXP, ''
+        @last_body.gsub! STATUS_REGEXP, ''
+        @last_body.gsub! XML_COMMENT_REGEXP, ''
+      end
 
+      def parse_object
+        parser = Nori.new :convert_tags_to => lambda { |tag| tag.snakecase.to_sym }
+        DataObject.new parser.parse @last_body
       end
 
   end
 
-end
+end

data/lib/backburner/data_object.rb

@@ -1,18 +1,55 @@
-require 'nori'
 require 'recursive_open_struct'
-require 'builder'
+require 'nokogiri'
 
 module Backburner
 
   class DataObject < RecursiveOpenStruct
 
-    def initialize xml=''
-      if xml
-        parser = Nori.new :convert_tags_to => lambda { |tag| tag.snakecase.to_sym }
-        super parser.parse(xml), recurse_over_arrays: true
-      else
-       super
+    TIME_REGEXP = /(^\d{4}\/\d{2}\/\d{2}-\d{2}:\d{2}:\d{2}:\d{4})(-\d{2})$/
+
+    def initialize h, args={recurse_over_arrays: true}
+      super
+    end
+
+    def new_ostruct_member(name)
+      name = name.to_sym
+      unless self.respond_to?(name)
+        class << self; self; end.class_eval do
+          define_method(name) do
+            v = @table[name]
+            if v.is_a?(Hash)
+              @sub_elements[name] ||= self.class.new(v, :recurse_over_arrays => @recurse_over_arrays)
+            elsif v.is_a?(Array) and @recurse_over_arrays
+              @sub_elements[name] ||= recurse_over_array v
+            elsif v.is_a?(String)
+              case v
+                when /^\d+$/ then v.to_i
+                when /^\d+\.\d+$/ then v.to_f
+                when TIME_REGEXP then parse_time(v)
+                when /^Yes$/i then true
+                when /^No$/i then false
+                else v
+              end
+            else
+              v
+            end
+          end
+          define_method("#{name}=") { |x| modifiable[name] = x }
+          define_method("#{name}_as_a_hash") { @table[name] }
+        end
       end
+      name
+    end
+
+    def parse_time(time_string)
+      res = time_string.scan TIME_REGEXP
+      time = Time.parse res[0][0]
+      time += res[0][1].to_i*60*60
+      time
+    end
+
+    def empty?
+      @table.empty?
     end
 
     def search name
@@ -32,7 +69,7 @@ module Backburner
     end
 
     def to_xml
-      builder = Nokogiri::XML::Builder.new do |xml|
+      builder = Nokogiri::XML::Builder.new(encoding: 'UTF-8') do |xml|
         xml.send @table.keys.first.to_s.camelize
       end
       document = builder.doc
@@ -68,6 +105,8 @@ module Backburner
           end
         elsif data.kind_of?(String)
           children << data
+        elsif data.kind_of?(Numeric)
+          children << data.to_s
         end
 
         children.each do |child|

data/lib/backburner/job.rb

@@ -8,6 +8,10 @@ module Backburner
       @client.job!(@handler)
     end
 
+    def create name, file_path, *parameters
+
+    end
+
   end
 
 end

data/lib/backburner/remote_command.rb

@@ -0,0 +1,32 @@
+module Backburner
+  class RemoteCommand
+    attr_accessor :command_stack
+
+    def initialize command_name, connection, command_stack=nil
+      @command_stack = command_stack || []
+      @command_stack.unshift command_name
+      @connection = connection
+    end
+
+    def method_missing method_name, *args
+      self.class.new method_name, @command_stack
+    end
+
+    def flatten
+      @command_stack.join ' '
+    end
+
+    %w{get set new del kill exit}.each do |method_name|
+
+      define_method method_name do |*args|
+        request_array = []
+        request_array << method_name
+        request_array << flatten
+        request_array += args||[]
+        @connection.send_text request_array.join(' ')
+      end
+
+    end
+
+  end
+end

data/lib/backburner/version.rb

@@ -1,3 +1,3 @@
 module Backburner
-  VERSION = "0.1.0"
+  VERSION = "0.2.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: autodesk_backburner_client
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -96,6 +96,7 @@ files:
 - lib/backburner/data_object.rb
 - lib/backburner/job.rb
 - lib/backburner/open_struct_data.rb
+- lib/backburner/remote_command.rb
 - lib/backburner/server.rb
 - lib/backburner/string/string_operations.rb
 - lib/backburner/version.rb