hq-log-monitor-server 0.2.2 → 0.3.0

data/features/event-page.feature

@@ -0,0 +1,67 @@
+Feature: View and manipulate a single event
+
+  Background:
+
+    Given the log monitor server config:
+      """
+      <log-monitor-server-config>
+        <server port="${port}"/>
+        <db host="${db-host}" port="${db-port}" name="${db-name}"/>
+        <icinga command-file="${command-file}">
+          <service name="service" icinga-host="host" icinga-service="service">
+            <type name="warning" level="warning"/>
+            <type name="critical" level="critical"/>
+          </service>
+        </icinga>
+      </log-monitor-server-config>
+      """
+
+    And the time is 100
+
+    And I submit the following event:
+      """
+      {
+        type: warning,
+        source: { class: class, host: host, service: service },
+        location: { file: logfile, line: 0 },
+        lines: {
+          before: [],
+          matching: WARNING blah,
+          after: [],
+        }
+      }
+      """
+
+  Scenario: View an event
+
+    When I visit /event/${event-id}
+
+    Then I should see the event
+    And the event status should be "unseen"
+
+    And the summary new should be 1
+    And the summary total should be 1
+    And the summary new for type "warning" should be 1
+    And the summary total for type "warning" should be 1
+
+Scenario: Mark as seen
+
+    Given the time is 200
+
+    When I visit /event/${event-id}
+    And I click "mark as seen"
+
+    Then I should see the event
+    And the event status should be "seen"
+
+    And the summary new should be 0
+    And the summary total should be 1
+    And the summary new for type "warning" should be 0
+    And the summary total for type "warning" should be 1
+
+    And icinga should receive:
+      """
+      [100] PROCESS_SERVICE_CHECK_RESULT;host;service;0;OK no new events
+      [100] PROCESS_SERVICE_CHECK_RESULT;host;service;1;WARNING 1 warning
+      [200] PROCESS_SERVICE_CHECK_RESULT;host;service;0;OK no new events
+      """

data/features/{overview.feature → overview-page.feature}

@@ -14,7 +14,7 @@ Feature: Log monitor server overview
 
   Scenario: No events
 
-    When I visit the overview page
+    When I visit /
 
     Then I should see no summaries
 
@@ -36,7 +36,7 @@ Feature: Log monitor server overview
       },
       """
 
-    When I visit the overview page
+    When I visit /
 
     Then I should see 1 summary
     And the 1st summary should be:

data/features/support/env.rb

@@ -33,6 +33,34 @@ def mongo_db name
 	mongo_conn[mongo_db_name name]
 end
 
+def get_event event_id
+
+	db =
+		mongo_db("logMonitorServer")
+
+	event =
+		db["events"].find({
+			"_id" => event_id,
+		}).first
+
+	return event
+
+end
+
+def get_summary source
+
+	db =
+		mongo_db("logMonitorServer")
+
+	summary =
+		db["summaries"].find({
+			"_id" => source,
+		}).first
+
+	return summary
+
+end
+
 After do
 
 	@db_names.each do

data/features/support/steps.rb

@@ -75,6 +75,16 @@ When /^I submit the following events?:$/ do
 
 	@submitted_events = events_data
 
+	# store information about the event (assuming it's the only one)
+
+	db = mongo_db("logMonitorServer")
+	event = db["events"].find.first
+
+	if event
+		@event_id = event["_id"]
+		@source = event["source"]
+	end
+
 end
 
 Then /^I should receive a (\d+) response$/ do
@@ -85,14 +95,15 @@ end
 Then /^the event should be in the database$/ do
 
 	db = mongo_db("logMonitorServer")
-
 	event = db["events"].find.first
 
 	event.should_not be_nil
 	event["timestamp"].should be_a Time
+	event["status"].should be_a String
 
 	event.delete "_id"
 	event.delete "timestamp"
+	event.delete "status"
 
 	event.should == @submitted_events.first
 
@@ -103,56 +114,69 @@ Then /^the summary should show:$/ do
 
 	expected_summary = YAML.load expected_string
 
-	db = mongo_db("logMonitorServer")
-
-	summary =
-		db["summaries"].find({
-			"_id" => expected_summary["_id"],
-		}).first
+	summary = get_summary expected_summary["_id"]
 
 	summary.should == expected_summary
 
 end
 
-# ui steps
+Then /^the event status should be "(.*?)"$/ do
+	|expected_status|
+
+	event = get_event @event_id
+
+	event["status"].should == expected_status
 
-When /^I visit the overview page$/ do
-	visit "/"
 end
 
-Then /^I should see no summaries$/ do
-	page.should have_content "No events have been logged"
+Then /^the summary new should be (\d+)$/ do
+	|count_str|
+
+	summary = get_summary @source
+
+	summary["combined"]["new"].should == count_str.to_i
+
 end
 
-Then /^I should see (\d+) summar(?:y|ies)$/ do
+Then /^the summary total should be (\d+)$/ do
 	|count_str|
-	count = count_str.to_i
-	find("#summaries").should have_css(".summary", :count => count)
+
+	summary = get_summary @source
+
+	summary["combined"]["total"].should == count_str.to_i
+
 end
 
-Then /^the (\d+(?:st|nd|rd|th)) summary should be:$/ do
-	|index_str, fields|
+Then /^the summary new for type "(.*?)" should be (\d+)$/ do
+	|type, count_str|
 
-	index = index_str.to_i
+	summary = get_summary @source
 
-	within "#summaries" do
+	summary["types"][type]["new"].should == count_str.to_i
 
-		fields.hashes.each do
-			|row|
-			find(".#{row["name"]}").text.should == row["value"]
-		end
+end
 
-	end
+Then /^the summary total for type "(.*?)" should be (\d+)$/ do
+	|type, count_str|
+
+	summary = get_summary @source
+
+	summary["types"][type]["total"].should == count_str.to_i
 
 end
 
 Then /^icinga should receive:$/ do
-	|expected_command|
+	|expected_commands|
 
 	command_contents =
 		File.new(@command_file).to_a.map { |line| line.strip }
 
-	command_contents.should \
-		include expected_command
+	expected_commands.split("\n").each do
+		|expected_command|
+
+		command_contents.should \
+			include expected_command
+
+	end
 
 end

data/features/support/ui-steps.rb

@@ -0,0 +1,50 @@
+When /^I visit (\/.*)$/ do
+	|url_raw|
+
+	url = url_raw.gsub /\$\{([-a-z]+)\}/ do
+		var_name = $1.gsub "-", "_"
+		instance_variable_get "@#{var_name}"
+	end
+
+	visit url
+
+end
+
+When /^I click "(.*?)"$/ do
+	|target_str|
+
+	click_on target_str
+
+end
+
+Then /^I should see no summaries$/ do
+	page.should have_content "No events have been logged"
+end
+
+Then /^I should see (\d+) summar(?:y|ies)$/ do
+	|count_str|
+	count = count_str.to_i
+	find("#summaries").should have_css(".summary", :count => count)
+end
+
+Then /^the (\d+(?:st|nd|rd|th)) summary should be:$/ do
+	|index_str, fields|
+
+	index = index_str.to_i
+
+	within "#summaries" do
+
+		fields.hashes.each do
+			|row|
+			find(".#{row["name"]}").text.should == row["value"]
+		end
+
+	end
+
+end
+
+Then /^I should see the event$/ do
+
+	find("#event #id td").text.should == @event_id.to_s
+
+end

data/lib/hq/log-monitor-server/do-checks.rb

@@ -0,0 +1,116 @@
+module HQ
+module LogMonitorServer
+
+class Script
+
+	def do_checks
+		@mutex.synchronize do
+
+			File.open @icinga_elem["command-file"], "a" do
+				|command_io|
+
+				summaries_by_service =
+					get_summaries_by_service
+
+				@icinga_elem.find("service").each do
+					|service_elem|
+
+					service_name = service_elem["name"]
+
+					critical_count = 0
+					warning_count = 0
+					unknown_count = 0
+
+					summaries =
+						summaries_by_service[service_name]
+
+					if summaries
+						summaries["types"].each do
+							|type_name, type_info|
+
+							type_elem =
+								service_elem.find_first("
+									type [@name = #{esc_xp type_name}]
+								")
+
+							if ! type_elem
+								unknown_count += type_info["new"]
+							elsif type_elem["level"] == "critical"
+								critical_count += type_info["new"]
+							elsif type_elem["level"] == "warning"
+								warning_count += type_info["new"]
+							else
+								unknown_count += type_info["new"]
+							end
+						end
+					end
+
+					status_int =
+						if critical_count > 0
+							2
+						elsif warning_count > 0
+							1
+						elsif unknown_count > 0
+							3
+						else
+							0
+						end
+
+					status_str =
+						if critical_count > 0
+							"CRITICAL"
+						elsif warning_count > 0
+							"WARNING"
+						elsif unknown_count > 0
+							"UNKNOWN"
+						else
+							"OK"
+						end
+
+					parts = []
+
+					if critical_count > 0
+						parts << "%d critical" % critical_count
+					end
+
+					if warning_count > 0
+						parts << "%d warning" % warning_count
+					end
+
+					if unknown_count > 0
+						parts << "%d unknown" % unknown_count
+					end
+
+					if parts.empty?
+						parts << "no new events"
+					end
+
+
+					command_io.print "[%s] %s\n" % [
+						Time.now.to_i,
+						[
+							"PROCESS_SERVICE_CHECK_RESULT",
+							service_elem["icinga-host"],
+							service_elem["icinga-service"],
+							status_int,
+							"%s %s" % [
+								status_str,
+								parts.join(", "),
+							]
+						].join(";"),
+					]
+
+				end
+
+			end
+
+			@next_check = Time.now + 60
+
+		end
+
+	end
+
+end
+
+end
+end

data/lib/hq/log-monitor-server/event-page.rb

@@ -0,0 +1,174 @@
+module HQ
+module LogMonitorServer
+
+class Script
+
+	def event_page env, context
+
+		req = Rack::Request.new env
+
+		# process form stuff
+
+		if req.request_method == "POST" \
+			&& req.params["mark-as-seen"]
+
+			mark_event_as_seen context[:event_id]
+
+		end
+
+		# read from database
+
+		event =
+			@db["events"]
+				.find_one({
+					"_id" => BSON::ObjectId.from_string(context[:event_id]),
+				})
+
+		# set headers
+
+		headers = {}
+
+		headers["content-type"] = "text/html; charset=utf-8"
+
+		# create page
+
+		html = []
+
+		title =
+			"Event %s \u2014 Log monitor" % [
+				context[:event_id],
+			]
+
+		html << "<!DOCTYPE html>\n"
+		html << "<html>\n"
+		html << "<head>\n"
+
+		html << "<title>%s</title>\n" % [
+			esc_ht(title),
+		]
+
+		html << "</head>\n"
+		html << "<body>\n"
+
+		html << "<h1>%s</h1>\n" % [
+			esc_ht(title),
+		]
+
+		unless event
+
+			html << "<p>Event id not recognised</p>\n"
+
+		else
+
+			html << "<table id=\"event\">\n"
+			html << "<tbody>\n"
+
+			html << "<tr id=\"id\">\n"
+			html << "<th>ID</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["_id"].to_s),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Timestamp</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["timestamp"].to_s),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Service</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["source"]["service"]),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Host</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["source"]["host"]),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Class</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["source"]["class"]),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Filename</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["location"]["file"]),
+			]
+			html << "</tr>\n"
+
+			html << "<tr>\n"
+			html << "<th>Line number</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht((event["location"]["line"] + 1).to_s),
+			]
+			html << "</tr>\n"
+
+			unless event["lines"]["before"].empty?
+
+				html << "<tr>\n"
+				html << "<th>Before</th>\n"
+				html << "<td>%s</td>\n" % [
+					event["lines"]["before"]
+						.map { |line| esc_ht(line) }
+						.join("<br>")
+				]
+
+			end
+
+			html << "<tr>\n"
+			html << "<th>Matching</th>\n"
+			html << "<td>%s</td>\n" % [
+				esc_ht(event["lines"]["matching"]),
+			]
+
+			unless event["lines"]["after"].empty?
+
+				html << "<tr>\n"
+				html << "<th>Before</th>\n"
+				html << "<td>%s</td>\n" % [
+					event["lines"]["after"]
+						.map { |line| esc_ht(line) }
+						.join("<br>")
+				]
+
+			end
+
+			html << "</tbody>\n"
+			html << "</table>\n"
+
+		end
+
+		html << "<form method=\"post\">\n"
+
+		html <<
+			"<p>" +
+			"<input " +
+				"type=\"submit\" " +
+				"name=\"mark-as-seen\" " +
+				"value=\"mark as seen\">" +
+			"</p>\n"
+
+		html << "</form>\n"
+
+		html << "</body>\n"
+		html << "</html>\n"
+
+		# return
+
+		return 200, headers, html
+
+	end
+
+end
+
+end
+end