1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/railties/lib/breakpoint_client.rb
2004-12-22 13:17:35 +00:00

172 lines
No EOL
4.8 KiB
Ruby

require 'breakpoint'
require 'optparse'
require 'timeout'
options = {
:ClientURI => nil,
:ServerURI => "druby://localhost:42531",
:RetryDelay => 1,
:Verbose => false
}
ARGV.options do |opts|
script_name = File.basename($0)
opts.banner = [
"Usage: ruby #{script_name} [options] [server uri]",
"",
"This tool lets you connect to a breakpoint service ",
"which was started via Breakpoint.activate_drb.",
"",
"The server uri defaults to druby://localhost:42531"
].join("\n")
opts.separator ""
opts.on("-c", "--client-uri=uri",
"Run the client on the specified uri.",
"This can be used to specify the port",
"that the client uses to allow for back",
"connections from the server.",
"Default: Find a good URI automatically.",
"Example: -c druby://localhost:12345"
) { |options[:ClientURI]| }
opts.on("-s", "--server-uri=uri",
"Connect to the server specified at the",
"specified uri.",
"Default: druby://localhost:42531"
) { |options[:ServerURI]| }
opts.on("-v", "--verbose",
"Report all connections and disconnections",
"Default: false"
) { |options[:Verbose]| }
opts.on("-R", "--retry-delay=delay", Integer,
"Automatically try to reconnect to the",
"server after delay seconds when the",
"connection failed or timed out.",
"A value of 0 disables automatical",
"reconnecting completely.",
"Default: 10"
) { |options[:RetryDelay]| }
opts.separator ""
opts.on("-h", "--help",
"Show this help message."
) { puts opts; exit }
opts.parse!
end
options[:ServerURI] = ARGV[0] if ARGV[0]
$running = true
trap("INT"){$running = false}
puts "Waiting for initial breakpoint..."
loop do
DRb.start_service(options[:ClientURI])
begin
service = DRbObject.new(nil, options[:ServerURI])
begin
timeout(10) { service.ping }
rescue Timeout::Error, DRb::DRbConnError
if options[:Verbose]
puts "",
" *** Breakpoint service didn't respond to ping request ***",
" This likely happened because of a misconfigured ACL (see the",
" documentation of Breakpoint.activate_drb, note that by default",
" you can only connect to a remote Breakpoint service via a SSH",
" tunnel), but might also be caused by an extremely slow connection.",
""
end
raise
end
begin
service.register_eval_handler do |code|
result = eval(code, TOPLEVEL_BINDING)
if result
DRbObject.new(result)
else
result
end
end
service.register_collision_handler do
msg = [
" *** Breakpoint service collision ***",
" Another Breakpoint service tried to use the",
" port already occupied by this one. It will",
" keep waiting until this Breakpoint service",
" is shut down.",
" ",
" If you are using the Breakpoint library for",
" debugging a Rails or other CGI application",
" this likely means that this Breakpoint",
" session belongs to an earlier, outdated",
" request and should be shut down via 'exit'."
].join("\n")
if RUBY_PLATFORM["win"] then
# This sucks. Sorry, I'm not doing this because
# I like funky message boxes -- I need to do this
# because on Windows I have no way of displaying
# my notification via puts() when gets() is still
# being performed on STDIN. I have not found a
# better solution.
begin
require 'tk'
root = TkRoot.new { withdraw }
Tk.messageBox('message' => msg, 'type' => 'ok')
root.destroy
rescue Exception
puts "", msg, ""
end
else
puts "", msg, ""
end
end
service.register_handler do |workspace, message|
puts message
IRB.start(nil, nil, workspace)
puts "", "Resumed execution. Waiting for next breakpoint...", ""
end
puts "Connection established. Waiting for breakpoint...", "" if options[:Verbose]
while $running
begin
service.ping
rescue DRb::DRbConnError => error
puts "Server exited. Closing connection..." if options[:Verbose]
break
end
sleep(0.5)
end
ensure
service.unregister_handler
end
rescue Exception => error
break unless $running
if options[:RetryDelay] > 0 then
puts "No connection to breakpoint service at #{options[:ServerURI]}:", " (#{error.inspect})" if options[:Verbose]
error.backtrace if $DEBUG
puts " Reconnecting in #{options[:RetryDelay]} seconds..." if options[:Verbose]
sleep options[:RetryDelay]
retry
else
raise
end
end
end