RubyGems - snowman-io - Versions diffs - 0.0.3 → 0.0.4 - Mend

snowman-io 0.0.3 → 0.0.4

Files changed (47) hide show

checksums.yaml +4 -4
data/CHANGELOG.md +5 -1
data/README.md +3 -17
data/bin/snowman +1 -0
data/lib/snowman-io/api/public/README.md +36 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap-theme.css +470 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap-theme.css.map +1 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap-theme.min.css +5 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap.css +6332 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap.css.map +1 -0
data/lib/snowman-io/api/public/bootstrap/css/bootstrap.min.css +5 -0
data/lib/snowman-io/api/public/bootstrap/fonts/glyphicons-halflings-regular.eot +0 -0
data/lib/snowman-io/api/public/bootstrap/fonts/glyphicons-halflings-regular.svg +229 -0
data/lib/snowman-io/api/public/bootstrap/fonts/glyphicons-halflings-regular.ttf +0 -0
data/lib/snowman-io/api/public/bootstrap/fonts/glyphicons-halflings-regular.woff +0 -0
data/lib/snowman-io/api/public/bootstrap/js/bootstrap.js +2320 -0
data/lib/snowman-io/api/public/bootstrap/js/bootstrap.min.js +7 -0
data/lib/snowman-io/api/public/bootstrap/js/npm.js +13 -0
data/lib/snowman-io/api/public/css/normalize.css +406 -0
data/lib/snowman-io/api/public/css/style.css +4 -0
data/lib/snowman-io/api/public/js/app.js +13 -0
data/lib/snowman-io/api/public/js/libs/ember-1.8.1.js +49740 -0
data/lib/snowman-io/api/public/js/libs/handlebars-v1.3.0.js +2746 -0
data/lib/snowman-io/api/public/js/libs/jquery-1.10.2.js +9789 -0
data/lib/snowman-io/api/public/tests/runner.css +14 -0
data/lib/snowman-io/api/public/tests/runner.js +13 -0
data/lib/snowman-io/api/public/tests/tests.js +30 -0
data/lib/snowman-io/api/public/tests/vendor/qunit-1.12.0.css +244 -0
data/lib/snowman-io/api/public/tests/vendor/qunit-1.12.0.js +2212 -0
data/lib/snowman-io/api/views/index.erb +26 -0
data/lib/snowman-io/api/views/layout.erb +24 -0
data/lib/snowman-io/api/views/login.erb +21 -0
data/lib/snowman-io/api/views/unpacking.erb +21 -0
data/lib/snowman-io/api.rb +61 -1
data/lib/snowman-io/check.rb +49 -0
data/lib/snowman-io/check_result.rb +15 -0
data/lib/snowman-io/checks/hosted_graphite.rb +23 -0
data/lib/snowman-io/handler.rb +27 -0
data/lib/snowman-io/launcher.rb +24 -0
data/lib/snowman-io/notifiers/slack.rb +76 -0
data/lib/snowman-io/options.rb +3 -14
data/lib/snowman-io/processor.rb +32 -0
data/lib/snowman-io/scheduler.rb +42 -0
data/lib/snowman-io/version.rb +1 -1
data/lib/snowman-io.rb +34 -5
data/snowman-io.gemspec +9 -2
metadata +131 -10

data/lib/snowman-io/api/views/index.erb ADDED Viewed

@@ -0,0 +1,26 @@
+<% content_for :head do %>
+  <meta content="<%= SnowmanIO::VERSION %>" name="snowman-io-version" />
+  <link rel="stylesheet" href="/css/style.css">
+<% end %>
+<script type="text/x-handlebars">
+<div class="container">
+  <h2>
+    Welcome to SnowmanIO
+    <small><a href="/logout">logout</a></small>
+  </h2>
+  {{outlet}}
+</div>
+</script>
+<script type="text/x-handlebars" id="index">
+  <p>Version: {{model.version}}</p>
+</script>
+<script src="/js/libs/jquery-1.10.2.js"></script>
+<script src="/js/libs/handlebars-v1.3.0.js"></script>
+<script src="/js/libs/ember-1.8.1.js"></script>
+<script src="/js/app.js"></script>
+<!-- to activate the test runner, add the "?test" query string parameter -->
+<script src="/tests/runner.js"></script>

data/lib/snowman-io/api/views/layout.erb ADDED Viewed

@@ -0,0 +1,24 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf-8">
+    <meta http-equiv="X-UA-Compatible" content="IE=edge">
+    <meta name="viewport" content="width=device-width, initial-scale=1">
+    <title>SnowmanIO: be a little snowy</title>
+    <!-- Bootstrap -->
+    <link href="/bootstrap/css/bootstrap.min.css" rel="stylesheet">
+    <!-- HTML5 shim and Respond.js for IE8 support of HTML5 elements and media queries -->
+    <!-- WARNING: Respond.js doesn't work if you view the page via file:// -->
+    <!--[if lt IE 9]>
+      <script src="https://oss.maxcdn.com/html5shiv/3.7.2/html5shiv.min.js"></script>
+      <script src="https://oss.maxcdn.com/respond/1.4.2/respond.min.js"></script>
+    <![endif]-->
+    <%= yield_content :head %>
+  </head>
+  <body>
+    <%= yield %>
+  </body>
+</html>

data/lib/snowman-io/api/views/login.erb ADDED Viewed

@@ -0,0 +1,21 @@
+<div class="container">
+  <div class="page-header">
+    <h1>SnowmanIO</h1>
+  </div>
+  <% if @wrong_password_warning %>
+    <div class="alert alert-danger">
+      Wrong password.
+    </div>
+  <% end %>
+  <div class="col-sm-6">
+    <form action="/login" method="POST" role="form">
+      <div class="form-group">
+        <label for=password>Password</label>
+        <input type=password name=password class="form-control">
+      </div>
+      <input type=submit value="Login" class="btn btn-default">
+    </form>
+  </div>
+</div>

data/lib/snowman-io/api/views/unpacking.erb ADDED Viewed

@@ -0,0 +1,21 @@
+<div class="container">
+  <div class="page-header">
+    <h1>SnowmanIO: Installation</h1>
+  </div>
+  <% if @empty_password_warning %>
+    <div class="alert alert-danger">
+      Empty password is not allowed.
+    </div>
+  <% end %>
+  <div class="col-sm-6">
+    <form action="/unpacking" method="POST" role="form">
+      <div class="form-group">
+        <label for=password>Admin password</label>
+        <input type=text name=password class="form-control">
+      </div>
+      <input type=submit value="Set Admin Password" class="btn btn-primary">
+    </form>
+  </div>
+</div>

data/lib/snowman-io/api.rb CHANGED Viewed

@@ -1,4 +1,5 @@
 require 'sinatra'
+require 'sinatra/content_for'
 module SnowmanIO
   class API < Sinatra::Base
@@ -8,8 +9,67 @@ module SnowmanIO
       run!(sinatra_options)
     end
+    enable :sessions
+    helpers Sinatra::ContentFor
+    set :public_folder, File.dirname(__FILE__) + "/api/public"
+    set :views, File.dirname(__FILE__) + "/api/views"
+    # TODO: fix it
+    set :session_secret, 'super secret'
+    ADMIN_PASSWORD_KEY = "admin_password_hash"
+    def admin_exists?
+      !!SnowmanIO.redis.get(ADMIN_PASSWORD_KEY)
+    end
+    def admin_authenticated?
+      admin_exists? && !!session[:user]
+    end
+    before do
+      if !admin_exists?
+        redirect to('/unpacking') if request.path_info != '/unpacking'
+      elsif !admin_authenticated?
+        redirect to('/login') if request.path_info != '/login'
+      end
+    end
     get "/" do
-      "SnowManIO Home Page"
+      erb :index
+    end
+    get "/login" do
+      erb :login
+    end
+    post "/login" do
+      if BCrypt::Password.new(SnowmanIO.redis.get(ADMIN_PASSWORD_KEY)) == params["password"]
+        session[:user] = "admin"
+        redirect to('/')
+      else
+        @wrong_password_warning = true
+        erb :login
+      end
+    end
+    get "/logout" do
+      session[:user] = nil
+      redirect to("/login")
+    end
+    get "/unpacking" do
+      erb :unpacking
+    end
+    post "/unpacking" do
+      if params["password"].empty?
+        @empty_password_warning = true
+        erb :unpacking
+      else
+        session[:user] = "admin"
+        SnowmanIO.redis.set(ADMIN_PASSWORD_KEY, BCrypt::Password.create(params["password"]))
+        redirect to('/')
+      end
     end
   end
 end

data/lib/snowman-io/check.rb ADDED Viewed

@@ -0,0 +1,49 @@
+require 'json'
+require 'open-uri'
+require "active_support/time"
+require 'snowman-io/notifiers/slack'
+module SnowmanIO
+  class Check
+    include Checks::HostedGraphite
+    DEFAULT_INTERVAL = 1.minute
+    class << self
+      def interval(value = nil)
+        if value
+          @interval = value
+        else
+          @interval || DEFAULT_INTERVAL
+        end
+      end
+      def human(value = nil)
+        if value
+          @human = value
+        else
+          self.name + ": #{@human}"
+        end
+      end
+      def notifiers
+        @notifiers ||= [Notifiers::Slack].select { |notifier| notifier.configured? }
+      end
+    end
+    def perform
+      if ok?
+        status = "success"
+        message = self.class.human + " - OK"
+      else
+        status = "failed"
+        message = self.class.human + " - FAIL"
+      end
+      CheckResult.new(self.class, status, message)
+    end
+    def ok?
+      raise "Implement ok? in check class"
+    end
+  end
+end

data/lib/snowman-io/check_result.rb ADDED Viewed

@@ -0,0 +1,15 @@
+module SnowmanIO
+  class CheckResult < Struct.new(:check, :status, :message)
+    def check_name
+      check.name
+    end
+    def fail?
+      status == "failed"
+    end
+    def serialize
+      JSON.dump({ status: status, message: message })
+    end
+  end
+end

data/lib/snowman-io/checks/hosted_graphite.rb ADDED Viewed

@@ -0,0 +1,23 @@
+module SnowmanIO
+  module Checks
+    # Gets last value from Hosted Graphite metric (https://www.hostedgraphite.com/)
+    module HostedGraphite
+      protected
+      def get_hg_value(metric, options = {})
+        access_key = ENV["HG_KEY"]
+        return nil unless access_key
+        base_url = "https://www.hostedgraphite.com#{access_key}/graphite/render"
+        from = options[:from] || "-10mins"
+        url = base_url + "?format=json&target=#{URI.escape metric}&from=#{from}"
+        handle = open(url)
+        raw_data = JSON.parse(handle.gets)
+        raw = raw_data.first
+        datapoints = raw['datapoints'].delete_if { |v| v.first.nil? }
+        if datapoints.last
+          datapoints.last.first
+        end
+      end
+    end
+  end
+end

data/lib/snowman-io/handler.rb ADDED Viewed

@@ -0,0 +1,27 @@
+module SnowmanIO
+  # Handler process result from checks, saves them
+  # to redis and fires notifications if needed.
+  class Handler
+    include Celluloid
+    def handle(result)
+      history_key = "history:#{result.check_name.underscore}"
+      SnowmanIO.redis.rpush(history_key, result.serialize)
+      history = SnowmanIO.redis.lrange(history_key, -4, -1).map { |result| JSON.load(result) }
+      previous_status_failed = if history.size < 4
+        false
+      else
+        ["failed", "exception"].include?(history.shift["status"])
+      end
+      if !previous_status_failed && history.size >= 3 && history.all? { |result| result["status"] == "failed" || result["status"] == "exception" }
+        notify_fail(result)
+      end
+    end
+    private
+    def notify_fail(result)
+      result.check.notifiers.each { |notifier| notifier.pool.async.notify(result) }
+    end
+  end
+end

data/lib/snowman-io/launcher.rb ADDED Viewed

@@ -0,0 +1,24 @@
+require 'snowman-io/scheduler'
+require 'snowman-io/handler'
+module SnowmanIO
+  class Launcher
+    include Celluloid
+    attr_reader :scheduler, :handler
+    def initialize(checks)
+      @handler = Handler.new_link
+      @scheduler = Scheduler.new_link(checks)
+      @scheduler.handler = handler
+    end
+    def start
+      scheduler.async.start
+    end
+    def stop
+      #TODO
+    end
+  end
+end

data/lib/snowman-io/notifiers/slack.rb ADDED Viewed

@@ -0,0 +1,76 @@
+module SnowmanIO
+  module Notifiers
+    class Slack
+      include Celluloid
+      class << self
+        def webhook_url
+          ENV["SLACK_WEBHOOK_URL"]
+        end
+        def channel
+          ENV["SLACK_CHANNEL"]
+        end
+        def bot_name
+          ENV["SLACK_BOT_NAME"]
+        end
+        def configured?
+          webhook_url.present?
+        end
+      end
+      def notify(result)
+        return unless self.class.configured?
+        post_data(result.message)
+      end
+      private
+      def configuration
+        {
+          :webhook_url => self.class.webhook_url,
+          :bot_name => self.class.bot_name || "snowman-io",
+          :channel => self.class.channel
+        }
+      end
+      def post_data(message)
+        uri = URI(configuration[:webhook_url])
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        req = Net::HTTP::Post.new("#{uri.path}?#{uri.query}")
+        req.body = payload(message).to_json
+        response = http.request(req)
+        verify_response(response)
+      end
+      def slack_uri(token)
+        url = "https://#{team_name}.slack.com/services/hooks/incoming-webhook?token=#{token}"
+        URI(url)
+      end
+      def verify_response(response)
+        case response
+        when Net::HTTPSuccess
+          true
+        else
+          raise response.error!
+        end
+      end
+      # TODO: include :icon_url
+      def payload(text)
+        {
+          :username    => configuration[:bot_name],
+          :attachments => [{
+            :text  => text,
+            :color => "#FF0000"
+          }]
+        }.tap { |payload| payload[:channel] = channel if configuration[:channel] }
+      end
+    end
+  end
+end

data/lib/snowman-io/options.rb CHANGED Viewed

@@ -9,26 +9,15 @@ module SnowmanIO
       options = default_options
       opt_parser = OptionParser.new do |opts|
-        opts.banner = "Usage: snowman COMMAND [options]"
+        opts.banner = "Usage: snowman [options]"
         opts.separator ""
-        opts.separator "Commands"
-        opts.separator "  server       Run SnowManIO server"
-        opts.separator ""
-        opts.separator "Server options:"
+        opts.separator "Options:"
         opts.on("-p", "--port PORT", "use PORT (default: 4567)") do |port|
           options[:port] = port.to_i
         end
-        if args.empty?
-          puts opts
-          exit
-        end
-        options[:command] = args.shift
-        unless AVAILABLE_COMMANDS.include?(options[:command])
-          puts "Error: Command '#{options[:command]}' not recognized"
+        opts.on("-h", "--help", "show this message") do
           puts opts
           exit
         end

data/lib/snowman-io/processor.rb ADDED Viewed

@@ -0,0 +1,32 @@
+module SnowmanIO
+  # Processor initiated by Scheduler, executes check and notifies
+  # Scheduler about the result of the check.
+  class Processor
+    include Celluloid
+    def initialize(scheduler)
+      @scheduler = scheduler
+    end
+    # TODO: logging can be extracted in some kind of middleware
+    def process(check)
+      begin
+        SnowmanIO.logger.info("Processing check #{check.human}, started at #{Time.now}")
+        result = check.new.perform
+      rescue Exception => e
+        result = result_from_exception(check, e)
+        raise
+      ensure
+        SnowmanIO.logger.info("Processing check #{check.human}, finished at #{Time.now}")
+        @scheduler.processor_done(current_actor, result)
+      end
+    end
+    private
+    def result_from_exception(check, e)
+      message = "Check #{check.human} was interruppted by exception: #{e.class}: #{e.message}"
+      CheckResult.new(check, "exception", message)
+    end
+  end
+end

data/lib/snowman-io/scheduler.rb ADDED Viewed

@@ -0,0 +1,42 @@
+require "snowman-io/processor"
+module SnowmanIO
+  # Scheduler schedules execution of checks.
+  class Scheduler
+    include Celluloid
+    trap_exit :processor_died
+    attr_accessor :handler
+    def initialize(checks)
+      @time = {}
+      checks.each do |check|
+        @time[check] = Time.now + check.interval
+      end
+    end
+    def start
+      every(1) { schedule_checks }
+    end
+    def processor_done(processor, result)
+      @handler.async.handle(result)
+      processor.terminate
+    end
+    private
+    def schedule_checks
+      @time.each do |check, time|
+        if time <= Time.now
+          Processor.new_link(current_actor).async.process(check)
+          @time[check] = Time.now + check.interval
+        end
+      end
+    end
+    def processor_died(_actor, _reason)
+      # TODO
+    end
+  end
+end

data/lib/snowman-io/version.rb CHANGED Viewed

@@ -1,3 +1,3 @@
 module SnowmanIO
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/snowman-io.rb CHANGED Viewed

@@ -1,17 +1,46 @@
+require 'logger'
+require 'redis'
+require 'bcrypt'
+require 'celluloid/autostart'
 require "snowman-io/version"
 require "snowman-io/api"
 require "snowman-io/options"
+require "snowman-io/checks/hosted_graphite"
+require "snowman-io/check"
+require "snowman-io/check_result"
+require "snowman-io/launcher"
 module SnowmanIO
   def self.start
     # parse options
     options = Options.new.parse!(ARGV)
-    case options[:command]
-    when "server"
-      API.start(options)
-    else
-      abort "Unreacheable point. Please report the bug to https://github.com/snowman-io/snowman-io/issues"
+    Celluloid.logger = (options[:verbose] ? SnowmanIO.logger : nil)
+    launcher = Launcher.new(load_checks)
+    launcher.start
+    # start web server on main thread
+    API.start(options)
+  end
+  def self.load_checks
+    checks = []
+    Dir[Dir.pwd + "/**/*_check.rb"].each do |path|
+      require path
+      klass = path.sub(Dir.pwd + "/checks" + "/", "").sub(/\.rb$/, '').camelize
+      checks.push Kernel.const_get(klass)
     end
+    checks
+  end
+  def self.redis
+    @redis ||= Redis.new(url: ENV["REDIS_URL"])
+  end
+  def self.logger
+    @logger ||= Logger.new(STDERR)
   end
 end

data/snowman-io.gemspec CHANGED Viewed

@@ -19,8 +19,15 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
-  spec.add_dependency "sinatra"
+  spec.add_dependency "sinatra", "~> 1.4"
+  spec.add_dependency "sinatra-contrib", "~> 1.4"
+  spec.add_dependency "celluloid", "~> 0.16.0"
+  spec.add_dependency "redis", "~> 3.1.0"
+  spec.add_dependency "activesupport", "~> 4.1.8"
+  spec.add_dependency "bcrypt", "~> 3.1"
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "rspec", "~> 3.1"
+  spec.add_development_dependency "capybara", "~> 2.4"
 end