blend 0.1.4

data/lib/blend/cli/juice.rb

@@ -0,0 +1,493 @@
+module Blend
+  module CLI
+    class Juice < Thor
+      desc "login", "Login to juice"
+      def login
+        puts "Logged in." if client.login
+      end
+
+      desc "logout", "Clear juice credentials"
+      def logout
+        puts "Juice credentials cleared." if client.logout
+      end
+
+      desc "open PROJECT", "Open up juice"
+      def open( name )
+        project_id = project_id_from_name name
+        if( project_id.nil? )
+          puts "#{name} not found"
+          return
+        end
+
+        system "open http://happyfunjuice.com/projects/#{project_id}"
+      end
+
+      desc "settings PROJECT", "Open up juice settings"
+      def settings(name)
+        project_id = project_id_from_name name
+        if( project_id.nil? )
+          puts "#{name} not found"
+          return
+        end
+
+        system "open http://happyfunjuice.com/projects/#{project_id}/overview"
+      end
+
+      desc "create PROJECT", "Create a juice project"
+      def create( name )
+        puts "TIP: you can also run check #{name} to set up everything"
+        client.create_project( name )
+      end
+
+      desc "projects", "List juice projects"
+      def projects
+        puts
+          printf "%-25s %-10s %-10s %-25s\n".blue, '', 'Commits', 'Deploys', '', ''
+          printf "%-25s %-10s %-10s %-25s\n".blue, '', 'this', 'this', '', ''
+          printf "%-25s %-10s %-10s %-25s %s\n".underline.blue, 'Name', 'wk', 'wk', 'Hipchat room', 'Teams'
+        client.summary.sort{|a,b| b['name'].to_i <=> a['name'].to_i}.each_with_index do |project,i|
+          printf "%-25s %-10s %-10s %-25s %s\n".try{|x| i%4==3 ? x.underline : x}, 
+            #project['id'], 
+            project['name'].projectize, 
+            (project['has_source_feeds'] ? project['commits_this_week'] : ''),
+            (project['has_server_feeds'] ? project['deploys_this_week'] : ''),
+            project['blend_config']['hipchat_room'],
+            project['blend_config']['teams'].join( ',' )
+        end
+        puts
+      end
+
+      desc "organizations", "Get a list of your organizations"
+      def organizations
+        puts
+        printf "%-5s %-25s %-25s\n".underline.blue, 'ID', 'Name', 'Domain name'
+        client.organizations.each do |o|
+          printf "%-5s %-25s %-25s\n", o['id'], o['name'], o['domain_name']
+        end
+        puts
+      end
+        
+
+      desc "feeds PROJECT", "Show the configured juice feeds"
+      def feeds(name)
+        puts
+        printf "%-30s %-20s %-30s\n".blue.underline, 'Feed', 'Environment', 'Namespace'
+        client.feeds( project_id_from_name( name ) ).sort do
+          |a,b| a['feed_name'] <=> b['feed_name']
+        end.each do |feed|
+          printf "%-30s %-20s %-30s\n", 
+            feed['feed_name'], 
+            (feed['environment'] || {})['name'],
+            [feed['namespace'],feed['name']].select {|x| x}.join( '/' )
+        end
+        puts
+      end
+
+      desc "users [PROJECT]", "Get a list of users"
+      def users( name=nil )
+
+        if name.nil?
+          _users = client.organization_users( 1 ) # Hard-code hfc org here
+        else
+          _users = client.project_users( project_id_from_name( name ) )
+        end
+
+        puts
+        printf "%-25s %35s %35s %35s %35s\n".blue.underline, 'Name', 'Email', 'Personal email', 'Heroku', 'Github'
+        _users.each do |u|
+          printf "%-25s %35s %35s %35s %35s\n", u['name'], u['email'], u['personal_email'], u['heroku_handle'], u['github_handle']
+        end
+        puts
+      end
+
+      desc "add_team PROJECT TEAM", "Add a github team to a project"
+      def add_team( name, team )
+        client.project_add_team( project_id_from_name( name ), team )
+        info( name )
+      end
+
+      desc "add_hipchat PROJECT ROOM", "Add a hipchat room to a project"
+      def add_hipchat( name, room )
+        client.project_add_hipchat( project_id_from_name( name ), room )
+        info( name )
+      end
+
+      desc "lookup_user NAME", "Looks up a user by email address"
+      def lookup_user( query )
+        pp client.lookup_user( query )
+      end
+
+      desc "search_users QUERY", "Look up a user by name, email, github, heroku, etc."
+      def search_users( query )
+        puts
+        printf "%-25s %35s %35s %35s %35s\n".blue.underline, 'Name', 'Email', 'Personal email', 'Heroku', 'Github'
+        client.search_users(query).each do |u|
+          printf "%-25s %35s %35s %35s %35s\n", u['name'], u['email'], u['personal_email'], u['heroku_handle'], u['github_handle']
+        end
+        puts
+      end
+
+      # desc "user_set FIELD VALUE", "Set the value of a particular field for a user"
+      # def user_set( field, value )
+      #   client
+      # end
+
+      desc "heroku_api TOKEN", "Sets the organization heroku token"
+      def heroku_api( token )
+        client.heroku_api token
+      end
+
+
+      desc "hipchat_api TOKEN", "Sets the organization hipchat token"
+      def hipchat_api( token )
+        client.hipchat_api token
+      end
+
+      desc "all_projects [--resolve]", "Show all project status"
+      option :resolve
+      def all_projects
+        client.projects.each do |p|
+          project( p['name'] )
+        end
+      end
+
+      desc "project NAME [--resolve]", "Get project status"
+      option :resolve
+      def project( name )
+        status = Blend::Status::Project.new( name, options[:resolve] )
+
+        status.header "#{name}: Juice Configuration"
+        status.check "Project Exists", :project_found
+        status.check "Hipchat Room", :hipchat
+        status.check "Github Teams", :github_teams
+        status.check "Repos Configured", :repos_setup
+        status.check "Source Control", :source_control
+        status.check "Bug Tracking", :bugtracking
+
+        status.header "#{name}: Environments"
+        status.check "Production", :production
+        status.check "Staging", :staging
+
+        status.header "#{name}: Team Configuration (#{status.github_teams.join(',')})" if status.github_teams.length > 0
+
+        status.check "Members", :github_members
+
+        status.check "User Matchup", :juice_users_synced
+        
+        status.github_members.each do |m|
+          member = m[:name]
+          access = m[:access]
+          # puts member
+          juice_user = client.user_from_github_user member
+          if access == :read
+            printf "%-20s %15s".yellow, member, "readonly"
+          else
+            printf "%-20s %15s".green, member, "fullaccess"
+          end
+
+          if juice_user
+            printf " %-20s %s\n", juice_user['name'], juice_user['email']
+          else
+            puts " Unknown to juice".red
+          end
+        end
+
+        status.repo_status.each do |repo|
+          status.header "#{repo.name} Configuration"
+
+          repo.check "Private", :private?
+          repo.check "Hipchat Deployhook", :hipchat_hook
+        end
+
+        status.environment_status.each do |env|
+          status.header "Server: #{env.server} Configuration"
+          
+          env.check "Dyno Redundancy", :dyno_redundancy
+          env.check "Database", :database
+          env.check "Backups", :backups
+          env.check "Stack", :stack
+          env.check "Exception Handling", :exception_handling
+          env.check "Deploy Hooks", :deployhooks
+          env.check "Log Monitoring", :log_monitoring
+          env.check "App Monitoring", :app_monitoring
+          env.check "SSL Addon", :ssl
+        end
+
+        status.domains_status.each do |domain|
+          status.header "DNS: #{domain.domain} configuration"
+
+          domain.check "Registered?", :registered?
+          domain.check "Expires", :expires
+          domain.check "Owner", :owner
+          domain.check "SSL Cert", :ssl_exists?
+          domain.check "SSL Expires", :ssl_valid_until
+          domain.check "SSL Common Name", :ssl_common_name
+        end
+
+      end
+=begin
+
+
+
+
+      desc "check [PROJECT]", "Check a project config (all the checks)"
+      def check( name )
+        check_project( name )
+        puts
+        check_hipchat( name )
+        puts
+        check_team( name )
+        puts
+        check_hooks( name )
+        puts
+        info( name )
+      end
+
+      desc "check_hooks [PROJECT]", "Check to see if the hooks are configured"
+      def check_hooks( name )
+        puts "Looking for github hooks".bold
+        ##
+        # Github Hooks
+        ##
+        project_id = project_id_from_name name
+        return if project_id.nil?
+
+        data = client.project project_id
+        config = data['blend_config']
+
+        teams = config['teams']
+
+        if teams.count > 0
+          teams.each do |team|
+            puts "Looking at repos for #{team}".bold
+            github_client.list_team_repos( team ).each do |repo|
+              printf "%-40s %s\n", repo['full_name'], repo['description']
+              found = {}
+              github_client.list_hooks( repo['full_name'] ).each do |hook|
+                puts "Found #{hook["name"]}"
+                found[hook['name']] = true
+              end
+
+              unless found['hipchat']
+                puts "Missing hipchat hook"
+                if config['hipchat_room']
+                  puts "Adding Hipchat Hook"
+                  Blend::CLI::Github.new.add_hipchat( repo['full_name'], config['hipchat_room'])
+                else
+                  puts "Missing hipchat room"
+                end
+              end
+            end
+          end
+        else
+          puts "Teams must be set up correctly in order to monitor hooks.".red
+        end
+
+        apps = client.heroku_apps( project_id ) 
+
+
+        if( apps['production'] )
+          puts "Production".blue
+          apps['production'].each do |app|
+            Blend::CLI::Heroku.new.check app['name']
+            a = heroku_client.addons(app['name'], /deployhooks/)
+
+            if( a.nil? || a.length == 0 )
+              puts "Adding deploy hook".yellow
+              system( "echo heroku addons:add deployhooks:hipchat --auth_token=#{client.hipchat_api} --room=\"#{config['hipchat_room']} --app #{app['name']}\"")
+              system( "heroku addons:add deployhooks:hipchat --auth_token=#{client.hipchat_api} --room=\"#{config['hipchat_room']}\" --app #{app['name']}")
+              Blend::Client::hipchat_client.post_message config['hipchat_room'], "Heroku app: #{app['name']} commit hook now added"
+            end
+          end
+        end
+        if( apps['staging'] )
+          puts "Staging".blue
+          apps['staging'].each do |app|
+            Blend::CLI::Heroku.new.check app['name']
+            a = heroku_client.addons(app['name'], /deployhooks/)
+
+            if( a.nil? || a.length == 0 )
+              puts "Adding deploy hook".yellow
+              system( "echo heroku addons:add deployhooks:hipchat --auth_token=#{client.hipchat_api} --room=\"#{config['hipchat_room']} --app #{app['name']}\"")
+              system( "heroku addons:add deployhooks:hipchat --auth_token=#{client.hipchat_api} --room=\"#{config['hipchat_room']} --app #{app['name']}\"")
+              Blend::Client::hipchat_client.post_message config['hipchat_room'], "Heroku app: #{app['name']} commit hook now added"
+            end
+          end
+        end
+
+      end
+=end
+
+      desc "hipchat_check", "Prints out all the rooms not assigned to rooms"
+      def hipchat_check
+        begin
+          client.hipchat_check.each do |x|
+            printf "%-30s %s\n", x[:room], x[:projects].join( "," )
+          end
+        rescue Exceptions::HipchatAuthenticationFailure
+          puts "Unable to connect to Hipchat. Is your key valid?".red
+        end
+      end
+
+      desc "github_team_check", "Prints out all the teams not assigned to projects"
+      option :fix
+      def github_team_check
+        client.github_team_check.each do |team,projects|
+          next if team == 'Owners'
+          next if projects.size > 0
+          puts "#{team} has no juice project".red
+          # printf "%-30s %s\n", team, projects.collect {|x| x['name'] }.join( "," )
+
+          if( options[:fix] )
+            choices = client.projects.collect { |x| x['name'] }
+
+            project = choose do |menu|
+              menu.header = "Select a github team action"
+              menu.prompt = "Please choose a unassigned juice project to associate #{team} with:"
+
+              menu.choice "Ignore" do
+                "ignore"
+              end
+
+              menu.choice "Create a project called #{team}" do
+                "create"
+              end
+
+              menu.choices *choices
+            end
+
+            if( project == "ignore" )
+              puts "Skip it"
+            else
+              if( project == "create" )
+                puts "TODO: Create a new project called #{team}"
+              else
+                add_team project, team
+                # puts "Attaching to: #{project}"
+              end
+            end
+          end
+        end
+      end
+
+      desc "activity PROJECT", "Shows recent project activity in last week or (default) current week [--lastweek] [--thisweek]"
+      option :lastweek
+      option :thisweek
+      def activity( name )
+        puts
+        project_id = project_id_from_name name
+        return if project_id.nil?
+
+        now = DateTime.now.to_date + 1
+        after = now - now.wday
+        before = Time.now
+
+        if( options[:lastweek] )
+          after -= 7
+          before = after + 7
+        end
+
+        summary = client.activities( project_id, after.to_time, before.to_time )
+
+        project = client.project( project_id )
+
+        puts "#{project['name']} activity".bold + " for #{summary[:after].strftime( "%Y-%m-%d %H:%M" )} (#{((Time.now-summary[:after])/3600/24).round(1)} days ago) - #{summary[:before].strftime( "%Y-%m-%d %H:%M" )} (now)"
+        puts
+        puts "Activity Summary".underline.blue
+        summary[:type].keys.sort.each do |x|
+          printf "%-6s %s\n", summary[:type][x].count, x
+        end
+
+        puts
+        puts "Activity Breakdown".underline.blue
+        summary[:actors_activites].keys.sort.each do |x|
+          summary[:actors_activites][x].keys.select { |x| x}.sort.each do |type|
+            printf "%-6s %-25s %s\n", summary[:actors_activites][x][type].count, type.strip_email, x
+          end
+        end
+
+        puts
+        puts "New Tickets".underline.blue
+        (summary[:type]['bugtracking:openticket'] || []).each do |activity|
+          printf "%-15s %-100s\n", activity['actor_identifier'], activity['description'][0..100].gsub( /\n/, " " )
+          #puts activity['description'][0..100].gsub( /\n/, " " )
+        end
+
+        puts
+        puts "Closed Tickets".underline.blue
+        (summary[:type]['bugtracking:closedticket'] || []).each do |activity|
+          printf "%-15s %-100s\n", activity['actor_identifier'], activity['description'][0..100].gsub( /\n/, " " )
+          #puts activity['description'][0..100].gsub( /\n/, " " )
+        end
+
+        puts
+        puts "Active Tickets".underline.blue
+        summary[:type].keys.select do |x|
+          x =~ /bugtracking/
+        end.collect do |x|
+          summary[:type][x].collect do |a|
+            a['description']
+          end
+        end.flatten.sort.uniq.each do |x|
+          puts x[0..100].gsub( /\n/, " " ) unless x =~ /^\[Changeset\]/
+        end
+
+        puts
+        puts "Commits".underline.blue
+        (summary[:type]['sourcecontrol:commit'] || []).each do |x|
+          printf "%-30s %s\n", x['actor_identifier'].strip_email, x['description'][0..100].gsub( /\n/, " " )
+        end
+      end
+
+      desc "report NAME", "Report for an individual project"
+      option :lastweek
+      def report(name)
+        info name
+        activity name
+      end
+
+      desc "report_dump", "Write out weekly reports"
+      option :lastweek
+      def report_dump
+        puts "Loading projects"
+
+        system "mkdir -p /tmp/juice_reports"
+        client.projects.each do |project|
+          File.open( "/tmp/juice_reports/#{project['name']}.txt", "w" ) do |out|
+            puts "Running report for #{project['name']}"
+            $stdout = out
+            info project['name']
+            activities project['name']
+            $stdout = STDOUT
+          end
+        end
+      end
+
+      no_commands do
+        def client
+          @client ||= Blend::Client.juice_client
+        end
+
+        def hipchat_client
+          Blend::Client.hipchat_client
+        end
+
+        def github_client
+          Blend::Client.github_client
+        end
+
+        def heroku_client
+          Blend::Client.heroku_client
+        end
+
+        def project_id_from_name( name )
+          client.project_id_from_name( name )
+        end
+
+        def set( s )
+          !s.nil? && s != ""
+        end
+      end
+    end
+  end
+end