require 'json'
require 'openssl'
require 'faraday'
require 'faraday_middleware'

module Shenzhen::Plugins
  module DeployGate
    class Client
      HOSTNAME = 'deploygate.com'

      def initialize(api_token, user_name)
        @api_token, @user_name = api_token, user_name
        @connection = Faraday.new(:url => "http://#{HOSTNAME}", :request => { :timeout => 120 }) do |builder|
          builder.request :multipart
          builder.request :json
          builder.response :json, :content_type => /\bjson$/
          builder.use FaradayMiddleware::FollowRedirects
          builder.adapter :net_http
        end
      end

      def upload_build(ipa, options)
        options.update({
          :token => @api_token,
          :file => Faraday::UploadIO.new(ipa, 'application/octet-stream'),
          :message => options[:message] || ''
        })

        @connection.post("/api/users/#{@user_name}/apps", options).on_complete do |env|
          yield env[:status], env[:body] if block_given?
        end

      rescue Faraday::Error::TimeoutError
        say_error "Timed out while uploading build. Check https://deploygate.com/ to see if the upload was completed." and abort
      end
    end
  end
end

command :'distribute:deploygate' do |c|
  c.syntax = "ipa distribute:deploygate [options]"
  c.summary = "Distribute an .ipa file over deploygate"
  c.description = ""
  c.option '-f', '--file FILE', ".ipa file for the build"
  c.option '-a', '--api_token TOKEN', "API Token. Available at https://deploygate.com/settings"
  c.option '-u', '--user_name USER_NAME', "User Name. Available at https://deploygate.com/settings"
  c.option '-m', '--message MESSAGE', "Release message for the build"

  c.action do |args, options|
    determine_file! unless @file = options.file
    say_error "Missing or unspecified .ipa file" and abort unless @file and File.exist?(@file)

    determine_deploygate_api_token! unless @api_token = options.api_token || ENV['DEPLOYGATE_API_TOKEN']
    say_error "Missing API Token" and abort unless @api_token

    determine_deploygate_user_name! unless @user_name = options.user_name || ENV['DEPLOYGATE_USER_NAME']
    say_error "Missing User Name" and abort unless @api_token

    @message = options.message

    parameters = {}
    parameters[:file] = @file
    parameters[:message] = @message
    parameters[:replace] = "true" if options.replace

    client = Shenzhen::Plugins::DeployGate::Client.new(@api_token, @user_name)
    response = client.upload_build(@file, parameters)
    case response.status
    when 200...300
      say_ok "Build successfully uploaded to DeployGate"
    else
      say_error "Error uploading to DeployGate: #{response.body}" and abort
    end
  end

  private

  def determine_deploygate_api_token!
    @api_token ||= ask "API Token:"
  end

  def determine_deploygate_user_name!
    @user_name ||= ask "User Name:"
  end
end