module Gtn::Supported

# File _plugins/gtn/supported.rb, line 21
def self.calculate(data, tool_list)
  # p "Calculating supported servers for this tool list"
  if data.nil? || data.empty? || tool_list.empty? || tool_list.nil?
    return {
      'exact' => [],
      'inexact' => Gtn::Usegalaxy.servers.map do |x|
                     x = x.transform_keys(&:to_s)
                     x['usegalaxy'] = true
                     x
                   end
    }
  end

  supported = { exact: {}, inexact: {} }
  tool_list.each do |tool|
    if tool.count('/') > 4
      # E.g. toolshed.g2.bx.psu.edu/repos/iuc/circos/circos/0.69.8+galaxy10
      tool_id = tool.split('/')[0..4].join('/')
      tool_version = tool.split('/')[5..].join('/')
      # p "Checking #{tool_id} #{tool_version}... "

      if data['tools'].key?(tool_id)
        supported[:exact][tool] = data['tools'][tool_id][tool_version] if data['tools'][tool_id].key?(tool_version)

        supported[:inexact][tool] = data['tools'][tool_id].map { |_, v| v }.flatten.uniq.sort
      end
    elsif data['tools'].key?(tool)
      # E.g. 'upload1'
      # p "Checking #{tool}... "
      supported[:inexact][tool] = data['tools'][tool].map { |_, v| v }.flatten.uniq.sort
      supported[:exact][tool] = data['tools'][tool].map { |_, v| v }.flatten.uniq.sort
    end
  end

  # Exactly supporting servers:
  # this is the set of intersections across supported[:exact][*]
  exact_support = (0..data['servers'].length - 1).to_a
  tool_list.each do |tool|
    exact_support &= (supported[:exact][tool] || [])
  end

  # Inexactly supporting servers
  # Set of intersections across (union of supported[:exact] and supported[:inexact])
  inexact_support = (0..data['servers'].length - 1).to_a
  tool_list.each do |tool|
    et = supported[:exact][tool] || []
    it = supported[:inexact][tool] || []
    inexact_support &= (et | it)
  end
  # Remove the exactly supported ones because that would be extraneous we
  # check here if it's an array because the above code will occasionally
  # generate a 'false' value when merging sets.
  inexact_support -= exact_support

  usegalaxy_server_urls = Gtn::Usegalaxy.servers.map { |x| x[:url] }

  {
    'exact' => (exact_support || []).map do |id|
      data['servers'][id].update(
        { 'usegalaxy' => usegalaxy_server_urls.include?(data['servers'][id]['url']) }
      )
    end,
    'inexact' => (inexact_support || []).map do |id|
      data['servers'][id].update(
        { 'usegalaxy' => usegalaxy_server_urls.include?(data['servers'][id]['url']) }
      )
    end
  }
end

# File _plugins/gtn/supported.rb, line 114
def self.calculate_matrix(data, tool_list)
  structure = {
    'servers' => [],
    'tools' => [],
  }
  # p "Calculating supported servers for this tool list"
  if data.nil? || data.empty? || tool_list.empty? || tool_list.nil?
    return structure
  end

  structure['servers'] = data['servers']

  tool_list.each do |tool|
    tool_for_server = {
      'id' => tool,
      'servers' => []
    }

    if tool.count('/') > 4
      # E.g. toolshed.g2.bx.psu.edu/repos/iuc/circos/circos/0.69.8+galaxy10
      tool_id = tool.split('/')[0..4].join('/')
      tool_version = tool.split('/')[5..].join('/')
      tool_for_server['version'] = tool_version
    else
      tool_id = tool
      tool_version = 'local'
      tool_for_server['version'] = tool_version
    end

    data['servers'].each.with_index do |server, s_index|
      # If we've never seen this tool anywhere
      if ! data['tools'].key?(tool_id)
        res = {'state' => 'missing'}
      else
        # If we've seen this exact version on the current server/index
        if data['tools'][tool_id].key?(tool_version) && data['tools'][tool_id][tool_version].include?(s_index)
          res = {'state' => 'exact', 'version' => tool_version}
        elsif data['tools'][tool_id]
          res = {
            'state' => 'inexact', 
            'versions' => data['tools'][tool_id].select{|k, v| v.include?(s_index)}.keys
          }

          if res['versions'].length == 1 && res['versions'][0] == '_'
            res = {
              'state' => 'local'
            }
          elsif res['versions'].length == 0
            res = {'state' => 'missing'}
          end
        end
      end

      res['server'] = server['url']
      tool_for_server['servers'].push(res)

    end

    structure['tools'].push(tool_for_server)
  end

  # TODO: would be nice to re-sort the servers by the number of exact matches
  structure
end