# -*- coding: utf-8 -*-
module Linkedin
  class Profile

    USER_AGENTS = ['Windows IE 6', 'Windows IE 7', 'Windows Mozilla', 'Mac Safari', 'Mac FireFox', 'Mac Mozilla', 'Linux Mozilla', 'Linux Firefox', 'Linux Konqueror']

    ATTRIBUTES = %w(name first_name last_name title location country industry summary picture linkedin_url education groups websites languages skills certifications organizations past_companies current_companies recommended_visitors)

    attr_reader :page, :linkedin_url

    def self.get_profile(url)
      begin
        Linkedin::Profile.new(url)
      rescue => e
        puts e
      end
    end

    def initialize(url)
      @linkedin_url = url
      @page         = http_client.get(url)
    end

    def name
      "#{first_name} #{last_name}"
    end

    def first_name
      @first_name ||= (@page.at('.given-name').text.strip if @page.at('.given-name'))
    end

    def last_name
      @last_name ||= (@page.at('.family-name').text.strip if @page.at('.family-name'))
    end

    def title
      @title ||= (@page.at('.headline-title').text.gsub(/\s+/, ' ').strip if @page.at('.headline-title'))
    end

    def location
      @location ||= (@page.at('.locality').text.split(',').first.strip if @page.at('.locality'))
    end

    def country
      @country ||= (@page.at('.locality').text.split(',').last.strip if @page.at('.locality'))
    end

    def industry
      @industry ||= (@page.at('.industry').text.gsub(/\s+/, ' ').strip if @page.at('.industry'))
    end

    def summary
      @summary ||= (@page.at('.description.summary').text.gsub(/\s+/, ' ').strip if @page.at('.description.summary'))
    end

    def picture
      @picture ||= (@page.at('#profile-picture/img.photo').attributes['src'].value.strip if @page.at('#profile-picture/img.photo'))
    end

    def skills
      @skills ||= (@page.search('.competency.show-bean').map{|skill| skill.text.strip if skill.text} rescue nil)
    end

    def past_companies
      @past_companies ||= get_companies('past')
    end

    def current_companies
      @current_companies ||= get_companies('current')
    end

    def education
      @education ||= @page.search('.position.education.vevent.vcard').map do |item|
        name   = item.at('h3').text.gsub(/\s+|\n/, ' ').strip      if item.at('h3')
        desc   = item.at('h4').text.gsub(/\s+|\n/, ' ').strip      if item.at('h4')
        period = item.at('.period').text.gsub(/\s+|\n/, ' ').strip if item.at('.period')

        {:name => name, :description => desc, :period => period}
      end
    end

    def websites
      @websites ||=  @page.search('.website').flat_map do |site|
        url = "http://www.linkedin.com#{site.at('a')['href']}"
        CGI.parse(URI.parse(url).query)['url']
      end

    end

    def groups
      @groups ||= @page.search('.group-data').map do |item|
        name = item.text.gsub(/\s+|\n/, ' ').strip
        link = "http://www.linkedin.com#{item.at('a')['href']}"
        {:name => name, :link => link}
      end
    end

    def organizations
      @organizations ||= @page.search('ul.organizations/li.organization').map do |item|
        name       = item.search('h3').text.gsub(/\s+|\n/, ' ').strip rescue nil
        start_date, end_date = item.search('ul.specifics li').text.gsub(/\s+|\n/, ' ').strip.split(' to ')
        start_date = Date.parse(start_date) rescue nil
        end_date   = Date.parse(end_date)   rescue nil
        {:name => name, :start_date => start_date, :end_date => end_date}
      end
    end

    def languages
      @languages ||= @page.search('ul.languages/li.language').map do |item|
        language    = item.at('h3').text rescue nil
        proficiency = item.at('span.proficiency').text.gsub(/\s+|\n/, ' ').strip rescue nil
        {:language=> language, :proficiency => proficiency }
      end
    end

    def certifications
        @certifications ||= @page.search('ul.certifications/li.certification').map do |item|
            name       = item.at('h3').text.gsub(/\s+|\n/, ' ').strip                         rescue nil
            authority  = item.at('.specifics/.org').text.gsub(/\s+|\n/, ' ').strip            rescue nil
            license    = item.at('.specifics/.licence-number').text.gsub(/\s+|\n/, ' ').strip rescue nil
            start_date = item.at('.specifics/.dtstart').text.gsub(/\s+|\n/, ' ').strip        rescue nil

            {:name => name, :authority => authority, :license => license, :start_date => start_date}
          end

    end


    def recommended_visitors
      @recommended_visitors ||= @page.search('.browsemap/.content/ul/li').map do |visitor|
        v = {}
        v[:link]    = visitor.at('a')['href']
        v[:name]    = visitor.at('strong/a').text
        v[:title]   = visitor.at('.headline').text.gsub('...',' ').split(' at ').first
        v[:company] = visitor.at('.headline').text.gsub('...',' ').split(' at ')[1]
        v
      end
    end

    def to_json
      require 'json'
      ATTRIBUTES.reduce({}){ |hash,attr| hash[attr.to_sym] = self.send(attr.to_sym);hash }.to_json
    end


    private

    def get_companies(type)
      companies = []
      if @page.search(".position.experience.vevent.vcard.summary-#{type}").first
        @page.search(".position.experience.vevent.vcard.summary-#{type}").each do |node|

          company               = {}
          company[:title]       = node.at('h3').text.gsub(/\s+|\n/, ' ').strip if node.at('h3')
          company[:company]     = node.at('h4').text.gsub(/\s+|\n/, ' ').strip if node.at('h4')
          company[:description] = node.at(".description.#{type}-position").text.gsub(/\s+|\n/, ' ').strip if node.at(".description.#{type}-position")

          start_date  = node.at('.dtstart')['title'] rescue nil
          company[:start_date] = parse_date(start_date) rescue nil

          end_date = node.at('.dtend')['title'] rescue nil
          company[:end_date] = parse_date(end_date) rescue nil

          company_link = node.at('h4/strong/a')['href'] if node.at('h4/strong/a')

          result = get_company_details(company_link)
          companies << company.merge!(result)
        end
      end
      companies
    end

    def parse_date(date)
      date = "#{date}-01-01" if date =~ /^(19|20)\d{2}$/
      Date.parse(date)
    end

    def get_company_details(link)
      result = {:linkedin_company_url => "http://www.linkedin.com#{link}"}
      page = http_client.get(result[:linkedin_company_url])

      result[:url] = page.at('.basic-info-about/ul/li/p/a').text if page.at('.basic-info-about/ul/li/p/a')
      node_2 = page.at('.basic-info-about/ul')
      if node_2
        node_2.search('p').zip(node_2.search('h4')).each do |value,title|
          result[title.text.gsub(' ','_').downcase.to_sym] = value.text.strip
        end
      end
      result[:address] = page.at('.vcard.hq').at('.adr').text.gsub("\n",' ').strip if page.at('.vcard.hq')
      result
    end

    def http_client
      Mechanize.new do |agent|
        agent.user_agent_alias = USER_AGENTS.sample
        agent.max_history = 0
      end
    end

  end
end