require 'slim'
require 'forwardable'

# Top level module / namespace.
module Neurohmmer
  # A class that holds methods related to Output
  class Output
    class <<self
      extend Forwardable
      def_delegators Neurohmmer, :conf

      def to_html(hmm_results)
        @html_results = format_seqs_for_html(hmm_results)
        template_path = File.expand_path(File.join(__FILE__, '../../../',
                                                   'template/contents.slim'))
        contents_temp = File.read(template_path)
        html_content = Slim::Template.new { contents_temp }.render(self)
        File.open(conf[:html_output], 'w') { |f| f.puts html_content }
      end

      def format_seqs_for_html(hmm_results)
        results = {}
        hmm_results.each do |query, hits|
          next if hits.length == 0
          results[query] = []
          hits.each do |hit|
            html_seq = format_html_seq(hit[:seq], hit[:flatseq])
            results[query] << { id: hit[:id], seq: html_seq }
          end
        end
        results
      end

      private

      def format_html_seq(seq, flatseq)
        seq.gsub!("\n", '')
        flatseq.each do |hsp|
          seq.gsub!(/#{hsp.gsub('-', '')}/i, '<span class=hsp>\0</span>')
        end
        seq.gsub(/KR|KK|RR/i, '<span class=clv>\0</span>')
          .gsub(/(K|R)<span class=hsp>(K|R)/i, '<span class=clv>\1</span>' \
                '<span class=clv_i>\2</span><span class=hsp>')
          .gsub('<span class=clv>R</span><span class=clv_i>K</span><span' \
                ' class=hsp>', 'R<span class=hsp>K')
          .gsub(/G<span class=clv>/, '<span class=gly>G</span><span class=clv>')
      end
    end
  end
end