naginata 0.1.0

data/lib/naginata/configuration/service.rb

@@ -0,0 +1,35 @@
+module Naginata
+  class Configuration
+    class Service
+      attr_reader :description, :hostname, :nagios
+
+      # @param [String] description
+      # @param [Hash] options
+      # @option options [String] :host hostname monitored by nagios
+      # @option options [String] :on nagios server name
+      def initialize(description, options = {})
+        ArgumentError if options[:host].nil?
+        ArgumentError if options[:on].nil?
+        @description = description
+        @hostname = options[:host]
+        @nagios = options[:on]
+      end
+
+      #def to_s
+      #  @description
+      #end
+
+    end
+
+    class Host < Service
+
+      # @param [String] hostname
+      # @param [Hash] options
+      # @option options [String] :nagios nagios server name
+      def initialize(hostname, options = {})
+        super(:ping, options.merge(host: hostname))
+      end
+
+    end
+  end
+end

data/lib/naginata/defaults.rb

@@ -0,0 +1,8 @@
+set_if_empty :format, :pretty
+set_if_empty :log_level, :warn
+set_if_empty :pty, false
+
+set_if_empty :nagios_server_options, {
+  command_file: '/usr/local/nagios/var/rw/nagios.cmd', 
+   status_file: '/usr/local/nagios/var/status.cmd', 
+}

data/lib/naginata/dsl.rb

@@ -0,0 +1,28 @@
+require 'naginata/configuration'
+
+module Naginata
+  module DSL
+
+    def nagios_server(name)
+      Configuration.env.nagios_server(name)
+    end
+
+    def fetch(key, default=nil, &block)
+      env.fetch(key, default, &block)
+    end
+
+    def env
+      Configuration.env
+    end
+
+    def set(key, value)
+      env.set(key, value)
+    end
+
+    def set_if_empty(key, value)
+      env.set_if_empty(key, value)
+    end
+
+  end
+end
+

data/lib/naginata/loader.rb

@@ -0,0 +1,39 @@
+require 'naginata/dsl'
+
+module Naginata
+  class Loader
+    #module_function
+
+    class << self
+      include Naginata::DSL
+
+      def load_configuration
+        instance_eval File.read(File.join(File.dirname(__FILE__), 'defaults.rb'))
+        instance_eval File.read 'Naginatafile'
+      end
+
+      def load_remote_objects
+        require 'naginata/cli/fetch'
+        # Refresh cached status.dat
+        CLI::Fetch.new.run
+
+        # Load status.dat
+        #
+        # @Note currently here reads all nagios servers' cache files. This
+        # means that users can not reduce nagios servers by --nagios= CLI 
+        # option. Below loop may be moved into initialization phase of CLI
+        # class.
+        Configuration.env.nagios_servers.each do |nagios_server|
+          status = ::Naginata::Status.find(nagios_server.hostname)
+          status.service_items.group_by{ |section| section.host_name }.each do |host, sections|
+            services = sections.map { |s| s.service_description }
+            Configuration.env.host(host, services: services, on: nagios_server.hostname)
+          end
+        end
+
+      end
+
+    end
+  end
+end
+

data/lib/naginata/runner.rb

@@ -0,0 +1,36 @@
+require 'naginata/configuration'
+require 'naginata/configuration/filter'
+require 'sshkit/coordinator'
+
+module Naginata
+  class Runner
+    def self.run(&block)
+      nagios_servers = Configuration.env.filter(Configuration.env.nagios_servers)
+      services = Configuration.env.filter_service(Configuration.env.services)
+      user = Configuration.env.fetch(:nagios_server_options)[:run_command_as]
+
+      run_backend(nagios_servers) do |nagios_server|
+        svcs = Configuration::Filter.new(:nagios_server, nagios_server).filter_service(services)
+        as(user) do
+          yield self, nagios_server, svcs
+        end
+      end
+    end
+
+    def self.run_locally(&block)
+      nagios_servers = Configuration.env.filter(Configuration.env.nagios_servers)
+      services = Configuration.env.filter_service(Configuration.env.services)
+      nagios_servers.each do  |nagios_server|
+        svcs = Configuration::Filter.new(:nagios_server, nagios_server).filter_service(services)
+        yield nagios_server, svcs
+      end
+    end
+
+    private
+
+    def self.run_backend(nagios_servers, options = {}, &block)
+      SSHKit::Coordinator.new(nagios_servers).each(options, &block)
+    end
+  end
+end
+ 

data/lib/naginata/status.rb

@@ -0,0 +1,82 @@
+require 'nagios_analyzer'
+require 'forwardable'
+require 'tempfile'
+
+module Naginata
+  class Status
+    extend Forwardable
+    def_delegators :@status, :service_items, :host_items, :items, :service_problems, :host_problems, :scopes
+    attr_accessor :status, :hostname
+
+    # Write status.dat into local disk
+    #
+    # @return [Naginata::Status] return self
+    # @return [false] if @status is nil
+    def save
+      return false if status.nil?
+      File.open(path, "w") do |f|
+        f.sync = true
+        Marshal.dump(status, f)
+      end
+      self
+    end
+
+    def purge_cache
+      if File.exist?(path)
+        FileUtils.rm path
+      end
+    end
+
+    # Path of marshal dumped status file
+    # 
+    # @return [String] path of marshal dumped status file
+    def path
+      raise "@hostname must be set" if hostname.nil?
+      @path ||= File.join(self.class.cache_dir, "#{hostname}.status.dat")
+    end
+
+    # Create a new instance from raw status.dat string
+    #
+    # @param [String] string raw status.dat text
+    # @param [String] nagios_server_hostname hostname to identify status.data data
+    # @return [Naginata::Status]
+    def self.build(string, nagios_server_hostname)
+      instance = new
+      instance.hostname = nagios_server_hostname
+
+      # @Note NagiosAnalizer::Status.new requires filename, so here
+      # needs to write text data into tempfile.
+      Tempfile.open(["naginata-#{nagios_server_hostname}", ".status.dat"]) do |temp|
+        temp.sync = true
+        temp.write string
+        instance.status = ::NagiosAnalyzer::Status.new(temp.path, include_ok: true)
+        temp.close!
+      end
+      instance
+    end
+
+    # Return a directory path for status.dat caching
+    # 
+    # @return [String] directory path for status.dat caching
+    def self.cache_dir
+      dir = (ENV['HOME'] && Dir.exist?(ENV['HOME'])) ? ENV['HOME'] : Dir.pwd
+      dir = File.join(dir, '.naginata/cache/status')
+      FileUtils.mkdir_p dir
+      return dir
+    end
+
+    # Find a instance from cache by nagios server hostname
+    #
+    # @return [Naginata::Status]
+    def self.find(nagios_server_hostname)
+      instance = new
+      instance.hostname = nagios_server_hostname
+      if File.exist?(instance.path)
+        File.open(instance.path) { |f| instance.status = Marshal.load(f) }
+        return instance
+      end
+    end
+
+  end
+end
+ 

data/lib/naginata/ui.rb

@@ -0,0 +1,6 @@
+require 'naginata/ui/shell'
+
+module Naginata
+  module UI
+  end
+end

data/lib/naginata/ui/shell.rb

@@ -0,0 +1,39 @@
+require 'thor/base'
+
+module Naginata
+  module UI
+    class Shell
+      LEVELS = %w(error warn info debug)
+
+      def initialize
+        @shell = Thor::Base.shell.new
+        @level = ENV['DEBUG'] ? "debug" : "info"
+      end
+
+      def info(msg, newline = nil)
+        @shell.say(msg, nil, newline) if level("info")
+      end
+
+      def warn(msg, newline = nil)
+        @shell.say(msg, :yellow, newline) if level("warn")
+      end
+
+      def error(msg, newline = nil)
+        @shell.say(msg, :red, newline) if level("error")
+      end
+
+      def debug(msg, newline = nil)
+        @shell.say(msg, nil, newline) if level("debug")
+      end
+
+      def level(name = nil)
+        name ? LEVELS.index(name) <= LEVELS.index(@level) : @level
+      end
+
+      def yes?(msg)
+        @shell.yes?(msg)
+      end
+
+    end
+  end
+end

data/lib/naginata/version.rb

@@ -0,0 +1,3 @@
+module Naginata
+  VERSION = "0.1.0"
+end

data/naginata.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'naginata/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "naginata"
+  spec.version       = Naginata::VERSION
+  spec.authors       = ["Nobuhiro Nikushi"]
+  spec.email         = ["deneb.ge@gmail.com"]
+
+  spec.summary       = "Remote multi nagios server control tool"
+  spec.description   = "Remote multi nagios server control tool over ssh connection"
+  spec.homepage      = "https://github.com/niku4i/naginata"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "bin"
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'sshkit', '~> 1.7'
+  spec.add_dependency 'thor'
+  spec.add_dependency 'nagios_analyzer', '~> 0.0.5'
+  spec.add_development_dependency "bundler", "~> 1.9"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3"
+end

metadata

@@ -0,0 +1,159 @@
+--- !ruby/object:Gem::Specification
+name: naginata
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Nobuhiro Nikushi
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-04-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sshkit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !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: nagios_analyzer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !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: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+description: Remote multi nagios server control tool over ssh connection
+email:
+- deneb.ge@gmail.com
+executables:
+- naginata
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CHANGELOG.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/naginata
+- integration-test/.gitignore
+- integration-test/Naginatafile
+- integration-test/Vagrantfile
+- lib/naginata.rb
+- lib/naginata/cli.rb
+- lib/naginata/cli/fetch.rb
+- lib/naginata/cli/notification.rb
+- lib/naginata/command/external_command.rb
+- lib/naginata/configuration.rb
+- lib/naginata/configuration/filter.rb
+- lib/naginata/configuration/nagios_server.rb
+- lib/naginata/configuration/service.rb
+- lib/naginata/defaults.rb
+- lib/naginata/dsl.rb
+- lib/naginata/loader.rb
+- lib/naginata/runner.rb
+- lib/naginata/status.rb
+- lib/naginata/ui.rb
+- lib/naginata/ui/shell.rb
+- lib/naginata/version.rb
+- naginata.gemspec
+homepage: https://github.com/niku4i/naginata
+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.4.5
+signing_key: 
+specification_version: 4
+summary: Remote multi nagios server control tool
+test_files: []