riemann-tail 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in riemann-tail.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Will Farrell
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,72 @@
+# Riemann::Tail
+
+Subscribe to (tail) [Riemann](http://riemann.io "Riemann") event streams from the console.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'riemann-tail'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install riemann-tail
+
+## Usage
+
+`riemann-tail --help`
+
+    Usage:
+      riemann-tail QUERY
+
+    Options:
+      h, [--host=HOST]                    # Riemann host
+                                          # Default: localhost
+      p, [--port=PORT]                    # Riemann port number
+                                          # Default: 5556
+      c, [--config=CONFIG]                # Path to config file
+                                          # Default: riemann-tail-config.rb
+      f, [--from-file], [--no-from-file]  # Read query from file instead of command
+
+    Subscribe to (tail) a riemann event stream based on QUERY
+
+Example configuration file (provided as `riemann-tail-config.rb`)
+
+    Riemann::Tail.configure do |c|
+      c.format :host, :blue
+      c.format :metric, [:to_s, :cyan]
+
+      c.format :tags, :light_black do |t|
+        t.join(", ")
+      end
+
+      c.format :time, :light_black do |t|
+        t.localtime.strftime("%T")
+      end
+
+      c.format :state, [:downcase, :bold] do |s|
+        case s
+        when /fail|err/i then s.red
+        when /warn/i then s.yellow
+        else s.green
+        end
+      end
+
+      c.line_format = "%{time}#{">".light_black} %{state} %{host} %{service} %{tags} %{metric}\n        #{">".light_black} %{description}"
+    end
+
+Example query file (provided as `query.clj`)
+
+    tagged "log"
+
+## Contributing
+
+1. Fork it ( http://github.com/<my-github-username>/riemann-tail/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/riemann-tail

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+
+require 'thor'
+require 'riemann/tail'
+
+Signal.trap 'SIGINT' do
+  puts "interrupted..."
+  Process.exit 1
+end
+
+class Riemann::Tail::CLI < Thor::Group
+  namespace ""
+  desc "Subscribe to (tail) a riemann event stream based on QUERY"
+  argument :query
+  class_option :host, :aliases => :h, :default => "localhost", :desc => "Riemann host"
+  class_option :port, :aliases => :p, :default => "5556", :desc => "Riemann port number"
+  class_option :config, :aliases => :c, :default => "riemann-tail-config.rb", :desc => "Path to config file"
+  class_option :from_file, :aliases => :f, :type => :boolean, :desc => "Read query from file instead of command"
+  def tail
+    require File.expand_path(options["config"])
+
+    Riemann::Tail.run_query(options["from_file"] ? File.read(query) : query, options)
+  end
+end
+
+Riemann::Tail::CLI.start(ARGV)

data/lib/riemann/tail.rb

@@ -0,0 +1,52 @@
+require 'uri'
+require 'json'
+require 'date'
+require 'colorize'
+require 'eventmachine'
+require 'faye/websocket'
+
+require "riemann/tail/config"
+require "riemann/tail/version"
+
+module Riemann
+  module Tail
+    def self.configure(&block)
+      yield @config = Config.new
+    end
+
+    def self.run_query(query, options)
+      url = "ws://#{options["host"]}:#{options["port"]}/index?subscribe=true&query=#{URI.encode(query)}"
+
+      EM.run do
+        ws = Faye::WebSocket::Client.new(url)
+
+        ws.on :error do |e|
+          puts "error... #{e.to_s}".red
+          Process.exit 1
+        end
+
+        ws.on :close do |e|
+          Process.exit 0
+        end
+
+        ws.on :message do |m|
+          event = JSON.parse(m.data).reduce(@config.defaults.clone) do |o, (k, v)|
+            if k == "time"
+              v = DateTime.parse(v).to_time
+            end
+
+            if fs = @config.formatters[k.to_sym]
+              v = fs[:block].call(v) if fs[:block]
+              v = fs[:methods].reduce(v) { |r, s| r.send(s) } if fs[:methods]
+            end
+
+            o[k.to_sym] = v
+            o
+          end
+
+          puts @config.line_format % event
+        end
+      end
+    end
+  end
+end

data/lib/riemann/tail/config.rb

@@ -0,0 +1,31 @@
+module Riemann
+  module Tail
+    class Config
+      def initialize
+        @defaults = {}
+        @formatters = {}
+      end
+
+      attr_accessor :line_format, :formatters, :defaults
+
+      def format(field, format, &block)
+        @formatters[field.to_sym] = {
+          methods: (format.is_a?(Array) ? format : [format.to_sym])
+        }
+        @formatters[field.to_sym][:block] = block if block_given?
+      end
+
+      def default(field, value)
+        @defaults[field.to_sym] = value.to_s
+      end
+
+      def line_format=(format)
+        # hacky way to prevent sprintf (%) from throwing key errors
+        format.scan(/%{([^}]+)}|%<([^>])>/).each do |(m, n)|
+          @defaults[(m||n).to_sym] ||= ""
+        end
+        @line_format = format
+      end
+    end
+  end
+end

data/lib/riemann/tail/version.rb

@@ -0,0 +1,5 @@
+module Riemann
+  module Tail
+    VERSION = "0.0.1"
+  end
+end

data/query.clj

@@ -0,0 +1 @@
+tagged "log"

data/riemann-tail-config.rb

@@ -0,0 +1,22 @@
+Riemann::Tail.configure do |c|
+  c.format :host, :blue
+  c.format :metric, [:to_s, :cyan]
+
+  c.format :tags, :light_black do |t|
+    t.join(", ")
+  end
+
+  c.format :time, :light_black do |t|
+    t.localtime.strftime("%T")
+  end
+
+  c.format :state, [:downcase, :bold] do |s|
+    case s
+    when /fail|err/i then s.red
+    when /warn/i then s.yellow
+    else s.green
+    end
+  end
+
+  c.line_format = "%{time}#{">".light_black} %{state} %{host} %{service} %{tags} %{metric}\n        #{">".light_black} %{description}"
+end

data/riemann-tail.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'riemann/tail/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "riemann-tail"
+  spec.version       = Riemann::Tail::VERSION
+  spec.authors       = ["Will Farrell"]
+  spec.email         = ["will@mojotech.com"]
+  spec.summary       = %q{Subscribe (tail) Riemann event streams from the console}
+  spec.description   = spec.summary
+  spec.homepage      = "https://github.com/wkf/riemann-tail"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency 'thor', '~> 0'
+  spec.add_runtime_dependency 'colorize', '~> 0'
+  spec.add_runtime_dependency 'faye-websocket', '~> 0'
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+end

metadata

@@ -0,0 +1,139 @@
+--- !ruby/object:Gem::Specification
+name: riemann-tail
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Will Farrell
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-03-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faye-websocket
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Subscribe (tail) Riemann event streams from the console
+email:
+- will@mojotech.com
+executables:
+- riemann-tail
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/riemann-tail
+- lib/riemann/tail.rb
+- lib/riemann/tail/config.rb
+- lib/riemann/tail/version.rb
+- query.clj
+- riemann-tail-config.rb
+- riemann-tail.gemspec
+homepage: https://github.com/wkf/riemann-tail
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23.2
+signing_key: 
+specification_version: 3
+summary: Subscribe (tail) Riemann event streams from the console
+test_files: []