Class: Invoker::Power::OsxSetup

Inherits:
Setup
  • Object
show all
Defined in:
lib/invoker/power/setup/osx_setup.rb

Constant Summary collapse

FIREWALL_PLIST_FILE = 
"/Library/LaunchDaemons/com.codemancers.invoker.firewall.plist"
RESOLVER_DIR = 
"/etc/resolver"

Instance Attribute Summary

Attributes inherited from Setup

#port_finder, #tld

Instance Method Summary collapse

Methods inherited from Setup

#check_if_setup_can_run?, #create_config_file, #drop_to_normal_user, #find_open_ports, #initialize, #install, install, installer_klass, #remove_resolver_file, #safe_remove_file, uninstall

Constructor Details

This class inherits a constructor from Invoker::Power::Setup

Instance Method Details

#build_power_config ⇒ Object 



36
37
38
39
40
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 36

def build_power_config
  config = super
  config[:dns_port] = port_finder.dns_port
  config
end

#firewall_command(http_port, https_port) ⇒ Object

Ripped from Pow code



101
102
103
104
105
106
107
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 101

def firewall_command(http_port, https_port)
  rules = [
    "rdr pass on lo0 inet proto tcp from any to any port 80 -> 127.0.0.1 port #{http_port}",
    "rdr pass on lo0 inet proto tcp from any to any port 443 -> 127.0.0.1 port #{https_port}"
  ].join("\n")
  "echo \"#{rules}\" | pfctl -a 'com.apple/250.InvokerFirewall' -f - -E"
end

#install_firewall(http_port, https_port) ⇒ Object 



49
50
51
52
53
54
55
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 49

def install_firewall(http_port, https_port)
  File.open(FIREWALL_PLIST_FILE, "w") { |fl|
    fl.write(plist_string(http_port, https_port))
  }
  unload_firewall_rule
  load_firewall_rule
end

#install_resolver(dns_port) ⇒ Object 



42
43
44
45
46
47
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 42

def install_resolver(dns_port)
  open_resolver_for_write { |fl| fl.write(resolve_string(dns_port)) }
rescue Errno::EACCES
  Invoker::Logger.puts("Running setup requires root access, please rerun it with sudo".colorize(:red))
  raise
end

#load_firewall_rule ⇒ Object 



57
58
59
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 57

def load_firewall_rule
  system("launchctl load -Fw #{FIREWALL_PLIST_FILE} 2>/dev/null")
end

#plist_string(http_port, https_port) ⇒ Object

Ripped from POW code



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 68

def plist_string(http_port, https_port)
  plist =<<-EOD
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
<key>Label</key>
<string>com.codemancers.invoker</string>
<key>ProgramArguments</key>
<array>
<string>sh</string>
<string>-c</string>
<string>#{firewall_command(http_port, https_port)}</string>
</array>
<key>RunAtLoad</key>
<true/>
<key>UserName</key>
<string>root</string>
</dict>
</plist>
  EOD
  plist
end

#resolve_string(dns_port) ⇒ Object 



92
93
94
95
96
97
98
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 92

def resolve_string(dns_port)
  string =<<-EOD
nameserver 127.0.0.1
port #{dns_port}
  EOD
  string
end

#resolver_file ⇒ Object 



7
8
9
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 7

def resolver_file
  File.join(RESOLVER_DIR, tld)
end

#setup_invoker ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 11

def setup_invoker
  if setup_resolver_file
    find_open_ports
    install_resolver(port_finder.dns_port)
    install_firewall(port_finder.http_port, port_finder.https_port)
    # Before writing the config file, drop down to a normal user
    drop_to_normal_user
    create_config_file
  else
    Invoker::Logger.puts("Invoker is not configured to serve from subdomains".colorize(:red))
  end
  self
end

#setup_resolver_file ⇒ Object 



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 109

def setup_resolver_file
  return true unless File.exist?(resolver_file)

  Invoker::Logger.puts "Invoker has detected an existing Pow installation. We recommend "\
    "that you uninstall pow and rerun this setup.".colorize(:red)
  Invoker::Logger.puts "If you have already uninstalled Pow, proceed with installation"\
    " by pressing y/n."
  replace_resolver_flag = Invoker::CLI::Question.agree("Replace Pow configuration (y/n) : ")

  if replace_resolver_flag
    Invoker::Logger.puts "Invoker has overwritten one or more files created by Pow. "\
    "If .#{tld} domains still don't resolve locally, try turning off the wi-fi"\
    " and turning it on. It'll force OS X to reload network configuration".colorize(:green)
  end
  replace_resolver_flag
end

#uninstall_invoker ⇒ Object 



25
26
27
28
29
30
31
32
33
34
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 25

def uninstall_invoker
  uninstall_invoker_flag = Invoker::CLI::Question.agree("Are you sure you want to uninstall firewall rules created by setup (y/n) : ")

  if uninstall_invoker_flag
    remove_resolver_file
    unload_firewall_rule(true)
    Invoker::Power::Config.delete
    Invoker::Logger.puts("Firewall rules were removed")
  end
end

#unload_firewall_rule(remove = false) ⇒ Object 



61
62
63
64
65
 
# File 'lib/invoker/power/setup/osx_setup.rb', line 61

def unload_firewall_rule(remove = false)
  system("pfctl -a com.apple/250.InvokerFirewall -F nat 2>/dev/null")
  system("launchctl unload -w #{FIREWALL_PLIST_FILE} 2>/dev/null")
  system("rm -rf #{FIREWALL_PLIST_FILE}") if remove
end