summaryrefslogtreecommitdiffstats
path: root/lib/puppet/provider/naginator.rb
blob: 5c610fb3186104c2d16cc357d0445be5d9c17de8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#  Created by Luke Kanies on 2007-11-27.
#  Copyright (c) 2007. All rights reserved.

require 'puppet'
require 'puppet/provider/parsedfile'
require 'puppet/external/nagios'

# The base class for all Naginator providers.
class Puppet::Provider::Naginator < Puppet::Provider::ParsedFile
  NAME_STRING = "## --PUPPET_NAME-- (called '_naginator_name' in the manifest)"
  # Retrieve the associated class from Nagios::Base.
  def self.nagios_type
    unless @nagios_type
      name = resource_type.name.to_s.sub(/^nagios_/, '')
      unless @nagios_type = Nagios::Base.type(name.to_sym)
        raise Puppet::DevError, "Could not find nagios type '#{name}'"
      end

      # And add our 'ensure' settings, since they aren't a part of
      # Naginator by default
      @nagios_type.send(:attr_accessor, :ensure, :target, :on_disk)
    end
    @nagios_type
  end

  def self.parse(text)
      Nagios::Parser.new.parse(text.gsub(NAME_STRING, "_naginator_name"))
  rescue => detail
      raise Puppet::Error, "Could not parse configuration for #{resource_type.name}: #{detail}"
  end

  def self.to_file(records)
    header + records.collect { |record| record.to_s }.join("\n").gsub("_naginator_name", NAME_STRING)
  end

  def self.skip_record?(record)
    false
  end

  def self.valid_attr?(klass, attr_name)
    nagios_type.parameters.include?(attr_name)
  end

  def initialize(resource = nil)
    if resource.is_a?(Nagios::Base)
      # We don't use a duplicate here, because some providers (ParsedFile, at least)
      # use the hash here for later events.
      @property_hash = resource
    elsif resource
      @resource = resource if resource
      # LAK 2007-05-09: Keep the model stuff around for backward compatibility
      @model = resource
      @property_hash = self.class.nagios_type.new
    else
      @property_hash = self.class.nagios_type.new
    end
  end
end