bpescatore-stash 0.3.2

data/lib/atlassian/stash/pull_request.rb

@@ -0,0 +1,153 @@
+require 'json'
+require 'net/https'
+require 'uri'
+require 'git'
+require 'launchy'
+
+include Atlassian::Util::TextUtil
+
+module Atlassian
+  module Stash
+    class CreatePullRequestResource
+      attr_accessor :resource
+
+      def initialize(sourceRepoInfo, targetRepoInfo, title, description, reviewers, source, target)
+        src_repository = {
+        'slug' => sourceRepoInfo.slug,
+        'project' => {
+          'key' => sourceRepoInfo.projectKey
+          }
+        }
+        target_repository = {
+        'slug' => targetRepoInfo.slug,
+        'project' => {
+          'key' => targetRepoInfo.projectKey
+          }
+        }
+        fromRef = {
+          'id' => source,
+          'repository' => src_repository
+        }
+        toRef = {
+          'id' => target,
+          'repository' => target_repository
+        }
+        @resource = {
+          'title' => title,
+          'fromRef' => fromRef,
+          'toRef' => toRef
+        }
+
+        @resource["description"] = description unless description.empty?
+
+        @resource["reviewers"] = reviewers.collect { |r|
+            {
+              'user' => {
+                'name' => r
+              }
+            }
+        } unless reviewers.empty?
+      end
+    end
+
+    class CreatePullRequest
+
+      def initialize(config)
+        @config = config
+      end
+
+      def create_pull_request(source, target, reviewers, options)
+        Process.exit if not target or not source
+
+        @source = source
+        @target = target
+
+        srcRepoInfo = RepoInfo.create(@config, options.src_remote)
+        targetRepoInfo = RepoInfo.create(@config, options.target_remote)
+
+        title, description = title_and_description(options)
+
+        resource = CreatePullRequestResource.new(srcRepoInfo, targetRepoInfo, title, description, reviewers, @source, @target).resource
+
+        username = @config["username"]
+        password = @config["password"]
+        proxy_addr, proxy_port = parse_proxy(@config["proxy"])
+
+        username = ask("Username: ") unless @config["username"]
+        password = ask("Password: ") { |q| q.echo = '*' } unless @config["password"]
+
+        uri = URI.parse(@config["stash_url"])
+        prPath = targetRepoInfo.repoPath + '/pull-requests'
+         
+        req = Net::HTTP::Post.new(uri.query.nil? ? "#{prPath}" : "#{prPath}?#{uri.query}", {'Content-Type' => 'application/json', 'Accept' => 'application/json'})
+        req.basic_auth username, password
+        req.body = resource.to_json
+        http = Net::HTTP.new(uri.host, uri.port, proxy_addr, proxy_port)
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE if @config["ssl_no_verify"]
+        http.use_ssl = uri.scheme.eql?("https")
+
+        response = http.start {|conn| conn.request(req) }
+
+        if not response.is_a? Net::HTTPCreated
+          responseBody = JSON.parse(response.body)
+          if responseBody['errors']
+            responseBody['errors'].collect { |error|
+              puts error['message']
+              if error['reviewerErrors']
+                error['reviewerErrors'].collect { |revError|
+                  puts revError['message']
+                }
+              end
+            }
+          elsif responseBody['message']
+            puts responseBody['message']
+          else
+            puts 'An unknown error occurred.'
+            puts response.code
+            puts response.body
+          end
+        else
+          responseBody = JSON.parse(response.body)
+          prUri = uri.clone
+          prUri.path = prPath + '/' + responseBody['id'].to_s
+          prUri.query = uri.query
+          puts prUri.to_s
+
+          if @config["open"] || options.open
+            Launchy.open prUri.to_s
+          end
+        end
+      end
+
+      private
+
+      def title_from_branch
+        convert_branch_name_to_sentence(@source) || "Merge '#{@source}' into '#{@target}'"
+      end
+
+      def git_commit_messages
+        @commit_messages ||= `git log --reverse --format=%s #{@target}..#{@source}`
+      end
+
+      def parse_proxy(conf)
+        return nil, nil unless conf
+
+        addr, port = conf.split(":")
+        if port =~ /\d+/
+          port = port.to_i
+        else
+          port = nil
+        end
+        [addr, port]
+      end
+
+      def title_and_description(options)
+        descr = (options.description or git_commit_messages)
+        title = (options.title or title_from_branch)
+
+        [title, descr]
+      end
+    end
+  end
+end
+

data/lib/atlassian/stash/repo_info.rb

@@ -0,0 +1,62 @@
+require 'git'
+
+module Atlassian
+  module Stash
+    class RepoInfo
+      def initialize(config, projectKey, slug)
+        @config = config
+        @projectKey = projectKey
+        @slug = slug
+      end
+
+      def projectKey
+        @projectKey
+      end
+
+      def slug
+        @slug
+      end
+
+      def repoPath
+        uri = URI.parse(@config["stash_url"])
+        repoPath = uri.path + '/projects/' + @projectKey + '/repos/' + @slug
+        repoPath
+      end
+
+      def repoUrl(suffix, branch)
+        uri = URI.parse(@config["stash_url"])
+        path = repoPath + (suffix.nil? ? '' : '/' + suffix)
+        uri.path = path
+        
+        if (!branch.nil? and !branch.empty?)
+            q = uri.query || ''
+            q = q + (q.empty? ? '' : '&') + 'at=' + branch unless branch.nil?
+            uri.query = q
+        end
+
+        uri.to_s
+      end
+
+      def self.create (config, remote=nil)
+        config = Hash.new if config.nil?
+        remote = config["remote"] if (remote.nil? || remote.empty?)
+        remoteUrl = Atlassian::Stash::Git.get_remote_url(remote)
+
+        if remoteUrl.nil?
+          remotes = Atlassian::Stash::Git.get_remotes
+          if remotes.empty?
+            raise "No git remotes found, could not determine Stash project URL"
+          else
+            remote = Atlassian::Stash::Git::DEFAULT_REMOTE if (remote.nil? || remote.empty?)
+            raise "Could not find requested git remote '#{remote}'. Remotes found: \r\n" + remotes
+          end
+        end
+
+        if m = remoteUrl.match(/\/([a-zA-Z~][a-zA-Z0-9_\-]*)\/([[:alnum:]][\w\-\.]*).git$/)
+          return RepoInfo.new(config, m[1], m[2])
+        end
+        raise "Repository does not seem to be hosted in Stash; Remote url: " + remoteUrl
+      end
+    end
+  end
+end

data/lib/atlassian/stash/version.rb

@@ -0,0 +1,9 @@
+
+module Atlassian
+  module Stash
+    module Version
+      STRING = IO.readlines(File.dirname(__FILE__) + "/../../../VERSION").first.strip
+    end
+  end
+end
+

data/lib/atlassian/util/text_util.rb

@@ -0,0 +1,31 @@
+
+module Atlassian
+  module Util
+    module TextUtil
+      def convert_branch_name_to_sentence(branch_name)
+        return '' if branch_name.nil?
+
+        branch_name = branch_name.to_s
+        return '' if branch_name.empty?
+
+        issue_key_regex = /([A-Z]{1,10}-\d+)/
+        branch_components = branch_name.split(issue_key_regex);
+
+        parts = branch_components.each_with_index.map { |value, index|
+          (index % 2 === 0) ? value.gsub(/[\-_]/, ' ') : value
+        }
+
+        to_sentence_case(parts.join(''))
+      end
+
+      def to_sentence_case(str)
+        return '' if str.nil?
+
+        str = str.to_s
+        return '' if str.empty?
+
+        str.slice(0, 1).upcase + str.slice(1, str.length)
+      end
+    end
+  end
+end

data/lib/stash_cli.rb

@@ -0,0 +1,3 @@
+
+Dir[File.join(File.dirname(__FILE__), "atlassian", "util", "*.rb")].sort.each {|f| require f}
+Dir[File.join(File.dirname(__FILE__), "atlassian", "stash", "*.rb")].sort.each {|f| require f}

data/test/helper.rb

@@ -0,0 +1,29 @@
+require 'rubygems'
+require 'bundler'
+
+if ENV["COVERAGE"]
+  require "simplecov"
+  SimpleCov.start :test_frameworks do
+    add_filter "/vendor/"
+  end
+end
+
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'minitest/autorun'
+require 'shoulda'
+require "mocha/mini_test"
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+# require 'atlassian-stash'
+require File.dirname(__FILE__) + "/../lib/stash_cli"
+
+
+class Minitest::Test
+end

data/test/test_stash-create-pull-request.rb

@@ -0,0 +1,84 @@
+require 'helper'
+
+include Atlassian::Stash
+include Atlassian::Stash::Git
+
+class TestStashCreatePullRequest < Minitest::Test
+
+  context "#parse_proxy" do
+    setup do
+      @cpr = CreatePullRequest.new nil
+    end
+
+    context 'when proxy_conf is nil' do
+      should 'returns [nil, nil]' do
+        assert_equal [nil, nil], @cpr.send(:parse_proxy, nil)
+      end
+    end
+
+    context 'when proxy_conf is blank' do
+      should 'returns [nil, nil]' do
+        assert_equal [nil, nil], @cpr.send(:parse_proxy, "")
+      end
+    end
+
+    context 'when proxy_conf is "proxy.example.com"' do
+      should 'returns ["proxy.example.com", nil]' do
+        assert_equal ["proxy.example.com", nil], @cpr.send(:parse_proxy, "proxy.example.com")
+      end
+    end
+
+    context 'when proxy_conf is "proxy.example.com:8080"' do
+      should 'returns ["proxy.example.com", 8080]' do
+        assert_equal ["proxy.example.com", 8080], @cpr.send(:parse_proxy, "proxy.example.com:8080")
+      end
+    end
+
+    context 'when proxy_conf is "proxy.example.com:foo"' do
+      should 'returns ["proxy.example.com", nil]' do
+        assert_equal ["proxy.example.com", nil], @cpr.send(:parse_proxy, "proxy.example.com:foo")
+      end
+    end
+  end
+
+  context '#title_and_description' do
+    setup do
+      @cpr = CreatePullRequest.new nil
+      def @cpr.title_from_branch; 'title_from_branch'; end
+      def @cpr.git_commit_messages; 'git_commit_messages'; end
+      @options = Struct.new(:title, :description)
+    end
+
+    context 'with no options' do
+      should 'sets default title and description' do
+        title, descr = @cpr.send(:title_and_description, @options.new(nil, nil))
+        assert_equal title, 'title_from_branch'
+        assert_equal descr, 'git_commit_messages'
+      end
+    end
+
+    context 'with title option' do
+      should 'sets custom title and default description' do
+        title, descr = @cpr.send(:title_and_description, @options.new('custom title', nil))
+        assert_equal title, 'custom title'
+        assert_equal descr, 'git_commit_messages'
+      end
+    end
+
+    context 'with description option' do
+      should 'sets default title and custom description' do
+        title, descr = @cpr.send(:title_and_description, @options.new(nil, 'custom description'))
+        assert_equal title, 'title_from_branch'
+        assert_equal descr, 'custom description'
+      end
+    end
+
+    context 'with both title and description options' do
+      should 'sets custom title and description' do
+        title, descr = @cpr.send(:title_and_description, @options.new('custom title', 'custom description'))
+        assert_equal title, 'custom title'
+        assert_equal descr, 'custom description'
+      end
+    end
+  end
+end

data/test/test_stash-git.rb

@@ -0,0 +1,96 @@
+require 'helper'
+
+include Atlassian::Stash
+include Atlassian::Stash::Git
+
+class TestGit < Minitest::Test
+
+	should "extract remote with ssh remote" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"origin	ssh://git@stash.atlassian.com:7999/STASH/stash.git (fetch)
+				origin	ssh://git@stash.atlassian.com:7999/STASH/stash.git (push)"
+				)
+		assert_equal 'ssh://git@stash.atlassian.com:7999/STASH/stash.git', Atlassian::Stash::Git.get_remote_url
+	end
+
+	should "extract push remote with different fetch and push urls" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"origin	ssh://git@github.com/~sebr/stash.git (fetch)
+				origin	ssh://git@stash.atlassian.com:7999/STASH/stash.git (push)"
+				)
+		assert_equal 'ssh://git@stash.atlassian.com:7999/STASH/stash.git', Atlassian::Stash::Git.get_remote_url
+	end
+
+	should "extract remote with http remote" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"origin     http://adam@sonoma:7990/stash/scm/QA/stash.git (fetch)
+				origin     http://adam@sonoma:7990/stash/scm/QA/stash.git (push)"
+				)
+		assert_equal 'http://adam@sonoma:7990/stash/scm/QA/stash.git', Atlassian::Stash::Git.get_remote_url
+	end
+
+	should "extract remote with multiple remote urls" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (fetch)
+				bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (push)
+				kostya  http://admin@kostya:7990/scm/CA/cylon.git (fetch)
+				kostya  http://admin@kostya:7990/scm/CA/cylon.git (push)
+				local   http://delirium:7990/git/STASH/stash.git (fetch)
+				local   http://delirium:7990/git/STASH/stash.git (push)
+				origin  ssh://git@stash.atlassian.com:7999/STASH/stash.git (fetch)
+				origin  ssh://git@stash.atlassian.com:7999/STASH/stash.git (push)
+				seb     http://adam@sonoma:7990/stash/scm/QA/stash.git (fetch)
+				seb     http://adam@sonoma:7990/stash/scm/QA/stash.git (push)
+				upstream        http://github-enterprise-11-10/stash/stash.git (fetch)
+				upstream        http://github-enterprise-11-10/stash/stash.git (push)")
+		assert_equal 'ssh://git@stash.atlassian.com:7999/STASH/stash.git', Atlassian::Stash::Git.get_remote_url
+	end
+
+	should "extract custom remote with multiple remote urls" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (fetch)
+				bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (push)
+				kostya  http://admin@kostya:7990/scm/CA/cylon.git (fetch)
+				kostya  http://admin@kostya:7990/scm/CA/cylon.git (push)
+				local   http://delirium:7990/git/STASH/stash.git (fetch)
+				local   http://delirium:7990/git/STASH/stash.git (push)
+				origin  ssh://git@stash.atlassian.com:7999/STASH/stash.git (fetch)
+				origin  ssh://git@stash.atlassian.com:7999/STASH/stash.git (push)
+				upstream  ssh://git@stash.atlassian.com:7999/ATLASSIAN/stash.git (fetch)
+				upstream  ssh://git@stash.atlassian.com:7999/ATLASSIAN/stash.git (push)
+				seb     http://adam@sonoma:7990/stash/scm/QA/stash.git (fetch)
+				seb     http://adam@sonoma:7990/stash/scm/QA/stash.git (push)
+				upstream        http://github-enterprise-11-10/stash/stash.git (fetch)
+				upstream        http://github-enterprise-11-10/stash/stash.git (push)")
+		assert_equal 'ssh://git@stash.atlassian.com:7999/ATLASSIAN/stash.git', Atlassian::Stash::Git.get_remote_url('upstream')
+	end
+
+	should "repo with no remotes returns nil" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns("")
+		assert_equal nil, Atlassian::Stash::Git.get_remote_url
+	end
+
+	should "repo with unfound remote returns nil" do
+		Atlassian::Stash::Git.stubs(:get_remotes).returns(
+				"bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (fetch)
+				bitbucket	git@bitbucket.org:atlassian/stash-command-line-tools.git (push)"
+		)
+		assert_equal nil, Atlassian::Stash::Git.get_remote_url('origin')
+	end
+
+        should "branches are matched" do
+                Atlassian::Stash::Git.stubs(:get_branches).returns(
+                                "  feature
+                                  temp
+                                * master
+                                  remotes/origin/master
+                                  remotes/origin/release/v1.0
+                                  remotes/origin/feature/awesome
+                                  remotes/upstream/master
+                                  remotes/upstream/release/v1.0
+                                  remotes/upstream/feature/Issue7")
+                assert_equal true, Atlassian::Stash::Git.is_branch?('master')
+                assert_equal true, Atlassian::Stash::Git.is_branch?('remotes/upstream/master')
+        end
+
+end