hotdog 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bc96b843200f93f4a812c164c690eead02717239
-  data.tar.gz: ee5bd2f02afa28d4b180d8c1392a01d33fc6e9eb
+  metadata.gz: 55a9182f8b6f286b99a6d76bd70db231d1e04945
+  data.tar.gz: e3eda56735a69185bf33faa2f1bc027b3573464c
 SHA512:
-  metadata.gz: 3a1a31cac900cbf90f39136308851a1545bd38ff736322dd58b48dd9744c7b495c8980743dfd8fb6c9948b28bcbde57205be3ce07d68b56603a0c8a2ffd307b9
-  data.tar.gz: cde383321ce521ffce0a7c6053160de969fab6756d243bf22256502bcf89db142409009a8f8d70c44dcff64283fb4c8643bd7d1b4e16c177a05ef15d4be3af4c
+  metadata.gz: 2fa9df74864b65781ddf36978fabfaed7c002e86173e9cb50b7a504addee6862d42d27b1d9b2ac397316080f9cb6e73529cc7bdea1904be69ff0222bf6b2174a
+  data.tar.gz: 28d70713ab309b7383f830f9d9fcf3431bf0f30c37a4ebfdc3b55592a4a0fd4dbe497712f5ed3dd0ded0da5568c4ea680484ada4e11d5aaff2ffc7361eacdcb0

data/lib/hotdog/application.rb

@@ -136,7 +136,8 @@ module Hotdog
           load library
           Hotdog::Commands.const_get(const_name(File.basename(library, ".rb")))
         else
-          raise(NameError.new("unknown command: #{name}"))
+          require "hotdog/commands/help"
+          Hotdog::Commands::Help
         end
       end
     end

data/lib/hotdog/commands/down.rb

@@ -0,0 +1,34 @@
+#!/usr/bin/env ruby
+
+require "fileutils"
+
+module Hotdog
+  module Commands
+    class Down < BaseCommand
+      def run(args=[])
+        args.each do |arg|
+          if arg.index(":").nil?
+            scope = "host:#{arg}"
+          else
+            scope = arg
+          end
+          start = Time.new
+          downtime = 86400 # TODO: make configurable
+          code, schedule = @dog.schedule_downtime(scope, :start => start.to_i, :end => (start+downtime).to_i)
+          logger.debug("dog.schedule_donwtime(%s, :start => %s, :end => %s) #==> [%s, %s]" % [scope.inspect, start.to_i, (start+downtime).to_i, code.inspect, schedule.inspect])
+          if code.to_i / 100 != 2
+            raise("dog.schedule_downtime(%s, ...) returns [%s, %s]" % [scope.inspect, code.inspect, schedule.inspect])
+          end
+        end
+
+        # Remove persistent.db to schedule update on next invocation
+        if not @db.nil?
+          @db.close
+          FileUtils.rm_f(File.join(confdir, PERSISTENT_DB))
+        end
+      end
+    end
+  end
+end
+
+# vim:set ft=ruby :

data/lib/hotdog/commands/up.rb

@@ -0,0 +1,42 @@
+#!/usr/bin/env ruby
+
+require "fileutils"
+
+module Hotdog
+  module Commands
+    class Up < BaseCommand
+      def run(args=[])
+        scopes = args.map { |arg|
+          if arg.index(":").nil?
+            "host:#{arg}"
+          else
+            arg
+          end
+        }
+        code, all_downtimes = @dog.get_all_downtimes()
+        if code.to_i / 100 != 2
+          raise("dog.get_all_downtimes() returns [%s, %s]" % [code.inspect, all_downtimes.inspect])
+        end
+
+        cancel_downtimes = all_downtimes.select { |downtime|
+          downtime["active"] and downtime["id"] and scopes.map { |scope| downtime.fetch("scope", []).include?(scope) }.any?
+        }
+
+        cancel_downtimes.each do |downtime|
+          code, cancel = @dog.cancel_downtime(downtime["id"])
+          if code.to_i / 100 != 2
+            raise("dog.cancel_downtime(%s) returns [%s, %s]" % [downtime["id"].inspect, code.inspect, cancel.inspect])
+          end
+        end
+
+        # Remove persistent.db to schedule update on next invocation
+        if not @db.nil?
+          @db.close
+          FileUtils.rm_f(File.join(confdir, PERSISTENT_DB))
+        end
+      end
+    end
+  end
+end
+
+# vim:set ft=ruby :

data/lib/hotdog/commands.rb

@@ -8,6 +8,8 @@ require "sqlite3"
 module Hotdog
   module Commands
     class BaseCommand
+      PERSISTENT_DB = "persistent.db"
+
       def initialize(options={})
         @application = options[:application]
         @confdir = options[:confdir]
@@ -58,6 +60,11 @@ module Hotdog
         s.index('*') or s.index('?') or s.index('[') or s.index(']')
       end
 
+      def host?(host_id)
+        host_id = execute("SELECT id FROM hosts WHERE name = %s LIMIT 1", s)
+        not host_id.nil?
+      end
+
       def get_hosts(hosts=[])
         update_db
         if 0 < tags.length
@@ -102,7 +109,7 @@ module Hotdog
       def update_db(options={})
         if @db.nil?
           FileUtils.mkdir_p(confdir)
-          persistent = File.join(confdir, "persistent.db")
+          persistent = File.join(confdir, PERSISTENT_DB)
 
           if not @force and File.exist?(persistent) and Time.new < File.mtime(persistent) + expiry
             begin

data/lib/hotdog/formatters/csv.rb

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+
+require "csv"
+
+module Hotdog
+  module Formatters
+    class Csv < BaseFormatter
+      def format(result, options={})
+        result = result.dup
+        if options[:headers] and options[:fields]
+          result.unshift(options[:fields])
+        end
+        CSV.generate { |csv|
+          result.each do |row|
+            csv << row
+          end
+        }
+      end
+    end
+  end
+end
+
+# vim:set ft=ruby :

data/lib/hotdog/formatters/ltsv.rb

@@ -0,0 +1,16 @@
+#!/usr/bin/env ruby
+
+module Hotdog
+  module Formatters
+    class Ltsv < BaseFormatter
+      def format(result, options={})
+        fields = options.fetch(:fields, [])
+        result.map { |row|
+          fields.zip(row).map { |column| column.join(":") }.join("\t")
+        }.join("\n") + "\n"
+      end
+    end
+  end
+end
+
+# vim:set ft=ruby :

data/lib/hotdog/formatters/tsv.rb

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+
+module Hotdog
+  module Formatters
+    class Tsv < BaseFormatter
+      def format(result, options={})
+        result = result.dup
+        if options[:headers] and options[:fields]
+          result.unshift(options[:fields])
+        end
+        result.map { |row|
+          row.join("\t")
+        }.join("\n") + "\n"
+      end
+    end
+  end
+end
+
+# vim:set ft=ruby :

data/lib/hotdog/version.rb

@@ -1,3 +1,3 @@
 module Hotdog
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hotdog
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Yamashita Yuu
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-16 00:00:00.000000000 Z
+date: 2015-01-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -97,14 +97,19 @@ files:
 - hotdog.gemspec
 - lib/hotdog/application.rb
 - lib/hotdog/commands.rb
+- lib/hotdog/commands/down.rb
 - lib/hotdog/commands/help.rb
 - lib/hotdog/commands/hosts.rb
 - lib/hotdog/commands/ls.rb
 - lib/hotdog/commands/search.rb
 - lib/hotdog/commands/tags.rb
+- lib/hotdog/commands/up.rb
 - lib/hotdog/formatters.rb
+- lib/hotdog/formatters/csv.rb
 - lib/hotdog/formatters/json.rb
+- lib/hotdog/formatters/ltsv.rb
 - lib/hotdog/formatters/plain.rb
+- lib/hotdog/formatters/tsv.rb
 - lib/hotdog/formatters/yaml.rb
 - lib/hotdog/version.rb
 homepage: https://github.com/yyuu/hotdog