genome/dgi-db

class DruggableGeneCategoryPresenter < Struct.new(:search_results, :source_db_names, :view_context)
  include Genome::Extensions

  def display_genes
    @display_genes ||= search_results.map do |result|
      sources = result.gene_claims
                  .map { |claim| claim.source }
                  .select { |source| source_db_names.include?(source.source_db_name)  }
                  .uniq
                  .sort_by { |s| CategoryResultSortOrder.sort_value(s.source_db_name) }
      DisplayGene.new(result.long_name, sources, result.name, view_context)
    end.sort_by { |display_gene| CategoryResultSortOrder.sort_value(display_gene.sources.first.source_db_name) }
  end

  private
  class DisplayGene < Struct.new(:gene_name, :sources, :short_name, :view_context)
    def source_links
      sources.map { |s| TrustLevelPresenter.source_link_green(view_context, s) }
        .join(' ')
        .html_safe
    end
  end
end