283
284
285
286

# File 'app/app.rb', line 283

after do
  return unless request.put?
  halt [400, @data.send("to_#{@format}") ] if ! @data[:result]
end

170
171
172
173

# File 'app/app.rb', line 170

before do
  @data = []
  @format = @output = nil
end

294
295
296

# File 'app/app.rb', line 294

after do
  @data = @data.values if @active_resource && @data.is_a?(Hash)
end

248
249
250
251
252
253
254

# File 'app/app.rb', line 248

before do
  if @format == :json
    ['callback','jscallback','jsonp','jsoncallback'].each do |x|
      @callback = params.delete(x) unless @callback
    end
  end
end

59
60
61
62
63
64
65
66
67

# File 'app/get/objects.rb', line 59

get "/_objects/:type/:name" do |type,name|
  begin
    @data = @objects[type.to_sym][name]
  rescue NoMethodError
    nil
  end

  nil
end

326
327
328
329

# File 'app/app.rb', line 326

get "/_api" do
  @data = self.api
  nil
end

20
21
22
23

# File 'app/get/config.rb', line 20

get "/_config" do
  @data = $nagios[:config].configuration
  nil
end

36
37
38
39
40
41
42
43
44
45
46

# File 'app/get/objects.rb', line 36

get "/_objects/:type" do |type|
  begin
    @data = @objects[type.to_sym]
    @data = @data.keys if @output == :list

  rescue NoMethodError
    nil
  end

  nil
end

16
17
18
19
20
21
22
23
24

# File 'app/get/objects.rb', line 16

get "/_objects" do

  @data = begin
            @output == :list ? @objects.keys : @objects
          rescue NoMethodError
            nil
          end
  nil
end

337
338
339
340
341
342
343
344
345
346
347
348
349

# File 'app/app.rb', line 337

get "/_runtime" do
  @data = {
    application: self.class,
    version: VERSION,
    runtime: {
      environment: Nagira.settings.environment,
      home: ENV['HOME'],
      user: ENV['LOGNAME'],
      nagiosFiles: $nagios.keys.map {  |x| {  x =>  $nagios[x].path }}
    }
  }
  nil
end

355
356
357
358
359
360
361
362
363

# File 'app/app.rb', line 355

get "/" do
  @data = {
    :application => self.class,
    :version => VERSION,
    :source => GITHUB,
    :apiUrl => request.url.sub(/\/$/,'') + "/_api",
  }
  nil
end

141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

# File 'app/get/status.rb', line 141

get /^\/_status(\/_hosts)?$/ do

  @data = @status.dup

  case @output
  when :state
    @data.each { |k,v| @data[k] = v['hoststatus'].slice("host_name", "current_state") }
  when :list
    @data = @data.keys
  when :full
    @data
  else
    @data.each { |k,v| @data[k] = v['hoststatus'] }
  end

  nil
end

17
18
19
20

# File 'app/get/status/hostgroups.rb', line 17

get "/_status/_hostgroup/:hostgroup" do |hostgroup|
  @data = Hostgroup.new(hostgroup).full
  nil
end

39
40
41
42

# File 'app/get/status/hostgroups.rb', line 39

get "/_status/_hostgroup/:hostgroup/_host" do |hostgroup|
  @data = Hostgroup.new(hostgroup).host_status
  nil
end

28
29
30
31

# File 'app/get/status/hostgroups.rb', line 28

get "/_status/_hostgroup/:hostgroup/_service" do |hostgroup|
  @data = Hostgroup.new(hostgroup).service_status
  nil
end

176
177
178
179
180
181
182
183
184
185
186
187

# File 'app/get/status.rb', line 176

get %r{^/_status/(?<hostname>#{hostname_regex})$} do |hostname|


  hostname = hostname.to_i if hostname =~ /^\d+$/
  @data =  @status[hostname]['hoststatus'].dup if @status.has_key? hostname

  if @output == :state
    @data = @data.slice("host_name", "current_state")
  end

  nil
end

94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

# File 'app/get/status.rb', line 94

get %r{^/_status/(?<hostname>#{hostname_regex})/_(?<service>(services|hostcomments|servicecomments))$} do |hostname,service|

  hostname = hostname.to_i if hostname =~ /^\d+$/
  key = case service
        when 'services'
          'servicestatus'
        else
          service
        end

  if @status && @status[hostname]
    case @output
    when :list
      @data = @status[hostname][key].keys
    when :state
      @data = @status.each { |k,v| @data[k] = v.slice("host_name", "service_description", "current_state") }
    else
      @data = @status[hostname][key]
    end
  end

  nil
end

65
66
67
68
69
70
71
72
73
74
75
76

# File 'app/get/status.rb', line 65

get "/_status/:hostname/_services/:service_name" do |hostname,service|

  hostname = hostname.to_i if hostname =~ /^\d+$/
  if @status && @status[hostname]
    if @output == :state
      @data = @status[hostname]['servicestatus'][service].slice("hostname", "service_description", "current_state")
    else
      @data = @status[hostname]['servicestatus'][service]
    end
  end
  nil
end

20
21
22
23
24
25
26

# File 'app/get/status/servicegroups.rb', line 20

get "/_status/_servicegroup/:servicegroup" do |group_name|

  servicegroup = Servicegroup.new(group_name)

  @data = servicegroup.send(@output)
  nil
end

266
267
268
269
270
271
272
273
274

# File 'app/app.rb', line 266

after do
  if ! @data || @data.empty?
    halt [404, {
            :message => "Object not found or bad request",
            :error => "HTTP::Notfound"
          }.send("to_#{@format}")
         ]
  end
end

10
11
12
13
14
15
16
17
18
19
20
21
22

# File 'app/put.rb', line 10

before do 
  if request.put?
    data = request.body.read
    @input = case @format
            when :json then JSON.parse    data
            when :xml  then Hash.from_xml data
            when :yaml then YAML.load     data
            end
    # Make sure we always return an Array
    @input = [@input] if @input.is_a? Hash
  end
  @input
end

125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162

# File 'app/app.rb', line 125

before do

  if Nagira.settings.start_background_parser
    unless $bg.alive?
      logger.warn "Background Parser is configured to run, but is not active"
      $nagios[:config].parse
      $nagios[:status].parse
      $nagios[:objects].parse
    end
    $use_inflight_status  ? @status = $nagios[:status_inflight].status['hosts']
                          : @status = $nagios[:status].status['hosts']
    $use_inflight_objects ? @objects  = $nagios[:objects_inflight].objects
                          : @objects  = $nagios[:objects].objects
  else
    $nagios[:config].parse
    $nagios[:status].parse
    $nagios[:objects].parse
    @status   = $nagios[:status].status['hosts']
    @objects  = $nagios[:objects].objects
  end


##
# TODO: This stuff breaks XML valid. Will have to wait.
#
#     idx = 0
#     @status.keys.uniq.each do |hostname|
#       @status[idx] = @status[hostname]
#       idx += 1
#     end
  #
  # Add plural keys to use ActiveResource with Nagira
  #
  @objects.keys.each do |singular|
    @objects[singular.to_s.pluralize.to_sym] = @objects[singular]
  end

end

47
48
49

# File 'app/put/host.rb', line 47

put "/_host_status/:host_name" do
  "Not implemented: TODO"
end

10
11
12

# File 'app/put/host.rb', line 10

put "/_status" do
  "TODO: Not implemented"
end

93
94
95
96

# File 'app/put/status.rb', line 93

put "/_status/:host_name/_services/:service_description/_return_code/:return_code/_plugin_output/:plugin_output" do 
  @data = update_service_status params
  nil
end

35
36
37
38
39
40

# File 'app/put/host.rb', line 35

put "/_status/:host_name" do
  @data = update_host_status @input.first.merge({
    'host_name' => params['host_name']
  })
  nil
end

54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

# File 'app/put/status.rb', line 54

put "/_status/:host_name/_services" do

  data, result = [], true

  @input.each do |datum|
    # FIXME - this calls update for each service. Should be batching them together
    update = update_service_status(
                                    datum.merge({ 
                                                  'host_name' => params['host_name']
                                                })
                                    )
    data << update[:object].first
    result &&= update[:result]
  end
  @data = { result: result, object: data }
  nil

end

313
314
315

# File 'app/app.rb', line 313

after do
  body( @callback ? "#{@callback.to_s} (#{@data.to_json})" : @data.send("to_#{@format}") )
end

61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

# File 'app/app.rb', line 61

configure do

  $nagios = {  }
  $nagios[:config]  = Nagios::Config.new Nagira.settings.nagios_cfg
  $nagios[:config].parse

  $nagios.merge!({
                  status: Nagios::Status.new(  Nagira.settings.status_cfg ||
                                               $nagios[:config].status_file
                                               ),
                  objects: Nagios::Objects.new( Nagira.settings.objects_cfg ||
                                                $nagios[:config].object_cache_file
                                                ),
                  commands: Nagios::ExternalCommands.new( Nagira.settings.command_file ||
                                                          $nagios[:config].command_file
                                                          )
                })

  $nagios.merge!({
                  status_inflight: Nagios::Status.new( Nagira.settings.status_cfg ||
                                                       $nagios[:config].status_file
                                                       ),
                  objects_inflight: Nagios::Objects.new( Nagira.settings.objects_cfg ||
                                                         $nagios[:config].object_cache_file
                                                       )
                }) if ::DEFAULT[:start_background_parser]

  puts "[#{Time.now}] -- Starting Nagira application"
  puts "[#{Time.now}] -- Version #{Nagira::VERSION}"
  puts "[#{Time.now}] -- Running in #{Nagira.settings.environment} environment"
  $nagios.keys.each do |x|
    puts "[#{Time.now}] -- Using nagios #{x} file: #{$nagios[x].path}"
  end

  $nagios[:status].parse
  $nagios[:objects].parse

  @status   = $nagios[:status].status['hosts']
  @objects  = $nagios[:objects].objects

  Nagios::BackgroundParser.new
end

194
195
196
197
198

# File 'app/app.rb', line 194

before do
  request.path_info.sub!(/#{settings.format_extensions}/, '')
  @format = ($1 || settings.format).to_sym
  content_type "application/#{@format.to_s}"
end

228
229
230
231

# File 'app/app.rb', line 228

before do
  request.path_info.sub!(/\/_(list|state|full)$/, '')
  @output = ($1 || :normal).to_sym
end

49
50
51

# File 'app/put.rb', line 49

def update_host_status params
  put_update :PROCESS_HOST_CHECK_RESULT, params
end

42
43
44

# File 'app/put.rb', line 42

def update_service_status params
  put_update :PROCESS_SERVICE_CHECK_RESULT, params
end