sensu-dashboard-sonian 0.9.3

data/.gitignore

@@ -0,0 +1,6 @@
+.DS_Store
+.sass-cache/
+.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in sensu-dashboard.gemspec
+gemspec

data/MIT-LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Sonian Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.org

@@ -0,0 +1,4 @@
+* Sonian Monitoring Dashboard
+- Lists current alerts from the monitoring server
+- Lists clients subscribed to AMQP queues
+- Uses WebSockets to receive new information

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/sensu-dashboard

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+begin
+  require 'sensu-dashboard/app.rb'
+rescue LoadError => e
+  require 'rubygems'
+  path = File.expand_path '../../lib', __FILE__
+  $:.unshift(path) if File.directory?(path) && !$:.include?(path)
+  require 'sensu-dashboard/app.rb'
+end

data/lib/sensu-dashboard/app.rb

@@ -0,0 +1,449 @@
+require 'eventmachine'
+require 'sinatra/async'
+require 'em-http-request'
+require 'em-websocket'
+require 'sass'
+require 'json'
+require 'sensu/config'
+
+options = Sensu::Config.read_arguments(ARGV)
+config = Sensu::Config.new(options)
+SETTINGS = config.settings
+
+EventMachine.run do
+
+  class DashboardServer < Sinatra::Base
+
+    register Sinatra::Async
+    set :root, File.dirname(__FILE__)
+    set :static, true
+    set :public_folder, Proc.new { File.join(root, "public") }
+
+    api_server = 'http://' + SETTINGS['api']['host'] + ':' + SETTINGS['api']['port'].to_s
+
+    use Rack::Auth::Basic do |user, password|
+      user == SETTINGS['dashboard']['user'] && password == SETTINGS['dashboard']['password']
+    end
+
+    before do
+      content_type 'application/json'
+    end
+
+    aget '/' do
+      content_type 'text/html'
+      @js = erb :event_templates, :layout => false
+      body erb :index
+    end
+
+    aget '/clients' do
+      content_type 'text/html'
+      @js = erb :client_templates, :layout => false
+      body erb :clients
+    end
+
+    aget '/stashes' do
+      content_type 'text/html'
+      @js = erb :stash_templates, :layout => false
+      body erb :stashes
+    end
+
+    aget '/css/sonian.css' do
+      content_type 'text/css'
+      body sass :sonian
+    end
+
+    # api proxy
+    aget '/events.json' do
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/events").get
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not retrieve events from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not retrieve events from the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    # api proxy
+    aget '/autocomplete.json' do
+      multi = EventMachine::MultiRequest.new
+
+      requests = [
+        "#{api_server}/events",
+        "#{api_server}/clients"
+      ]
+
+      requests.each do |url|
+        multi.add EventMachine::HttpRequest.new(url).get
+      end
+
+      multi.callback do
+        events = {}
+        clients = []
+
+        multi.responses[:succeeded].each do |request|
+          body = JSON.parse(request.response)
+          case body
+          when Hash
+            events = body
+          when Array
+            clients = body
+          end
+        end
+
+        if events && clients
+          autocomplete = []
+          statuses = {:warning => [], :critical => [], :unknown => []}
+          subscriptions = {}
+          checks = []
+
+          # searching by client
+          clients.each do |client|
+            client_name = client['name']
+            if events.include?(client_name)
+              autocomplete.push({:value => [client_name], :type => 'client', :name => client_name})
+              client['subscriptions'].each do |subscription|
+                subscriptions[subscription] ||= []
+                subscriptions[subscription].push(client_name)
+              end
+              events[client_name].each do |check, event|
+                
+                case event['status']
+                when 1
+                  statuses[:warning].push(event['status'])
+                when 2
+                  statuses[:critical].push(event['status'])
+                else
+                  statuses[:unknown].push(event['status'])
+                end
+                checks.push(check)
+              end
+            end
+          end
+
+          # searching by subscription
+          subscriptions.each do |k, v|
+            autocomplete.push({:value => v.uniq, :type => 'subscription', :name => k})
+          end
+
+          # searching by status
+          statuses.each do |k, v|
+            autocomplete.push({:value => v.uniq, :type => 'status', :name => k})
+          end
+
+          # searching by check
+          checks.uniq.each do |v|
+            autocomplete.push({:value => [v], :type => 'check', :name => v})
+          end
+
+          body autocomplete.to_json
+        else
+          status 404
+          body '{"error":"could not retrieve events and/or clients from the sensu api"}'
+        end
+      end
+    end
+
+    aget '/clients.json' do
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/clients").get
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not retrieve clients from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not retrieve clients from the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    # api proxy
+    aget '/clients/autocomplete.json' do
+      multi = EventMachine::MultiRequest.new
+
+      requests = [
+        "#{api_server}/clients"
+      ]
+
+      requests.each do |url|
+        multi.add EventMachine::HttpRequest.new(url).get
+      end
+
+      multi.callback do
+        events = {}
+        clients = []
+
+        multi.responses[:succeeded].each do |request|
+          body = JSON.parse(request.response)
+          case body
+          when Array
+            clients = body
+          end
+        end
+
+        if clients
+          autocomplete = []
+          subscriptions = {}
+
+          # searching by client
+          clients.each do |client|
+            client_name = client['name']
+            autocomplete.push({:value => [client_name], :type => 'client', :name => client_name})
+            client['subscriptions'].each do |subscription|
+              subscriptions[subscription] ||= []
+              subscriptions[subscription].push(client_name)
+            end
+          end
+
+          # searching by subscription
+          subscriptions.each do |k, v|
+            autocomplete.push({:value => v.uniq, :type => 'subscription', :name => k})
+          end
+
+          body autocomplete.to_json
+        else
+          status 404
+          body '{"error":"could not retrieve clients from the sensu api"}'
+        end
+      end
+    end
+
+    aget '/client/:id.json' do |id|
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/client/#{id}").get
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not retrieve client from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not retrieve client from the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    adelete '/client/:id.json' do |id|
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/client/#{id}").delete
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not delete client from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not delete client from the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    aget '/stash/*.json' do |path|
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/stash/#{path}").get
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not retrieve a stash from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not retrieve a stash from the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    apost '/stash/*.json' do |path|
+      begin
+        request_options = {
+          :body => {'timestamp' => Time.now.to_i}.to_json,
+          :head => {
+            'content-type' => 'application/json'
+          }
+        }
+        http = EventMachine::HttpRequest.new("#{api_server}/stash/#{path}").post request_options
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not create a stash with the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not create a stash with the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    adelete '/stash/*.json' do |path|
+      begin
+        http = EventMachine::HttpRequest.new("#{api_server}/stash/#{path}").delete
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not delete a stash with the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not delete a stash with the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    apost '/event/resolve.json' do
+      begin
+        request_options = {
+          :body => request.body.read,
+          :head => {
+            'content-type' => 'application/json'
+          }
+        }
+        http = EventMachine::HttpRequest.new("#{api_server}/event/resolve").post request_options
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not resolve an event with the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not resolve an event with the sensu api"}'
+      end
+
+      http.callback do
+        status http.response_header.status
+        body http.response
+      end
+    end
+
+    aget '/stashes.json' do
+      begin
+        request_options = {
+#          :body => request.body.read,
+          :head => {
+            'content-type' => 'application/json'
+          }
+        }
+        http = EventMachine::HttpRequest.new("#{api_server}/stashes").get request_options
+      rescue => e
+        puts e
+        status 404
+        body '{"error":"could not retrieve a list of stashes from the sensu api"}'
+      end
+
+      http.errback do
+        status 404
+        body '{"error":"could not retrieve a list of stashes from the sensu api"}'
+      end
+
+      http.callback do
+        resp = http.response
+        puts resp
+        status http.response_header.status
+        body resp
+      end
+    end
+
+    apost '/stashes.json' do
+      begin
+        request_options = {
+          :body => request.body.read,
+          :head => {
+            'content-type' => 'application/json'
+          }
+        }
+        http = EventMachine::HttpRequest.new("#{api_server}/stashes").post request_options
+      rescue => e
+        puts e
+        status 404
+        body '{\"error\":\"could not retrieve a list of stashes from the sensu api\"}'
+      end
+
+      http.errback do
+        status 404
+        body '{\"error\":\"could not retrieve a list of stashes from the sensu api\"}'
+      end
+
+      http.callback do
+        resp = http.response
+        puts resp
+        status http.response_header.status
+        body resp
+      end
+    end
+
+    websocket_connections = Array.new
+    EventMachine::WebSocket.start(:host => "0.0.0.0", :port => 9000) do |websocket|
+      websocket.onopen do
+        websocket_connections.push websocket
+        puts 'client connected to websocket'
+      end
+      websocket.onclose do
+        websocket_connections.delete websocket
+        puts 'client disconnected from websocket'
+      end
+    end
+
+    apost '/events.json' do
+      unless websocket_connections.empty?
+        websocket_connections.each do |websocket|
+          websocket.send '{"update":"true"}'
+        end
+      end
+      body '{"success":"triggered dashboard refresh"}'
+    end
+  end
+
+  DashboardServer.run!({:port => SETTINGS['dashboard']['port']})
+
+  #
+  # Recognize exit command
+  #
+  Signal.trap("INT") do
+    EM.stop
+  end
+  Signal.trap("TERM") do
+    EM.stop
+  end
+
+end