lib/ukemi/cli.rb in ukemi-0.1.0 vs lib/ukemi/cli.rb in ukemi-0.2.0

@@ -4,18 +4,32 @@
 require "thor"
 
 module Ukemi
   class CLI < Thor
     desc "lookup [IP|DOMAIN]", "Lookup passive DNS services"
+    method_option :order_by, type: :string, desc: "Ordering of the passve DNS resolutions (last_seen or first_seen)", default: "-last_seen"
     def lookup(data)
       data = refang(data)
+      set_ordering options["order_by"]
+
       result = Moderator.lookup(data)
       puts JSON.pretty_generate(result)
     end
 
     no_commands do
       def refang(data)
         data.gsub("[.]", ".").gsub("(.)", ".")
+      end
+
+      def set_ordering(order_by)
+        parts = order_by.split("-")
+        ordering_key = parts.last
+        sort_order = parts.length == 2 ? "DESC" : "ASC"
+
+        Ukemi.configure do |config|
+          config.ordering_key = ordering_key
+          config.sort_order = sort_order
+        end
       end
     end
   end
 end