neetob 0.1.0

data/lib/neetob/cli/github/labels/upsert.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class Upsert < Base
+          attr_accessor :apps, :required_labels_json_file_path, :sandbox
+
+          def initialize(apps, required_labels_json_file_path = "", sandbox = false)
+            super()
+            @apps = apps
+            @required_labels_json_file_path = required_labels_json_file_path
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :github, sandbox)
+            inform_about_default_labels_file
+            matching_apps.each do |app|
+              ui.info("\n Working on #{app} repo \n")
+              begin
+                required_labels = read_json_file(required_labels_json_file_path || default_labels_file_path)
+                required_labels.each do |label|
+                  create_or_update_label(app, label)
+                end
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+          end
+
+          private
+
+            def create_or_update_label(app, label_details)
+              begin
+                client.update_label(app, label_details["name"], label_details)
+                ui.success("Label \"#{label_details["name"]}\" updated successfully")
+              rescue Octokit::NotFound
+                client.add_label(app, label_details["name"], label_details["color"], label_details)
+                ui.success("Label \"#{label_details["name"]}\" created successfully")
+              end
+            end
+
+            def default_labels_file_path
+              File.expand_path("../../../../../data/github-labels.json", __dir__)
+            end
+
+            def inform_about_default_labels_file
+              if required_labels_json_file_path.nil?
+                ui.info("Upserting labels from the \"neetob/data/github-labels.json\" file")
+              end
+            end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/login.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Github
+      class Login < Base
+        def initialize
+          super()
+          Auth.new(**read_and_parse_auth_params_from_env).start_oauth2_device_flow
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/make_pr/base.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module MakePr
+        class Base < Github::Base
+          BRANCH_NAME = "neeto_compliance"
+          PR_TITLE = "Neeto Compliance"
+          attr_accessor :branch_name, :pr_title
+
+          def initialize(pr_title = PR_TITLE, branch_name = BRANCH_NAME)
+            super()
+            @branch_name = branch_name || BRANCH_NAME
+            @pr_title = pr_title || PR_TITLE
+          end
+
+          private
+
+            def delete_and_create_temp_neetob_dir
+              `rm -rf /tmp/neetob`
+              `mkdir /tmp/neetob`
+            end
+
+            def app_name_without_org_suffix(app)
+              app.split("/").last
+            end
+
+            def clone_app_in_tmp_dir(app)
+              `git clone --quiet git@github.com:#{app}.git /tmp/neetob/#{app_name_without_org_suffix(app)}`
+            end
+
+            def add_commmit_and_push_changes(app)
+              `#{cd_to_app_in_tmp_dir(app)} && git add . && git commit -m "#{pr_title}" && git push -u origin #{branch_name} --force`
+            end
+
+            def delete_local_feature_branch(app)
+              `#{cd_to_app_in_tmp_dir(app)} && git checkout main && git branch -D #{branch_name}`
+            end
+
+            def checkout_to_new_branch(app)
+              `#{cd_to_app_in_tmp_dir(app)} && git checkout -b #{branch_name}`
+            end
+
+            def checkout_to_feature_branch(app)
+              `#{cd_to_app_in_tmp_dir(app)} && git checkout #{branch_name}`
+            end
+
+            def delete_remote_feature_branch(app)
+              `#{cd_to_app_in_tmp_dir(app)} && git push origin --delete #{branch_name}`
+            end
+
+            def check_and_delete_remote_branch(app)
+              checkout_to_feature_branch(app)
+              if $?.success?
+                ui.info("Remote branch found with the name \"#{branch_name}\"")
+                delete_remote_feature_branch(app)
+                delete_local_feature_branch(app)
+              end
+              checkout_to_new_branch(app)
+            end
+
+            def cd_to_app_in_tmp_dir(app)
+              "cd /tmp/neetob/#{app_name_without_org_suffix(app)}"
+            end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/make_pr/commands.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "compliance_fix"
+require_relative "script"
+require_relative "../../sub_command_base"
+
+module Neetob
+  class CLI
+    module Github
+      module MakePr
+        class Commands < SubCommandBase
+          class_option :apps,
+            {
+              type: :array, aliases: "-a", default: ["*"],
+              desc: "Github app names. Can be matched using the '*' wildcard. Example: \"neeto*\" \"neeto-cal-web\""
+            }
+
+          desc "compliance-fix", "Create PRs for fixing neeto compliance in all products"
+          def compliance_fix
+            ComplianceFix.new(options[:apps], options[:sandbox]).run
+          end
+
+          desc "script", "Create PRs after running the given script for each product."
+          option :path, type: :string, aliases: "-p",
+            desc: "Path to UNIX executable script which will be run for each product", required: true
+          option :title, type: :string, aliases: "-t", desc: "Title for the PR"
+          option :branch, type: :string, aliases: "-b", desc: "Feature branch name for the PR"
+          def script
+            Script.new(options[:apps], options[:path], options[:title], options[:branch], options[:sandbox]).run
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/make_pr/compliance_fix.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Github
+      module MakePr
+        class ComplianceFix < Base
+          attr_accessor :apps, :sandbox
+
+          def initialize(apps, sandbox = false)
+            super()
+            @apps = apps
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :github, sandbox)
+            delete_and_create_temp_neetob_dir
+            matching_apps.each do |app|
+              ui.info("\n  Working on #{app}  \n")
+              begin
+                clone_app_in_tmp_dir(app)
+                check_and_delete_remote_branch(app)
+                bundle_install(app)
+                ui.info(add_commmit_and_push_changes(app))
+                delete_local_feature_branch(app)
+                res = client.create_pull_request(app, "main", BRANCH_NAME, PR_TITLE)
+                ui.success("PR created in \"#{app}\" project successfully.")
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+            `rm -rf /tmp/neetob`
+          end
+
+          private
+
+            def bundle_install(app)
+              `#{cd_to_app_in_tmp_dir(app)} && bundle`
+            end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/make_pr/script.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Github
+      module MakePr
+        class Script < Base
+          attr_accessor :apps, :sandbox, :path_to_the_script_file
+
+          def initialize(apps, path_to_the_script_file, pr_title, branch_name, sandbox = false)
+            super(pr_title, branch_name)
+            @apps = apps
+            @sandbox = sandbox
+            @path_to_the_script_file = path_to_the_script_file
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :github, sandbox)
+            delete_and_create_temp_neetob_dir
+            matching_apps.each do |app|
+              ui.info("\n  Working on #{app}  \n")
+              begin
+                clone_app_in_tmp_dir(app)
+                check_and_delete_remote_branch(app)
+                update_script_file_permissions
+                execute_script(app)
+                ui.info(add_commmit_and_push_changes(app))
+                delete_local_feature_branch(app)
+                res = client.create_pull_request(app, "main", branch_name, pr_title)
+                ui.success("PR created in \"#{app}\" project successfully.")
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+            `rm -rf /tmp/neetob`
+          end
+
+          private
+
+            def update_script_file_permissions
+              `chmod +x #{path_to_the_script_file}`
+            end
+
+            def execute_script(app)
+              `#{cd_to_app_in_tmp_dir(app)} && #{path_to_the_script_file}`
+            end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/protect_branch.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Github
+      class ProtectBranch < Base
+        attr_accessor :branch_name, :required_rules_json_file_path, :apps, :sandbox
+
+        def initialize(branch_name, apps, required_rules_json_file_path = "", sandbox = false)
+          super()
+          @branch_name = branch_name
+          @required_rules_json_file_path = required_rules_json_file_path
+          @apps = apps
+          @sandbox = sandbox
+        end
+
+        def run
+          matching_apps = find_all_matching_apps(apps, :github, sandbox)
+          inform_about_default_rules_file
+          matching_apps.each do |app|
+            ui.info("\n Working on \"#{app}\" repo")
+            ui.info(" Updating \"#{branch_name}\" branch protection rules")
+            rules = read_json_file(required_rules_json_file_path || default_rules_file_path)
+            rules_with_symbol_keys = rules.transform_keys(&:to_sym)
+            client.protect_branch(app, branch_name, rules_with_symbol_keys)
+            ui.success("Branch protection rules updated successfully")
+          end
+        end
+
+        private
+
+          def default_rules_file_path
+            File.expand_path("../../../../data/branch-protection-rules.json", __dir__)
+          end
+
+          def inform_about_default_rules_file
+            if required_rules_json_file_path.nil?
+              ui.info("Updating protection rules from the \"neetob/data/branch-protection-rules.json\" file")
+            end
+          end
+      end
+    end
+  end
+end

data/lib/neetob/cli/github/search.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Github
+      class Search < Base
+        attr_accessor :apps, :key_to_search, :search_file_path, :sandbox
+
+        def initialize(apps, key_to_search, search_file_path, sandbox = false)
+          super()
+          @apps = apps
+          @key_to_search = key_to_search
+          @search_file_path = search_file_path
+          @sandbox = sandbox
+        end
+
+        def run
+          matching_apps = find_all_matching_apps(apps, :github, sandbox)
+          matching_apps.each do |app|
+            begin
+              ui.info("\n Searching in \"#{app}/#{search_file_path}\" for \"#{key_to_search}\"\n")
+              content = Base64.decode64(client.contents(app, path: search_file_path).content)
+              ui.error("Keyword not found") and next unless content.include? key_to_search
+
+              content.each_line do |line|
+                ui.success(line) if line.include?(key_to_search)
+              end
+            rescue StandardError => e
+              ExceptionHandler.new(e).process
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/access/add.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require_relative "../../base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module Access
+        class Add < Base
+          attr_accessor :apps, :new_users, :sandbox
+
+          def initialize(apps, new_users = [], sandbox = false)
+            super()
+            @apps = apps
+            @new_users = new_users
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :heroku, sandbox)
+            matching_apps.each do |app|
+              `heroku access -a #{app}`
+              unless $?.success?
+                ui.error("There is a problem in accessing the app with name \"#{app}\" in your account.")
+                ui.error("Please check the specified app name and ensure you're authorized to view that app.")
+                next
+              end
+
+              new_users.each do |user|
+                ui.info(`heroku access:add #{user} -a #{app}`)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/access/commands.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "list"
+require_relative "add"
+require_relative "remove"
+require_relative "../../sub_command_base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module Access
+        class Commands < SubCommandBase
+          class_option :apps,
+            {
+              type: :array, aliases: "-a", required: true,
+              desc: "Heroku app names. Can be matched using the '*' wildcard. Example: \"neeto*\" \"neeto-cal-web-staging\""
+            }
+
+          desc "list", "List all the users of the Heroku apps"
+          def list
+            List.new(options[:apps], options[:sandbox]).run
+          end
+
+          desc "add", "Add new users to the Heroku apps"
+          option :users, type: :array, required: true, desc: "Users' emails that need to be added to Heroku apps"
+          def add
+            Add.new(options[:apps], options[:users], options[:sandbox]).run
+          end
+
+          desc "remove", "Remove the users from the Heroku apps"
+          option :users, type: :array, required: true, desc: "Users' emails that need to be removed from the Heroku apps"
+          def remove
+            Remove.new(options[:apps], options[:users], options[:sandbox]).run
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/access/list.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative "../../base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module Access
+        class List < Base
+          attr_accessor :apps, :sandbox
+
+          def initialize(apps, sandbox = false)
+            super()
+            @apps = apps
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :heroku, sandbox)
+            matching_apps.each do |app|
+              ui.info("\n  Users for \"#{app}\" app\n")
+              users = `heroku access -a #{app}`
+              unless $?.success?
+                ui.error("There is a problem in accessing the app with name \"#{app}\" in your account.")
+                ui.error("Please check the specified app name and ensure you're authorized to view that app.")
+                next
+              end
+
+              ui.success(users)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/access/remove.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require_relative "../../base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module Access
+        class Remove < Base
+          attr_accessor :apps, :users_to_be_removed, :sandbox
+
+          def initialize(apps, users_to_be_removed = [], sandbox = false)
+            super()
+            @apps = apps
+            @users_to_be_removed = users_to_be_removed
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :heroku, sandbox)
+            matching_apps.each do |app|
+              `heroku access -a #{app}`
+              unless $?.success?
+                ui.error("There is a problem in accessing the app with name \"#{app}\" in your account.")
+                ui.error("Please check the specified app name and ensure you're authorized to view that app.")
+                next
+              end
+
+              users_to_be_removed.each do |user|
+                ui.info(`heroku access:remove #{user} -a #{app}`)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/commands.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require "thor"
+require_relative "config_vars/commands"
+require_relative "access/commands"
+require_relative "execute"
+
+module Neetob
+  class CLI
+    module Heroku
+      class Commands < Thor
+        desc "execute", "Run the given Heroku CLI command or Rails console command on Heroku apps"
+        option :command, type: :string, aliases: "-c", required: true, desc: "Command to run on Heroku apps"
+        option :rails, type: :boolean, aliases: "-r", default: false, desc: "Use this flag to run the given command in the Rails console"
+        option :apps, type: :array, aliases: "-a", required: true, desc: "Heroku app names. Can be matched using the '*' wildcard. Example: \"neeto*\" \"neeto-cal-web-staging\""
+        def execute
+          Execute.new(options[:apps], options[:command], options[:rails], options[:sandbox]).run
+        end
+
+        desc "config_vars", "Interact with the config(env) variables on Heroku apps"
+        subcommand "config_vars", ConfigVars::Commands
+
+        desc "access", "Interact with the users of the Heroku apps"
+        subcommand "access", Access::Commands
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/config_vars/audit.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require "json"
+require "terminal-table"
+
+require_relative "base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module ConfigVars
+        class Audit < Base
+          attr_accessor :apps, :expected_config_vars_json_file_path, :sandbox
+
+          def initialize(apps, expected_config_vars_json_file_path = "", sandbox = false)
+            super()
+            @apps = apps
+            @expected_config_vars_json_file_path = expected_config_vars_json_file_path
+            @sandbox = sandbox
+          end
+
+          def run
+            matching_apps = find_all_matching_apps(apps, :heroku, sandbox)
+            table_rows = []
+            expected_config = read_json_file(expected_config_vars_json_file_path || default_config_vars_json_file_path)
+            return unless expected_config
+
+            matching_apps.each do |app|
+              actual_config = `heroku config -a #{app} --json`
+              unless $?.success?
+                ui.error("There is a problem in accessing the app with name \"#{app}\" in your account.")
+                ui.error("Please check the specified app name and ensure you're authorized to view that app.")
+                next
+              end
+
+              actual_config_json = JSON.parse(actual_config)
+              table_rows.concat(compare_config_and_build_rows(app, expected_config, actual_config_json))
+            end
+            table = Terminal::Table.new headings: ["App", "Config var", "Expected", "Result"], rows: table_rows
+            ui.success(table)
+          end
+
+          private
+
+            def compare_config_and_build_rows(app, expected_config, actual_config)
+              expected_config.map do |key, val|
+                row = [app, key.to_s]
+                row << (value_set_dynamically_by_heroku?(val) ? "Heroku has dynamically set the value" : val.to_s)
+                row << (actual_config[key] == val ? "✅" : (actual_config[key] || "No such key"))
+              end
+            end
+
+            def value_set_dynamically_by_heroku?(value)
+              value == { "isDynamic" => true }
+            end
+
+            def default_config_vars_json_file_path
+              File.expand_path("../../../../../data/config-vars-audit.json", __dir__)
+            end
+        end
+      end
+    end
+  end
+end

data/lib/neetob/cli/heroku/config_vars/base.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require_relative "../../base"
+
+module Neetob
+  class CLI
+    module Heroku
+      module ConfigVars
+        class Base < CLI::Base
+          attr_accessor :client
+
+          def initialize
+            super()
+          end
+        end
+      end
+    end
+  end
+end