bombshell 0.0.0 → 0.1.0

data/VERSION

@@ -1 +1 @@
-0.0.0
+0.1.0

data/bombshell.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{bombshell}
-  s.version = "0.0.0"
+  s.version = "0.1.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Andy Rossmeissl"]
-  s.date = %q{2011-01-24}
+  s.date = %q{2011-03-07}
   s.description = %q{Custom IRB consoles made easy}
   s.email = %q{andy@rossmeissl.net}
   s.extra_rdoc_files = [
@@ -25,7 +25,13 @@ Gem::Specification.new do |s|
     "Rakefile",
     "VERSION",
     "bombshell.gemspec",
+    "doc/brainstorm.rb",
     "lib/bombshell.rb",
+    "lib/bombshell/completor.rb",
+    "lib/bombshell/environment.rb",
+    "lib/bombshell/irb.rb",
+    "lib/bombshell/shell.rb",
+    "lib/bombshell/shell/commands.rb",
     "spec/bombshell_spec.rb",
     "spec/spec_helper.rb"
   ]

data/doc/brainstorm.rb

@@ -0,0 +1,56 @@
+$:.push File.join(File.dirname(__FILE__), '..', 'lib')
+require 'bombshell'
+
+module Foo
+  class Shell < Bombshell::Environment
+    include Bombshell::Shell
+    
+    before_launch do
+      @punctuation = '!'
+    end
+    
+    before_launch do |arg|
+      puts arg if arg
+    end
+    
+    having_launched do
+      @msg = 'Done' + self.class.punctuation
+    end
+    
+    having_launched do
+      puts 'Welcome to FooShell'
+    end
+    
+    prompt_with do
+      punctuation
+    end
+    
+    def do_something
+      puts @msg
+    end
+    
+    def sub
+      Subshell.launch
+    end
+    
+    class << self
+      def punctuation
+        @punctuation
+      end
+    end
+  end
+end
+
+module Foo
+  class Subshell < Bombshell::Environment
+    include Bombshell::Shell
+    
+    def do_something_else
+      puts '... and done'
+    end
+    
+    prompt_with '[foo::subshell]'
+  end
+end
+
+Bombshell.launch(Foo::Shell)

data/lib/bombshell.rb

@@ -0,0 +1,22 @@
+require 'irb'
+
+require 'bombshell/shell'
+require 'bombshell/completor'
+require 'bombshell/environment'
+require 'bombshell/irb'
+
+module Bombshell
+  def launch(shell)
+    begin
+      failure = shell.launch(ARGV.dup)
+      Kernel.exit(failure ? 1 : 0)
+    rescue SystemExit => e
+      Kernel.exit(e.status)
+    rescue Exception => e
+      STDERR.puts("#{e.message} (#{e.class})")
+      STDERR.puts(e.backtrace.join("\n"))
+      Kernel.exit(1)
+    end
+  end
+  module_function :launch
+end

data/lib/bombshell/completor.rb

@@ -0,0 +1,18 @@
+module Bombshell
+  class Completor
+    def initialize(shell)
+      @shell = shell
+    end
+    
+    attr_reader :shell
+    
+    def complete(fragment)
+      self.class.filter(shell.instance_methods).grep Regexp.new(Regexp.quote(fragment))
+    end
+    
+    def self.filter(m)
+      m - Bombshell::Environment.instance_methods - Bombshell::Shell::Commands::HIDE
+    end
+  end
+end
+

data/lib/bombshell/environment.rb

@@ -0,0 +1,8 @@
+module Bombshell
+  class Environment
+    instance_methods.each do |m|
+      undef_method m if m.to_s !~ /(?:^__|^nil\?$|^send$|^instance_eval$|^define_method$|^class$|^object_id|^instance_methods$)/
+    end
+  end
+end
+

data/lib/bombshell/irb.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+
+module IRB
+  def self.start_session(binding)
+    unless @__initialized
+      args = ARGV
+      ARGV.replace(ARGV.dup)
+      IRB.setup(nil)
+      ARGV.replace(args)
+      @__initialized = true
+    end
+
+    workspace = WorkSpace.new(binding)
+    
+    @CONF[:PROMPT][:CARBON] = {
+      :PROMPT_I => "%m> ",
+      :PROMPT_S => "%m\"> ",
+      :PROMPT_C => "%m…>",
+      :PROMPT_N => "%m→>",
+      :RETURN => ''
+    }
+    @CONF[:PROMPT_MODE] = :CARBON
+
+    irb = Irb.new(workspace)
+
+    ::Readline.completion_proc = Bombshell::Completor.new(eval('self.class', binding)).method(:complete)
+    
+    @CONF[:IRB_RC].call(irb.context) if @CONF[:IRB_RC]
+    @CONF[:MAIN_CONTEXT] = irb.context
+    catch(:IRB_EXIT) do
+      irb.eval_input
+    end
+  end
+end
+

data/lib/bombshell/shell.rb

@@ -0,0 +1,62 @@
+require 'bombshell/shell/commands'
+
+module Bombshell
+  module Shell
+    def self.included(base)
+      base.extend ClassMethods
+      base.instance_variable_set :@bombshell_callbacks, :before_launch => [], :having_launched => []
+    end
+    
+    def to_s
+      _prompt
+    end
+    
+    include Commands
+    
+    def _prompt
+      if self.class.bombshell_prompt.is_a? String
+        self.class.bombshell_prompt
+      elsif self.class.bombshell_prompt.is_a? Proc and self.class.bombshell_prompt.arity == 1
+        self.class.bombshell_prompt.call self
+      elsif self.class.bombshell_prompt.is_a? Proc
+        self.class.bombshell_prompt.call
+      else
+        '[Bombshell]'
+      end
+    end
+    
+    def get_binding
+      binding
+    end
+    
+    module ClassMethods
+      def launch(arguments = [])
+        @bombshell_callbacks[:before_launch].each do |callback|
+          callback.call(*arguments.first(callback.arity))
+        end
+        shell = new
+        @bombshell_callbacks[:having_launched].each do |callback|
+          shell.instance_eval &callback
+        end
+        ::IRB.start_session(shell.get_binding)
+      end
+    
+      def before_launch(&callback)
+        @bombshell_callbacks[:before_launch] << callback
+      end
+
+      def having_launched(&callback)
+        @bombshell_callbacks[:having_launched] << callback
+      end
+      
+      def prompt_with(p = nil, &prompt)
+        @bombshell_prompt = p || prompt
+      end
+      
+      def bombshell_prompt
+        @bombshell_prompt
+      end
+    end
+  end
+end
+

data/lib/bombshell/shell/commands.rb

@@ -0,0 +1,17 @@
+module Bombshell
+  module Shell
+    module Commands
+      HIDE = [:method_missing, :get_binding, :_prompt, :to_s]
+      
+      def quit
+        exit
+      end
+      def method_missing(*args)
+        return if [:extend, :respond_to?].include? args.first
+        puts "Unknown command #{args.first}"
+      end
+    end
+  end
+end
+
+

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: bombshell
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 27
   prerelease: false
   segments: 
   - 0
+  - 1
   - 0
-  - 0
-  version: 0.0.0
+  version: 0.1.0
 platform: ruby
 authors: 
 - Andy Rossmeissl
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-01-24 00:00:00 -05:00
+date: 2011-03-07 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -98,7 +98,13 @@ files:
 - Rakefile
 - VERSION
 - bombshell.gemspec
+- doc/brainstorm.rb
 - lib/bombshell.rb
+- lib/bombshell/completor.rb
+- lib/bombshell/environment.rb
+- lib/bombshell/irb.rb
+- lib/bombshell/shell.rb
+- lib/bombshell/shell/commands.rb
 - spec/bombshell_spec.rb
 - spec/spec_helper.rb
 has_rdoc: true