class Spaceship::DUClient

def self.hostname
  "https://du-itc.itunes.apple.com"
end

def parse_upload_response(response)
  content = response.body
  if !content['statusCode'].nil? && content['statusCode'] != 200
    error_codes = ""
    error_codes = content['errorCodes'].join(',') unless content['errorCodes'].nil?
    error_message = "[#{error_codes}] #{content['localizedMessage']}"
    raise UnexpectedResponse.new, error_message
  end
  content
end

def picture_type_map
  # rubocop:enable Style/ExtraSpacing
  {
    watch:        "MZPFT.SortedN27ScreenShot",
    ipad:         "MZPFT.SortedTabletScreenShot",
    iphone6:      "MZPFT.SortedN61ScreenShot",
    iphone6Plus:  "MZPFT.SortedN56ScreenShot",
    iphone4:      "MZPFT.SortedN41ScreenShot",
    iphone35:     "MZPFT.SortedScreenShot"
  }
end

def screenshot_picture_type(device)
  device = device.to_sym
  raise "Unknown picture type for device: #{device}" unless picture_type_map.key? device
  picture_type_map[device]
end

def upload_file(app_version, upload_file, path, content_provider_id, sso_token, du_validation_rule_set = nil)
  raise "File #{upload_file.file_path} is empty" if upload_file.file_size == 0
  version = app_version.version
  app_id = app_version.application.apple_id
  app_type = app_version.app_type
  referrer = app_version.application.url
  r = request(:post) do |req|
    req.url "#{self.class.hostname}#{path}"
    req.body = upload_file.bytes
    req.headers['Accept'] = 'application/json, text/plain, */*'
    req.headers['Content-Type'] = upload_file.content_type
    req.headers['X-Apple-Upload-Referrer'] = referrer
    req.headers['Referrer'] = referrer
    req.headers['X-Apple-Upload-AppleId'] = app_id
    req.headers['X-Apple-Jingle-Correlation-Key'] = "#{app_type}:AdamId=#{app_id}:Version=#{version}"
    req.headers['X-Apple-Upload-itctoken'] = sso_token
    req.headers['X-Apple-Upload-ContentProviderId'] = content_provider_id
    req.headers['X-Original-Filename'] = upload_file.file_name
    req.headers['X-Apple-Upload-Validation-RuleSets'] = du_validation_rule_set if du_validation_rule_set
    req.headers['Content-Length'] = "#{upload_file.file_size}"
    req.headers['Connection'] = "keep-alive"
  end
  if r.status == 500 and r.body.include?("Server Error")
    return upload_file(app_version, upload_file, path, content_provider_id, sso_token, du_validation_rule_set)
  end
  parse_upload_response(r)
end

def upload_geojson(app_version, upload_file, content_provider_id, sso_token_for_image)
  upload_file(app_version, upload_file, '/upload/geo-json', content_provider_id, sso_token_for_image)
end

def upload_large_icon(app_version, upload_file, content_provider_id, sso_token_for_image)
  upload_file(app_version, upload_file, '/upload/image', content_provider_id, sso_token_for_image, 'MZPFT.LargeApplicationIcon')
end

def upload_screenshot(app_version, upload_file, content_provider_id, sso_token_for_image, device)
  upload_file(app_version, upload_file, '/upload/image', content_provider_id, sso_token_for_image, screenshot_picture_type(device))
end

def upload_trailer(app_version, upload_file, content_provider_id, sso_token_for_video)
  upload_file(app_version, upload_file, '/upload/purple-video', content_provider_id, sso_token_for_video)
end

def upload_video_preview(app_version, upload_file, content_provider_id, sso_token_for_image)
  upload_file(app_version, upload_file, '/upload/app-screenshot-image', content_provider_id, sso_token_for_image)
end

def upload_watch_icon(app_version, upload_file, content_provider_id, sso_token_for_image)
  upload_file(app_version, upload_file, '/upload/image', content_provider_id, sso_token_for_image, 'MZPFT.GizmoAppIcon')
end