require 'nokogiri'
require 'date'
require 'active_support/core_ext/hash/conversions'

class RIXML

  def initialize(document)
    @attrs = Hash.from_xml(document.to_s)
  end

  def product_id
    @attrs['Research']['Product']['productID']
  end

  def status
    @attrs['Research']['Product']['StatusInfo']['statusType'].try(:downcase).try(:to_sym) || :published
  end

  def publication_date
    DateTime.strptime(@attrs['Research']['Product']['StatusInfo']['statusDateTime'])
  end

  def authors
    org = @attrs['Research']['Product']['Source']['Organization']
    if org.is_a? Array
      org = org.find { |v| v['primaryIndicator'] == 'Yes' } || org.first
    end
    authors = org['PersonGroup']['PersonGroupMember']
    authors = [authors] unless authors.is_a? Array

    authors.map do |author|
      person = author['Person']
      {
        name: person['DisplayName'],
        first_name: person['GivenName'],
        middle_name: person['MiddleName'],
        last_name: person['FamilyName'],
        job_title: person['JobTitle'],
        email: person['ContactInfo'].try(:[], 'Email').try(:downcase)
      }
    end
  end

  def report_info
    content = @attrs['Research']['Product']['Content']
    {
      title: content['Title'],
      abstract: content['Abstract'],
      file_name: content['Resource']['Name'],
      pages: content['Resource']['Length'].to_i
    }
  end

  def context
    context = @attrs['Research']['Product']['Context']
    context_info = {
      companies: RIXML.extract_companies_from_context(context),
      sectors: RIXML.extract_sectors_from_context(context)
    }
  end

  def self.parse data
    RIXML.new(Nokogiri::XML(data).root)
  end

  def self.parse_from_file filename
    self.parse self.read_file(filename)
  end

  private

  def self.read_file filename
    body = ''
    File.open(filename, 'r') do |infile|
      while (line = infile.gets)
        body << line
      end
    end
    body
  end

  def self.extract_companies_from_context context
    companies = []
    list = context['IssuerDetails'].try(:[], 'Issuer')
    return [] if list.nil?
    list = [list] unless list.is_a? Array
    list.select { |c| c['issuerType'] == 'Corporate' }.each do |company|
      securities = company['SecurityDetails']['Security']['SecurityID']
      securities = [securities] unless securities.is_a? Array
      isin = securities.find { |security| security['idType'] == 'ISIN' }
      companies << { isin: isin['idValue'] } unless isin.nil?
    end
    companies
  end

  def self.extract_sectors_from_context context
    sectors = []
    list = context['ProductClassifications'].try(:[], 'SectorIndustry')
    return [] if list.nil?
    list = [list] unless list.is_a? Array
    list.select { |s| s['classificationType'] == 'GICS' }.map do |v|
      { code: v['code'].to_i, focus: v['focusLevel'].try(:downcase) == 'yes' }
    end
  end
end