This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
class Pry
|
||||
class Command
|
||||
class SaveFile < Pry::ClassCommand
|
||||
match 'save-file'
|
||||
group 'Input and Output'
|
||||
description 'Export to a file using content from the REPL.'
|
||||
|
||||
banner <<-'BANNER'
|
||||
Usage: save-file [OPTIONS] --to [FILE]
|
||||
|
||||
Export to a file using content from the REPL.
|
||||
|
||||
save-file my_method --to hello.rb
|
||||
save-file -i 1..10 --to hello.rb --append
|
||||
save-file show-method --to my_command.rb
|
||||
save-file sample_file.rb --lines 2..10 --to output_file.rb
|
||||
BANNER
|
||||
|
||||
def options(opt)
|
||||
CodeCollector.inject_options(opt)
|
||||
|
||||
opt.on :to=, "Specify the output file path"
|
||||
opt.on :a, :append, "Append output to file"
|
||||
end
|
||||
|
||||
def process
|
||||
@cc = CodeCollector.new(args, opts, pry_instance)
|
||||
raise CommandError, "Found no code to save." if @cc.content.empty?
|
||||
|
||||
if !file_name
|
||||
display_content
|
||||
else
|
||||
save_file
|
||||
end
|
||||
end
|
||||
|
||||
def file_name
|
||||
opts[:to] || nil
|
||||
end
|
||||
|
||||
def save_file
|
||||
File.open(file_name, mode) do |f|
|
||||
f.puts @cc.content
|
||||
end
|
||||
output.puts "#{file_name} successfully saved"
|
||||
end
|
||||
|
||||
def display_content
|
||||
output.puts @cc.content
|
||||
output.puts "\n\n--\nPlease use `--to FILE` to export to a file."
|
||||
output.puts "No file saved!\n--"
|
||||
end
|
||||
|
||||
def mode
|
||||
opts.present?(:append) ? "a" : "w"
|
||||
end
|
||||
end
|
||||
|
||||
Pry::Commands.add_command(Pry::Command::SaveFile)
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user