Class: VagrantDNS::Service

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant-dns/service.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(tmp_path) ⇒ Service

Returns a new instance of Service.



8
9
10
# File 'lib/vagrant-dns/service.rb', line 8

def initialize(tmp_path)
  self.tmp_path = tmp_path
end

Instance Attribute Details

#tmp_pathObject

Returns the value of attribute tmp_path.



6
7
8
# File 'lib/vagrant-dns/service.rb', line 6

def tmp_path
  @tmp_path
end

Instance Method Details

#config_fileObject



81
82
83
# File 'lib/vagrant-dns/service.rb', line 81

def config_file
  File.join(tmp_path, "config")
end

#restart!(start_opts = {}) ⇒ Object



55
56
57
58
# File 'lib/vagrant-dns/service.rb', line 55

def restart!(start_opts = {})
  stop!
  start!(start_opts)
end

#run!(run_options) ⇒ Object



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# File 'lib/vagrant-dns/service.rb', line 30

def run!(run_options)
  Daemons.run_proc("vagrant-dns", run_options) do
    require 'rubydns'
    require 'async/dns/system'

    registry = YAML.load(File.read(config_file))
    std_resolver = RubyDNS::Resolver.new(Async::DNS::System.nameservers)

    RubyDNS::run_server(VagrantDNS::Config.listen) do
      registry.each do |pattern, ip|
        match(Regexp.new(pattern), Resolv::DNS::Resource::IN::A) do |transaction, match_data|
          transaction.respond!(ip)
        end
      end

      otherwise do |transaction|
        transaction.passthrough!(std_resolver) do |reply, reply_name|
          puts reply
          puts reply_name
        end
      end
    end
  end
end

#runoptsObject



71
72
73
74
75
76
77
78
79
# File 'lib/vagrant-dns/service.rb', line 71

def runopts
  daemon_dir = File.join(tmp_path, "daemon")
  {
    :dir_mode   => :normal,
    :dir        => daemon_dir,
    :log_output => true,
    :log_dir    => daemon_dir
 }
end

#show_configObject



60
61
62
63
64
65
66
67
68
69
# File 'lib/vagrant-dns/service.rb', line 60

def show_config
  registry = YAML.load(File.read(config_file)) if File.exists?(config_file)
  if registry && registry.any?
    registry.each do |pattern, ip|
      puts format("%s => %s", Regexp.new(pattern).inspect, ip)
    end
  else
    puts "Configuration missing or empty."
  end
end

#start!(opts = {}) ⇒ Object



12
13
14
15
16
17
18
# File 'lib/vagrant-dns/service.rb', line 12

def start!(opts = {})
  run_options = {
    :ARGV => ["start"],
    :ontop => opts[:ontop]
  }.merge!(runopts)
  run!(run_options)
end

#status!Object



25
26
27
28
# File 'lib/vagrant-dns/service.rb', line 25

def status!
  run_options = {:ARGV => ["status"]}.merge(runopts)
  run!(run_options)
end

#stop!Object



20
21
22
23
# File 'lib/vagrant-dns/service.rb', line 20

def stop!
  run_options = {:ARGV => ["stop"]}.merge(runopts)
  run!(run_options)
end