dash.rb

require "rubygems"
require "namespace"

require "config"
require "erb"
require "helpers"
require "models"
require "rack"
require "sinatra/base"
require "json"
require "open-uri"
require "yaml"
require "chronic"
require "chronic_duration"
require "optparse"
require "rubyfixes"

$:.unshift(File.dirname(__FILE__))

module Dash
  class App < Sinatra::Base
    include Dash::Models
    helpers Dash::Helpers
    config = Dash::Config.new
    set :config, config
    set :port, config.global_config[:port]
    set :run, true
    use Rack::Session::Cookie, :expire_after => 126227700 # 4 years
    set :root, File.dirname(__FILE__)
    set :static, true
    set :logging, true
    set :erb, :trim => '-'

    def initialize(settings={})
      super
    end

    before do
      session[:not] #fixme kludge is back
      @request_time = Time.now
      @dashboards = Dashboard.all
      @no_graphs = false
      # time stuff
      start = param_lookup("start")
      duration = param_lookup("duration")
      @stime = Chronic.parse(start)
      if @stime
        @stime -= @stime.sec unless @params["noq"]
      end
      if duration
        @duration = ChronicDuration.parse(duration) || 0
      else
        @duration = @request_time.to_i - @stime.to_i
      end

      unless @params["noq"]
        @duration -= (@duration % settings.config.global_config[:quantum]||1)
      end

      if @stime
        @etime = Time.at(@stime + @duration)
        @etime = @request_time if @etime > @request_time
      else
        @etime = @request_time
      end

      session[:stime] = @stime.to_i.to_s
      session[:etime] = @etime.to_i.to_s
      # fixme reload hosts after some expiry
    end

    get %r[^/(dash/?)?$] do
      @no_graphs = true
      if settings.config.clusters.size == 1
        redirect append_query_string("/dash/#{settings.config.clusters.first}")
      else
        redirect append_query_string('/dash/global')
      end
    end

    get '/dash/:cluster/:dashboard/:zoom/?' do
      @cluster = params[:cluster]
      @dash = Dashboard.find(params[:dashboard])
      raise "Unknown dashboard: #{params[:dashboard].inspect}" unless @dash

      @zoom = nil
      @dash.graphs.each do |graph|
        @zoom = graph if graph.name == params[:zoom]
      end
      raise "Unknown zoom parameter: #{params[:zoom]}" unless @zoom

      @title = "dashboard :: #{@cluster} :: #{@dash['title']} :: #{params[:zoom]}"

      if @cluster == "global"
        erb :'dash-global-zoom'
      else
        erb :'dash-cluster-zoom'
      end
    end

    get '/dash/:cluster/:dashboard/?' do
      @cluster = params[:cluster]
      @dash = Dashboard.find(params[:dashboard])
      raise "Unknown dashboard: #{params[:dashboard].inspect}" unless @dash

      @title = "dashboard :: #{@cluster} :: #{@dash['title']}"

      if @cluster == "global"
        erb :'dash-global'
      else
        erb :'dash-cluster'
      end
    end

    get '/dash/:cluster/?' do
      @no_graphs = true
      @cluster = params[:cluster]
      if @cluster == "global"
        @title = "Overview"
        erb :global
      else
        @title = "cluster :: #{params[:cluster]}"
        erb :cluster
      end
    end

    get '/host/:cluster/:host/?' do
      @host = Host.find_by_name_and_cluster(params[:host], params[:cluster])
      @cluster = @host.cluster
      raise "Unknown host: #{params[:host]} in #{params[:cluster]}" unless @host

      @title = "#{@host.cluster} :: host :: #{@host.name}"

      erb :host
    end

    get '/process' do
      case params["action"]
      when "Save"
        # fixme make sure not to save shitty values for :start
        puts 'saving prefs'
        params.each do |k ,v|
          next if [:etime, :stime, :duration].member?(k.to_sym)
          session[k] = v unless v.empty?
        end
        redirect URI.parse(request.referrer).path
      when "Clear"
        puts URI.parse(request.referrer).path
        redirect URI.parse(request.referrer).path
      when "Reset"
        puts "clearing prefs"
        session.clear
        redirect URI.parse(request.referrer).path
      when "Submit"
        # fixme offensive to sensibility
        redirect URI.parse(request.referrer).path + "?" + \
        request.query_string.sub("&action=Submit", "").sub("?action=Submit", "")
      end
    end
  end # Dash::App
end # Dash