barkdog 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9b638cff71a732aae5f42cbd4f60f30e4e674d36
+  data.tar.gz: 89044b32af12f4aee82413d5d8848132582e5479
+SHA512:
+  metadata.gz: b2577c9e00dc2e8758a315bc74cfd25457d395b581c9ec404d6fc388476b44fb2f82d8bf554589eb002c0599b052e9fcda2f1fcbc7a21c86760b24b1250a2c03
+  data.tar.gz: a6bc116e8fc6bffdd0003658713ce24359564959147e71a993768125e77374ad41c1f802713117c916bf31253089c780aa0e017c2eab8e0f2a0fda23a48f5ae6

data/.gitignore

@@ -0,0 +1,17 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log
+test.rb
+Barkfile
+*.bark

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Genki Sugawara
+
+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,64 @@
+# Barkdog
+
+Barkdog is a tool to manage [Datadog monitors](http://docs.datadoghq.com/guides/monitoring/).
+
+It defines Datadog monitors using Ruby DSL, and updates monitors according to DSL.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'barkdog'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install barkdog
+
+## Usage
+
+```sh
+export BARKDOG_API_KEY=...
+export BARKDOG_APP_KEY=...
+
+barkdog -e -o Barkfile
+vi Barkfile
+barkdog -a --dry-run
+barkdog -a
+```
+
+## Help
+
+```
+Usage: barkdog [options]
+        --api-key API_KEY
+        --app-key APP_KEY
+    -a, --apply
+    -f, --file FILE
+        --dry-run
+    -e, --export
+    -o, --output FILE
+        --no-color
+        --debug
+    -h, --help
+```
+
+## Barkfile example
+
+```ruby
+monitor "Check load avg", :type=>"metric alert" do
+  query "avg(last_5m):avg:ddstat.load_avg.1m{host:i-XXXXXXXX} > 1"
+  message "@winebarrel@example.net"
+  options do
+    notify_no_data true
+    no_data_timeframe 2
+    notify_audit true
+    silenced({})
+  end
+end
+```

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler/gem_tasks'
+

data/barkdog.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'barkdog/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'barkdog'
+  spec.version       = Barkdog::VERSION
+  spec.authors       = ['Genki Sugawara']
+  spec.email         = ['sgwr_dts@yahoo.co.jp']
+  spec.summary       = %q{Barkdog is a tool to manage Datadog monitors.}
+  spec.description   = %q{Barkdog is a tool to manage Datadog monitors.}
+  spec.homepage      = 'https://github.com/winebarrel/barkdog'
+  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_dependency 'dogapi'
+  spec.add_dependency 'term-ansicolor'
+
+  spec.add_development_dependency 'bundler'
+  spec.add_development_dependency 'rake'
+end

data/bin/barkdog

@@ -0,0 +1,94 @@
+#!/usr/bin/env ruby
+$: << File.expand_path("#{File.dirname __FILE__}/../lib")
+
+require 'rubygems'
+require 'barkdog'
+require 'optparse'
+
+Version = Barkdog::VERSION
+DEFAULT_FILENAME = 'Barkfile'
+
+mode = nil
+file = DEFAULT_FILENAME
+output_file = '-'
+
+options = {
+  :api_key         => ENV['BARKDOG_API_KEY'],
+  :application_key => ENV['BARKDOG_APP_KEY'],
+  :dry_run         => false,
+  :color           => true,
+  :debug           => false,
+}
+
+ARGV.options do |opt|
+  begin
+    opt.on(''  , '--api-key API_KEY') {|v| options[:api_key]          = v       }
+    opt.on(''  , '--app-key APP_KEY') {|v| options[:application_key]  = v       }
+    opt.on('-a', '--apply')           {    mode                       = :apply   }
+    opt.on('-f', '--file FILE')       {|v| file                       = v        }
+    opt.on(''  , '--dry-run')         {    options[:dry_run]          = true     }
+    opt.on('-e', '--export')          {    mode                       = :export  }
+    opt.on('-o', '--output FILE')     {|v| output_file                = v        }
+    opt.on(''  , '--no-color')        {    options[:color]            = false    }
+    opt.on(''  , '--debug')           {    options[:debug]            = true     }
+
+    opt.on('-h', '--help') do
+      puts opt.help
+      exit 1
+    end
+
+    opt.parse!
+
+    unless mode
+      puts opt.help
+      exit 1
+    end
+
+    if not options[:api_key] or not options[:application_key]
+      raise 'api-key and app-key is required'
+    end
+  rescue => e
+    $stderr.puts("[ERROR] #{e.message}")
+    exit 1
+  end
+end
+
+String.colorize = options[:color]
+
+begin
+  logger = Barkdog::Logger.instance
+  logger.set_debug(options[:debug])
+  client = Barkdog::Client.new(options)
+
+  case mode
+  when :export
+    if output_file == '-'
+      logger.info('# Export Datadog monitors')
+      puts client.export
+    else
+      logger.info("Export Datadog monitors to `#{output_file}`")
+      open(output_file, 'wb') {|f| f.puts client.export }
+    end
+  when :apply
+    unless File.exist?(file)
+      raise "No Barkfile found (looking for: #{file})"
+    end
+
+    msg = "Apply `#{file}` to Datadog monitors"
+    msg << ' (dry-run)' if options[:dry_run]
+    logger.info(msg)
+
+    updated = client.apply(file)
+
+    unless updated
+      logger.info('No change'.intense_blue)
+    end
+  end
+rescue => e
+  if options[:debug]
+    raise e
+  else
+    $stderr.puts("[ERROR] #{e.message}".red)
+    exit 1
+  end
+end

data/lib/barkdog.rb

@@ -0,0 +1,19 @@
+require 'dogapi'
+require 'logger'
+require 'singleton'
+require 'term/ansicolor'
+
+module Barkdog; end
+
+require 'barkdog/ext/string_ext'
+require 'barkdog/constants'
+require 'barkdog/logger'
+require 'barkdog/client'
+require 'barkdog/driver'
+require 'barkdog/dsl'
+require 'barkdog/dsl/context'
+require 'barkdog/dsl/context/monitor'
+require 'barkdog/dsl/context/monitor/options'
+require 'barkdog/dsl/converter'
+require 'barkdog/exporter'
+require 'barkdog/version'

data/lib/barkdog/client.rb

@@ -0,0 +1,83 @@
+class Barkdog::Client
+  include Barkdog::Logger::Helper
+
+  def initialize(options = {})
+    @options = options
+
+    api_key, app_key = @options.values_at(:api_key, :application_key)
+    raise 'API Key does not exist' unless api_key
+    raise 'Application Key does not exist' unless app_key
+
+    @dog = Dogapi::Client.new(api_key, app_key)
+    @driver = Barkdog::Driver.new(@dog, @options)
+  end
+
+  def export(export_options = {})
+    exported = Barkdog::Exporter.export(@dog, @options)
+    Barkdog::DSL.convert(exported, @options)
+  end
+
+  def apply(file)
+    walk(file)
+  end
+
+  private
+
+  def walk(file)
+    expected = load_file(file)
+    actual = Barkdog::Exporter.export(@dog, @options)
+    walk_monitors(expected, actual)
+  end
+
+  def walk_monitors(expected, actual)
+    updated = false
+
+    expected.each do |name, expected_monitor|
+      actual_monitor = actual.delete(name)
+
+      if actual_monitor
+        updated = walk_monitor(name, expected_monitor, actual_monitor) || updated
+      else
+        updated = @driver.create_monitor(name, expected_monitor) || updated
+      end
+    end
+
+    actual.each do |name, actual_monitor|
+      updated = @driver.delete_monitor(name, actual_monitor) || updated
+    end
+
+    updated
+  end
+
+  def walk_monitor(name, expected, actual)
+    updated = false
+
+    Barkdog::FIXED_OPTION_KEYS.each do |key|
+      if expected[key] != actual[key]
+        log(:warn, "#{name}: `#{key}` can not be changed (Changes has been ignored)", :color => :yellow)
+        return updated
+      end
+    end
+
+    actual_without_id = actual.dup
+    monitor_id = actual_without_id.delete('id')
+
+    if expected != actual_without_id
+      updated = @driver.update_monitor(name, expected.merge('id' => monitor_id)) || updated
+    end
+
+    updated
+  end
+
+  def load_file(file)
+    if file.kind_of?(String)
+      open(file) do |f|
+        Barkdog::DSL.parse(f.read, file)
+      end
+    elsif [File, Tempfile].any? {|i| file.kind_of?(i) }
+      Barkdog::DSL.parse(file.read, file.path)
+    else
+      raise TypeError, "can't convert #{file} into File"
+    end
+  end
+end

data/lib/barkdog/constants.rb

@@ -0,0 +1,5 @@
+module Barkdog
+  FIXED_OPTION_KEYS = %w(
+    type
+  )
+end

data/lib/barkdog/driver.rb

@@ -0,0 +1,63 @@
+class Barkdog::Driver
+  include Barkdog::Logger::Helper
+
+  def initialize(dog, options = {})
+    @dog = dog
+    @options = options
+  end
+
+  def create_monitor(name, attrs)
+    updated = false
+    log(:info, "Create Monitor: #{name}", :color => :cyan)
+
+    unless @options[:dry_run]
+      @dog.monitor(
+        attrs['type'],
+        attrs['query'],
+        :name => name,
+        :message => attrs['message'],
+        :options => attrs['options']
+      )
+
+      updated = true
+    end
+
+    updated
+  end
+
+  def delete_monitor(name, attrs)
+    updated = false
+    log(:info, "Delete Monitor: #{name}", :color => :red)
+
+    unless @options[:dry_run]
+      @dog.delete_monitor(attrs['id'])
+      updated = true
+    end
+
+    updated
+  end
+
+  def update_monitor(name, attrs)
+    updated = false
+    log(:info, "Update Monitor: #{name}", :color => :green)
+
+    attrs.each do |key, value|
+      next if key == 'id'
+      log(:info, " set #{key}=#{value}", :color => :green)
+    end
+
+    unless @options[:dry_run]
+      @dog.update_monitor(
+        attrs['id'],
+        attrs['query'],
+        :name => name,
+        :message => attrs['message'],
+        :options => attrs['options']
+      )
+
+      updated = true
+    end
+
+    updated
+  end
+end

data/lib/barkdog/dsl.rb

@@ -0,0 +1,9 @@
+class Barkdog::DSL
+  def self.convert(exported, options = {})
+    Barkdog::DSL::Converter.convert(exported, options)
+  end
+
+  def self.parse(dsl, path, options = {})
+    Barkdog::DSL::Context.eval(dsl, path, options).result
+  end
+end

data/lib/barkdog/dsl/context.rb

@@ -0,0 +1,48 @@
+class Barkdog::DSL::Context
+  def self.eval(dsl, path, options = {})
+    self.new(path, options) {
+      eval(dsl, binding, path)
+    }
+  end
+
+  attr_reader :result
+
+  def initialize(path, options = {}, &block)
+    @path = path
+    @options = options
+    @result = {}
+    instance_eval(&block)
+  end
+
+  private
+
+  def require(file)
+    barkfile = (file =~ %r|\A/|) ? file : File.expand_path(File.join(File.dirname(@path), file))
+
+    if File.exist?(barkfile)
+      instance_eval(File.read(barkfile), barkfile)
+    elsif File.exist?(barkfile + '.rb')
+      instance_eval(File.read(barkfile + '.rb'), barkfile + '.rb')
+    else
+      Kernel.require(file)
+    end
+  end
+
+  def monitor(name, fixed_options = {}, &block)
+    name = name.to_s
+
+    if @result[name]
+      raise "Monitor `#{name}` is already defined"
+    end
+
+    fixed_options.keys.each do |key|
+      unless Barkdog::FIXED_OPTION_KEYS.include?(key.to_s)
+        raise "Monitor `#{name}`: `#{key}` is invalid fixed option key"
+      end
+    end
+
+    fixed_options = Hash[fixed_options.map {|k, v| [k.to_s, v] }]
+    attrs = Barkdog::DSL::Context::Monitor.new(name, &block).result
+    @result[name] = fixed_options.merge(attrs)
+  end
+end

data/lib/barkdog/dsl/context/monitor.rb

@@ -0,0 +1,23 @@
+class Barkdog::DSL::Context::Monitor
+  def initialize(name, &block)
+    @monitor_name = name
+    @result = {}
+    instance_eval(&block)
+  end
+
+  attr_reader :result
+
+  private
+
+  def query(value)
+    @result['query'] = value.to_s
+  end
+
+  def message(value)
+    @result['message'] = value.to_s
+  end
+
+  def options(&block)
+    @result['options'] = Barkdog::DSL::Context::Monitor::Options.new(&block).result
+  end
+end

data/lib/barkdog/dsl/context/monitor/options.rb

@@ -0,0 +1,19 @@
+class Barkdog::DSL::Context::Monitor::Options
+  def initialize(&block)
+    @result = {}
+    instance_eval(&block)
+  end
+
+  attr_reader :result
+
+  private
+
+  def method_missing(method_name, *args)
+    if args.length == 1
+      value = args.first
+      @result[method_name.to_s] = value
+    else
+      super
+    end
+  end
+end

data/lib/barkdog/dsl/converter.rb

@@ -0,0 +1,84 @@
+class Barkdog::DSL::Converter
+  def self.convert(exported, options = {})
+    self.new(exported, options).convert
+  end
+
+  def initialize(exported, options = {})
+    @exported = exported
+    @options = options
+  end
+
+  def convert
+    output_monitors(@exported)
+    #[
+    #  output_users(@exported[:users]),
+    #  output_groups(@exported[:groups]),
+    #  output_roles(@exported[:roles]),
+    #  output_instance_profiles(@exported[:instance_profiles]),
+    #].join("\n")
+  end
+
+  private
+
+  def output_monitors(monitor_by_name)
+    monitor_by_name.sort_by {|k, v| k }.map {|monitor_name, attrs|
+      next unless target_matched?(monitor_name)
+      output_monitor(monitor_name, attrs)
+    }.select {|i| i }.join("\n")
+  end
+
+  def output_monitor(monitor_name, attrs)
+    fixed_options = Hash[Barkdog::FIXED_OPTION_KEYS.map {|k| [k.to_sym, attrs[k]] }]
+    query = attrs['query']
+    message = attrs['message']
+    monitor_options = attrs['options'] || {}
+
+    if monitor_options.empty?
+      monitor_options = ''
+    else
+      monitor_options = "\n" + output_monitor_options(monitor_options)
+    end
+
+    <<-EOS
+monitor #{monitor_name.inspect}, #{unbrace(fixed_options.inspect)} do
+  query #{query.inspect}
+  message #{message.inspect}#{
+  monitor_options}
+end
+    EOS
+  end
+
+  def output_monitor_options(monitor_options)
+    options_body = monitor_options.map {|key, value|
+      value_is_hash = value.is_a?(Hash)
+      value = value.inspect
+
+      if value_is_hash
+        value = unbrace(value).strip
+        value = value.empty? ? '({})' : " #{value}"
+      else
+        value = " #{value}"
+      end
+
+      "#{key}#{value}"
+    }.join("\n    ")
+
+    <<-RUBY.chomp
+  options do
+    #{options_body}
+  end
+    RUBY
+  end
+
+  def unbrace(str)
+    str.sub(/\A\s*\{/, '').sub(/\}\s*\z/, '').strip
+  end
+
+  def target_matched?(name)
+    if @options[:target]
+      name =~ @options[:target]
+    else
+      true
+    end
+  end
+end

data/lib/barkdog/exporter.rb

@@ -0,0 +1,47 @@
+class Barkdog::Exporter
+  EXCLUDE_KEYS = %w(
+    overall_state
+    creator
+    org_id
+    multi
+  )
+
+  class << self
+    def export(dog, opts = {})
+      self.new(dog, opts).export
+    end
+  end # of class methods
+
+  def initialize(dog, options = {})
+    @dog = dog
+    @options = options
+  end
+
+  def export
+    monitors = @dog.get_all_monitors[1]
+    normalize(monitors)
+  end
+
+  private
+
+  def normalize(monitors)
+    monitor_by_name = {}
+
+    monitors.each do |m|
+      name = m.delete('name')
+
+      if monitor_by_name[name]
+        raise "Duplicate monitor name exists: #{name}"
+      end
+
+      EXCLUDE_KEYS.each do |key|
+        m.delete(key)
+      end
+
+      monitor_by_name[name] = m
+    end
+
+    monitor_by_name
+  end
+end
+

data/lib/barkdog/ext/string_ext.rb

@@ -0,0 +1,25 @@
+class String
+  @@colorize = false
+
+  class << self
+    def colorize=(value)
+      @@colorize = value
+    end
+
+    def colorize
+      @@colorize
+    end
+  end # of class methods
+
+  Term::ANSIColor::Attribute.named_attributes.map do |attribute|
+    class_eval(<<-EOS, __FILE__, __LINE__ + 1)
+      def #{attribute.name}
+        if @@colorize
+          Term::ANSIColor.send(#{attribute.name.inspect}, self)
+        else
+          self
+        end
+      end
+    EOS
+  end
+end

data/lib/barkdog/logger.rb

@@ -0,0 +1,30 @@
+class Barkdog::Logger < ::Logger
+  include Singleton
+
+  def initialize
+    super($stdout)
+
+    self.formatter = proc do |severity, datetime, progname, msg|
+      "#{msg}\n"
+    end
+
+    self.level = INFO
+  end
+
+  def set_debug(value)
+    self.level = value ? DEBUG : INFO
+  end
+
+  module Helper
+    def log(level, message, opts = {})
+      opts = (@options || {}).merge(opts)
+
+      message = "[#{level.to_s.upcase}] #{message}" unless level == :info
+      message << ' (dry-run)' if opts[:dry_run]
+      message = message.send(opts[:color]) if opts[:color]
+
+      logger = opts[:logger] || Barkdog::Logger.instance
+      logger.send(level, message)
+    end
+  end
+end

data/lib/barkdog/version.rb

@@ -0,0 +1,3 @@
+module Barkdog
+  VERSION = '0.0.1'
+end

metadata

@@ -0,0 +1,121 @@
+--- !ruby/object:Gem::Specification
+name: barkdog
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Genki Sugawara
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-05-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dogapi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: term-ansicolor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Barkdog is a tool to manage Datadog monitors.
+email:
+- sgwr_dts@yahoo.co.jp
+executables:
+- barkdog
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- barkdog.gemspec
+- bin/barkdog
+- lib/barkdog.rb
+- lib/barkdog/client.rb
+- lib/barkdog/constants.rb
+- lib/barkdog/driver.rb
+- lib/barkdog/dsl.rb
+- lib/barkdog/dsl/context.rb
+- lib/barkdog/dsl/context/monitor.rb
+- lib/barkdog/dsl/context/monitor/options.rb
+- lib/barkdog/dsl/converter.rb
+- lib/barkdog/exporter.rb
+- lib/barkdog/ext/string_ext.rb
+- lib/barkdog/logger.rb
+- lib/barkdog/version.rb
+homepage: https://github.com/winebarrel/barkdog
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.14
+signing_key: 
+specification_version: 4
+summary: Barkdog is a tool to manage Datadog monitors.
+test_files: []