# encoding: utf-8

require 'thread'
require 'oddb2xml/builder'
require 'oddb2xml/downloader'
require 'oddb2xml/extractor'
require 'oddb2xml/compressor'

module Oddb2xml
  class Cli
    SUBJECTS  = %w[product article]
    LANGUAGES = %w[DE FR] # EN does not exist
    def initialize(args)
      @options = args
      #@mutex = Mutex.new
      @items = {} # Items from Preparations.xml in BAG
      @index = {} # Base index from swissINDEX
      LANGUAGES.each do |lang|
        @index[lang] = {}
      end
    end
    def run
      # Sometimes nokogiri crashes with ruby in Threads.
      #threads = []
      # bag_xml
      #threads << Thread.new do
        downloader = BagXmlDownloader.new
        xml = downloader.download
        @items = BagXmlExtractor.new(xml).to_hash
      #end
      LANGUAGES.each do |lang|
        # swissindex
        types.each do |type|
          #threads << Thread.new do
            downloader = SwissIndexDownloader.new(type)
            xml = downloader.download_by(lang)
            hsh = SwissIndexExtractor.new(xml, type).to_hash
            #@mutex.synchronize do
              @index[lang][type] = hsh
            #end
          #end
        end
      end
      #threads.map(&:join)
      build
      report
    end
    private
    def build
      files = {}
      SUBJECTS.each{ |sbj| files[sbj] = "oddb_#{sbj}.xml" }
      begin
        files.each_pair do |sbj, file|
          builder = Builder.new do |builder|
            index = {}
            LANGUAGES.each do |lang|
              index[lang] = {} unless index[lang]
              types.each do |type|
                index[lang].merge!(@index[lang][type])
              end
            end
            builder.subject    = sbj
            builder.index      = index
            builder.items      = @items
            builder.tag_suffix = @options[:tag_suffix]
          end
          xml = builder.to_xml
          File.open(file, 'w:utf-8'){ |fh| fh << xml }
        end
        if @options[:compress]
          compressor = Compressor.new(@options[:compress])
          files.values.each do |file|
            compressor.contents << file
          end
          compressor.finalize!
        end
      rescue Interrupt
        files.values.each do |file|
          if File.exist? file
            File.unlink file
          end
        end
        raise Interrupt
      end
    end
    def report
      lines = []
      LANGUAGES.each do |lang|
        lines << lang
        types.each do |type|
          key = (type == :nonpharma ? 'NonPharma' : 'Pharma')
          lines << sprintf("\t#{key} products: %i", @index[lang][type].values.length)
        end
      end
      puts lines.join("\n")
    end
    def types # swissindex
      if @options[:nonpharma]
        [:pharma, :nonpharma]
      else
        [:pharma]
      end
    end
  end
end