/bin/stash

#!/usr/bin/env ruby

require 'rubygems'
require File.dirname(__FILE__) + "/../lib/stash_cli"
#require 'stash_cli'
require 'commander/import'
require 'yaml'

program :name, "Atlassian Stash CLI"
program :version, Atlassian::Stash::Version::STRING
program :description, "Provides convenient functions for interacting with Atlassian Stash through the command line"

include Atlassian::Stash
include Atlassian::Stash::Git

$configFile = File.join(Etc.getpwuid.dir, ".stashconfig.yml")

def load_config
  raise "No Stash configuration found; please run configure" unless File.exists? $configFile
  config = YAML.load_file($configFile)
  raise "Stash configuration file is incomplete, please re-run configure" unless config['username'] and config['password'] and config['stash_url']
  config
end

command 'configure' do |c|
  c.syntax = 'configure'
  c.description = 'Setup configuration details to your Stash instance'
  c.example 'stash configure --username sebr --password s3cre7 --stash_url http://stash.mycompany.com', 'Setup Stash CLI with credentials to the Stash server'
  c.option '--username user', String, 'Writes your Stash username to the configuration file'
  c.option '--password password', String, 'Writes your Stash user password to the configuration file. If omitted, password will be prompted to be entered'
  c.option '--stashUrl', String, 'Writes the Stash server url to the configuration file'
  c.action do |args, options|
    username = options.username ? options.username : ask("Stash Username: ")
    password = options.password ? options.password : ask("Stash Password: ") { |q| q.echo = "*" }
    stashUrl = options.stashUrl ? options.stashUrl : ask("Stash URL: ")

    c = {
      'username' => username.to_s,
      'password' => password.to_s,
      'stash_url' => stashUrl.to_s
    }

    File.open($configFile, 'w') do |out|
      YAML.dump(c, out)
    end

    File.chmod 0600, $configFile

    create_git_alias if agree "Create a git alias 'git create-pull-request'? "
  end
end

command 'pull-request' do |c|
  def extract_reviewers(args = [])
    args.collect { |user|
      user[1..-1] if user.start_with?("@")
    }.compact
  end

  c.syntax = 'pull-request [sourceBranch] targetBranch [@reviewer1 @reviewer2]'
  c.description = 'Create a pull request in Stash'
  c.example 'stash pull-request topicBranch master @michael', "Create a pull request from branch 'topicBranch' into 'master' with 'michael' added as a reviewer"
  c.example 'stash pull-request master', "Create a pull request from the current git branch into 'master'"
  c.action do |args, options|
    if args.length == 0
      command(:help).run('pull-request')
      Process.exit
    end

    source = args.shift
    if args.empty? or args.first.start_with?("@")
      target = source
      source = get_current_branch()
      reviewers = extract_reviewers args
    else
      target = args.shift
      reviewers = extract_reviewers args
    end

    ensure_within_git! do
      cpr = CreatePullRequest.new(load_config)
      cpr.create_pull_request source, target, reviewers
    end
  end
end

default_command :help
program :help_formatter, :compact
program :help, 'Authors', 'Seb Ruiz <sruiz@atlassian.com> with others'
program :help, 'Website', 'https://bitbucket.org/atlassian/stash-command-line-tools'


# vim set ft=ruby