dietrb 0.1.0

data/.gitignore

@@ -0,0 +1 @@
+pkg

data/README

@@ -0,0 +1 @@
+IRB on a diet, for MacRuby / Ruby 1.9

data/Rakefile

@@ -0,0 +1,25 @@
+task :default => :spec
+
+desc "Run the specs"
+task :spec do
+  sh "macbacon #{FileList['spec/**/*_spec.rb'].join(' ')}"
+end
+
+desc "Run specs with Kicker"
+task :kick do
+  sh "kicker -e 'rake spec' lib spec"
+end
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gemspec|
+    gemspec.name = "dietrb"
+    gemspec.summary = gemspec.description = "IRB on a diet, for MacRuby / Ruby 1.9"
+    gemspec.email = "eloy.de.enige@gmail.com"
+    gemspec.homepage = "http://github.com/alloy/dietrb"
+    gemspec.authors = ["Eloy Duran"]
+    
+    gemspec.required_ruby_version = ::Gem::Requirement.new("~> 1.9")
+  end
+rescue LoadError
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/dietrb

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'irb'
+
+IRB::Context.new(self).run

data/dietrb.gemspec

@@ -0,0 +1,56 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{dietrb}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Eloy Duran"]
+  s.date = %q{2010-02-07}
+  s.default_executable = %q{dietrb}
+  s.description = %q{IRB on a diet, for MacRuby / Ruby 1.9}
+  s.email = %q{eloy.de.enige@gmail.com}
+  s.executables = ["dietrb"]
+  s.extra_rdoc_files = [
+    "README"
+  ]
+  s.files = [
+    ".gitignore",
+     "README",
+     "Rakefile",
+     "VERSION",
+     "bin/dietrb",
+     "dietrb.gemspec",
+     "lib/irb.rb",
+     "lib/irb/context.rb",
+     "lib/irb/source.rb",
+     "spec/context_spec.rb",
+     "spec/source_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/alloy/dietrb}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.required_ruby_version = Gem::Requirement.new("~> 1.9")
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{IRB on a diet, for MacRuby / Ruby 1.9}
+  s.test_files = [
+    "spec/context_spec.rb",
+     "spec/source_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end
+

data/lib/irb/context.rb

@@ -0,0 +1,78 @@
+require 'readline'
+
+module IRB
+  class Context
+    attr_reader :object, :binding, :line, :source
+    
+    def initialize(object)
+      @object  = object
+      @binding = object.instance_eval { binding }
+      @line    = 1
+      clear_buffer
+    end
+    
+    def evaluate(source)
+      result = eval("_ = (#{source})", @binding)
+      puts format_result(result)
+      result
+    rescue Exception => e
+      puts format_exception(e)
+    end
+    
+    def readline
+      Readline.readline(prompt, true)
+    end
+    
+    def run
+      while line = readline
+        continue = process_line(line)
+        break unless continue
+      end
+    end
+    
+    # Returns whether or not the user wants to continue the current runloop.
+    # This can only be done at a code block indentation level of 0.
+    #
+    # For instance, this will continue:
+    #
+    #   process_line("def foo") # => true
+    #   process_line("quit") # => true
+    #   process_line("end") # => true
+    #
+    # But at code block indentation level 0, `quit' means exit the runloop:
+    #
+    #   process_line("quit") # => false
+    def process_line(line)
+      @source << line
+      return false if @source.to_s == "quit"
+      
+      if @source.valid?
+        evaluate(@source)
+        clear_buffer
+      end
+      @line += 1
+      
+      true
+    end
+    
+    PROMPT = "irb(%s):%03d:%d> "
+    
+    def prompt
+      PROMPT % [@object.inspect, @line, @source.level]
+    end
+    
+    def format_result(result)
+      "=> #{result.inspect}"
+    end
+    
+    def format_exception(e)
+      "#{e.class.name}: #{e.message}\n\t#{e.backtrace.join("\n\t")}"
+    end
+    
+    private
+    
+    def clear_buffer
+      @source = Source.new
+    end
+  end
+end

data/lib/irb/source.rb

@@ -0,0 +1,103 @@
+require 'ripper'
+
+module IRB
+  class Source
+    attr_reader :buffer
+    
+    def initialize(buffer = [])
+      @buffer = buffer
+    end
+    
+    # Adds a source line to the buffer and flushes the cached reflection.
+    def <<(source)
+      @reflection = nil
+      @buffer << source.chomp
+    end
+    
+    # Returns the accumulated source as a string, joined by newlines.
+    def source
+      @buffer.join("\n")
+    end
+    
+    alias_method :to_s, :source
+    
+    # Reflects on the accumulated source and returns the current code block
+    # indentation level.
+    def level
+      reflect.level
+    end
+    
+    # Reflects on the accumulated source to see if it's a valid code block.
+    def valid?
+      reflect.valid?
+    end
+    
+    # Returns a Reflector for the accumulated source and caches it.
+    def reflect
+      @reflection ||= Reflector.new(source)
+    end
+    
+    class Reflector < Ripper::SexpBuilder
+      def initialize(source)
+        super
+        @level = 0
+        @valid = !parse.nil?
+      end
+      
+      # Returns the code block indentation level.
+      #
+      #   Reflector.new("").level # => 0
+      #   Reflector.new("class Foo").level # => 1
+      #   Reflector.new("class Foo; def foo").level # => 2
+      #   Reflector.new("class Foo; def foo; end").level # => 1
+      #   Reflector.new("class Foo; def foo; end; end").level # => 0
+      attr_reader :level
+      
+      # Returns whether or not the source is a valid code block, but does not
+      # take syntax errors into account.
+      #
+      # For example, this is not a valid full code block:
+      #
+      #   def foo; p :ok
+      #
+      # This however is:
+      #
+      #   def foo; p :ok; end
+      def valid?
+        @valid
+      end
+      
+      INCREASE_LEVEL_KEYWORDS = %w{ class module def begin if unless case while for do }
+      
+      def on_kw(token) #:nodoc:
+        case token
+        when *INCREASE_LEVEL_KEYWORDS
+          @level += 1
+        when "end"
+          @level -= 1
+        end
+        super
+      end
+      
+      def on_lbracket(token) #:nodoc:
+        @level += 1
+        super
+      end
+      
+      def on_rbracket(token) #:nodoc:
+        @level -= 1
+        super
+      end
+      
+      def on_lbrace(token) #:nodoc:
+        @level += 1
+        super
+      end
+      
+      def on_rbrace(token) #:nodoc:
+        @level -= 1
+        super
+      end
+    end
+  end
+end

data/lib/irb.rb

@@ -0,0 +1,2 @@
+require 'irb/context'
+require 'irb/source'

data/spec/context_spec.rb

@@ -0,0 +1,162 @@
+require File.expand_path('../spec_helper', __FILE__)
+require 'tempfile'
+
+main = self
+
+describe "IRB::Context" do
+  before do
+    @context = IRB::Context.new(main)
+  end
+  
+  it "initializes with an object and stores a copy of its binding" do
+    @context.object.should == main
+    eval("self", @context.binding).should == main
+    eval("x = :ok", @context.binding)
+    eval("y = x", @context.binding)
+    eval("y", @context.binding).should == :ok
+  end
+  
+  it "initializes with an 'empty' state" do
+    @context.line.should == 1
+    @context.source.should.be.instance_of IRB::Source
+    @context.source.to_s.should == ""
+  end
+  
+  it "does not use the same binding copy of the top level object" do
+    lambda { eval("x", @context.binding) }.should.raise NameError
+  end
+  
+  it "returns a prompt string, displaying line number and code indentation level" do
+    @context.prompt.should == "irb(main):001:0> "
+    @context.instance_variable_set(:@line, 23)
+    @context.prompt.should == "irb(main):023:0> "
+    @context.source << "def foo"
+    @context.prompt.should == "irb(main):023:1> "
+  end
+  
+  it "describes the context's object in the prompt" do
+    @context.prompt.should == "irb(main):001:0> "
+    o = Object.new
+    IRB::Context.new(o).prompt.should == "irb(#{o.inspect}):001:0> "
+  end
+  
+  it "returns a formatted exception message" do
+    begin; DoesNotExist; rescue NameError => e; exception = e; end
+    @context.format_exception(exception).should ==
+      "NameError: uninitialized constant Bacon::Context::DoesNotExist\n\t#{exception.backtrace.join("\n\t")}"
+  end
+end
+
+describe "IRB::Context, when evaluating source" do
+  before do
+    @context = IRB::Context.new(main)
+    def @context.puts(string); @printed = string; end
+  end
+  
+  it "evaluates code with the object's binding" do
+    @context.evaluate("self").should == main
+  end
+  
+  it "prints the result" do
+    @context.evaluate("Hash[:foo, :foo]")
+    printed = @context.instance_variable_get(:@printed)
+    printed.should == "=> {:foo=>:foo}"
+  end
+  
+  it "assigns the result to the local variable `_'" do
+    result = @context.evaluate("Object.new")
+    @context.evaluate("_").should == result
+    @context.evaluate("_").should == result
+  end
+  
+  it "coerces the given source to a string first" do
+    o = Object.new
+    def o.to_s; "self"; end
+    @context.evaluate(o).should == main
+  end
+  
+  it "rescues any type of exception" do
+    lambda {
+      @context.evaluate("DoesNotExist")
+      @context.evaluate("raise Exception")
+    }.should.not.raise
+  end
+  
+  it "prints the exception that occurs" do
+    @context.evaluate("DoesNotExist")
+    printed = @context.instance_variable_get(:@printed)
+    printed.should.match /^NameError: uninitialized constant DoesNotExist/
+  end
+end
+
+class << Readline
+  attr_reader :received
+  
+  def stub_input(*input)
+    @input = input
+  end
+  
+  def readline(prompt, history)
+    @received = [prompt, history]
+    @input.shift
+  end
+end
+
+describe "IRB::Context, when receiving input" do
+  before do
+    @context = IRB::Context.new(main)
+  end
+  
+  it "prints the prompt, reads a line, saves it to the history and returns it" do
+    Readline.stub_input("def foo")
+    @context.readline.should == "def foo"
+    Readline.received.should == ["irb(main):001:0> ", true]
+  end
+  
+  it "processes the output" do
+    Readline.stub_input("def foo")
+    def @context.process_line(line); @received = line; false; end
+    @context.run
+    @context.instance_variable_get(:@received).should == "def foo"
+  end
+  
+  it "adds the received code to the source buffer" do
+    @context.process_line("def foo")
+    @context.process_line("p :ok")
+    @context.source.to_s.should == "def foo\np :ok"
+  end
+  
+  it "increases the current line number" do
+    @context.line.should == 1
+    @context.process_line("def foo")
+    @context.line.should == 2
+    @context.process_line("p :ok")
+    @context.line.should == 3
+  end
+  
+  it "evaluates the buffered source once it's a valid code block" do
+    def @context.evaluate(source); @evaled = source; end
+    
+    @context.process_line("def foo")
+    @context.process_line(":ok")
+    @context.process_line("end; p foo")
+    
+    source = @context.instance_variable_get(:@evaled)
+    source.to_s.should == "def foo\n:ok\nend; p foo"
+  end
+  
+  it "returns whether or not the runloop should continue, but only if the level is 0" do
+    @context.process_line("def foo").should == true
+    @context.process_line("quit").should == true
+    @context.process_line("end").should == true
+    
+    @context.process_line("quit").should == false
+  end
+  
+  it "exits the runloop if the user wishes so" do
+    Readline.stub_input("quit", "def foo")
+    def @context.process_line(line); @received = line; super; end
+    @context.run
+    @context.instance_variable_get(:@received).should.not == "def foo"
+  end
+end

data/spec/source_spec.rb

@@ -0,0 +1,137 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe "IRB::Source" do
+  before do
+    @source = IRB::Source.new
+  end
+  
+  it "initializes with an empty buffer" do
+    @source.buffer.should == []
+  end
+  
+  it "appends source to the buffer, removing trailing newlines" do
+    @source << "foo\n"
+    @source << "bar\r\n"
+    @source.buffer.should == %w{ foo bar }
+  end
+  
+  it "returns the full buffered source, joined by newlines" do
+    @source.source.should == ""
+    @source << "foo\n"
+    @source.source.should == "foo"
+    @source << "bar\r\n"
+    @source.source.should == "foo\nbar"
+  end
+  
+  it "aliases #to_s to #source" do
+    @source << "foo"
+    @source << "bar"
+    @source.to_s.should == "foo\nbar"
+  end
+  
+  it "returns that the accumulated source is a valid code block" do
+    [
+      ["def foo", "p :ok", "end"],
+      ["class A; def", "foo(x); p x", "end; end"]
+    ].each do |buffer|
+      IRB::Source.new(buffer).should.be.valid
+    end
+  end
+  
+  it "returns that the accumulated source is not a valid code block" do
+    [
+      ["def foo", "p :ok"],
+      ["class A; def", "foo(x); p x", "end"]
+    ].each do |buffer|
+      IRB::Source.new(buffer).should.not.be.valid
+    end
+  end
+  
+  it "returns the current code block indentation level" do
+    @source.level.should == 0
+    @source << "class A"
+    @source.level.should == 1
+    @source << "  def foo"
+    @source.level.should == 2
+    @source << "    p :ok"
+    @source.level.should == 2
+    @source << "  end"
+    @source.level.should == 1
+    @source << "  class B"
+    @source.level.should == 2
+    @source << "    def bar"
+    @source.level.should == 3
+    @source << "      p :ok; end"
+    @source.level.should == 2
+    @source << "  end; end"
+    @source.level.should == 0
+  end
+  
+  it "caches the reflection when possible" do
+    @source << "def foo"
+    reflection = @source.reflect
+    @source.level
+    @source.valid?
+    @source.reflect.should == reflection
+    
+    @source << "end"
+    @source.level
+    new_reflection = @source.reflect
+    new_reflection.should.not == reflection
+    @source.valid?
+    @source.reflect.should == new_reflection
+  end
+end
+
+describe "IRB::Source::Reflector" do
+  def reflect(source)
+    IRB::Source::Reflector.new(source)
+  end
+  
+  it "returns whether or not the source is a valid code block" do
+    reflect("def foo").should.not.be.valid
+    reflect("def foo; p :ok").should.not.be.valid
+    reflect("def foo; p :ok; end").should.be.valid
+  end
+  
+  it "returns the code block indentation level" do
+    reflect("").level.should == 0
+    reflect("class A").level.should == 1
+    reflect("class A; def foo").level.should == 2
+    reflect("class A; def foo; p :ok").level.should == 2
+    reflect("class A; def foo; p :ok; end").level.should == 1
+    reflect("class A; class B").level.should == 2
+    reflect("class A; class B; def bar").level.should == 3
+    reflect("class A; class B; def bar; p :ok; end").level.should == 2
+    reflect("class A; class B; def bar; p :ok; end; end; end").level.should == 0
+  end
+  
+  it "correctly increases and decreases the code block indentation level for keywords" do
+    [
+      "class A",
+      "module A",
+      "def foo",
+      "begin",
+      "if x == :ok",
+      "unless x == :ko",
+      "case x",
+      "while x",
+      "for x in xs",
+      "x.each do"
+    ].each do |open|
+      reflect(open).level.should == 1
+      reflect("#{open}\nend").level.should == 0
+    end
+  end
+  
+  it "correctly increases and decreases the code block indentation level for literals" do
+    [
+      ["lambda { |x|", "}"],
+      ["{", "}"],
+      ["[", "]"]
+    ].each do |open, close|
+      reflect(open).level.should == 1
+      reflect("#{open}\n#{close}").level.should == 0
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'rubygems'
+require 'bacon'
+
+ROOT = File.expand_path('../../', __FILE__)
+$:.unshift File.join(ROOT, 'lib')
+
+require 'irb'

metadata

@@ -0,0 +1,68 @@
+--- !ruby/object:Gem::Specification 
+name: dietrb
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Eloy Duran
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-07 00:00:00 +01:00
+default_executable: dietrb
+dependencies: []
+
+description: IRB on a diet, for MacRuby / Ruby 1.9
+email: eloy.de.enige@gmail.com
+executables: 
+- dietrb
+extensions: []
+
+extra_rdoc_files: 
+- README
+files: 
+- .gitignore
+- README
+- Rakefile
+- VERSION
+- bin/dietrb
+- dietrb.gemspec
+- lib/irb.rb
+- lib/irb/context.rb
+- lib/irb/source.rb
+- spec/context_spec.rb
+- spec/source_spec.rb
+- spec/spec_helper.rb
+has_rdoc: true
+homepage: http://github.com/alloy/dietrb
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ~>
+    - !ruby/object:Gem::Version 
+      version: "1.9"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: IRB on a diet, for MacRuby / Ruby 1.9
+test_files: 
+- spec/context_spec.rb
+- spec/source_spec.rb
+- spec/spec_helper.rb