spinal_tap 0.0.1 → 0.1.0

data/lib/spinal_tap/client_helpers.rb

@@ -1,8 +1,137 @@
 module SpinalTap
 
   module ClientHelpers
-    def prompt
-      write('> ')
+    def setup(server)
+      @server = server
+
+      @buffer = ''
+      @cursor_pos = 1
+      @history = []
+      @history_pos = 0
+
+      setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+
+      suppress_go_ahead = [255, 251, 3].pack('C3')
+      write(suppress_go_ahead)
+
+      echo = [255, 251, 1].pack('C3')
+      write(echo)
+
+      6.times { getc }
+    end
+
+    def process_loop
+      while (line = read_parsed_line)
+        command = line[:command]
+        args = line[:args]
+
+        case command
+        when 'help' then exec_help
+        when 'eval' then exec_eval(args.join(' '))
+        when 'quit'
+          close
+          break
+        else
+          write("Unknown command\r\n")
+        end
+      end
+    ensure
+      close unless closed?
+      @server.unregister(Thread.current)
+    end
+
+    def read_parsed_line
+      @buffer = ''
+      @cursor_pos = 1
+
+      redraw_cmd_line
+
+      while byte = getbyte
+        byte_s = '' << byte
+
+        # Delete Char.
+        if byte == 127
+          if @buffer.length > 0 && @cursor_pos > 1
+            @cursor_pos -= 1
+            @buffer.slice!(@cursor_pos - 1)
+          end
+
+        # Null Char.
+        elsif byte == 0
+          next
+
+        # Escape Char.
+        elsif byte == 27
+          if (byte = getbyte) == 91 # [ Char.
+            case getbyte
+            when 65 # A Char - Up Arrow.
+              if @history_pos > 0
+                @history_pos -= 1
+                @buffer = @history[@history_pos].to_s
+              end
+            when 66 # B Char - Down Arrow.
+              if @history_pos < @history.length
+                @history_pos += 1
+                @buffer = @history[@history_pos].to_s
+              end
+            when 67 # C Char - Right Arrow.
+              if @cursor_pos < @buffer.length
+                @cursor_pos += 1
+              end
+            when 68 # D Char - Left Arrow.
+              if @cursor_pos > 0
+                @cursor_pos -= 1
+              end
+            end
+          end
+
+        # Carriage Return Char.
+        elsif byte == 13
+          write("\r\n")
+
+          tokens = @buffer.split(' ')
+          command = tokens.first
+          args = tokens[1..-1]
+
+          if @buffer.length > 0
+            @history.push(@buffer)
+            @history_pos = @history.length
+          end
+
+          return {:command => command, :args => args}
+
+        # Other Chars.
+        elsif byte >= 32 && byte <= 126
+          @buffer.insert(@cursor_pos - 1, byte_s)
+          @cursor_pos += 1
+
+        # Ignore all other special characters.
+        else
+        end
+
+        redraw_cmd_line
+      end
+    end
+
+    def redraw_cmd_line
+      write("\e[2K\r>\e[s #{@buffer}\e[u\e[#{@cursor_pos}C")
+    end
+
+    def exec_help
+      write("Commands: help quit eval\r\n")
+    end
+
+    def exec_eval(code)
+      begin
+        result = eval(code)
+        write("=> #{result.to_s}\r\n")
+      rescue Exception => e
+        write(exception_to_s(e))
+      end
+    end
+
+    def exception_to_s(e)
+      "#{e.message}\r\n#{e.backtrace.join("\r\n")}"
     end
   end
 

data/lib/spinal_tap/server.rb

@@ -1,4 +1,5 @@
 require 'socket'
+require 'debugger'
 
 module SpinalTap
 
@@ -7,7 +8,7 @@ module SpinalTap
       @host = params[:host] || '127.0.0.1'
       @port = params[:port] || 9000
 
-      @run = false
+      @running = false
       @workers = {}
       @workers_lock = Mutex.new
     end
@@ -23,16 +24,14 @@ module SpinalTap
         while true
           Thread.new(@server_sock.accept) do |client|
             begin
-              @workers_lock.synchronize do
-                @workers[Thread.current] = client
-              end
+              register(Thread.current, client)
 
-              client.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
               client.extend(SpinalTap::ClientHelpers)
+              client.setup(self)
+              client.process_loop
 
-              process(client)
             rescue Exception => e
-              puts("WORKER DIED: #{exception_to_s(e)}")
+              puts("WORKER DIED: #{e.message}\n#{e.backtrace.join("\n")}")
             end
           end
         end
@@ -56,53 +55,17 @@ module SpinalTap
       end
     end
 
-    private
-
-    def process(client)
-      client.prompt
-
-      while(line = client.gets)
-        line.chomp!
-
-        tokens = line.split(' ')
-        command = tokens.first
-
-        case command
-        when 'help' then exec_help(client)
-        when 'eval' then exec_eval(client, tokens[1..-1].join(' '))
-        when 'quit'
-          client.close
-          break
-        else
-          client.puts('Unknown command')
-        end
-
-        client.prompt
-      end
-
-    ensure
-      client.close unless client.closed?
-
+    def register(thread, client)
       @workers_lock.synchronize do
-        @workers.delete(Thread.current)
+        @workers[thread] = client
       end
     end
 
-    def exec_help(client)
-      client.puts('Commands: help quit eval')
-    end
-
-    def exec_eval(client, code)
-      begin
-        result = eval(code)
-        client.puts(result.to_s)
-      rescue Exception => e
-        client.puts(exception_to_s(e))
+    def unregister(thread)
+      @workers_lock.synchronize do
+        @workers.delete(thread)
       end
     end
 
-    def exception_to_s(e)
-      "#{e.message}\r\n#{e.backtrace.join("\r\n")}"
-    end
   end
 end

data/lib/spinal_tap/version.rb

@@ -1,3 +1,3 @@
 module SpinalTap
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/spinal_tap.gemspec

@@ -20,4 +20,5 @@ Gem::Specification.new do |gem|
   gem.add_development_dependency('guard', ['= 1.2.3'])
   gem.add_development_dependency('guard-rspec', ['= 1.2.0'])
   gem.add_development_dependency('growl', ['= 1.0.3'])
+  gem.add_development_dependency('debugger', ['>= 1.1.4'])
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: spinal_tap
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-10 00:00:00.000000000 Z
+date: 2012-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -91,6 +91,22 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.0.3
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.1.4
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.1.4
 description: Backdoor into your long running ruby processes.
 email:
 - chad@remesch.com
@@ -124,7 +140,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 4269159806787806500
+      hash: 3767051084925650736
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -133,7 +149,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 4269159806787806500
+      hash: 3767051084925650736
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24