blend 0.1.4

data/lib/blend/cli.rb

@@ -0,0 +1,38 @@
+require 'blend/cli/hipchat'
+require 'blend/cli/github'
+require 'blend/cli/juice'
+require 'blend/cli/heroku'
+
+module Blend
+  module CLI
+    class CLI < Thor
+
+      desc "logout", "Clear juice credentials"
+      def logout
+        Blend::CLI::Juice.new.logout
+      end
+
+      desc "login", "Acquire juice credentials"
+      def login
+        Blend::CLI::Juice.new.login
+      end
+
+      desc 'projects', 'List of Juice projects'
+      def projects
+        Blend::CLI::Juice.new.projects
+      end
+
+      desc "hipchat COMMANDS", "Hipchat Control Module"
+      subcommand "hipchat", Blend::CLI::Hipchat
+      
+      desc "github COMMANDS", "Github Control Module"
+      subcommand "github", Blend::CLI::Github
+
+      desc "juice COMMANDS", "Juice Control Module"
+      subcommand "juice", Blend::CLI::Juice
+
+      desc "heroku COMMANDS", "Heroku Control Module"
+      subcommand "heroku", Blend::CLI::Heroku
+    end
+  end
+end

data/lib/blend/cli/github.rb

@@ -0,0 +1,140 @@
+module Blend
+  module CLI
+    class Github < Thor
+      desc "hooks REPO", "Lists out service hooks for a given repo"
+      def hooks( repo )
+        puts
+        printf "%-20s %-60s\n".blue.underline, 'Name', 'Config'
+        client.list_hooks( repo ).each do |hook|
+          printf "%-20s", hook.name
+          cfg = hook.config.collect{|k,v| [k,v]}
+          if hook.config.count > 0
+            cfg.each_with_index do |(k,v),i|
+              if i>0
+                printf "%-20s %-20s: %-20s\n", '', k, v.inspect
+              else
+                printf " %-20s: %-20s\n", k, v.inspect
+              end
+            end
+          else
+            puts "\n"
+          end
+          puts
+        end
+      end
+
+      desc "add_hipchat REPO ROOM", "Add hipchat hook to repo"
+      def add_hipchat( repo, room )
+        client.add_hook( repo, "hipchat", { 
+          auth_token: Blend::Client::get_token( :hipchat ), 
+          notify: "1",
+          restrict_to_branch: "",
+          room: room } )
+        Blend::Client::hipchat_client.post_message room, "#{repo} commit hook now added"
+      end
+
+      desc "repos", "Lists out the repos"
+      def repos
+        puts
+        printf "%-50s %-40s\n".blue.underline, 'Name', 'Description'
+        client.list_repos.each do |repo|
+          printf "%-50s %-40s\n", repo['full_name'], repo['description']
+        end
+        puts
+      end
+
+      desc "repo_search NAME", "Lists out matching repos"
+      def repo_search( name )
+        puts
+        printf "%-50s %-40s\n".blue.underline, 'Name', 'Description'
+        client.list_repos(name).each do |repo|
+          printf "%-50s %-40s\n", repo['full_name'], repo['description']
+        end
+        puts
+      end
+
+      desc "repo_create TEAM, NAME", "Create team repo"
+      def repo_create( team, name )
+        client.repo_create( team, name )
+        team( team )
+      end
+
+
+      desc "collaborators REPO", "List out collaborators"
+      def collaborators( repo )
+        client.list_collaborators( repo ).each do |collaborator|
+          puts collaborator['login']
+        end
+      end
+
+      desc "teams", "List out organizations team"
+      def teams
+        client.list_teams.each do |team|
+          puts team.name
+        end
+      end
+
+      desc "team_repos TEAM", "List out a teams repos"
+      def team_repos( team )
+        puts
+        printf "%-40s %-40s\n".blue.underline, 'Name', 'Description'
+        client.list_team_repos( team ).each do |repo|
+          printf "%-40s %-40s\n", repo['full_name'], repo['description']
+        end
+        puts
+      end
+
+      desc "team TEAM", "List out the team"
+      def team( team )  
+        puts
+        printf "%-40s %-40s\n".blue.underline, 'Repo name', 'Description'
+        client.list_team_repos( team ).each do |repo|
+          printf "%-40s %s\n", repo['full_name'], repo['description']
+        end
+        puts
+        printf "%-30s\n".blue.underline, 'Github users'
+        client.list_team_members( team ).each do |user|
+          printf "%s\n", user['login']
+        end
+        puts
+      end
+
+      desc "team_create TEAM", "Create a team"
+      def team_create( team )
+        client.create_team( team )
+        team( team )
+      end
+      
+      desc "team_add TEAM, USER", "Add a user to a team"
+      def team_add( team, user )
+        client.add_team_member( team, user )
+        team( team )
+      end
+
+      desc "team_rm TEAM, USER", "Remove a user from a team"
+      def team_rm( team, user)
+        client.remove_team_member( team, user )
+        team( team )
+      end
+
+      desc "team_repo_add TEAM, REPO", "Add a repo to a team"
+      def team_repo_add( team, repo )
+        client.add_team_repo( team, repo )
+        team( team )
+      end
+
+      desc "team_repo_rm TEAM, REPO", "Remove a repo from a team"
+      def team_repo_rm( team, repo )
+        client.remove_team_repo( team, repo )
+        team( team )
+      end
+
+      no_commands do
+        def client
+          @client ||= Blend::Client.github_client
+        end
+      end
+
+    end
+  end
+end

data/lib/blend/cli/heroku.rb

@@ -0,0 +1,232 @@
+module Blend
+  module CLI
+    class Heroku < Thor
+
+      desc "info [APP]", "Prints info for a given app"
+      def info( app )
+        puts
+        _info = client.info(app)
+        _info.reject{|x| %w(domain_name).include? x}.each do |k,v|
+          printf "%35s: %-25s\n", k, v
+        end
+        puts
+      end
+
+      desc 'domains [APP]', 'Print domain info'
+      def domains(app)
+        d = client.domains(app)
+        d.each do |params|
+          puts "\n#{params['domain']}:".blue
+          params.each do |k,v|
+            printf "%15s: %-25s\n", k, v
+          end
+        end
+        puts
+      end
+
+      desc 'check_domains [APP]', 'Check domains'
+      def check_domains(app)
+        client.nonheroku_domains(app).each do |domain|
+          puts "\nChecking domain #{domain}:".blue
+          result = DomainChecker.check(domain)
+
+          if result.include? :error
+            puts result[:error].red
+            return
+          end
+          
+          # Check registration
+          if result[:registered]
+            r = result[:registrar]
+
+            # Registrar:
+            if r.nil?
+              printf "%25s: %s", 'Registrar', format_result(:warn, "DNS not configured correctly for #{domain}")
+            else
+              registrar = [r['name'], r['organization'], r['url']].reject(&:nil?).join(';  ')
+              printf "%25s: %s", 'Registrar', format_result(:pass, registrar)
+            end
+          else
+            printf "%25s: %s", 'Registrar', format_result(:fail)
+          end
+
+
+          unless result[:expires].nil?
+            # Expiration:
+            days_to_expiration = ((result[:expires] - Time.now)/(3600.0*24.0)).to_i 
+            formatted_dte = "#{result[:expires].strftime("%F")} (#{days_to_expiration} days from now)"
+            if days_to_expiration < 30
+              state = :fail
+            elsif days_to_expiration < 30*3
+              state = :warn
+            else
+              state = :pass
+            end
+
+            printf "%25s: %s", 'Expiration date', format_result(state, formatted_dte)
+          end
+
+          printf "%25s:\n", 'DNS'
+          result[:dns].each do |k,v|
+            v[:domains].each do |record|
+              if record[:type] == 'MX'
+                description = "#{record[:data][0]}: #{record[:data][1].to_s}"
+              else
+                description = record[:data].to_s
+              end
+              printf "%25s  ", ''
+              printf "%5s".green, record[:type].to_s
+              printf "( ttl=%i ): %s\n", record[:ttl], description
+            end
+          end
+
+        end
+      end
+
+      desc 'check_log_monitoring_addon [APP]', 'Check for log monitoring add-on'
+      def check_log_monitoring_addon(app)
+        a = client.addons(app, /(logentries|keen|papertrail|loggly|flydata)/)
+        names = a.collect{|x| x['name']}.join(', ')
+        result = a.count > 0 ? format_result(:pass, names) : result = format_result(:fail)
+        printf "%25s: %s", 'Log monitoring add-on', result
+      end
+
+      desc 'check_app_monitoring_addon [APP]', 'Check for app monitoring add-on'
+      def check_app_monitoring_addon(app)
+        a = client.addons(app, /(relic)/)
+        names = a.collect{|x| x['name']}.join(', ')
+        result = a.count > 0 ? format_result(:pass, names) : result = format_result(:fail)
+        printf "%25s: %s", 'App monitoring add-on', result
+      end
+
+      desc 'check_exception_handling_addon [APP]', 'Check for exception handling add-on'
+      def check_exception_handling_addon(app)
+        a = client.addons(app, /(airbrake|exceptional)/)
+        names = a.collect{|x| x['name']}.join(', ')
+        result = a.count > 0 ? format_result(:pass, names) : format_result(:fail)
+        printf "%25s: %s", 'Exception handling add-on', result
+      end
+
+      desc 'check_deployhooks_appon [APP]', 'Check for deployhooks handling add-on'
+      def check_deployhooks_appon(app)
+        a = client.addons(app, /deploynooks/)
+        names = a.collect{|x| x['name']}.join(', ')
+        result = a.count > 0 ? format_result(:pass, names) : format_result(:fail)
+        printf "%25s: %s", 'Deploy hooks add-on', result
+      end
+
+
+      desc 'check_ssl_addon [APP]', 'Check for SSL addon'
+      def check_ssl_addon(app)
+        a = client.addons(app, /ssl/)
+        names = a.collect{|x| x['name']}.join(', ')
+        result = a.count > 0 ? format_result(:pass, names) : format_result(:warn)
+        printf "%25s: %s", 'SSL add-on', result
+      end
+
+
+      desc "addons [APP]", "Print addons for a given app"
+      def addons(app)
+        _addons = client.addons(app)
+        _addons.each do |addon|
+          puts addon['name']
+        end
+      end
+
+      desc "addon_price [APP]", "Show the price of app addons"
+      def addon_price(app)
+        _addons = client.addons(app)
+        _addons.each do |addon|
+          printf "%-35s: $%10.2f / %-10s\n", addon['name'], addon['price']['cents'].to_i*0.01, addon['price']['unit']
+        end
+      end
+
+      desc "check_backups [APP]", "Check to make sure the app is properly set up"
+      def check_backups(app)
+        printf "%25s: ", "Database backups"
+        b = client.backups(app)
+        if b.count==0
+          printf format_result(:fail)
+        else
+          printf format_result(:pass, b.collect{|x| x[:plan]}.join(', '))
+        end
+      end
+
+      desc 'check_database [APP]', 'Make sure we\'re running a production database'
+      def check_database(app)
+        printf "%25s: ", "Databases"
+        b = client.databases(app)
+        unique_plans = b.collect{|x| x[:plan]}.uniq
+        if b.nil?
+          printf format_result(:fail)
+        elsif unique_plans.count==1 and unique_plans[0]=='dev'
+          printf format_result(:warn, b.collect{|x| x[:plan]}.join(', '))
+        else
+          printf format_result(:pass, b.collect{|x| x[:plan]}.join(', '))
+        end
+      end
+
+      desc 'check_dyno_redundancy [APP]', 'Make sure more than one dyno is assigned'
+      def check_dyno_redundancy(app)
+        printf "%25s: ", "Dyno redundancy"
+        d = client.dynos(app).to_i
+        if d > 1
+          printf format_result(:pass, "#{d} dynos")
+        else
+          printf format_result(:fail, "#{d} dyno")
+        end
+      end
+
+      desc 'check_stack [APP]', 'Make sure app is on cedar stack'
+      def check_stack(app)
+        printf "%25s: ", "Cedar stack"
+        s = client.stack(app)
+        if s == 'cedar'
+          printf format_result(:pass, s)
+        else
+          printf format_result(:fail, s)
+        end
+      end
+        
+
+      desc 'check [APP]', 'Run production check for app'
+      def check(app)
+        puts
+        puts "Checking app #{app}:".blue
+        check_dyno_redundancy(app)
+        check_database(app)
+        check_backups(app)
+        check_stack(app)
+        check_exception_handling_addon(app)
+        check_deployhooks_appon(app)
+        check_log_monitoring_addon(app)
+        check_app_monitoring_addon(app)
+        check_ssl_addon(app)
+        check_domains(app)
+        puts
+      end
+
+
+
+      no_commands do
+        def client
+          @client ||= Blend::Client.heroku_client
+        end
+      end
+
+      no_tasks do
+        def format_result( state, message=nil )
+          case state
+            when :pass
+              sprintf "\u2713 #{message}\n".encode('utf-8').green
+            when :warn
+              sprintf "\u2718 #{message}\n".encode('utf-8').yellow
+            when :fail
+              sprintf "\u2718 #{message}\n".encode('utf-8').red
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/blend/cli/hipchat.rb

@@ -0,0 +1,61 @@
+module Blend
+  module CLI
+    class Hipchat < Thor
+      desc "rooms", "Show all hipchat rooms"
+      def rooms
+        client.rooms.sort do
+          |b,a| a['last_active'].to_i <=> b['last_active'].to_i
+        end.each do |r|
+          printf "%6s %-35s %10s %s\n", r['room_id'], r['name'], r['last_active'], r['topic']
+        end
+      end
+
+      desc "room ROOM", "Show room info"
+      def room( room )
+        room_info = client.room_info( room )
+        puts "Name        : #{room_info['name']}"
+        puts "ID          : #{room_info['room_id']}"
+        puts "Topic       : #{room_info['topic']}"
+        puts "Last active : #{room_info['last_active']}"
+        puts "Participants:"
+        room_info['participants'].each do |p|
+          puts p['name']
+        end
+      end
+
+      desc "users", "Show all users"
+      def users
+        client.users.sort do
+          |b,a| a['last_active'].to_i <=> b['last_active'].to_i
+        end.each do |u|
+          printf "%6s %-25s %-15s %-25s %-17s %-19s %s\n", u['user_id'], u['name'], u['mention_name'], u['email'], u['last_active'], u['timezone'], u['title']
+        end
+      end
+
+      desc "create ROOM", "Create a hipchat room"
+      def create( room )
+        puts "Creating room: #{room}"
+
+        client.create_room( room, "Let's talk about #{room}!" )
+
+        rooms
+      end
+
+      desc "topic ROOM TOPIC", "Set a topic of a hipchat room"
+      def topic( room, topic )
+        client.set_topic room, topic
+      end
+
+      desc "message ROOM MESSAGE", "Send a room a message"
+      def message( room, message )
+        client.post_message room, message
+      end
+
+      no_commands do
+        def client
+          @client ||= Blend::Client.hipchat_client
+        end
+      end
+    end
+  end
+end