Merge pull request #9 from ninoseki/add-ping

feat: add Ping class

lib/rogue_one.rb

@@ -4,6 +4,7 @@
 
 require "rogue_one/resolver"
 require "rogue_one/detector"
+require "rogue_one/ping"
 require "rogue_one/cli"
 
 module RogueOne

lib/rogue_one/cli.rb

@@ -8,6 +8,8 @@ class CLI < Thor
     desc "report [DNS_SERVER]", "Show a report of a given DNS server"
     def report(dns_server)
       with_error_handling do
+        Ping.pong? dns_server
+
         detector = Detector.new(target: dns_server)
         puts JSON.pretty_generate(detector.report)
       end
@@ -17,7 +19,8 @@ def report(dns_server)
       def with_error_handling
         yield
       rescue StandardError => e
-        puts "Warning: #{e}"
+        message = { error: e.to_s }
+        puts JSON.pretty_generate(message)
       end
     end
   end

lib/rogue_one/ping.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module RogueOne
+  class Ping
+    attr_reader :resolver
+
+    def initialize(nameserver)
+      @resolver = Resolver.new(nameserver: nameserver)
+    end
+
+    def pong?
+      result = resolver.dig("example.com", "A")
+      raise Error, "DNS resolve error: there is no resopnse from #{resolver.nameserver}" unless result
+
+      true
+    end
+
+    def self.pong?(target)
+      new(target).pong?
+    end
+  end
+end

lib/rogue_one/resolver.rb

@@ -20,6 +20,8 @@ def dig(domain, type)
 
     def _resolver
       @_resolver ||= Resolv::DNS.new(nameserver: [nameserver])
+      @_resolver.timeouts = 5
+      @_resolver
     end
 
     def resource_by_type(type)

spec/ping_spec.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+RSpec.describe RogueOne::Ping do
+  subject { described_class.new("1.1.1.1") }
+
+  describe "#pong?" do
+    it do
+      expect(subject.pong?).to eq(true)
+    end
+  end
+end