# 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]
    ADDITIONS = %w[substance limitation interaction code]
    OPTIONALS = %w[fi fi_product]
    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
      @flags = {} # narcotics flag from ywesee
      @infos = {} # [option] FI from SwissmedicInfo
      @actions = [] # [addition] interactions from epha
      @orphans = [] # [addition] Orphaned drugs from Swissmedic xls
      @fridges = [] # [addition] ReFridge drugs from Swissmedic xls
      LANGUAGES.each do |lang|
        @index[lang] = {}
      end
    end
    def run
      threads = []
      # swissmedic-info
      if @options[:fi]
        threads << Thread.new do
          downloader = SwissmedicInfoDownloader.new
          xml = downloader.download
          @mutex.synchronize do
            hsh = SwissmedicInfoExtractor.new(xml).to_hash
            @infos = hsh
          end
        end
      end
      # swissmedic
      [:orphans, :fridges].each do |type|
        threads << Thread.new do
          downloader = SwissmedicDownloader.new(type)
          bin = downloader.download
          self.instance_variable_set("@#{type.to_s}", SwissmedicExtractor.new(bin, type).to_arry)
        end
      end
      # epha
      threads << Thread.new do
        downloader = EphaDownloader.new
        str = downloader.download
        @mutex.synchronize do
          @actions = EphaExtractor.new(str).to_arry
        end
      end
      # bag
      threads << Thread.new do
        downloader = BagXmlDownloader.new
        xml = downloader.download
        @mutex.synchronize do
          hsh = BagXmlExtractor.new(xml).to_hash
          @items = hsh
        end
      end
      # ywesee
      threads << Thread.new do
        downloader = YweseeBMDownloader.new
        str = downloader.download
        @mutex.synchronize do
          @flags = YweseeBMExtractor.new(str).to_hash
        end
      end
      ssl = true
      catch :ssl do
      LANGUAGES.each do |lang|
          # swissindex
          types.each do |type|
            begin
              thread = thread.new do
                downloader = SwissIndexDownloader.new(type, lang)
                xml = downloader.download
                @mutex.synchronize do
                  hsh = SwissIndexExtractor.new(xml, type).to_hash
                  @index[lang][type] = hsh
                end
              end
              thread.abort_on_exception = true
              threads << thread
            rescue Exception
              ssl = false
              throw :ssl
            end
          end
        end
      end
      unless ssl
        puts
        puts "(Aborted)"
        puts "Please install SSLv3 CA certificates on your machine."
        puts "You can check with `ruby -ropenssl -e 'p OpenSSL::X509::DEFAULT_CERT_FILE'`."
        puts "See README."
        exit
      end
      threads.map(&:join)
      build
      compress if @options[:compress_ext]
      report
    end
    private
    def build
      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]) if @index[lang][type]
              end
            end
            builder.subject = sbj
            builder.index   = index
            builder.items   = @items
            builder.flags   = @flags
            # additions
            %w[actions orphans fridges].each do |addition|
              builder.send("#{addition}=".intern, self.instance_variable_get("@#{addition}"))
            end
            # optionals
            builder.infos = @infos
            builder.tag_suffix = @options[:tag_suffix]
          end
          xml = builder.to_xml
          File.open(file, 'w:utf-8'){ |fh| fh << xml }
        end
      rescue Interrupt
        files.values.each do |file|
          if File.exist? file
            File.unlink file
          end
        end
        raise Interrupt
      end
    end
    def compress
      compressor = Compressor.new(prefix, @options[:compress_ext])
      files.values.each do |file|
        if File.exists?(file)
          compressor.contents << file
        end
      end
      compressor.finalize!
    end
    def files
      unless @_files
        @_files = {}
        ##
        # building order
        #   1. addtions
        #   2. subjects
        #   3. optionals
        _files = (ADDITIONS + SUBJECTS)
        _files += OPTIONALS if @options[:fi]
        _files.each do|sbj|
          @_files[sbj] = "#{prefix}_#{sbj.to_s}.xml"
        end
      end
      @_files
    end
    def prefix
      @_prefix ||= (@options[:tag_suffix] || 'oddb').gsub(/^_|_$/, '').downcase
    end
    def report
      lines = []
      LANGUAGES.each do |lang|
        lines << lang
        types.each do |type|
          key = (type == :nonpharma ? 'NonPharma' : 'Pharma')
          if @index[lang][type]
            lines << sprintf(
              "\t#{key} products: %i", @index[lang][type].values.length)
          end
        end
      end
      puts lines.join("\n")
    end
    def types # swissindex
      @_types ||=
        if @options[:nonpharma]
          [:pharma, :nonpharma]
        else
          [:pharma]
        end
    end
  end
end