diff --git a/lib/irb/color_printer.rb b/lib/irb/color_printer.rb index 31644aa7f..c0961a33f 100644 --- a/lib/irb/color_printer.rb +++ b/lib/irb/color_printer.rb @@ -8,8 +8,8 @@ class ColorPrinter < ::PP METHOD_INSPECT = Object.instance_method(:inspect) class << self - def pp(obj, out = $>, width = screen_width) - q = ColorPrinter.new(out, width) + def pp(obj, out = $>, width = screen_width, colorize: true) + q = ColorPrinter.new(out, width, colorize: colorize) q.guard_inspect_key {q.pp obj} q.flush out << "\n" @@ -24,6 +24,12 @@ def screen_width end end + def initialize(out, width, colorize: true) + @colorize = colorize + + super(out, width) + end + def pp(obj) if String === obj # Avoid calling Ruby 2.4+ String#pretty_print that splits a string by "\n" @@ -46,9 +52,9 @@ def text(str, width = nil) when ',', '=>', '[', ']', '{', '}', '..', '...', /\A@\w+\z/ super(str, width) when /\A#' - super(Color.colorize(str, [:GREEN]), width) + super(@colorize ? Color.colorize(str, [:GREEN]) : str, width) else - super(Color.colorize_code(str, ignore_error: true), width) + super(@colorize ? Color.colorize_code(str, ignore_error: true) : str, width) end end end diff --git a/lib/irb/command/copy.rb b/lib/irb/command/copy.rb new file mode 100644 index 000000000..841593f26 --- /dev/null +++ b/lib/irb/command/copy.rb @@ -0,0 +1,58 @@ +# frozen_string_literal: true + +module IRB + module Command + class Copy < Base + category "Workspace" + description "Copy command output to clipboard" + + help_message(<<~HELP) + Usage: copy (command) + HELP + + def execute(arg) + value = irb_context.workspace.binding.eval(arg) + output = irb_context.inspect_method.inspect_value(value, colorize: false) + + if clipboard_available? + copy_to_clipboard(output) + else + warn "System clipboard not found" + end + rescue StandardError => e + warn "Error: #{e}" + warn "Is IRB.conf[:COPY_COMMAND] set to a bad value?" + end + + private + + def copy_to_clipboard(text) + IO.popen(clipboard_program, 'w') do |io| + io.write(text) + end + + raise IOError.new("Copying to clipboard failed") unless $? == 0 + + puts "Copied to system clipboard" + end + + def clipboard_program + @clipboard_program ||= if IRB.conf[:COPY_COMMAND] + IRB.conf[:COPY_COMMAND] + elsif executable?("pbcopy") + "pbcopy" + elsif executable?("xclip") + "xclip -selection clipboard" + end + end + + def executable?(command) + system("which #{command} > /dev/null 2>&1") + end + + def clipboard_available? + !!clipboard_program + end + end + end +end diff --git a/lib/irb/context.rb b/lib/irb/context.rb index c65628192..0a1b09f49 100644 --- a/lib/irb/context.rb +++ b/lib/irb/context.rb @@ -308,6 +308,8 @@ def irb_path=(path) attr_reader :use_autocomplete # A copy of the default IRB.conf[:INSPECT_MODE] attr_reader :inspect_mode + # Inspector for the current context + attr_reader :inspect_method # A copy of the default IRB.conf[:PROMPT_MODE] attr_reader :prompt_mode diff --git a/lib/irb/default_commands.rb b/lib/irb/default_commands.rb index 768fbee9d..e56aaf7b4 100644 --- a/lib/irb/default_commands.rb +++ b/lib/irb/default_commands.rb @@ -9,6 +9,7 @@ require_relative "command/chws" require_relative "command/context" require_relative "command/continue" +require_relative "command/copy" require_relative "command/debug" require_relative "command/delete" require_relative "command/disable_irb" @@ -249,6 +250,7 @@ def load_command(command) ) register(:cd, Command::CD) + register(:copy, Command::Copy) end ExtendCommand = Command diff --git a/lib/irb/init.rb b/lib/irb/init.rb index d474bd41d..695649e81 100644 --- a/lib/irb/init.rb +++ b/lib/irb/init.rb @@ -194,6 +194,8 @@ def IRB.init_config(ap_path) :'$' => :show_source, :'@' => :whereami, } + + @CONF[:COPY_COMMAND] = ENV.fetch("IRB_COPY_COMMAND", nil) end def IRB.set_measure_callback(type = nil, arg = nil, &block) diff --git a/lib/irb/inspector.rb b/lib/irb/inspector.rb index 8046744f8..a1679b20f 100644 --- a/lib/irb/inspector.rb +++ b/lib/irb/inspector.rb @@ -93,8 +93,8 @@ def init end # Proc to call when the input is evaluated and output in irb. - def inspect_value(v) - @inspect.call(v) + def inspect_value(v, colorize: true) + @inspect.call(v, colorize: colorize) rescue => e puts "An error occurred when inspecting the object: #{e.inspect}" @@ -110,11 +110,11 @@ def inspect_value(v) end Inspector.def_inspector([false, :to_s, :raw]){|v| v.to_s} - Inspector.def_inspector([:p, :inspect]){|v| - Color.colorize_code(v.inspect, colorable: Color.colorable? && Color.inspect_colorable?(v)) + Inspector.def_inspector([:p, :inspect]){|v, colorize: true| + Color.colorize_code(v.inspect, colorable: colorize && Color.colorable? && Color.inspect_colorable?(v)) } - Inspector.def_inspector([true, :pp, :pretty_inspect], proc{require_relative "color_printer"}){|v| - IRB::ColorPrinter.pp(v, +'').chomp + Inspector.def_inspector([true, :pp, :pretty_inspect], proc{require_relative "color_printer"}){|v, colorize: true| + IRB::ColorPrinter.pp(v, +'', colorize: colorize).chomp } Inspector.def_inspector([:yaml, :YAML], proc{require "yaml"}){|v| begin diff --git a/test/irb/command/test_copy.rb b/test/irb/command/test_copy.rb new file mode 100644 index 000000000..c35df6c3e --- /dev/null +++ b/test/irb/command/test_copy.rb @@ -0,0 +1,70 @@ +# frozen_string_literal: true + +require 'irb' + +require_relative "../helper" + +module TestIRB + class CopyTest < IntegrationTestCase + def setup + super + @envs['IRB_COPY_COMMAND'] = "ruby -e \"puts 'foo' + STDIN.read\"" + end + + def test_copy_with_pbcopy + write_ruby <<~'ruby' + class Answer + def initialize(answer) + @answer = answer + end + end + + binding.irb + ruby + + output = run_ruby_file do + type "copy Answer.new(42)" + type "exit" + end + + assert_match(/foo# "1\n", + "a\nb" => %["a\\nb"\n], + IRBTestColorPrinter.new('test') => "#\n", + Ripper::Lexer.new('1').scan => "[#]\n", + Class.new{define_method(:pretty_print){|q| q.text("[__FILE__, __LINE__, __ENCODING__]")}}.new => "[__FILE__, __LINE__, __ENCODING__]\n", + }.each do |object, result| + actual = with_term { IRB::ColorPrinter.pp(object, '', colorize: false) } + assert_equal(result, actual, "Case: IRB::ColorPrinter.pp(#{object.inspect}, '')") + end + end + private def with_term