Class: Ddig::Cli

Inherits:
Object
  • Object
show all
Defined in:
lib/ddig/cli.rb

Instance Method Summary collapse

Constructor Details

#initialize(args) ⇒ Cli

Returns a new instance of Cli.



7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/ddig/cli.rb', line 7

def initialize(args)
  @args = args
  @options = {
    dns_type: 'all',
    format: 'text',
  }

  parse_options

  unless valid_options?
    puts @option_parser
    exit
  end
end

Instance Method Details

#exec ⇒ Object 



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/ddig/cli.rb', line 62

def exec
  if @options[:ipv4] || @options[:ipv6]
    @use_ipv4 = @options[:ipv4] || false
    @use_ipv6 = @options[:ipv6] || false
  end

  if @options[:ddr]
    resolve_ddr
    exit
  end

  case @options[:dns_type]
  when "all"
    resolve_all
  when "do53"
    resolve_do53
  when "dot"
    resolve_dot
  when "doh_h1"
    resolve_doh_h1
  end
end

#parse_options ⇒ Object 



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
47
48
 
# File 'lib/ddig/cli.rb', line 22

def parse_options
  @option_parser = OptionParser.new do |opts|
    opts.banner = "Usage: ddig [options] hostname"

    opts.on("-d", "--dns-type={all|do53|dot|doh_h1}", "resolve type (default: all)") { |v| @options[:dns_type] = v }
    opts.on("--udp", "use resolve type of udp(do53)") { |v| @options[:dns_type] = 'do53' }
    opts.on("--dot", "use resolve type of dot") { |v| @options[:dns_type] = 'dot' }
    opts.on("--doh-h1", "use resolve type of doh (http/1.1)") { |v| @options[:dns_type] = 'doh_h1' }
    opts.on("--doh-path=doh-path", "doh service path") { |v| @options[:doh_path] = v }
    opts.on("--ddr", "discover designated resolvers via ddr (discovery of designated resolvers)") { |v| @options[:ddr] = v }
    opts.on("-4", "--ipv4", "use IPv4 query transport only") { |v| @options[:ipv4] = v }
    opts.on("-6", "--ipv6", "use IPv6 query transport only") { |v| @options[:ipv6] = v }
    opts.on("-@", "--nameserver=ipaddress|doh-hostname", "nameserver") { |v| @options[:nameserver] = v }
    opts.on("-p", "--port=port", "port") { |v| @options[:port] = v }
    opts.on("--format={text|json}", "output format (default: text)") { |v| @options[:format] = v }

    opts.separator ""

    opts.on("-v", "--verbose", "run verbosely") { |v| @options[:verbose] = v }

    opts.on("-h", "--help", "show this help message.") { puts opts; exit }
    opts.on("--version", "show version.") { puts Ddig::VERSION; exit }
  end
  @option_parser.parse!

  @hostname = @args[0]
end

#resolve_all ⇒ Object 



85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
 
# File 'lib/ddig/cli.rb', line 85

def resolve_all
  @ddig = Ddig.lookup(@hostname, nameservers: @options[:nameserver], use_ipv4: @use_ipv4, use_ipv6: @use_ipv6)

  if @options[:format] == 'json'
    puts @ddig.to_json

  else
    unless @ddig[:do53][:ipv4].nil?
      puts "# Do53 (IPv4)"
      @ddig[:do53][:ipv4].to_cli
      puts
    end

    unless @ddig[:do53][:ipv6].nil?
      puts "# Do53 (IPv6)"
      @ddig[:do53][:ipv6].to_cli
      puts
    end

    unless @ddig[:ddr].nil?
      puts "# DDR"
      @ddig[:ddr].each_with_index do |designated_resolver, index|
        puts "## DDR (##{index}) - #{designated_resolver.to_s}"
        designated_resolver.to_cli
        puts
      end
    end
  end
end

#resolve_ddr ⇒ Object 



144
145
146
147
148
149
 
# File 'lib/ddig/cli.rb', line 144

def resolve_ddr
  ip = Ddig::Ip.new(use_ipv4: @use_ipv4, use_ipv6: @use_ipv6)
  ddr = Ddig::Ddr.new(nameservers: @options[:nameserver], ip: ip.ip_type)

  ddr.to_cli
end

#resolve_do53 ⇒ Object 



115
116
117
118
119
120
121
122
123
124
125
 
# File 'lib/ddig/cli.rb', line 115

def resolve_do53
  ip = Ddig::Ip.new(use_ipv4: @use_ipv4, use_ipv6: @use_ipv6)
  do53 = Ddig::Resolver::Do53.new(hostname: @hostname, nameservers: @options[:nameserver], ip: ip.ip_type).lookup

  if do53.nil?
    puts "Error: Could not lookup wit nameserver: #{@options[:nameserver]}"
    exit
  end

  do53.to_cli
end

#resolve_doh_h1 ⇒ Object 



133
134
135
136
137
138
139
140
141
142
 
# File 'lib/ddig/cli.rb', line 133

def resolve_doh_h1
  if @options[:nameserver].nil? || @options[:doh_path].nil?
    puts 'ddig: doh needs option of --doh-path=doh-path'
    exit
  end

  doh = Ddig::Resolver::DohH1.new(hostname: @hostname, server: @options[:nameserver], dohpath: @options[:doh_path], port: @options[:port]).lookup

  doh.to_cli
end

#resolve_dot ⇒ Object 



127
128
129
130
131
 
# File 'lib/ddig/cli.rb', line 127

def resolve_dot
  dot = Ddig::Resolver::Dot.new(hostname: @hostname, server: @options[:nameserver], port: @options[:port]).lookup

  dot.to_cli
end

#valid_options? ⇒ Boolean

Returns:

  • (Boolean) 


50
51
52
53
54
55
56
57
58
59
60
 
# File 'lib/ddig/cli.rb', line 50

def valid_options?
  if @hostname.nil?
    if @options[:ddr]
      return true
    end

    return false
  end

  return true
end