abtion-scripts 1.1.1

data/lib/abtion_scripts/begin.rb

@@ -0,0 +1,195 @@
+require 'json'
+require 'tempfile'
+
+module AbtionScripts
+  class Pivotal
+    attr_reader :token
+
+    def initialize(token)
+      @token = token
+    end
+
+    def request(url, method: "GET", body: nil)
+      cmd = %Q{curl --silent -X #{method} -H "X-TrackerToken: #{token}"}
+      cmd += %Q{ -H "Content-Type: application/json"}
+
+      if body
+        cmd += %Q{ -d '#{body}'}
+      end
+
+      cmd += %Q{ "#{url}"}
+
+      result = `#{cmd}`
+
+      JSON.parse(result.strip)
+    end
+  end
+
+  class Begin < Base
+    def self.description
+      "Starts your Pivotal Tracker story and opens a new PR on Github"
+    end
+
+    def self.help
+      <<-EOF
+abtion begin #{colorize(:light_blue, "[story id] [branch name]")}
+
+Example: $ #{colorize(:light_blue, 'abtion begin "#133717234"')}
+
+This command will start your Pivotal Tracker story for you, open a pull
+request on Github, and copy over the Pivotal Tracker story description to
+the Github pull request description. As well, any tasks in your
+Pivotal Tracker story will automatically become [x] checkbox tasks on the
+Github PR.
+
+* All branch names will be auto-converted to
+  kebab-case, lowercase
+
+* Passing story id/branch name as arguments are optional - if
+  they are missing, you'll be prompted
+EOF
+    end
+
+    def run
+      story_id, branch_name = argv
+
+      if !command?("hub")
+        abort <<-EOF
+You need to install `hub` before you can use this program.
+
+brew install hub
+EOF
+      end
+
+      pivotal = Pivotal.new(git_config("user.pivotalApiToken"))
+      config_abort_if_blank!("user.pivotalApiToken", pivotal.token)
+
+      project_id = git_config("user.pivotalProjectId")
+      config_abort_if_blank!("user.pivotalProjectId", project_id)
+
+      story_id ||= prompt("Please paste the Pivotal story ID here")
+      story_id = story_id.gsub(/[^\d]/, '')
+
+      story_url = "https://www.pivotaltracker.com/services/v5"\
+        "/projects/#{project_id}/stories/#{story_id}"
+
+      story = pivotal.request(story_url)
+
+      default_branch_name = normalized_branch_name(story['name'])
+
+      branch_name ||= prompt("Please enter a branch name (#{default_branch_name})")
+
+      if branch_name.strip == ""
+        branch_name = default_branch_name
+      else
+        branch_name = normalized_branch_name(branch_name)
+      end
+
+      # Start the story
+      pivotal.request(story_url, method: "PUT", body: '{ "current_state":"started" }')
+
+      silent "git checkout master",
+        "git fetch origin",
+        "git reset --hard origin/master"
+
+      puts "==> Checking out #{branch_name}"
+
+      silent "git checkout -b #{branch_name}",
+        'git commit --allow-empty -m "Initial commit for story #' + story_id + '"',
+        "git push origin #{branch_name}",
+        "git branch --set-upstream #{branch_name} origin/#{branch_name}"
+
+      tasks = pivotal.request(
+      "https://www.pivotaltracker.com/services/v5"\
+        "/projects/#{project_id}/stories/#{story_id}/tasks"
+      )
+
+      story_description = story['description']
+
+      if story_description.nil?
+        story_description = "_No story description given in Pivotal_"
+      end
+
+      description = <<-EOF
+#{story['name']}
+
+#{story['url']}
+
+# Description
+
+#{story_description}
+EOF
+
+      description.strip!
+
+      unless tasks.empty?
+        description += "\n\n## TODO\n\n"
+
+        tasks.each do |task|
+          description += "- [ ] #{task['description']}\n"
+        end
+
+        description.strip!
+      end
+
+      puts "==> Opening pull request on GitHub"
+
+      tempfile = Tempfile.new('begin_pull_request')
+
+      begin
+        tempfile.write(description)
+        tempfile.close
+
+        labels = ['WIP', story['story_type']].join(',')
+
+        url = `hub pull-request -F #{tempfile.path} -l "#{labels}" -a "" -o`
+
+        # Copy it to your clipboard
+        system("echo #{url} | pbcopy")
+        puts url
+      ensure
+        tempfile.unlink
+      end
+    end
+
+    private
+
+    def silent(*cmds)
+      cmds.each { |cmd| system("#{cmd} >/dev/null 2>&1") }
+    end
+
+    def command?(name)
+      `which #{name}`
+      $?.success?
+    end
+
+    def git_config(key)
+      `git config --local --get #{key}`.strip
+    end
+
+    def config_abort_if_blank!(key, value)
+      if value.strip == ""
+        abort <<-EOF
+You need to set the #{key} value in your git config!
+
+git config --local --add #{key} [value]
+EOF
+      end
+    end
+
+    def prompt(msg)
+      print "#{msg} > "
+      value = STDIN.gets.strip
+      puts
+      value
+    end
+
+    def normalized_branch_name(branch_name)
+      branch_name
+        .gsub(/[^\w\s-]/, '')
+        .gsub(/\s+/, '-')
+        .downcase
+        .gsub(/-*$/, '') # trailing dashes
+    end
+  end
+end

data/lib/abtion_scripts/colorize.rb

@@ -0,0 +1,39 @@
+module AbtionScripts::Colorize
+  extend self
+
+  def self.included(base)
+    colorize = self
+
+    base.class_eval do
+      extend colorize
+    end
+  end
+
+  COLOR_CODES = {
+    black: 30,
+    blue: 34,
+    brown: 33,
+    cyan: 36,
+    dark_gray: 90,
+    green: 32,
+    light_blue: 94,
+    light_cyan: 96,
+    light_gray: 37,
+    light_green: 92,
+    light_purple: 95,
+    light_red: 91,
+    light_yellow: 93,
+    purple: 35,
+    red: 31,
+    white: 97,
+    yellow: 33,
+
+    command: 96,
+    error: 91,
+    info: 93,
+  }
+
+  def colorize(color, string)
+    "\e[#{COLOR_CODES[color]}m#{string}\e[0m"
+  end
+end

data/lib/abtion_scripts/diff.rb

@@ -0,0 +1,29 @@
+class Diff < AbtionScripts::Base
+  def self.name
+    "diff"
+  end
+
+  DAY = 86400
+
+  def run
+    yesterday = (Time.now - DAY).strftime("%Y-%m-%d")
+
+    last_sha = `git log --after "#{yesterday} 00:00" --before "#{yesterday} 23:59" --format="format:%H"`
+      .strip
+      .split("\n")
+      .last
+
+    start_sha = `git log "#{last_sha}^" --format="format:%H"`
+      .strip
+      .split("\n")
+      .first
+
+    repo = `git remote -v | grep github | awk '{ print $2 }' | head -n 1`.strip
+    repo.gsub!(/\.git/, "")
+    repo.gsub!(/git@github.com:/, "")
+
+    github_repo = repo.split("/")[-2..-1].join("/")
+
+    system("open https://github.com/#{github_repo}/compare/#{start_sha}...master")
+  end
+end

data/lib/abtion_scripts/doctor.rb

@@ -0,0 +1,200 @@
+class AbtionScripts::Doctor < AbtionScripts::Base
+  class Check
+    include AbtionScripts::Colorize
+
+    attr_reader :name, :command, :remedy, :problems
+
+    def initialize(name:, command:, remedy:)
+      @name = name
+      @command = command
+      @remedy = remedy
+      @problems = []
+    end
+
+    def run!
+      print "Checking: #{name}... "
+
+      success = if command.respond_to?(:call)
+        command.call
+      else
+        system "#{command} > /dev/null 2>&1"
+      end
+
+      if success
+        puts 'OK'
+      else
+        print colorize(:error, 'F')
+        fix = remedy.respond_to?(:join) ? remedy.join(" ") : remedy
+        puts "\n  To fix: #{fix}\n\n"
+
+        problems << name
+      end
+    end
+  end
+
+  def self.description
+    "Checks the health of your development environment"
+  end
+
+  def initialize(*args)
+    super
+    @checks = []
+  end
+
+  def run
+    case argv.first
+    when "list"
+      list_default_checks
+    else
+      run_doctor
+    end
+  end
+
+  def self.help
+    "doctor - helps you diagnose any setup issues with this application\n"
+  end
+
+  def self.help_subcommands
+    {
+      "abtion doctor" => "runs health checks and gives a report",
+      "abtion doctor list" => "prints a list of default checks you can use when overriding doctor checks in your app"
+    }
+  end
+
+  def run_doctor
+    run_checks
+    report
+  end
+
+  def list_default_checks
+    puts "These default checks are available for use in your overrides:"
+    puts
+
+    default_checks.each do |name|
+      puts "  - #{colorize(:light_blue, name)}"
+    end
+
+    puts
+  end
+
+  def check(**options)
+    check = Check.new(options)
+    @checks << check
+
+    check.run!
+  end
+
+  private
+
+  def run_checks
+    run_default_checks
+  end
+
+  def run_default_checks
+    default_checks.each do |check|
+      send(check)
+    end
+  end
+
+  def default_checks
+    %i[
+      check_envrc_file_exists
+      check_env_file_exists
+      check_direnv_installed
+      check_gemfile_dependencies
+      check_postgres_launchctl
+      check_postgres_running
+      check_postgres_role
+      check_db_exists
+      check_db_migrated
+      check_phantomjs_installed
+    ]
+  end
+
+  def check_postgres_launchctl
+    check \
+      name: "postgres launchctl script is linked",
+      command: "ls -1 ~/Library/LaunchAgents/homebrew.mxcl.postgresql.plist",
+      remedy: command("ln -sfv /usr/local/opt/postgresql/*.plist ~/Library/LaunchAgents")
+  end
+
+  def check_postgres_running
+    check \
+      name: "postgres is running",
+      command: "psql -l",
+      remedy: command("launchctl load ~/Library/LaunchAgents/homebrew.mxcl.postgresql.plist")
+  end
+
+  def check_postgres_role
+    check \
+      name: "postgres role exists",
+      command: "psql -U postgres -l",
+      remedy: command("createuser --superuser postgres")
+  end
+
+  def check_gemfile_dependencies
+    check \
+      name: "Gemfile dependencies are up to date",
+      command: "bundle check",
+      remedy: command("bundle")
+  end
+
+  def check_db_exists
+    check \
+      name: "Development database exists",
+      command: "source .envrc && rails runner -e development 'ActiveRecord::Base.connection'",
+      remedy: command("rake db:setup")
+
+    check \
+      name: "Test database exists",
+      command: "source .envrc && rails runner -e test 'ActiveRecord::Base.connection'",
+      remedy: command("rake db:setup")
+  end
+
+  def check_db_migrated
+    check \
+      name: "DB is migrated",
+      command: "source .envrc && rails runner 'ActiveRecord::Migration.check_pending!'",
+      remedy: command("rake db:migrate db:test:prepare")
+  end
+
+  def check_direnv_installed
+    check \
+      name: "direnv installed",
+      command: "which direnv",
+      remedy: command("brew install direnv")
+  end
+
+  def check_phantomjs_installed
+    check \
+      name: "PhantomJS installed",
+      command: "which phantomjs",
+      remedy: command("brew install phantomjs")
+  end
+
+  def check_envrc_file_exists
+    check \
+      name: ".envrc file exists",
+      command: "stat .envrc",
+      remedy: command("cp .envrc.sample .envrc")
+  end
+
+  def check_env_file_exists
+    check \
+      name: ".env file exists (for 'heroku local')",
+      command: "stat .env",
+      remedy: command("ln -s .envrc .env")
+  end
+
+  def problems
+    @checks.map(&:problems).flatten
+  end
+
+  def report
+    exit problems.size
+  end
+
+  def command(s)
+    "run #{colorize :command, s}"
+  end
+end