Permalink
Fetching contributors…
Cannot retrieve contributors at this time
152 lines (133 sloc) 4.22 KB
require 'socket'
require 'json'
require 'tmpdir'
module Vagrant
module Notify
class Server
HTTP_RESPONSE = "Hi! You just reached the vagrant notification server"
def self.run(id, port, bind_ip, sender_app, sender_params_str, sender_params_escape, machine_name='default', provider='virtualbox')
#id = env[:machine].id
#machine_name = env[:machine].name
#provider = env[:machine].provider_name
#sender_app = env[:machine].config.sender_app
#sender_params_str = env[:machine].config.sender_params_str
#sender_params_escape = env[:machine].config.sender_params_escape
if __FILE__ == $0
begin
tcp_server = TCPServer.open(bind_ip, port)
rescue
exit 1
end
server = self.new(id, sender_app, sender_params_str, sender_params_escape, machine_name, provider)
# Have to wrap this in a begin/rescue block so we can be certain the server is running at all times.
begin
loop {
Thread.start(tcp_server.accept) { |client|
Thread.handle_interrupt(Interrupt => :never) {
server.receive_data(client)
}
}
}
rescue Interrupt
retry
end
end
end
def initialize(id, sender_app, sender_params_str, sender_params_escape, machine_name = :default, provider = :virtualbox)
@id = id
@machine_name = machine_name
@provider = provider
@sender_app = sender_app
@sender_params_str = sender_params_str
@sender_params_escape = sender_params_escape
end
def receive_data(client)
args = read_args(client)
if http_request?(args)
client.puts HTTP_RESPONSE
else
json_data=JSON.parse(args)
parsed_args=map_params_str(json_data)
fix_icon_path! parsed_args
system "#{@sender_app} #{parsed_args}"
end
client.close
rescue => ex
log ex.message
end
private
# Maps params str with values
#
#@param data [Map] Array values map
#
#@return [String]
def map_params_str(data)
cmd=@sender_params_str + ''
cmd.gsub! '%', '%%'
replace=[]
cmd.scan(/\[[^\]]+\]/).each do |part|
variable=part[/\{[^\}]+\}/][1..-2]
if data.key? variable
replace << part[1..-2].sub('{' + variable + '}', escape_param(data[variable]))
cmd.sub! part, '%'+replace.length.to_s+'$s'
else
cmd.sub! part, ''
end
end
cmd.scan(/\{[^\}]+\}/).each do |part|
variable=part[1..-2]
if data.key? variable
replace << escape_param(data[variable])
cmd.sub! part, '%'+replace.length.to_s+'$s'
end
end
cmd % replace
end
def log(message)
File.open(@log_path, 'a+') do |log|
log.puts "#{message}"
end
end
# Escapes param
#
#@param param [String] Param
#
#@return [String]
def escape_param(param)
return param unless @sender_params_escape
'"' + param.gsub('"', "\\\"").gsub("'", "\\'").gsub("\\", "\\\\") + '"'
end
# Gets log path
#
#@return [String]
def log_path
File.join Dir.tmpdir(), "vagrant-notify-error-#{@id}.log"
end
def read_args(client)
''.tap do |args|
while tmp = client.gets and tmp !~ /^\s*$/
args << tmp
end
end
end
def http_request?(args)
args =~ /^GET/
end
def fix_icon_path!(args)
return unless args =~ /-i '([^']+)'/
icon = $1
# TODO: Handle system icons
host_file = "/tmp/vagrant-notify/#{@id}/#{icon.gsub('/', '-')}"
args.gsub!(icon, host_file)
end
end
end
end
# Ghetto
id = ARGV[0]
port = ARGV[1]
bind_ip = ARGV[2]
sender_app = ARGV[3]
sender_params_str = ARGV[4]
sender_params_escape = ARGV[5]
Vagrant::Notify::Server.run id, port, bind_ip, sender_app, sender_params_str, sender_params_escape