pry-command_result 0.0.3 → 0.0.4

data/lib/pry-command_result/kernel_ext.rb

@@ -1,17 +1,31 @@
 require 'stringio'
 
 module Kernel
-  def command_result(command)
-    output = StringIO.new
+  def command_result(command, target = binding)
+    is_command = Pry.commands.to_a.map(&:first).any? do |cmd|
+      case cmd
+      when Regexp
+        cmd =~ command.split(' ').first
+      when String
+        cmd == command.split(' ').first
+      end
+    end
+
     pager = Pry.config.pager
     color = Pry.config.color
     Pry.config.pager = false
     Pry.config.color = false
-    Pry.run_command(command, output: output, show_output: true)
-    output.string
+
+    if is_command
+      output = StringIO.new
+      Pry.run_command(command, output: output, show_output: true)
+      output.string
+    else
+      eval(command, target)
+    end
   ensure
     Pry.config.color = color
     Pry.config.pager = pager
-    output.close
+    output.close if output.respond_to? :close
   end
 end

data/lib/pry-command_result/version.rb

@@ -1,3 +1,3 @@
 module PryCommandResult
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -2,7 +2,7 @@
 name: pry-command_result
 version: !ruby/object:Gem::Version
   prerelease: 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Takatoshi Matsumoto