duoconsole 0.1.0

data/bin/duoconsole

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+APP_PATH = File.join(Dir.pwd, 'config', 'application')
+
+begin
+  require File.join(Dir.pwd, 'config', 'boot')
+rescue LoadError
+  abort "Duoconsole must be run from the root directory of a Rails project."
+end
+
+require 'duoconsole'
+Duoconsole.start

data/lib/duoconsole.rb

@@ -0,0 +1,162 @@
+require 'socket'
+
+class Duoconsole
+
+  def self.start
+    new.start
+  end
+
+  attr_accessor :child_socket, :parent_socket
+
+  def start
+    preload_gems
+    create_socket_pair
+    fork_child
+    load_application
+    initialize_command_client
+    start_console
+  end
+
+  def preload_gems
+    require 'rails/all'
+
+    if defined?(Bundler)
+      Bundler.require(:default, :assets)
+    end
+
+    require 'commands'
+  end
+
+  def create_socket_pair
+    self.child_socket, self.parent_socket = Socket.pair(:UNIX, :DGRAM, 0)
+  end
+
+  def fork_child
+    child_pid = fork do
+      ENV['RAILS_ENV'] = 'test'
+
+      if defined?(Rails)
+        Rails.env = 'test'
+      end
+
+      load_application
+      monkeypatch_test_environment
+
+      trap(:INT) {
+        # Ignore. This process needs to stay alive until the parent process exits
+      }
+
+      CommandServer.new(child_socket).start
+    end
+
+    # cleanup before exiting
+    at_exit {
+      Process.kill(:QUIT, child_pid)
+      parent_socket.close
+      child_socket.close
+    }
+  end
+
+  def load_application
+    require APP_PATH
+    Rails.application.require_environment!
+  end
+
+  def initialize_command_client
+    ConsoleDelegation.command_client = CommandClient.new(parent_socket)
+  end
+
+  def start_console
+    require 'rails/commands/console'
+    require 'rails/console/app'
+    Rails::ConsoleMethods.send :include, ConsoleDelegation
+    Rails::Console.start(Rails.application)
+  end
+
+  def monkeypatch_test_environment
+    Rails::Commands::TestEnvironment.module_eval do
+      # Overriding this method to add the following behavior:
+      #   1. fix issue with Postgres adapter and forking behavior
+      #   2. trap INT signal and exit
+      def fork
+        ActiveRecord::Base.clear_active_connections!
+        Rails::Commands::Environment.fork do
+          setup_for_test
+          trap(:INT) { exit(1) }
+
+          yield
+          ActiveRecord::Base.clear_active_connections!
+        end
+      end
+    end
+  end
+
+
+  class CommandClient
+    attr_reader :socket
+
+    def initialize socket
+      @socket = socket
+    end
+
+    def send msg
+      socket.write(msg)
+      socket.recv(1000)
+    end
+
+    def method_missing(m, *args, &block)
+      send "#{m} #{args.join(' ')}"
+    end
+  end
+
+
+  class CommandServer
+    attr_reader :socket
+
+    def initialize socket
+      @socket = socket
+    end
+
+    def start
+      loop do
+        msg = socket.recv(1000)
+        command, args = get_command_and_args msg
+
+        retval = if commander.respond_to?(command)
+          commander.send(command, args)
+        else
+          'Unrecognized command'
+        end
+
+        socket.write(retval)
+      end
+    end
+
+    def get_command_and_args msg
+      parts = msg.split(' ')
+      command = parts.shift
+      args = parts.join(' ')
+      args = nil unless args.match(/\S/)
+      [command, args]
+    end
+
+    def commander
+      @commander ||= Rails::Commands::Commander.new
+    end
+  end
+
+
+  module ConsoleDelegation
+    def self.command_client= client
+      @@command_client = client
+    end
+
+    def testenv
+      @@command_client
+    end
+
+    def test *args
+      testenv.test *args
+    end
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification
+name: duoconsole
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Geoff Buesing
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+- !ruby/object:Gem::Dependency
+  name: commands
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.2.1
+description: 
+email: gbuesing@gmail.com
+executables:
+- duoconsole
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/duoconsole.rb
+- bin/duoconsole
+homepage: https://github.com/gbuesing/duoconsole
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Launch Rails development console with test environment as a child process
+test_files: []
+has_rdoc: