homeq 1.1.4

data/lib/homeq/cli.rb

@@ -0,0 +1,35 @@
+#############################################################################
+#
+# $Id: homeq.rb 38 2008-10-21 02:38:26Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+module HomeQ
+  module CLI
+    def self.execute(dest_dir, rakefile)
+      args = ARGV.dup
+      args += ["-I", "#{HOMEQ_ROOT}/lib/tasks"]
+      args += ["-f", rakefile]
+      ENV['HOMEQ_CLI_DEST'] = dest_dir
+      exec('rake', *args)
+    end
+  end
+end

data/lib/homeq/cp/commands.rb

@@ -0,0 +1,71 @@
+#############################################################################
+#
+# $Id: commands.rb 48 2008-11-19 05:11:59Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+# TODO: info 
+#
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+module HomeQ
+  
+  module CP
+    
+    module Commands
+      
+      def shutdown(*args)
+        if !args || !args.include?('really')
+          return "Really??"
+        end
+        logger.info {
+          "Shutdown initiated by control port."
+        }
+        EventMachine::add_timer(3) do
+          HomeQ::Base::System.instance.stop
+        end
+        EventMachine::add_timer(1) do
+          close
+        end
+        "OK"
+      end
+      
+      def close
+        close_connection
+      end
+
+      alias :quit :close
+      alias :exit :close
+
+      def help
+        show_commands
+      end
+      
+    end # Commands
+
+  end # CP
+
+end # HomeQ
+
+module HomeQ::Base::Commando::InstanceMethods
+  include HomeQ::CP::Commands
+  document_command "help", "A little love"
+  document_command "quit|exit|close", "Close control port connection"
+  document_command "shutdown", "Shut down this homeq process"
+end

data/lib/homeq/cp/connection.rb

@@ -0,0 +1,97 @@
+#############################################################################
+#
+# $Id: connection.rb 40 2008-10-23 14:20:06Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+# TODO: info 
+#
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+module HomeQ
+  
+  module CP
+
+    include Base::Logging
+
+    class Connection < EventMachine::Connection
+      
+      include HomeQ
+
+      TRACE_REXP = /^#{Regexp.escape(__FILE__)}/
+
+      attr :server, true
+      
+      def receive_data data
+        if (data == "\004" || # ^D
+            data == "\377\364\377\375\006") # ^C
+          close_connection
+          return
+        end
+        
+        begin
+          send_data("#{eval(data)}\r\n")
+        rescue SyntaxError, StandardError, Exception
+          trace = []
+          stack = $!.backtrace
+          trace << stack.pop until stack.first =~ TRACE_REXP
+          send_data([$!, $!.message, trace].join("\n\t"))
+        end
+        
+      end
+
+      def unbind
+        server.connections.delete(self) if server
+        logger.info {
+          "Connection has terminated to control port client " +
+            "#{@peer[1]}:#{@peer[0]}."
+        }
+      end
+      
+      def post_init
+        @peer = Socket.unpack_sockaddr_in(get_peername)
+        logger.info {
+          "Control port connection received from " +
+            "#{@peer[1]}:#{@peer[0]}."
+        }
+        
+        send_data issue
+
+        # Add all the commands as methods
+        self.class.class_eval {
+          include HomeQ::Base::Commando::InstanceMethods
+        }
+      end
+
+      protected
+
+      def issue
+        str = "\r\n"
+        str << "\r\n" + $0.center(79)
+        str << "\r\n" + "Queue: #{config.queue_name}".center(79)
+        str << "This is a live ruby interpreter.".center(79) + "\r\n"
+        str << "Careful, you can seriously mess this process up.".center(79)
+        str << "\r\n\r\n"
+      end
+
+    end # Connection
+
+  end # CP
+
+end # HomeQ

data/lib/homeq/cp/cp.rb

@@ -0,0 +1,30 @@
+#############################################################################
+#
+# $Id: cp.rb 42 2008-10-24 05:53:35Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+# TODO: info 
+#
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+require 'homeq/cp/commands'
+require 'homeq/cp/connection'
+require 'homeq/cp/server'
+

data/lib/homeq/cp/server.rb

@@ -0,0 +1,105 @@
+#############################################################################
+#
+# $Id: server.rb 40 2008-10-23 14:20:06Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+# TODO: info 
+#
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+module HomeQ
+  
+  module CP
+
+    class Server
+
+      DEFAULT_PORT = 56001
+      DEFAULT_HOST = 'localhost'
+
+      include Base::Logging
+      include Base::Configuration
+
+      attr_accessor :connections
+
+      # Make our config info available
+      module HomeQ::Base::Commando::InstanceMethods
+        config_accessor :cp_host
+        config_accessor :cp_port
+      end
+      
+      def initialize(host=nil, port=nil)
+        @host = host || DEFAULT_HOST
+        @port = port || DEFAULT_PORT
+        @connections = []
+      end
+
+      def start
+        begin
+          @signature = EventMachine.start_server(@host,
+                                                 @port,
+                                                 CP::Connection) { |conn|
+            conn.server = self
+          }
+        rescue Exception => e
+          msg = "Error starting CP listener #{@host}:#{@port}. " +
+            "Likely it's a config error. " +
+            "Original error: #{e.message}"
+          Base::System.instance.die(msg)
+        else
+          logger.info {
+            "Control port running on #{@host}:#{@port}"
+          }
+        end
+      end
+
+      def stop
+        EventMachine.stop_server(@signature)
+        unless wait_for_connections
+          # Still some running
+          EventMachine.add_periodic_timer(1) {
+            logger.info {
+              '.'
+            }
+            wait_for_connections
+          }
+        end
+      end
+
+      protected
+
+      def wait_for_connections
+        if @connections.empty?
+          true
+        else
+          logger.info {
+            "Waiting for #{@connections.length} connections to finish."
+          }
+          @connections.each { |c|
+            c.close_connection_after_writing
+          }
+          false
+        end
+      end
+      
+    end # Server
+
+  end # CP
+
+end # HomeQ

data/lib/homeq/sobs/client.rb

@@ -0,0 +1,119 @@
+#############################################################################
+#
+# $Id: client.rb 40 2008-10-23 14:20:06Z colin $
+#
+# Author:: Colin Steele (colin@colinsteele.org)
+# Homepage::  
+# 
+# TODO: info 
+#
+#----------------------------------------------------------------------------
+#
+#      Copyright (C) 2008 by Colin Steele. All Rights Reserved.
+#                        colin@colinsteele.org
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+#############################################################################
+
+module HomeQ
+  
+  module SOBS
+    
+    class Client < SOBS::Connection
+
+      attr :queue, true      # the queue that owns this connection
+      attr :host, true
+      attr :port, true
+      attr :retry, true
+
+      def self.connect(host, port, interval, server, queue)
+        EventMachine::connect(host,
+                              port,
+                              self) { |conn|
+          conn.host   = host
+          conn.port   = port
+          conn.retry  = interval
+          conn.server = server
+          conn.queue  = queue
+        }
+      end
+
+      def initialize
+        super
+        @host = @port = @retry = nil
+      end
+
+      
+      ######################################################################
+      # SOBS::Connection entry points
+      ######################################################################
+
+      def opened
+        @queue.opened(self) if @queue
+        subscribe if readable?
+      end
+      
+      def closed
+        @queue.closed if @queue
+        retry_connection_after_interval(@retry)
+      end
+
+      
+      ######################################################################
+      # High level message interface - receiving
+      ######################################################################
+
+      def receive_reserved(message)
+        @queue.create_job(message)
+      end
+
+      def receive_inserted(message)
+        @queue.ack(message)
+      end
+      
+      def receive_deleted(message)
+        @queue.deleted(message)
+      end
+
+      
+      ######################################################################
+      # Other
+      ######################################################################
+
+      def readable?
+        @queue.readable?
+      end
+
+      def writeable?
+        @queue.writeable?
+      end
+
+      protected
+
+      def get_next_message
+        reserve if readable?
+      end
+
+      def retry_connection_after_interval(interval)
+        EventMachine::add_timer(interval) do
+          logger.info {
+            "Trying connection to #{host}:#{port}."
+          }
+          Client.connect(@host, @port, @retry, @server, @queue)
+        end
+      end
+
+    end # Client
+
+  end # module SOBS
+  
+end # module HomeQ