wagemage 1.0.0

data/lib/wagemage/cli.rb

@@ -0,0 +1,191 @@
+module Wagemage
+  class CLI
+    include Wagemage::Helpers
+
+    def initialize(args)
+      @options = Slop.parse(args) do |option|
+        option.bool '-h', '--help', 'print this help'
+        option.on '-v', '--version', 'print the version'
+
+        option.string '-o', '--org', 'github org'
+        option.string '-r', '--repo', 'regex against which to match repo names'
+        option.string '-b', '--branch', 'regex against which to match branches'
+
+        option.path '-s', '--script', "the script to run on each repo's branch"
+
+        option.bool '--first-branch', 'operate only on the "oldest" branch'
+        option.array '--reviewers', 'array of github users to put on the PR'
+        option.string(
+          '--branch-prefix',
+          'prefix of the new branch',
+          default: 'wagemage'
+        )
+
+        option.bool '--debug', "don't push or issue PR, keep the tmp directory"
+      end
+
+      validate_options!
+
+      token = ENV['WAGEMAGE_GITHUB_TOKEN'] || request_token
+      @okclient = Octokit::Client.new(access_token: token)
+
+      @tmpdir = Dir.mktmpdir
+      @script_path = @options[:script].expand_path.to_s
+    end
+
+    def run
+      raise Error, 'No repos found' if repos.empty?
+
+      say "#{repos.size} repo(s) found:", space: true
+      repos.each { |r| say "* #{r.name}", color: :green }
+
+      say 'Would you like to clone these repos? (Y/n)', space: true
+      abort if ask.casecmp?('n')
+
+      begin
+        clone_repos
+        display_branch_list
+
+        say <<~MESSAGE, space: true, color: :yellow
+          You're about to run this script against the aforementioned list:
+          => #{@script_path}
+        MESSAGE
+
+        if @options[:debug]
+          say "(--debug flag enabled. No code will be pushed.)", color: :green
+        else
+          say "(--debug flag NOT enabled. Code may be pushed.)", color: :red
+        end
+
+        say "Would you like to execute this script? (Y/n)"
+        abort if ask.casecmp?('n')
+
+        repos.each do |repo|
+          repo.branches.each_with_index do |branch, index|
+            if index > 0
+              say "=> Skipping #{repo.name}:#{branch}", color: :yellow
+              next
+            end
+
+            new_branch = [
+              @options[:branch_prefix],
+              branch,
+              Time.now.to_i
+            ].join('/')
+
+            repo.checkout! branch
+            repo.checkout! new_branch, create: true
+
+            say "=> Running script on #{repo.name}:#{new_branch}"
+            script_result = command(
+              [@script_path, repo.clone_dir, repo.name, branch].join(' ')
+            )
+
+            unless script_result[:stderr].empty?
+              say script_result[:stderr], color: :red
+            end
+
+            if script_result[:status].success? && !repo.has_changed?
+              say 'SCRIPT SUCCEEDED BUT NO CHANGES TO COMMIT!', color: :yellow
+              next
+            elsif !script_result[:status].success?
+              say 'SCRIPT FAILED!', color: :red
+              next
+            end
+
+            say 'SCRIPT SUCCEEDED! COMMITTING CHANGES!', color: :green
+
+            repo.add_all!
+            repo.commit! script_result[:stdout]
+
+            if @options[:debug]
+              say 'DEBUG ENABLED! SKIPPING PUSH & PULL REQUEST!', color: :yellow
+              next
+            end
+
+            repo.push!
+
+            pr_result = repo.pull_request!(branch, @options[:reviewers])
+
+            if pr_result[:status].success?
+              say "=> PULL REQUEST URL: #{pr_result[:stdout]}", color: :green
+            else
+              say "=> PULL REQUEST FAILED!", color: :red
+            end
+          end
+        end
+      ensure
+        if @options[:debug]
+          say <<~MESSAGE, space: :true, color: :yellow
+            The temporary directory has been retained because you have specified
+            the --debug flag. You can view it here:
+            => #{@tmpdir}
+          MESSAGE
+        else
+          FileUtils.remove_entry(@tmpdir)
+        end
+      end
+    end
+
+    private
+
+    def repos
+      @repos ||= begin
+        repos = @options[:org].nil? ?
+          @okclient.repos :
+          @okclient.org_repos(@options[:org])
+
+        last_response = @okclient.last_response
+        while last_response.rels[:next] do
+          repos.concat(last_response.rels[:next].get.data)
+          last_response = last_response.rels[:next].get
+        end
+
+        repos.map! { |r| Wagemage::Repo.new(r, @tmpdir, @options[:branch]) }
+
+        return repos if @options[:repo].nil?
+
+        repo_name_regex = Regexp.new(@options[:repo])
+        repos.select { |r| r.name =~ repo_name_regex }
+      end
+    end
+
+    def clone_repos
+      repos.each do |repo|
+        say "=> Cloning #{repo.name} to #{repo.clone_dir}", color: :light_blue
+        repo.clone!
+      end
+    end
+
+    def display_branch_list
+      repos.each do |repo|
+        say "* #{repo.name}:", space: true
+
+        if repo.branches.empty?
+          say ' - NONE FOUND', color: :red
+          next
+        end
+
+        if @options[:first_branch]
+          say " - #{repo.branches.first}", color: :green
+          repo.branches[1..-1].each { |b| say " - #{b}" }
+        else
+          repo.branches.each { |b| say " - #{b}", color: :green }
+        end
+      end
+    end
+
+    def request_token
+      say 'Github Personal Access Token missing', color: :red
+      say 'Please supply it now:'
+      ask
+    end
+
+    def validate_options!
+      abort(@options.to_s) if @options.help?
+      abort("Wagemage v#{Wagemage::VERSION}") if @options.version?
+
+      raise OptionError if @options[:script].nil?
+    end
+  end
+end

data/lib/wagemage/helpers.rb

@@ -0,0 +1,30 @@
+module Wagemage
+  module Helpers
+    def say(message, space: false, color: :white)
+      puts if space
+      puts message.colorize(color.to_sym)
+    end
+
+    def ask
+      STDIN.gets.chomp
+    end
+
+    def warning(message)
+      say(message, color: :red)
+    end
+
+    def command(cmd, chdir: Dir.pwd, error: false)
+      stdout, stderr, status = Open3.capture3(cmd, chdir: chdir)
+
+      unless status.success?
+        error ? (raise Error, stderr) : warning(stderr)
+      end
+
+      {
+        stdout: stdout,
+        stderr: stderr,
+        status: status
+      }
+    end
+  end
+end

data/lib/wagemage/repo.rb

@@ -0,0 +1,78 @@
+module Wagemage
+  class Repo
+    attr_reader :clone_dir
+
+    def initialize(info, dir, branch_pattern)
+      @info = info
+      @clone_dir = [dir, info[:full_name]].join('/')
+      @branch_pattern = branch_pattern
+    end
+
+    def name
+      @info[:full_name]
+    end
+
+    def url
+      @info[:ssh_url]
+    end
+
+    def clone!
+      Wagemage.command("git clone #{url} #{clone_dir}", error: true)
+    end
+
+    def branches
+      @branches ||= begin
+        result = Wagemage.command('git branch -a', chdir: @clone_dir)
+
+        return [] unless result[:status].success?
+
+        branch_list =
+          result[:stdout]
+            .split("\n")
+            .select { |b| b.include?('remotes/origin/') }
+            .reject { |b| b.include?('->') }
+            .map { |b| b.split('/')[2..-1].join('/') }
+
+        if branch_list.include?('master')
+          branch_list
+            .reject! { |b| b == 'master' }
+            .push('master')
+        end
+
+        return branch_list if @branch_pattern.nil?
+
+        branch_name_regex = Regexp.new(@branch_pattern)
+        branch_list.select { |b| b =~ branch_name_regex }
+      end
+    end
+
+    def checkout!(ref, create: false)
+      cmd = create ? "git checkout -b #{ref}" : "git checkout #{ref}"
+      Wagemage.command(cmd, chdir: @clone_dir)
+    end
+
+    def add_all!
+      Wagemage.command('git add .', chdir: @clone_dir)
+    end
+
+    def commit!(message)
+      Wagemage.command(%Q[git commit -m "#{message}"], chdir: @clone_dir)
+    end
+
+    def push!
+      Wagemage.command('git push origin HEAD', chdir: @clone_dir)
+    end
+
+    def pull_request!(base_branch, reviewers = [])
+      cmd = "hub pull-request --no-edit -b #{base_branch}"
+      cmd = [cmd, '-r', reviewers.join(',')].join(' ') unless reviewers.empty?
+
+      Wagemage.command(cmd, chdir: @clone_dir)
+    end
+
+    def has_changed?
+      result = Wagemage.command('git status -s', chdir: @clone_dir)
+      !result[:stdout].empty?
+    end
+  end
+end

data/lib/wagemage/version.rb

@@ -0,0 +1,3 @@
+module Wagemage
+  VERSION = "1.0.0"
+end

data/wagemage.gemspec

@@ -0,0 +1,36 @@
+lib = File.expand_path("lib", __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "wagemage/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "wagemage"
+  spec.version       = Wagemage::VERSION
+  spec.authors       = ["Curt Howard"]
+  spec.email         = ["curt@portugly.com"]
+
+  spec.summary       = "A CLI for making changes to many Github repos"
+  spec.description   = "A CLI for making changes to many Github repos"
+  spec.homepage      = "https://github.com/meowsus/wagemage"
+  spec.license       = "MIT"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/meowsus/wagemage"
+
+  spec.files = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`
+      .split("\x0")
+      .reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "rake", "~> 12.3.3"
+  spec.add_development_dependency "minitest", "~> 5.0"
+
+  spec.add_dependency "slop", "~> 4.7"
+  spec.add_dependency "colorize", "~> 0.8"
+  spec.add_dependency "octokit", "~> 4.14"
+end

metadata

@@ -0,0 +1,157 @@
+--- !ruby/object:Gem::Specification
+name: wagemage
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Curt Howard
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2020-11-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 12.3.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 12.3.3
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: slop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.7'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+- !ruby/object:Gem::Dependency
+  name: octokit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.14'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.14'
+description: A CLI for making changes to many Github repos
+email:
+- curt@portugly.com
+executables:
+- wagemage
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/tests.yml"
+- ".gitignore"
+- ".rubocop.yml"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/rake
+- bin/setup
+- examples/add_test_file
+- examples/consolidate_pushes_during_release
+- examples/enable_teaspoon_tests
+- examples/find_decorated_test
+- examples/find_teaspoon_tests
+- examples/ignore_remote_rubocop_config
+- examples/nextyear
+- exe/wagemage
+- lib/wagemage.rb
+- lib/wagemage/cli.rb
+- lib/wagemage/helpers.rb
+- lib/wagemage/repo.rb
+- lib/wagemage/version.rb
+- wagemage.gemspec
+homepage: https://github.com/meowsus/wagemage
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/meowsus/wagemage
+  source_code_uri: https://github.com/meowsus/wagemage
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.2
+signing_key:
+specification_version: 4
+summary: A CLI for making changes to many Github repos
+test_files: []