hutils 0.2.3 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d6377c61f4cb6224f93e56cb1c0c0a3a2ebec5ec
-  data.tar.gz: 2f92ed9f0c685c64604dca8f3c9a924e3ab12f50
+  metadata.gz: ae97daa9f753f4d2ca5e40dac56ae46403d49f13
+  data.tar.gz: ede5236d2a555615a6e4cf90f5386074f403326a
 SHA512:
-  metadata.gz: 7e2e50000c379d0d51d45984ac8fb86f83bccfb175c49858d83dcc636c9bcfcbd4f8afb923178ba1acf1a28ff34b2e691d8750cbd5c486b5081a55b134a5534a
-  data.tar.gz: b7494f90c22b8ae2856804b273450bed52cc7bdda0a5534a40f195526adf4793cee8dd4df21684e20bb47622766d255ff4e23f5f2d0c6467220162e3a64106d7
+  metadata.gz: eda70d9d6e3defea2d158964a9a9c78014d0afadbd19b22d928e40089bc4623ac2cb710b561b112d094abca8cc18264eea0c9e6c9d441050d0a4dd9071c185bd
+  data.tar.gz: 09ef57610dd6dd8edf8ed516f96a5bbc4da8adbda33c2ce6903746d7e4846039df1ee0b8227ed9044994f7265647568ccf5b5b674718d9a85db6a2f4030487d8

data/bin/ltap

@@ -30,8 +30,12 @@ end
 opts.parse!
 
 # load a profile from ~/.ltap if one was specified
-if conf.profile
-  conf.load_section(conf.profile)
+begin
+  if conf.profile
+    conf.load_section(conf.profile)
+  end
+rescue Hutils::Ltap::Conf::SectionNotFound
+  abort("No such section: #{conf.profile}")
 end
 
 unless conf.type
@@ -50,6 +54,8 @@ if !(earliest = parser.parse(conf.earliest, from: Time.now.getutc))
 end
 
 drainer = case conf.type
+when "lvat"
+  Hutils::Ltap::LvatDrainer
 when "papertrail"
   Hutils::Ltap::PaperTrailDrainer
 when "splunk"
@@ -58,15 +64,19 @@ else
   abort("Unknown type: #{conf.type}")
 end
 
-drainer = drainer.new(
-  earliest: earliest,
-  key: conf.key,
-  query: ARGV.first,
-  timeout: conf.timeout,
-  timestamps: conf.timestamps,
-  url: conf.url,
-  verbose: conf.verbose
-)
+begin
+  drainer = drainer.new(
+    earliest: earliest,
+    key: conf.key,
+    query: ARGV.first,
+    timeout: conf.timeout,
+    timestamps: conf.timestamps,
+    url: conf.url,
+    verbose: conf.verbose
+  )
+rescue ArgumentError
+  abort($!.message)
+end
 
 thread = Thread.start do
   $stdout.puts drainer.run

data/lib/hutils/ltap.rb

@@ -1,4 +1,5 @@
 require_relative "ltap/conf"
+require_relative "ltap/lvat_drainer"
 require_relative "ltap/paper_trail_drainer"
 require_relative "ltap/splunk_drainer"
 require_relative "ltap/time_bound_parser"

data/lib/hutils/ltap/conf.rb

@@ -24,16 +24,21 @@ module Hutils::Ltap
     end
 
     def load_section(name)
-      if section = @ini && @ini[name]
-        load_value(section, :earliest)
-        load_value(section, :key)
-        load_value(section, :profile)
-        load_value(section, :timeout)
-        load_value(section, :timestamps)
-        load_value(section, :type)
-        load_value(section, :url)
-        load_value(section, :verbose)
-      end
+      return unless @ini
+      raise SectionNotFound unless @ini.has_section?(name)
+
+      section = @ini[name]
+      load_value(section, :earliest)
+      load_value(section, :key)
+      load_value(section, :profile)
+      load_value(section, :timeout)
+      load_value(section, :timestamps)
+      load_value(section, :type)
+      load_value(section, :url)
+      load_value(section, :verbose)
+    end
+
+    class SectionNotFound < StandardError
     end
 
     private

data/lib/hutils/ltap/lvat_drainer.rb

@@ -0,0 +1,28 @@
+require "json"
+
+module Hutils::Ltap
+  class LvatDrainer
+    def initialize(earliest:, key:, timeout:, query:, timestamps:, url:, verbose:)
+      @query = query
+
+      if timestamps
+        raise ArgumentError, "lvat does not supported `timestamps` option"
+      end
+
+      @api = Excon.new(url, read_timeout: timeout)
+    end
+
+    def run
+      resp = @api.get(
+        path: "/messages",
+        expects: [200, 404],
+        query: {
+          query: @query
+        })
+
+      return [] if resp.status == 404
+
+      JSON.parse(resp.body)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hutils
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.3.0
 platform: ruby
 authors:
 - Brandur
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-20 00:00:00.000000000 Z
+date: 2014-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: excon
@@ -84,6 +84,7 @@ files:
 - "./lib/hutils/curses_visualizer.rb"
 - "./lib/hutils/ltap.rb"
 - "./lib/hutils/ltap/conf.rb"
+- "./lib/hutils/ltap/lvat_drainer.rb"
 - "./lib/hutils/ltap/paper_trail_drainer.rb"
 - "./lib/hutils/ltap/splunk_drainer.rb"
 - "./lib/hutils/ltap/time_bound_parser.rb"