chihuahua 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 17b3368c2f4e8d6e09f7f09ffd9be09b495af7e2
+  data.tar.gz: d3e1d6d403c878481b803d5fa0e05528c4f6a584
+SHA512:
+  metadata.gz: f59911e85a752e51ace8e4dad46440264aa2faa6c74a6a382fa0d2c1c5a585fdd2cb65e829f1961dceaa18e115abc77b69d2b4bd381ec67955eee1ba96ef1b5a
+  data.tar.gz: 918100d36e89194f47df7c28fe823eb39375016af61f48825ee744a798170f508877d2b6706e9f7aaa391dc56fc6d595b24623a43a212a88c2a12f593ff95919

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/
+/monitors/

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2017 Yohei Kawahara
+
+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,193 @@
+# Chihuahua
+
+![chihuahua](https://raw.githubusercontent.com/inokappa/chihuahua/master/images/chihuahua.png)
+
+Chihuahua は [Datadog monitors](http://docs.datadoghq.com/guides/monitoring/) を管理するツールです.
+
+## Caution
+
+* Datadog monitors を YAML DSL で管理してみる試みです
+* Monitor Name や Monitor Tags で絞り込んで任意の Monitor のみ抽出して管理出来る筈です
+* 基本的には [codenize-tools/barkdog](https://github.com/codenize-tools/barkdog) を使いましょう
+
+## Installation
+
+```sh
+git clone ...
+cd chihuahua
+bundle install --path vendor/bundle
+```
+
+## Usage
+
+```sh
+export DATADOG_API_KEY=...
+export DATADOG_APP_KEY=...
+
+bundle exec ./bin/chihuahua init
+bundle exec ./bin/chihuahua export --project=your_project_name --tags=project:foo,stage:production
+vi ./monitors/your_project_name/monitors.yml
+bundle exec ./bin/chihuahua apply --project=your_project_name --dry-run
+bundle exec ./bin/chihuahua apply --project=your_project_name
+```
+
+## Help
+
+```
+$ bundle exec ./bin/chihuahua --help
+Commands:
+  chihuahua apply           # Monitor 設定を apply する
+  chihuahua export          # Monitor 設定を export する
+  chihuahua help [COMMAND]  # Describe available commands or one specific command
+  chihuahua init            # Project の Root ディレクトリ(./monitors)を作成する.
+  chihuahua version         # version 情報を出力.
+```
+
+## Chihuahua example
+
+### 書き出し
+
+```sh
+#
+# 初回の書き出し
+#
+$ bundle exec ./bin/chihuahua export --project=foo --tags=host:vagrant-ubuntu-trusty-64
+Export...
+4 monitors output done.
+
+$ tree -a monitors/
+monitors/
+└── foo
+    ├── .filter.yml
+    └── monitors.yml
+
+1 directory, 2 files
+
+#
+# 2 回目以降、絞込の条件（--name や --tags に変更が無い場合）
+#
+$ bundle exec ./bin/chihuahua export --project=foo
+
+```
+
+
+
+### 新規作成
+
+- YAML 定義を ./monitors/your_project_name/monitors.yml に追記
+
+```yaml
+- query: avg(last_1m):avg:system.load.5{host:vagrant-ubuntu-trusty-64} > 1
+  message: |-
+    CPU load is very high on {{host.name}}
+    @slack-datadog-notification
+  name: Test 5 [{{#is_alert}}CRITICAL{{/is_alert}}{{#is_warning}}WARNING{{/is_warning}}]
+    CPU load is very high on {{host.name}}
+  type: metric alert
+  options:
+    thresholds:
+      critical: 1.0
+      warning: 0.8
+```
+
+詳細は http://docs.datadoghq.com/ja/api/?lang=ruby#monitor-create を御確認ください.
+
+- dry-run
+
+```sh
+$ bundle exec ./bin/chihuahua apply --project=foo --dry-run
+Apply...(dry-run)
+Check add line.
+---
+query: avg(last_1m):avg:system.load.5{host:vagrant-ubuntu-trusty-64} > 1
+message: |-
+  CPU load is very high on {{host.name}}
+  @slack-datadog-notification
+name: Test 5 [{{#is_alert}}CRITICAL{{/is_alert}}{{#is_warning}}WARNING{{/is_warning}}]
+  CPU load is very high on {{host.name}}
+type: metric alert
+options:
+  thresholds:
+    critical: 1.0
+    warning: 0.8
+```
+
+- apply
+
+```sh
+$ bundle exec ./bin/chihuahua apply --project=foo
+Apply...
+Add line.
+...
+... Latest Monitors List ...
+...
+done.
+```
+
+### 更新
+
+- thresholds を追記
+
+```yaml
+- tags: []
+  query: avg(last_1m):avg:system.load.5{host:vagrant-ubuntu-trusty-64} > 1
+  message: |-
+    CPU load is very high on {{host.name}}
+    @slack-datadog-notification
+  id: 12345678
+  name: Test3 [{{#is_alert}}CRITICAL{{/is_alert}}{{#is_warning}}WARNING{{/is_warning}}]
+    CPU load is very high on {{host.name}}
+  type: metric alert
+  options:
+    notify_audit: false
+    locked: false
+    silenced: {}
+    new_host_delay: 300
+    require_full_window: true
+    notify_no_data: false
+    thresholds:
+      critical: 1.0
+      warning: 0.8
+```
+
+詳細は http://docs.datadoghq.com/ja/api/?lang=ruby#monitor-edit を御確認ください.
+
+- dry-run
+
+```sh
+$ bundle exec ./bin/chihuahua apply --project=foo --dry-run
+Apply...(dry-run)
+Check update line.
+ ---
+ tags: []
+ query: avg(last_1m):avg:system.load.5{host:vagrant-ubuntu-trusty-64} > 1
+ message: |-
+   CPU load is very high on {{host.name}}
+   @slack-datadog-notification
+ id: 12345678
+ name: Test3 [{{#is_alert}}CRITICAL{{/is_alert}}{{#is_warning}}WARNING{{/is_warning}}]
+   CPU load is very high on {{host.name}}
+ type: metric alert
+ options:
+   notify_audit: false
+   locked: false
+   silenced: {}
+   new_host_delay: 300
+   require_full_window: true
+   notify_no_data: false
++  thresholds:
++    critical: 1.0
++    warning: 0.8
+```
+
+- apply
+
+```sh
+$ bundle exec ./bin/chihuahua apply --project=foo
+Apply...
+Update line.
+...
+... Latest Monitors List ...
+...
+done.
+```

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "chihuahua"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/chihuahua.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'chihuahua/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "chihuahua"
+  spec.version       = Chihuahua::VERSION
+  spec.authors       = ["kapp"]
+  spec.email         = ["inokara@gmail.com"]
+
+  spec.summary       = %q{Chihuahua is a tool to manage Datadog monitors.}
+  spec.description   = %q{Chihuahua is a tool to manage Datadog monitors.}
+  spec.homepage      = "https://github.com/inokappa/chihuahua"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.11"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_dependency 'thor'
+  spec.add_dependency 'diffy'
+  spec.add_dependency 'dogapi'
+  spec.add_dependency 'highline'
+end

data/exe/chihuahua

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'chihuahua'
+
+Chihuahua::CLI.start

data/lib/chihuahua.rb

@@ -0,0 +1,10 @@
+require "chihuahua/version"
+require "chihuahua/cli"
+require "chihuahua/client"
+require "chihuahua/common"
+require "chihuahua/export"
+require "chihuahua/update"
+
+module Chihuahua
+  # Your code goes here...
+end

data/lib/chihuahua/cli.rb

@@ -0,0 +1,66 @@
+# coding: utf-8
+require 'thor'
+require 'dogapi'
+require 'yaml'
+require 'json'
+require 'diffy'
+require 'highline'
+
+module Chihuahua
+  class CLI < Thor
+    default_command :version
+    # class_option :api_key
+    # class_option :app_key
+
+    desc 'version', 'version 情報を出力.'
+    def version
+      puts Chihuahua::VERSION
+    end
+
+    desc 'export', 'Monitors 定義を export する'
+    option :project, type: :string, aliases: '-p', desc: 'Project を指定.(Project 名でディレクトリを作成する)'
+    option :name, type: :string, aliases: '-n', desc: 'Monitor を name キーで絞り込む.'
+    option :tags, type: :array, aliases: '-t', desc: 'Monitor を tags キーで絞り込む.'
+    def export
+      unless options[:project] then
+        puts 'Project 名がセットされていません. (--project xxxxxx)'
+        exit 1
+      end
+
+      puts 'Export...'
+      # exporter = Chihuahua::Export.new(options[:project])
+      args = {}
+      args['name'] = options[:name]
+      args['tags'] = options[:tags]
+      p args.class
+      exporter = Chihuahua::Export.new(options[:project], args)
+      monitors_data = exporter.export_monitors(options[:project])
+      exporter.store_monitors_data(monitors_data)
+    end
+#
+    desc 'apply', 'Monitors 定義を apply する'
+    option :project, type: :string, aliases: '-p', desc: 'Project を指定.'
+    option :dry_run, type: :boolean, aliases: '-d', desc: 'apply 前の試行.'
+    def apply
+      if options[:dry_run] then
+        puts 'Apply...(dry-run)'
+      else
+        puts 'Apply...'
+      end
+      updater = Chihuahua::Update.new(options[:project])
+      updater.update_monitors(options[:dry_run])
+    end
+
+    desc 'generate', 'Monitors 定義の雛形を生成する.'
+    option :project, type: :string, aliases: '-p', desc: 'Project を指定.'
+    def generate
+      #
+    end
+
+    desc 'init', 'Project の Root ディレクトリ(./monitors)を作成する.'
+    def init
+      Chihuahua::Helper.create_project_root_dir
+    end
+
+  end
+end

data/lib/chihuahua/client.rb

@@ -0,0 +1,16 @@
+# coding: utf-8
+module Chihuahua
+  class Client
+    def initialize
+      raise 'API Key がセットされていません.' unless ENV['DATADOG_API_KEY']
+      raise 'Application Key がセットされていません.' unless ENV['DATADOG_APP_KEY']
+      @api_key = ENV['DATADOG_API_KEY']
+      @app_key = ENV['DATADOG_APP_KEY']
+    end
+
+    def dog
+      Dogapi::Client.new(@api_key, @app_key)
+    end
+
+  end
+end

data/lib/chihuahua/common.rb

@@ -0,0 +1,29 @@
+# coding: utf-8
+module Chihuahua
+  module Helper
+    def create_project_root_dir
+      FileUtils.mkdir_p('./monitors') unless FileTest.exist?('./monitors')
+      puts 'done.'
+    end
+
+    def filter_monitor(monitor)
+      filter = %w[tags query type message id name options]
+      filterd_monitor = monitor.select { |key, _| filter.include? key }
+      return filterd_monitor
+    end
+
+    def get_filter(project)
+      filter = open('./monitors/' + project + '/.filter.yml', 'r') { |f| YAML.load(f) }
+      return filter
+    end
+
+    def hl
+      HighLine.new
+    end
+
+    module_function :create_project_root_dir
+    module_function :filter_monitor
+    module_function :get_filter
+    module_function :hl
+  end
+end

data/lib/chihuahua/export.rb

@@ -0,0 +1,73 @@
+# coding: utf-8
+module Chihuahua
+  class Export
+
+    include Chihuahua::Helper
+
+    def initialize(project, *args)
+      @dog = Chihuahua::Client.new.dog
+      @project = project
+      @args = args.last
+      @project_dir = './monitors/' + @project
+      @monitors_file_path = @project_dir + '/monitors.yml'
+      @filter_file_path = @project_dir + '/.filter.yml'
+    end
+
+    def export_result_display(name)
+      puts hl.color(name, :magenta) + ' を export しました.'
+    end
+
+    def export_monitors(project)
+      call_from = caller[0][/`([^']*)'/, 1]
+
+      if FileTest.exist?(@filter_file_path) && @args == nil then
+        filter = get_filter(project)
+        name = filter['name']
+        tags = filter['tags']
+      else
+        if FileTest.exist?(@filter_file_path) && @args['name'] == nil && @args['tags'] == nil then
+          filter = get_filter(@project)
+          name = filter['name']
+          tags = filter['tags']
+        else
+          name = @args['name']
+          tags = @args['tags']
+        end
+      end
+
+      filterd_monitors = []
+      begin
+        @dog.get_all_monitors({:name => name, :tags => tags}).last.each do |monitor|
+          export_result_display(monitor['name']) unless call_from == 'update_monitors'
+          filterd_monitors << filter_monitor(monitor)
+        end
+      rescue => e
+        puts e
+      end
+      return filterd_monitors
+    end
+
+    def store_monitors_data(monitors_data)
+      FileUtils.mkdir_p(@project_dir) unless FileTest.exist?(@project_dir)
+
+      unless FileTest.exist?(@filter_file_path) and @args == nil or \
+        FileTest.exist?(@filter_file_path) && @args['name'] == nil && @args['tags'] == nil then
+        filter = {}
+        filter['name'] = @args['name']
+        filter['tags'] = @args['tags']
+        begin
+          YAML.dump(filter, File.open(@filter_file_path, 'w'))
+        rescue => e
+          puts e
+        end
+      end
+
+      begin
+        YAML.dump(monitors_data, File.open(@monitors_file_path, 'w'))
+      rescue => e
+        puts e
+      end
+      puts monitors_data.length.to_s + ' 件の Monitors 定義を出力しました.'
+    end
+  end
+end

data/lib/chihuahua/update.rb

@@ -0,0 +1,83 @@
+# coding: utf-8
+module Chihuahua
+  class Update
+
+    include Chihuahua::Helper
+
+    def initialize(project)
+      @dog = Chihuahua::Client.new.dog
+      @exporter = Chihuahua::Export.new(project)
+      @project = project
+      @project_dir = './monitors/' + @project
+      @monitors_file_path = @project_dir + '/monitors.yml'
+      @filter_file_path = @project_dir + '/.filter.yml'
+    end
+
+    def apply_result_display(res)
+      result = res.last
+      puts hl.color(result['name'], :light_cyan) + ' を apply しました.'
+    end
+
+    def update_monitor(data)
+      puts hl.color('Update line.', :yellow, :underline)
+      begin
+        res = @dog.update_monitor(data['id'], data['query'], :message => data['message'], :name => data['name'], :options => data['options'])
+      rescue => e
+        puts e
+      end
+      apply_result_display(res)
+    end
+
+    def create_monitor(data)
+      puts hl.color('Add line.', :yellow, :underline)
+      begin
+        res = @dog.monitor(data['type'], data['query'], :message => data['message'], :name => data['name'], :options => data['options'])
+      rescue => e
+        puts e
+      end
+      apply_result_display(res)
+    end
+
+    def update_monitors(dry_run)
+      filter = get_filter(@project)
+      # current_monitors = @exporter.export_monitors(filter['name'], filter['tags']) { |f| YAML.load(f) }
+      current_monitors = @exporter.export_monitors(@project) { |f| YAML.load(f) }
+      datas = open(@monitors_file_path) { |f| YAML.load(f) }
+      datas.each do |data|
+        # 新規登録 or 更新のチェック(id キーが有れば更新)
+        if data.has_key?('id') then
+          current_monitor = YAML.dump(current_monitors.select {|m| m['id'] == data['id']}.last)
+          latest_monitor = YAML.dump(data)
+          diff = Diffy::Diff.new(current_monitor, latest_monitor).to_s(:color)
+          # 差分の有無をチェック(diff != "\n" であれば差分が有ると判断)
+          if diff != "\n" then
+            # --dry-run フラグのチェック
+            if dry_run != nil then
+              puts hl.color('Check update line.', :gray, :underline)
+              puts diff
+              puts ''
+            else
+              update_monitor(data)
+              monitors_data = @exporter.export_monitors(@project)
+              @exporter.store_monitors_data(monitors_data)
+            end
+          end
+        else
+          # 新規登録
+          # --dry-run フラグのチェック
+          if dry_run == nil then
+            create_monitor(data)
+            monitors_data = @exporter.export_monitors(@project)
+            @exporter.store_monitors_data(monitors_data)
+          else
+            puts hl.color('Check add line.', :gray, :underline)
+            puts hl.color(YAML.dump(data), :light_cyan)
+            puts ''
+          end
+        end
+      end
+      puts 'done.'
+    end
+
+  end
+end

data/lib/chihuahua/version.rb

@@ -0,0 +1,3 @@
+module Chihuahua
+  VERSION = "0.0.3"
+end

metadata

@@ -0,0 +1,145 @@
+--- !ruby/object:Gem::Specification
+name: chihuahua
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- kapp
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-03-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  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: diffy
+  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: 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: highline
+  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'
+description: Chihuahua is a tool to manage Datadog monitors.
+email:
+- inokara@gmail.com
+executables:
+- chihuahua
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- chihuahua.gemspec
+- exe/chihuahua
+- images/chihuahua.png
+- lib/chihuahua.rb
+- lib/chihuahua/cli.rb
+- lib/chihuahua/client.rb
+- lib/chihuahua/common.rb
+- lib/chihuahua/export.rb
+- lib/chihuahua/update.rb
+- lib/chihuahua/version.rb
+homepage: https://github.com/inokappa/chihuahua
+licenses: []
+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.5.1
+signing_key: 
+specification_version: 4
+summary: Chihuahua is a tool to manage Datadog monitors.
+test_files: []