Class: RubyYacht::Runner::UpdateHosts

Inherits:
Command
  • Object
show all
Defined in:
lib/ruby_yacht/runner/update_hosts.rb

Overview

This class provides a command for updating the hosts file.

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Command

#backtick, #default_project, #docker, #get_machine_info, #log, #option_parser, #parse_positional_arguments, #project_named, #projects, short_script_name, #system

Class Method Details

.commandObject

The name of the command.


5
# File 'lib/ruby_yacht/runner/update_hosts.rb', line 5

def self.command; 'update_hosts'; end

.descriptionObject

The description of the command.


8
9
10
# File 'lib/ruby_yacht/runner/update_hosts.rb', line 8

def self.description
  "Add entries for your app domains to your hosts file"
end

Instance Method Details

#current_host_contentsObject

The current contents of the hosts file.


13
14
15
# File 'lib/ruby_yacht/runner/update_hosts.rb', line 13

def current_host_contents
  File.read('/etc/hosts')
end

#hosts_file_entries(project, ip_address) ⇒ Object

This method gets the contents of the hosts file for a project.

Parameters

  • project: RubyYacht::Project The project we are working on.
  • ip_address: String The IP address that docker listens on.

Returns

An Array with the new lines for the hosts file.


58
59
60
61
62
63
64
65
66
67
68
69
70
# File 'lib/ruby_yacht/runner/update_hosts.rb', line 58

def hosts_file_entries(project, ip_address)
  system_prefix = project.system_prefix
  main_domain = project.domain

  domains = [main_domain]
  apps = project.apps.select { |app| app.name != system_prefix }
  domains += apps.map { |app| "#{app.name}.#{main_domain}" }

  header = "# #{system_prefix} docker containers"
  new_hosts = ["", header]
  new_hosts += domains.map { |domain| "#{ip_address}    #{domain}" }
  new_hosts
end

#runObject

This method runs the logic of the command.


18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# File 'lib/ruby_yacht/runner/update_hosts.rb', line 18

def run
  ip_address = get_machine_info('.Driver.IPAddress')
  if ip_address == ""
    ip_address = "127.0.0.1"
  end
  
  current_hosts = current_host_contents.split("\n")
  new_hosts = current_hosts.select do |entry|
    projects.none? do |project|
      entry.include?(project.domain) || entry.include?("#{project.system_prefix} docker containers")
    end
  end
  
  new_hosts.pop while new_hosts.any? && new_hosts.last.length == 0
  
  new_hosts += projects.map { |p| self.hosts_file_entries(p, ip_address) }.flatten
  
  FileUtils.mkdir_p('tmp')
  File.open(File.join('tmp', 'hosts'), 'w') do |file|
    file.write(new_hosts.join("\n"))
  end
  
  timestamp = Time.now.utc.strftime('%Y%m%d%H%M%S')
  log "Please enter your password so that we can update the hosts file"
  system "sudo cp /etc/hosts /etc/hosts.#{timestamp}.backup"
  system "sudo mv tmp/hosts /etc/hosts"
  
  true
end