require 'open-uri'

module Nzbs
  class Nfo
  
    attr_accessor :srt, :imdb_link
  
    def initialize(agent, url)
      @page = agent.get(url.gsub(/\/$/,''))
      @srt = []
    
      parse_nfo
      @srt.uniq!
    end
  
  private
  
    # searching for subtitles languages info & imdb link
    def parse_nfo
      pre = @page.search('pre.nfo')
      pre.to_s.split(/\n/).each do |line|
        case line
        when /subtitle|sub/i
          @srt << 'fr' if line =~ /fr|fre|french/i
          @srt << 'en' if line =~ /en|eng|english/i
          # TODO add more language parsing here
          @srt << 'none' if line =~ /none/i
        when /imdb\.com\/title\//
          @imdb_link = (matched = line.match(/imdb.com\/title\/(tt[0-9]+)/)) && "http://imdb.com/title/#{matched[1]}"
        end
      end
      @srt << 'unknown' if @srt.empty?
    end
  
  end
end