pry--pry/lib/pry/commands/gist.rb

103 lines
2.8 KiB
Ruby

class Pry
class Command::Gist < Pry::ClassCommand
match 'gist'
group 'Misc'
description 'Upload code, docs, history to https://gist.github.com/.'
command_options requires_gem: "gist"
banner <<-'BANNER'
Usage: gist [OPTIONS] [--help]
The gist command enables you to gist code from files and methods to github.
gist -i 20 --lines 1..3
gist Pry#repl --lines 1..-1
gist Rakefile --lines 5
BANNER
def setup
require 'gist'
end
def options(opt)
CodeCollector.inject_options(opt)
opt.on :login, "Authenticate the gist gem with GitHub"
opt.on :p, :public, "Create a public gist (default: false)", default: false
opt.on :clip, "Copy the selected content to clipboard instead, do NOT gist it", default: false
end
def process
return ::Gist.login! if opts.present?(:login)
cc = CodeCollector.new(args, opts, _pry_)
if cc.content =~ /\A\s*\z/
raise CommandError, "Found no code to gist."
end
if opts.present?(:clip)
clipboard_content(cc.content)
else
# we're overriding the default behavior of the 'in' option (as
# defined on CodeCollector) with our local behaviour.
content = opts.present?(:in) ? input_content : cc.content
gist_content content, cc.file
end
end
def clipboard_content(content)
::Gist.copy(content)
output.puts "Copied content to clipboard!"
end
def input_content
content = ""
CodeCollector.input_expression_ranges.each do |range|
input_expressions = _pry_.input_ring[range] || []
Array(input_expressions).each_with_index do |code, index|
corrected_index = index + range.first
if code && code != ""
content << code
if code !~ /;\Z/
content << "#{comment_expression_result_for_gist(_pry_.config.gist.inspecter.call(_pry_.output_ring[corrected_index]))}"
end
end
end
end
content
end
def comment_expression_result_for_gist(result)
content = ""
result.lines.each_with_index do |line, index|
if index == 0
content << "# => #{line}"
else
content << "# #{line}"
end
end
content
end
def gist_content(content, filename)
response = ::Gist.gist(content, filename: filename || "pry_gist.rb", public: !!opts[:p])
if response
url = response['html_url']
message = "Gist created at URL #{url}"
begin
::Gist.copy(url)
message << ", which is now in the clipboard."
rescue ::Gist::ClipboardError
end
output.puts message
end
end
end
Pry::Commands.add_command(Pry::Command::Gist)
Pry::Commands.alias_command 'clipit', 'gist --clip'
end