github-changelog-generator/lib/github_changelog_generator/parser.rb

#!/usr/bin/env ruby
require "optparse"
require "pp"
require_relative "version"

module GitHubChangelogGenerator
  class Parser
    def self.parse_options
      options = {
        tag1: nil,
        tag2: nil,
        date_format: "%Y-%m-%d",
        output: "CHANGELOG.md",
        issues: true,
        add_issues_wo_labels: true,
        add_pr_wo_labels: true,
        pulls: true,
        filter_issues_by_milestone: true,
        author: true,
        unreleased: true,
        unreleased_label: "Unreleased",
        compare_link: true,
        include_labels: %w(bug enhancement),
        exclude_labels: %w(duplicate question invalid wontfix),
        max_issues: nil,
        simple_list: false,
        verbose: true,

        merge_prefix: "**Merged pull requests:**",
        issue_prefix: "**Closed issues:**",
        bug_prefix: "**Fixed bugs:**",
        enhancement_prefix: "**Implemented enhancements:**",
        git_remote: "origin"
      }

      parser = OptionParser.new do |opts|
        opts.banner = "Usage: github_changelog_generator [options]"
        opts.on("-u", "--user [USER]", "Username of the owner of target GitHub repo") do |last|
          options[:user] = last
        end
        opts.on("-p", "--project [PROJECT]", "Name of project on GitHub") do |last|
          options[:project] = last
        end
        opts.on("-t", "--token [TOKEN]", "To make more than 50 requests per hour your GitHub token is required. You can generate it at: https://github.com/settings/tokens/new") do |last|
          options[:token] = last
        end
        opts.on("-f", "--date-format [FORMAT]", "Date format. Default is %Y-%m-%d") do |last|
          options[:date_format] = last
        end
        opts.on("-o", "--output [NAME]", "Output file. Default is CHANGELOG.md") do |last|
          options[:output] = last
        end
        opts.on("--[no-]issues", "Include closed issues in changelog. Default is true") do |v|
          options[:issues] = v
        end
        opts.on("--[no-]issues-wo-labels", "Include closed issues without labels in changelog. Default is true") do |v|
          options[:add_issues_wo_labels] = v
        end
        opts.on("--[no-]pr-wo-labels", "Include pull requests without labels in changelog. Default is true") do |v|
          options[:add_pr_wo_labels] = v
        end
        opts.on("--[no-]pull-requests", "Include pull-requests in changelog. Default is true") do |v|
          options[:pulls] = v
        end
        opts.on("--[no-]filter-by-milestone", "Use milestone to detect when issue was resolved. Default is true") do |last|
          options[:filter_issues_by_milestone] = last
        end
        opts.on("--[no-]author", "Add author of pull-request in the end. Default is true") do |author|
          options[:author] = author
        end
        opts.on("--unreleased-only", "Generate log from unreleased closed issues only.") do |v|
          options[:unreleased_only] = v
        end
        opts.on("--[no-]unreleased", "Add to log unreleased closed issues. Default is true") do |v|
          options[:unreleased] = v
        end
        opts.on("--unreleased-label [label]", "Add to log unreleased closed issues. Default is true") do |v|
          options[:unreleased_label] = v
        end
        opts.on("--[no-]compare-link", "Include compare link (Full Changelog) between older version and newer version. Default is true") do |v|
          options[:compare_link] = v
        end
        opts.on("--include-labels  x,y,z", Array, 'Only issues with the specified labels will be included in the changelog. Default is \'bug,enhancement\'') do |list|
          options[:include_labels] = list
        end
        opts.on("--exclude-labels  x,y,z", Array, 'Issues with the specified labels will be always excluded from changelog. Default is \'duplicate,question,invalid,wontfix\'') do |list|
          options[:exclude_labels] = list
        end
        opts.on("--max-issues [NUMBER]", Integer, "Max number of issues to fetch from GitHub. Default is unlimited") do |max|
          options[:max_issues] = max
        end
        opts.on("--github-site [URL]", "The Enterprise Github site on which your project is hosted.") do |last|
          options[:github_site] = last
        end
        opts.on("--github-api [URL]", "The enterprise endpoint to use for your Github API.") do |last|
          options[:github_endpoint] = last
        end
        opts.on("--simple-list", "Create simple list from issues and pull requests. Default is false.") do |v|
          options[:simple_list] = v
        end
        opts.on("--future-release [RELEASE-VERSION]", "Put the unreleased changes in the specified release number.") do |future_release|
          options[:future_release] = future_release
        end
        opts.on("--[no-]verbose", "Run verbosely. Default is true") do |v|
          options[:verbose] = v
        end
        opts.on("-v", "--version", "Print version number") do |_v|
          puts "Version: #{GitHubChangelogGenerator::VERSION}"
          exit
        end
        opts.on("-h", "--help", "Displays Help") do
          puts opts
          exit
        end
      end

      parser.parse!

      detect_user_and_project(options)

      if !options[:user] || !options[:project]
        puts parser.banner
        exit
      end

      if options[:verbose]
        puts "Performing task with options:"
        pp options
        puts ""
      end

      options
    end

    def self.detect_user_and_project(options)
      options[:user], options[:project] = user_project_from_option(ARGV[0], ARGV[1], options[:github_site])
      if !options[:user] || !options[:project]
        if ENV["RUBYLIB"] =~ /ruby-debug-ide/
          options[:user] = "skywinder"
          options[:project] = "changelog_test"
        else
          remote = `git config --get remote.#{options[:git_remote]}.url`
          options[:user], options[:project] = user_project_from_remote(remote)
        end
      end
    end

    # Try to find user and project name from git remote output
    #
    # @param [String] output of git remote command
    # @return [Array] user and project
    def self.user_project_from_option(arg0, arg2, github_site = "github.com")
      user = nil
      project = nil

      if arg0 && !arg2
        # this match should parse  strings such "https://github.com/skywinder/Github-Changelog-Generator" or "skywinder/Github-Changelog-Generator" to user and name
        match = /(?:.+#{Regexp.escape(github_site)}\/)?(.+)\/(.+)/.match(arg0)

        begin
          param = match[2].nil?
        rescue
          puts "Can't detect user and name from first parameter: '#{arg0}' -> exit'"
          exit
        end
        if param
          exit
        else
          user = match[1]
          project = match[2]
        end
      end
      [user, project]
    end

    # Try to find user and project name from git remote output
    #
    # @param [String] output of git remote command
    # @return [Array] user and project
    def self.user_project_from_remote(remote)
      # try to find repo in format:
      # origin	git@github.com:skywinder/Github-Changelog-Generator.git (fetch)
      # git@github.com:skywinder/Github-Changelog-Generator.git
      regex1 = /.*(?:[:\/])((?:-|\w|\.)*)\/((?:-|\w|\.)*)(?:\.git).*/

      # try to find repo in format:
      # origin	https://github.com/skywinder/ChangelogMerger (fetch)
      # https://github.com/skywinder/ChangelogMerger
      regex2 = /.*\/((?:-|\w|\.)*)\/((?:-|\w|\.)*).*/

      remote_structures = [regex1, regex2]

      user = nil
      project = nil
      remote_structures.each do |regex|
        matches = Regexp.new(regex).match(remote)

        if matches && matches[1] && matches[2]
          puts "Detected user:#{matches[1]}, project:#{matches[2]}"
          user = matches[1]
          project = matches[2]
        end

        break unless matches.nil?
      end

      [user, project]
    end
  end

  if __FILE__ == $PROGRAM_NAME
    remote = "invalid reference to project"
    p user_project_from_option(ARGV[0], ARGV[1], remote)
  end
end