neetob-ud 0.1.0

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

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class Delete < Base
+          attr_accessor :apps, :sandbox, :labels, :all_neeto_repos
+
+          def initialize(apps, labels, sandbox = false, all_neeto_repos = false)
+            super()
+            @labels = labels
+            @apps = apps
+            @sandbox = sandbox
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            matching_apps.each do |app|
+              ui.info("\n Deleting labels from #{app} repo \n")
+              labels.each do |label|
+                delete_label(app, label)
+              end
+            end
+          end
+
+          private
+
+            def delete_label(app, label)
+              if client.delete_label!(app, label)
+                ui.success("The \"#{label}\" label deleted successfully")
+              else
+                ui.error("The \"#{label}\" label can't be deleted. Please check and try again")
+              end
+            end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class DeleteAll < Base
+          attr_accessor :apps, :sandbox, :all_neeto_repos
+
+          def initialize(apps, sandbox = false, all_neeto_repos = false)
+            super()
+            @apps = apps
+            @sandbox = sandbox
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            matching_apps.each do |app|
+              ui.info("\n Working on #{app} repo \n")
+              begin
+                labels = client.labels(app)
+                labels.each do |label|
+                  delete_label(app, label)
+                end
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+          end
+
+          private
+
+            def delete_label(app, label_details)
+              if client.delete_label!(app, label_details[:name])
+                ui.success("The \"#{label_details[:name]}\" label deleted successfully")
+              else
+                ui.error("The \"#{label_details[:name]}\" label can't be deleted. Please try again")
+              end
+            end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class List < Base
+          attr_accessor :apps, :sandbox, :all_neeto_repos
+
+          def initialize(apps, sandbox = false, all_neeto_repos = false)
+            super()
+            @apps = apps
+            @sandbox = sandbox
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            matching_apps.each do |app|
+              ui.info("\n  Labels of #{app}  \n")
+              begin
+                labels = client.labels(app)
+                ui.success(labels)
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class Show < Base
+          attr_accessor :apps, :label_name, :sandbox, :all_neeto_repos
+
+          def initialize(apps, label_name, sandbox = false, all_neeto_repos = false)
+            super()
+            @apps = apps
+            @label_name = label_name
+            @sandbox = sandbox
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            matching_apps.each do |app|
+              ui.info("\n \"#{label_name}\" label details from #{app}  \n")
+              begin
+                label_details = client.label(app, label_name)
+                ui.success(label_details.inspect)
+              rescue StandardError => e
+                ExceptionHandler.new(e).process
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "thor"
+
+require_relative "../base"
+
+module Neetob
+  class CLI
+    module Github
+      module Labels
+        class Update < Base
+          attr_accessor :apps, :sandbox, :old_name, :new_name, :all_neeto_repos
+
+          def initialize(apps, old_name, new_name, sandbox = false, all_neeto_repos = false)
+            super()
+            @apps = apps
+            @sandbox = sandbox
+            @old_name = old_name
+            @new_name = new_name
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            matching_apps.each do |app|
+              ui.info("\n Updating label for #{app} repo \n")
+              update_label!(app, old_name, new_name)
+            end
+          end
+
+          private
+
+            def update_label!(app, old_name, new_name)
+              client.update_label(app, old_name, { name: new_name })
+              ui.success("Label \"#{old_name}\" updated to \"#{new_name}\" successfully")
+            end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+# 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, :all_neeto_repos
+
+          def initialize(apps, required_labels_json_file_path = "", sandbox = false, all_neeto_repos = false)
+            super()
+            @apps = apps
+            @required_labels_json_file_path = required_labels_json_file_path
+            @sandbox = sandbox
+            @all_neeto_repos = all_neeto_repos
+          end
+
+          def run
+            check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+            matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+            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,48 @@
+# 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, :all_neeto_repos
+
+        def initialize(branch_name, apps, required_rules_json_file_path = "", sandbox = false, all_neeto_repos = false)
+          super()
+          @branch_name = branch_name
+          @required_rules_json_file_path = required_rules_json_file_path
+          @apps = apps
+          @sandbox = sandbox
+          @all_neeto_repos = all_neeto_repos
+        end
+
+        def run
+          check_for_apps_and_all_neeto_repos_option(apps, all_neeto_repos)
+          matching_apps = find_all_matching_apps(apps, :github, sandbox, false, all_neeto_repos)
+          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