Class: Vagrant::Notify::Action::StartServer

Inherits:
Object
  • Object
show all
Includes:
Util::IsPortOpen
Defined in:
lib/vagrant-notify/action/start_server.rb

Instance Method Summary collapse

Constructor Details

#initialize(app, env) ⇒ StartServer

Returns a new instance of StartServer.



11
12
13
 
# File 'lib/vagrant-notify/action/start_server.rb', line 11

def initialize(app, env)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



15
16
17
18
19
20
21
22
23
 
# File 'lib/vagrant-notify/action/start_server.rb', line 15

def call(env)
  @env = env

  port = next_available_port
  env[:notify_data][:pid]  = Server.run(env, port)
  env[:notify_data][:port] = port

  @app.call env
end

#next_available_port ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/vagrant-notify/action/start_server.rb', line 25

def next_available_port
  # Determine a list of usable ports for us to use
  usable_ports = Set.new(@env[:machine].config.vm.usable_port_range)

  # Pass one, remove all defined host ports from usable ports
  with_forwarded_ports do |options|
    usable_ports.delete(options[:host])
  end

  # Pass two, remove ports used by other processes
  with_forwarded_ports do |options|
    host_port = options[:host]
    usable_ports.delete(options[:host]) if is_port_open?("127.0.0.1", host_port)
  end

  # If we have no usable ports then we can't use the plugin
  raise 'No usable ports available!' if usable_ports.empty?

  # Set the port up to be the last one since vagrant's port collision handler
  # will use the first as in:
  #   https://github.com/mitchellh/vagrant/blob/master/lib/vagrant/action/builtin/handle_forwarded_port_collisions.rb#L84
  usable_ports.to_a.sort.reverse.find do |port|
    return port unless is_port_open?("127.0.0.1", port)
  end
end

#with_forwarded_ports ⇒ Object 



51
52
53
54
55
56
57
58
 
# File 'lib/vagrant-notify/action/start_server.rb', line 51

def with_forwarded_ports
  @env[:machine].config.vm.networks.each do |type, options|
    # Ignore anything but forwarded ports
    next if type != :forwarded_port

    yield options
  end
end