ontohub/lib/ontology_search.rb

	ontology_search.rb revision e532e91ba3c45c82dc542f135171ee71610bdc86
require 'json'

#
# Beware! This is not tested well.
#
class OntologySearch

  def initialize
    @limit = 20
  end

  def make_repository_keyword_list_json(repository, prefix)
    JSON.generate(make_repository_keyword_list(repository, prefix))
  end

  def make_global_keyword_list_json(prefix)
    JSON.generate(make_global_keyword_list(prefix))
  end

  def make_repository_keyword_list(repository, prefix)
    text_list = Set.new

    unless repository.ontologies.where("name = :prefix", prefix: prefix, repository_id: repository).empty?
      text_list.add(prefix)
    end

    unless Entity.where("name = :prefix", prefix: prefix).empty?
      #TODO Search only symbols of ontologies of the repository
      #text_list.add(prefix)
    end

    repository.ontologies.select(:name).where("name ILIKE :prefix", prefix: "#{prefix}%").group("name").limit(5).each do |ontology|
      text_list.add(ontology.name)
    end

    ontologies = repository.ontologies
    ontology_ids = Set.new
    ontologies.each do |ontology|
      ontology.entities.select(:text).where("text ILIKE :prefix", prefix: "#{prefix}%").group("text").limit(5).each do |symbol|
        text_list.add(symbol.text)
      end
      ontology_ids.add(ontology.id)
    end

    Logic.where("name ILIKE :prefix", prefix: "#{prefix}%").limit(5).each do |logic|
       logic_ontology_ids = Set.new
       logic.ontologies.each do |ontology|
         logic_ontology_ids.add(ontology.id)
       end
       if (ontology_ids & logic_ontology_ids).size != 0
         text_list.add(logic.name)
       end
    end

    text_list.to_a.sort.map { |x| {text: x} }
  end

  def make_global_keyword_list(prefix)
    text_list = Set.new

    unless Ontology.where("name = :prefix", prefix: prefix).empty?
      text_list.add(prefix)
    end

    unless Entity.where("name = :prefix", prefix: prefix).empty?
      text_list.add(prefix)
    end

    Ontology.select(:name).where("name ILIKE :prefix", prefix: "#{prefix}%").group("name").limit(5).each do |ontology|
      text_list.add(ontology.name)
    end

    Entity.select(:name).where("name ILIKE :prefix", prefix: "#{prefix}%").group("name").limit(5).each do |symbol|
      text_list.add(symbol.name)
    end

    Logic.where("name ILIKE :prefix", prefix: "#{prefix}%").limit(5).each do |logic|
      if logic.ontologies.size != 0
        text_list.add(logic.name)
      end
    end

    text_list.to_a.sort.map { |x| {text: x} }
  end

  def make_repository_bean_list_json(repository, keyword_list, page)
    JSON.generate(make_repository_bean_list_response(repository, keyword_list, page))
  end

  def make_global_bean_list_json(keyword_list, page)
    JSON.generate(make_global_bean_list_response(keyword_list, page))
  end

  def make_repository_bean_list_response(repository, keyword_list, page)
    ontology_hash = Hash.new
    index = 0

    # Display all repository ontologies for empty keyword list
    if keyword_list.size == 0
      offset = (page - 1) * @limit
      bean_list_factory = OntologyBeanListFactory.new
      repository.ontologies.limit(@limit).offset(offset).each do |ontology|
        bean_list_factory.add_small_bean(ontology)
      end
      return {
        page: page,
        resultsInPage: @limit,
        resultsInSet: repository.ontologies.count,
        results: bean_list_factory.bean_list
      }
    end

    keyword_list.each do |keyword|
      keyword_hash = Hash.new

      repository.ontologies.where("name = :name", name: "#{keyword}").limit(50).each do |ontology|
        keyword_hash[ontology.id] ||= ontology
      end

      Entity.where("name = :name", name: "#{keyword}").limit(50).each do |symbol|
        if repository.id == symbol.ontology.repository.id
          keyword_hash[symbol.ontology.id] ||= symbol.ontology
        end
      end

      if logic = Logic.find_by_name(keyword)
        logic.ontologies.each { |o| keyword_hash[o.id] ||= o }
      end

      if index == 0
        ontology_hash = keyword_hash
      else
        hash = Hash.new

        keyword_hash.each_key do |key|
          hash[key] ||= ontology_hash[key] if ontology_hash[key]
        end

        ontology_hash = hash
      end

      index += 1
    end

    count = ontology_hash.size
    max = page * @limit
    min = max - @limit
    index = 0
    bean_list_factory = OntologyBeanListFactory.new
    ontology_hash.each_value do |ontology|
      if index >= min && index < max
        bean_list_factory.add_small_bean(ontology)
      end
      index = index + 1
    end

    {
      page: page,
      resultsInPage: @limit,
      resultsInSet: count,
      results: bean_list_factory.bean_list
    }
  end

  def make_global_bean_list_response(keyword_list, page)
    {
      page: 0,
      resultsInPage: 50,
      resultsInSet: 0,
      results: make_global_bean_list(keyword_list, page)
    }
  end

  def make_global_bean_list(keyword_list, page)
    ontology_hash = Hash.new
    index = 0

    keyword_list.each do |keyword|
      keyword_hash = Hash.new

      Ontology.search_by_keyword(keyword).each do |ontology|
        keyword_hash[ontology.id] ||= ontology
      end

      if index == 0
        ontology_hash = keyword_hash
      else
        hash = Hash.new

        keyword_hash.each_key do |key|
          hash[key] ||= ontology_hash[key] if ontology_hash[key]
        end

        ontology_hash = hash
      end

      index += 1
    end

    bean_list_factory = OntologyBeanListFactory.new
    ontology_hash.each_value do |ontology|
      bean_list_factory.add_small_bean(ontology)
    end

    bean_list_factory.bean_list
  end

end