vadv/kurchatov

# encoding: utf-8

require 'kurchatov/plugin'
require 'kurchatov/mashie'
require 'timeout'

module Kurchatov
  module Plugins
    class Riemann < Kurchatov::Plugin

      include Kurchatov::Mixin::Ohai
      include Kurchatov::Mixin::Event
      include Kurchatov::Mixin::Command
      include Kurchatov::Mixin::Http
      include Kurchatov::Mixin::RunitService
      include Kurchatov::Mixin::Queue
      include Kurchatov::Mixin::Monitor

      attr_accessor :run_if, :collect, :run, :always_start, :required, :ignore_errors, :interval, :plugin

      def initialize(name = '')
        super(name)
        @run_if = Proc.new { true }
        @required = Proc.new { true }
        @run = nil
        @plugin = Mashie.new
        @always_start = false
        @ignore_errors = false
        @collect = nil
        @interval = 60.0
      end

      def plugin_config
        plugin
      end

      def start!
        super
        run.nil? ? start_collect : start_run
      end

      def start_collect
        loop do
          tick = Time.now
          Timeout::timeout(interval) { self.instance_eval(&collect) }
          tick = interval - (Time.now - tick).to_f
          tick > 0.0 ? sleep(tick) : raise(Timeout::Error, "Execution timeout. Exec time: #{tick}, interval: #{interval}")
        end
      end

      def start_run
        self.instance_eval(&run)
      end

      def respond_to_ohai?(opts = {})
        opts.each { |k, v| return false unless ohai[k] == v }
        true
      end

      def runnable_by_required?
        begin
          self.instance_eval(&required)
        rescue LoadError
          return
        end
        true
      end

      def runnable_by_config?
        Log.info("Plugin '#{self.name}' disabled by run and collect nil") and return if (collect.nil? && run.nil?)
        Log.info("Plugin '#{self.name}' disabled in config") and return if (plugin[:disable] == true)
        Log.info("Plugin '#{self.name}' not started by run_if condition ") and
            return unless self.instance_eval(&run_if)
        Log.error("Plugin '#{self.name}' not started by required block") and return unless runnable_by_required?
        @plugin[:service] ||= name
        true
      end

    end
  end
end