setup_keys.rb

require 'watir'
require 'nokogiri'
require 'mechanize'
require 'pry'
require 'pry-byebug'
require 'open-uri'
require 'selenium-webdriver'
require '../global_utilities/global_utilities'


class SetupKeys
  attr_reader :agent 
  def initialize
    @initial_wait  = true
    @work          = parse_args
    @agent         = Watir::Browser.new :chrome
    @sf_domain     = CredService.creds.salesforce.production.instance_url
    @sf_username   = CredService.creds.user.salesforce.production.username
    @sf_password   = CredService.creds.user.salesforce.production.password
    @ngrok_page    = 'http://localhost:4040/status'
    @download_page = @sf_domain + '/02u?retURL=%2Fui%2Fsetup%2FSetup%3Fsetupid%3DDevTools&setupid=TabSet'
    @box_page      = 'https://reedhein.app.box.com/developers/services'
    @ngrok_tunnel  = find_tunnel
    begin
      do_work
    rescue Net::ReadTimeout
      retry
    rescue => e
      puts e
      binding.pry
    ensure
      @agent.close
    end
  end

  def find_tunnel
    agent.goto(@ngrok_page)
    document = agent.html
    doc = Nokogiri::HTML(document)
    doc.at('table:first tr:first td:last').text.strip
  end

  def salesforce
    @agent.goto(@download_page)
    @agent.text_field(id: 'username').when_present.set @sf_username
    @agent.text_field(id: 'password').set @sf_password
    @agent.button(name: 'Login').click
    Watir::Wait.until { @agent.div(class: 'content').wait_until_present }
    @agent.links(text: 'UtilityApp').first.click
    a = Nokogiri::HTML(@agent.html)
    current_value = URI.parse(a.search('td.dataCol.last:last').text.squish)
    new_value     = URI.parse(@ngrok_tunnel)
    unless new_value.host == current_value.host
      @agent.button(text: 'Edit').when_present.click
      Watir::Wait.until { @agent.textarea(id: /callback/).wait_until_present }
      callback_text_field = @agent.textarea(id: /callback/)
      callback_text_field.value = @ngrok_tunnel.to_s + '/auth/salesforce/callback'
      sleep 2
      agent.button(text: "Save").click
      sleep 2
      begin
        agent.button(text: "Continue").when_present.click
      rescue
      end
    end
  end

  def box
    @agent.goto(@box_page)
    @agent.text_field(name: 'login').when_present.set  CredService.creds.user.box.username
    @agent.text_field(name: 'password').set CredService.creds.user.box.password
    @agent.button(type: 'submit').click
    @agent.button(id: /button_edit_application_/).when_present.click
    Watir::Wait.until { @agent.text_field(id: 'field_oauth2_redirect_uri').wait_until_present }
    box_field = @agent.text_field(id: 'field_oauth2_redirect_uri')
    current_value = box_field.value
    unless URI.parse(@ngrok_tunnel).host == URI.parse(current_value).host
      sleep 1
      box_field.set(@ngrok_tunnel.to_s + '/auth/box/callback')
      @agent.button(id: 'save_service_button').when_present.click
    end
    authenticate_box
  end

  def do_work
    case @work
    when 'all'
      salesforce
      authenticate_salesforce
      box
      authenticate_box
    when 'sf'
      salesforce
      authenticate_salesforce
    when 'sandbox'
      @sf_domain     = CredService.creds.salesforce.sandbox.instance_url
      @download_page = @sf_domain + '/02u?retURL=%2Fui%2Fsetup%2FSetup%3Fsetupid%3DDevTools&setupid=TabSet'
      @sf_username   = CredService.creds.user.salesforce.sandbox.username
      @sf_password   = CredService.creds.user.salesforce.sandbox.password
      salesforce
      authenticate_salesforce
    when 'box'
      box
      authenticate_box
    else
      salesforce
      authenticate_salesforce
      box
      authenticate_box
    end
  end

  private

  def authenticate_salesforce
    if @initial_wait == true
      puts 'sleeping for 10 minutes'
      sleep 600
    end
    @initial_wait = false
    @agent.goto(@ngrok_tunnel)
    Watir::Wait.until { @agent.h1(text: 'Reed Hein Oauth service').wait_until_present }
    while (URI.parse(@agent.url).host == URI.parse(@ngrok_tunnel).host) && agent.text.include?('You are NOT authenticated in salesforce')  do
      @agent.button(id: 'salesforce_auth').when_present.click
      puts 'waiting for salesforce to update'
      sleep 55
      @agent.goto(@ngrok_tunnel) if URI.parse(@agent.url).host != URI.parse(@ngrok_tunnel).host
      sleep 5
    end
    `say sales force authorization token updated` if RbConfig::CONFIG['host_os'] =~ /darwin/
  end

  def authenticate_box
    @agent.goto(@ngrok_tunnel)
    sleep 2
    while (URI.parse(@agent.url).host == URI.parse(@ngrok_tunnel).host) && agent.text.include?('You are NOT authenticated in box')  do
      @agent.button(id: 'box_auth').when_present.click
      sleep 2
      @agent.button(id: 'consent_accept_button').when_present.click
      @agent.goto(@ngrok_tunnel) if URI.parse(@agent.url).host != URI.parse(@ngrok_tunnel).host
      sleep 5
    end

    `say box authorization token updated` if RbConfig::CONFIG['host_os'] =~ /darwin/
  end

  def parse_args
    puts ARGV[0]
    case ARGV[0]
    when "all"
      'all'
    when "box"
      'box'
    when /salesforce|sf/
      'sf'
    when 'sandbox'
      'sandbox'
    else
      # 'all'
      'sandbox'
    end
  end
end

a = SetupKeys.new
puts 'finished'