opswatch 0.1.0

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2018 Gianluca Rizzo
+
+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.md

@@ -0,0 +1,60 @@
+# Opswatch [![CircleCI](https://circleci.com/gh/gianlucarizzo/opswatch.svg?style=svg)](https://circleci.com/gh/gianlucarizzo/opswatch)
+
+Version 0.1.0 (first release)
+
+OpsWatch lets you track your AWS OpsWorks stacks status in real time on your traybar with a single glance.
+Easy to configure and to use. This first version has no fancy features, but new features and ideas are welcome!
+
+## Installation
+
+To install simply use
+```
+gem install opswatch
+```
+
+## Usage
+
+First rum will ask you to input Amazon AWS stack information:
+```
+opswatch
+Let's set things up
+Stack Name (can be anything):
+Production
+Region (e.g. eu-central-1):
+eu-central-1
+IAM User key:
+XXXXXXXXX
+IAM User secret:
+XXXXXXXXX
+OpsWorks Stack ID:
+id1234567
+Insert other Stack? (Y/n)
+n
+```
+a file `.opswatch.yml` in your $HOME path will be created to store your configuration, and the traybar will be shown on your UI.
+
+After this, you need to simply run
+``
+opswatch
+``
+again to use the last settings.
+
+If you want to start from scratch you can just delete the `.opswatch.yml` file.
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/gianlucarizzo/opswatch. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the Opswatch project’s codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/opswatch/blob/master/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList["test/**/*_test.rb"]
+end
+
+task :default => :test

data/bin/console

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "opswatch"
+
+require "irb"
+IRB.start(__FILE__)

data/bin/opswatch

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+GEM_ROOT = File.expand_path("../..", __FILE__)
+
+require 'opswatch'
+
+Opswatch::Main.start ARGV

data/bin/setup

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install

data/icons/hold.svg

@@ -0,0 +1,23 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Generator: Adobe Illustrator 17.0.1, SVG Export Plug-In . SVG Version: 6.00 Build 0)  -->
+<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
+<svg version="1.1" id="Livello_1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px"
+	 width="566.93px" height="566.93px" viewBox="0 0 566.93 566.93" enable-background="new 0 0 566.93 566.93" xml:space="preserve">
+<rect x="340.33" y="291.464" fill="#E8CB51" width="212.598" height="212.598"/>
+<path fill="#E8CB51" d="M46.79,243.211l-33.732-13.238c-4.846,16.065-7.845,32.93-8.745,50.351h35.946
+	C41.124,267.532,43.352,255.115,46.79,243.211z"/>
+<path fill="#E8CB51" d="M47.639,162.468c-10.862,14.211-19.979,29.828-27.017,46.537l33.702,13.226
+	c5.882-13.608,13.438-26.319,22.38-37.905L47.639,162.468z"/>
+<path fill="#E8CB51" d="M113.333,105.613c-19.074,10.626-36.345,24.099-51.231,39.852l29.322,22.052
+	c11.716-11.777,25.082-21.907,39.714-30.033L113.333,105.613z"/>
+<path fill="#E8CB51" d="M199.473,79.566c-23.294,1.86-45.526,7.474-66.096,16.238l17.93,32.095
+	c15.138-6.05,31.304-10.071,48.166-11.729V79.566z"/>
+<g>
+	<path fill="#E8CB51" d="M46.743,341.536c-3.587-12.468-5.849-25.497-6.607-38.93H4.312c0.842,16.305,3.533,32.12,7.846,47.251
+		L46.743,341.536z"/>
+	<path fill="#E8CB51" d="M227.896,79.166v36.477c92.77,5.63,166.27,82.636,166.27,176.822c0,97.846-79.32,177.165-177.165,177.165
+		c-72.915,0-135.522-44.056-162.698-106.995l-34.877,8.392c31.494,77.995,107.894,133.037,197.174,133.037
+		c117.415,0,212.598-95.184,212.598-212.598C429.197,177.84,340.058,85.039,227.896,79.166z"/>
+</g>
+<polygon fill="#E7CA50" points="248,162.468 213.52,296.477 102,375 255,326.231 "/>
+</svg>

data/icons/negative.svg

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Generator: Adobe Illustrator 17.0.1, SVG Export Plug-In . SVG Version: 6.00 Build 0)  -->
+<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
+<svg version="1.1" id="Livello_1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px"
+	 width="566.93px" height="566.93px" viewBox="0 0 566.93 566.93" enable-background="new 0 0 566.93 566.93" xml:space="preserve">
+<path fill="#E94F36" d="M216.598,78.867C99.184,78.867,4,174.05,4,291.465s95.184,212.598,212.598,212.598
+	s212.598-95.184,212.598-212.598S334.013,78.867,216.598,78.867z M217,469.63c-97.846,0-177.165-79.32-177.165-177.165
+	S119.154,115.3,217,115.3s177.165,79.32,177.165,177.165S314.846,469.63,217,469.63z"/>
+<rect x="340.33" y="291.464" fill="#E94F36" width="212.598" height="212.598"/>
+<polygon fill="#E94F36" points="123.5,375.465 197.5,275.465 230.5,314.465 "/>
+<polygon fill="#E94F36" points="304.5,214.465 230.5,314.465 197.5,275.465 "/>
+<polygon fill="#E94F36" points="293,384.465 193,310.465 232,277.465 "/>
+<polygon fill="#E94F36" points="132,203.465 232,277.465 193,310.465 "/>
+</svg>

data/icons/off.svg

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Generator: Adobe Illustrator 17.0.1, SVG Export Plug-In . SVG Version: 6.00 Build 0)  -->
+<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
+<svg version="1.1" id="Livello_1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px"
+	 width="566.93px" height="566.93px" viewBox="0 0 566.93 566.93" enable-background="new 0 0 566.93 566.93" xml:space="preserve">
+<path fill="#BBBBBB" d="M216.598,78.867C99.184,78.867,4,174.05,4,291.465s95.184,212.598,212.598,212.598
+	s212.598-95.184,212.598-212.598S334.013,78.867,216.598,78.867z M217,469.63c-97.846,0-177.165-79.32-177.165-177.165
+	S119.154,115.3,217,115.3s177.165,79.32,177.165,177.165S314.846,469.63,217,469.63z"/>
+<rect x="340.33" y="291.464" fill="#BCBCBB" width="212.598" height="212.598"/>
+<path fill="#BBBBBB" d="M218.598,221.599c-39.138,0-70.866,31.728-70.866,70.866c0,39.138,31.728,70.866,70.866,70.866
+	c39.138,0,70.866-31.728,70.866-70.866C289.465,253.327,257.737,221.599,218.598,221.599z M218.598,336.52
+	c-23.483,0-42.52-19.037-42.52-42.52s19.037-42.52,42.52-42.52c23.483,0,42.52,19.037,42.52,42.52S242.081,336.52,218.598,336.52z"
+	/>
+</svg>

data/icons/positive.svg

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Generator: Adobe Illustrator 17.0.1, SVG Export Plug-In . SVG Version: 6.00 Build 0)  -->
+<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
+<svg version="1.1" id="Livello_1" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" x="0px" y="0px"
+	 width="566.93px" height="566.93px" viewBox="0 0 566.93 566.93" enable-background="new 0 0 566.93 566.93" xml:space="preserve">
+<path fill="#57B35F" d="M216.598,78.867C99.184,78.867,4,174.05,4,291.465s95.184,212.598,212.598,212.598
+	s212.598-95.184,212.598-212.598S334.013,78.867,216.598,78.867z M217,469.63c-97.846,0-177.165-79.32-177.165-177.165
+	S119.154,115.3,217,115.3s177.165,79.32,177.165,177.165S314.846,469.63,217,469.63z"/>
+<polygon fill="#57B35F" points="91,275 197,388 343,212 195,306 "/>
+<rect x="340.332" y="291.465" fill="#57B35F" width="212.598" height="212.598"/>
+</svg>

data/lib/aws_poller.rb

@@ -0,0 +1,28 @@
+require 'aws-sdk'
+
+module Opswatch
+  class AwsPoller
+    def initialize(params)
+      credentials = Aws::Credentials.new(params['iam_user'], params['iam_secret'])
+      @client = Aws::OpsWorks::Client.new(region: params['region'], credentials: credentials)
+      @stack_id = params['stack_id']
+    end
+
+    def poll
+      ret = {}
+      begin
+        stack_resp = @client.describe_stack_summary(stack_id: @stack_id)
+        deploy_resp = @client.describe_deployments(stack_id: @stack_id).deployments.first
+        ret[:online] = stack_resp.stack_summary.instances_count.online
+        ret[:running_setup] = stack_resp.stack_summary.instances_count.running_setup
+        ret[:setup_failed] = stack_resp.stack_summary.instances_count.setup_failed
+        ret[:start_failed] = stack_resp.stack_summary.instances_count.start_failed
+        ret[:deploy_state] = deploy_resp.status
+        ret
+      rescue StandardError
+        ret[:exception] = true
+        ret
+      end
+    end
+  end
+end

data/lib/configuration.rb

@@ -0,0 +1,45 @@
+module Opswatch
+  class Configuration
+    def self.load_file(filename)
+      local_path = "#{GEM_ROOT}/#{filename}"
+      global_path = File.join(ENV['HOME'], filename)
+      if File.exist?(local_path)
+        YAML.load_file(local_path)
+      elsif File.exist?(global_path)
+        YAML.load_file(global_path)
+      else
+        create_configuration
+      end
+    end
+
+    def self.create_configuration
+      settings = {}
+      continue = true
+      puts "Let's set things up"
+
+      while continue
+        puts 'Stack Name (can be anything):'
+        name = gets.chomp
+        puts 'Region (e.g. eu-central-1):'
+        region = gets.chomp
+        puts 'IAM User key:'
+        iam_user = gets.chomp
+        puts 'IAM User secret:'
+        iam_secret = gets.chomp
+        puts 'OpsWorks Stack ID:'
+        stack_id = gets.chomp
+
+        settings[name] = { 'region' => region,
+                           'iam_user' => iam_user,
+                           'iam_secret' => iam_secret,
+                           'stack_id' => stack_id }
+
+        puts 'Insert other Stack? (Y/n)'
+        continue = gets.chomp != 'n'
+      end
+
+      File.write(File.join(ENV['HOME'], '.opswatch.yml'), settings.to_yaml)
+      settings
+    end
+  end
+end

data/lib/icon.rb

@@ -0,0 +1,40 @@
+require 'rmagick'
+require 'tempfile'
+
+class Icon
+  include Magick
+
+  attr_reader :stack_name
+
+  def initialize(stack_name, source_path)
+    @stack_name = stack_name
+    @source_path = source_path
+    @temp_file = Tempfile.new(stack_name)
+    draw_initial
+  end
+
+  def path
+    @temp_file.path
+  end
+
+  private
+
+  def stack_initial
+    @stack_name[0].upcase
+  end
+
+  def draw_initial
+    img = ImageList.new(@source_path)
+    drawer = Magick::Draw.new
+    drawer.annotate(img, 0, 0, 60, 10, stack_initial) do
+      self.font = "#{GEM_ROOT}/fonts/arial.ttf"
+      self.pointsize = 512
+      self.font_weight = Magick::BoldWeight
+      self.fill = 'white'
+      self.gravity = Magick::SouthEastGravity
+      self.stroke = 'black'
+      self.stroke_width = 4
+    end
+    img.write(path)
+  end
+end

data/lib/icon_pack.rb

@@ -0,0 +1,23 @@
+require 'rmagick'
+require 'tempfile'
+require_relative 'icon'
+
+module Opswatch
+  class IconPack
+    def initialize(stack_name, icon_paths = nil)
+      icon_paths = icon_paths || %w[hold negative off positive].map { |base_icon| [base_icon, "#{GEM_ROOT}/icons/#{base_icon}.png"] }.to_h
+      @stack_name = stack_name
+      @file_paths = generate_icons(icon_paths)
+    end
+
+    def fetch(key)
+      @file_paths[key].path
+    end
+
+    private
+
+    def generate_icons(icon_paths)
+      icon_paths.map { |name, path| [name, Icon.new(@stack_name, path)] }.to_h
+    end
+  end
+end

data/lib/ops_tray.rb

@@ -0,0 +1,43 @@
+require 'ruby-libappindicator'
+
+module Opswatch
+  class OpsTray < AppIndicator::AppIndicator
+    def initialize(name, icon_pack)
+      super(name, "dialog-question", AppIndicator::Category::APPLICATION_STATUS)
+      set_menu(Gtk::Menu.new)
+      set_status(AppIndicator::Status::ACTIVE)
+      @icon_pack = icon_pack
+    end
+
+    def update_status(resp)
+      set_icon_status(compute_status(resp))
+    end
+
+    private
+
+    def compute_status(resp)
+      ret = 'UNKNOWN_STATE'
+      ret = 'ONLINE' if resp[:online] && resp[:deploy_state] == 'successful'
+      ret = 'FAILED' if resp[:setup_failed] || resp[:start_failed] || resp[:deploy_state] == 'failed'
+      ret = 'OFFLINE' if resp[:online].nil? && resp[:setup_failed].nil? && resp[:start_failed].nil?
+      ret = 'DEPLOYING' if resp[:running_setup] || resp[:deploy_state] == 'running'
+      ret = 'INVALID_STATE' if resp[:exception]
+      ret
+    end
+
+    def set_icon_status(status)
+      case status
+      when 'ONLINE'
+        set_icon @icon_pack.fetch('positive')
+      when 'OFFLINE'
+        set_icon @icon_pack.fetch('off')
+      when 'DEPLOYING'
+        set_icon @icon_pack.fetch('hold')
+      when 'FAILED'
+        set_icon @icon_pack.fetch('negative')
+      when 'INVALID_STATE'
+        set_icon 'dialog-question'
+      end
+    end
+  end
+end

data/lib/opswatch.rb

@@ -0,0 +1,37 @@
+require "opswatch/version"
+require 'yaml'
+
+require 'ops_tray'
+require 'aws_poller'
+require 'icon_pack'
+require 'configuration'
+
+module Opswatch
+  class Main
+    def initialize
+      settings = Configuration.load_file(".opswatch.yml")
+      @groups = {}
+      settings.each do |k, v|
+        @groups[k] = { tray: OpsTray.new(k, IconPack.new(k)), poller: AwsPoller.new(v) }
+      end
+
+      Gtk.timeout_add(5000) { poll }
+    end
+
+    def self.start(_argv = nil)
+      @instance = new
+
+      Gtk.main
+    end
+
+    private
+
+    def poll
+      @groups.each do |_k, g|
+        res = g[:poller].poll
+        g[:tray].update_status(res)
+      end
+      true
+    end
+  end
+end

data/lib/opswatch/version.rb

@@ -0,0 +1,3 @@
+module Opswatch
+  VERSION = "0.1.0"
+end

data/opswatch.gemspec

@@ -0,0 +1,32 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "opswatch/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "opswatch"
+  spec.version       = Opswatch::VERSION
+  spec.authors       = ["Gianluca Rizzo"]
+  spec.email         = ["ing.rizzo.gianluca@gmail.com"]
+
+  spec.summary       = 'Simple tray application that notifies the status of your Opswork stacks.'
+  spec.description   = 'Simple tool that helps an AWS OpsWorks user to be constantly informed about the status of multiple stacks.
+                        Uses the appindicator library to show status icon on the OS traybar.'
+  spec.homepage      = 'https://github.com/gianlucarizzo/opswatch'
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = 'bin'
+  spec.executable    = 'opswatch'
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency "bundler", "~> 1.16"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "minitest", "~> 5.0"
+
+  spec.add_runtime_dependency('aws-sdk', "~> 3.0.1")
+  spec.add_runtime_dependency('rmagick', "~> 2.16.0")
+  spec.add_runtime_dependency('ruby-libappindicator', "~> 0.1.5")
+  spec.add_runtime_dependency('app_configuration', "~> 0.0.3")
+end