#!/usr/bin/env ruby
require 'open3'
require 'io/console'

unless `which heroku` =~ /heroku/
  puts "It looks like the heroku command line tool hasn't been installed yet.  Please install"
  puts "the Heroku Toolbelt from https://toolbelt.heroku.com, run 'heroku auth:login', and then"
  puts "run this script again."
  exit 1
end

def capture(cmd, opts = {})
  if opts.delete(:no_stderr)
    o, s = Open3.capture2(cmd, opts)
  else
    o, s = Open3.capture2e(cmd, opts)
  end
  o.strip
end

def ask(question, opts = {})
  print question + " "
  STDOUT.flush
  (opts[:noecho] ? STDIN.noecho(&:gets) : gets).strip
end

def nag(question, opts = {})
  answer = ''
  while answer.length == 0
    answer = ask(question, opts)
  end
  answer
end

def yes?(question)
  ask(question + " (y/n)") =~ /^y/i
end

def grab_heroku_config!
  config_data = capture("heroku config -s", no_stderr: true)
  $config = {}
  if config_data !~ /has no config vars/
    config_data.split("\n").map do |line|
      next if line =~ /^\s*(#|$)/ # skip comments and empty lines
      first_equal_sign = line.index('=')
      $config[line.slice(0, first_equal_sign)] = line.slice(first_equal_sign + 1, line.length)
    end
  end
end

def set_value(key, value, options = {})
  if $config[key].nil? || $config[key] == '' || ($config[key] != value && options[:force] != false)
    puts "Setting #{key} to #{value}" unless options[:silent]
    puts capture("heroku config:set #{key}=#{value}")
    $config[key] = value
  end
end

unless File.exist?(File.expand_path("~/.netrc")) && File.read(File.expand_path("~/.netrc")) =~ /heroku/
  puts "It looks like you need to log in to Heroku.  Please run 'heroku auth:login' before continuing."
  exit 1
end

puts "Welcome #{`heroku auth:whoami`.strip}!  It looks like you're logged into Heroku."
puts

info = capture("heroku info")
if info =~ /No app specified/i
  puts "It looks like you don't have a Heroku app set up yet for this repo."
  puts "You can either exit now and run 'heroku create', or I can do it for you."
  if yes?("Would you like me to create a Heroku app for you now in this repo?")
    puts `heroku create`
    info = capture("heroku info")
  else
    puts "Okay, exiting so you can do it."
    exit 0
  end
end

app_name = info.scan(/http:\/\/([\w\d-]+)\.herokuapp\.com/).flatten.first

unless yes?("Your Heroku app name is #{app_name}.  Is this correct?")
  puts "Well, then I'm not sure what to do here, sorry."
  exit 1
end

if (root_id = `git rev-list --max-parents=0 HEAD`.chomp) != '620acffa5a302c6a27165d3214cf3da6be6c1d0d'
  if (`git remote`.split - %w[heroku]).empty?
    puts "You don't seem to have cantino/huginn set up as upstream repository."
    if yes?("Would you like me to set this work tree up for you?")
      if system('git remote add origin https://github.com/cantino/huginn.git') &&
         system('git remote update origin')
        rebase_command = "git rebase #{root_id} --onto origin/master"
        if system(rebase_command)
          puts "Done!"
        else
          system('git rebase --abort')
          puts "Rebasing your work tree onto the upstream master failed."
          puts "Please run the following command and merge your local changes by yourself."
          puts "\t#{rebase_command}"
          exit 1
        end
      else
        exit 1
      end
    end
  end
end

grab_heroku_config!

if $config.length > 0
  puts
  puts "Your current Heroku config:"
  $config.each do |key, value|
    puts '  ' + key + ' ' * (25 - [key.length, 25].min) + '= ' + value
  end
end

unless $config['APP_SECRET_TOKEN']
  puts "Setting up APP_SECRET_TOKEN..."
  puts capture("heroku config:set APP_SECRET_TOKEN=`rake secret`")
end

unless $config['DOMAIN']
  set_value 'DOMAIN', "#{app_name}.herokuapp.com", force: false
  first_time = true
end

set_value 'BUILDPACK_URL', "https://github.com/ddollar/heroku-buildpack-multi.git"
set_value 'PROCFILE_PATH', "deployment/heroku/Procfile.heroku", force: false
set_value 'ON_HEROKU', "true"
set_value 'FORCE_SSL', "true"
set_value 'USE_GRAPHVIZ_DOT', 'dot'

unless $config['INVITATION_CODE']
  puts "You need to set an invitation code for your Huginn instance.  If you plan to share this instance, you will"
  puts "tell this code to anyone who you'd like to invite.  If you won't share it, then just set this to something"
  puts "that people will not guess."

  invitation_code = nag("What code would you like to use?")
  set_value 'INVITATION_CODE', invitation_code
end

unless $config['SMTP_DOMAIN'] && $config['SMTP_USER_NAME'] && $config['SMTP_PASSWORD'] && $config['SMTP_SERVER'] && $config['EMAIL_FROM_ADDRESS']
  puts "Okay, let's setup outgoing email settings.  The simplest solution is to use the free sendgrid Heroku addon."
  puts "If you'd like to use your own server, or your Gmail account, please see .env.example and set"
  puts "SMTP_DOMAIN, SMTP_USER_NAME, SMTP_PASSWORD, and SMTP_SERVER with 'heroku config:set'."
  if yes?("Should I enable the free sendgrid addon?")
    puts capture("heroku addons:add sendgrid")

    set_value 'SMTP_SERVER', "smtp.sendgrid.net", silent: true
    set_value 'SMTP_DOMAIN', "heroku.com", silent: true

    grab_heroku_config!
    set_value 'SMTP_USER_NAME', $config['SENDGRID_USERNAME'], silent: true
    set_value 'SMTP_PASSWORD', $config['SENDGRID_PASSWORD'], silent: true
  else
    puts "Okay, you'll need to set SMTP_DOMAIN, SMTP_USER_NAME, SMTP_PASSWORD, and SMTP_SERVER with 'heroku config:set' manually."
  end

  unless $config['EMAIL_FROM_ADDRESS']
    email = nag("What email address would you like email to appear to be sent from?")
    set_value 'EMAIL_FROM_ADDRESS', email
  end
end

branch = capture("git rev-parse --abbrev-ref HEAD")
if yes?("Should I push your current branch (#{branch}) to heroku?")
  puts "This may take a moment..."
  puts capture("git push heroku #{branch}:master -f")

  puts "Running database migrations..."
  puts capture("heroku run rake db:migrate")
end

if first_time
  puts "Restarting..."
  puts capture("heroku restart")
  puts "Done!"
  puts
  puts
  puts "I can make an admin user on your new Huginn instance and setup some example Agents."
  if yes?("Should I create a new admin user and some example Agents?")
    done = false
    while !done
      seed_email = nag "Okay, what is your email address?"
      seed_username = nag "And what username would you like to login as?"
      seed_password = nag "Finally, what password would you like to use?", noecho: true
      puts "\nJust a moment..."

      result = capture("heroku run rake db:seed SEED_EMAIL=#{seed_email} SEED_USERNAME=#{seed_username} SEED_PASSWORD=#{seed_password}")
      if result =~ /Validation failed/
        puts "ERROR:"
        puts
        puts result
        puts
      else
        done = true
      end
    end
    puts
    puts
    puts "Okay, you should be all set!  Visit https://#{app_name}.herokuapp.com and login as '#{seed_username}' with your password."
    puts
    puts "If you'd like to make more users, you can visit https://#{app_name}.herokuapp.com/users/sign_up and use the invitation code:"
  else
    puts
    puts "Visit https://#{app_name}.herokuapp.com/users/sign_up and use the invitation code shown below:"
  end
  puts
  puts "\t#{$config['INVITATION_CODE']}"
end

puts
puts "Done!"