app.rb

require 'bundler/setup'
require 'sinatra'
require 'json'
require 'open-uri'

set :bind, '0.0.0.0'
set :protection, except: [:frame_options, :host_authorization]
set :host_authorization, { permitted_hosts: [] }

# required ENV variables, can be set in .env file
ENV['ROR_API'] ||= 'https://api.ror.org/v2'.freeze
ENV['ROR_RECON'] ||= 'https://reconcile.ror.org'.freeze

MAX_RESULTS = 5

helpers do
  def extract_ror_id(ror)
    if ror.include?('ror.org/')
      ror.split('ror.org/').last
    else
      ror
    end
  end

  def get_display_name(record)
    names = record['names'] || []
    display = names.find { |n| n['types']&.include?('ror_display') }
    display ? display['value'] : names.first&.dig('value')
  end

  def get_country_name(record)
    record.dig('locations', 0, 'geonames_details', 'country_name')
  end

  def search_ror(test_org_name)
    test_org_name = test_org_name.tr(';', '')
    test_org_name = test_org_name.tr('&', '%26')
    test_org_name = test_org_name.tr('/', '%2F')
    test_org_name = test_org_name.tr('?', '%3F')
    uri = "#{ENV['ROR_API']}/organizations?query=#{URI.encode_www_form_component(test_org_name)}"
    res = URI.open(uri).read
    JSON.parse(res)
  end

  def get_ror(ror)
    ror_id = extract_ror_id(ror)
    uri = "#{ENV['ROR_API']}/organizations/#{URI.encode_www_form_component(ror_id)}"
    res = URI.open(uri).read
    JSON.parse(res)
  end
end

get '/heartbeat/?', provides: %i[html json] do
  status = { max_results: MAX_RESULTS, named: 'ROR Reconciler', status: 'OK', pid: $PROCESS_ID.to_s, ruby_version: RUBY_VERSION.to_s, phusion: defined?(PhusionPassenger) ? true : false }
  status.to_json
end

get '/preview/*', provides: [:html] do
  ror = params['splat'].first
  ror_record = get_ror(ror)

  display_name = get_display_name(ror_record)
  country_name = get_country_name(ror_record)

  erb :preview, locals: { ror_api: ENV['ROR_API'], ror_record: ror_record, display_name: display_name, country_name: country_name }
end

post '/reconcile/?', provides: %i[html json] do
  content_type :json
  queries = JSON.parse params['queries']
  return {}.to_json unless queries['q0'].key?('type')

  results = {}
  queries.each_pair do |key, q|
    hits = search_ror(q['query'])
    results[key] = {}
    results[key]['result'] = []
    if hits['items']
      score = hits['items'].length
      type = { 'id' => '/ror/organization', 'name' => 'Organization' }
        hits['items'][0, MAX_RESULTS].each do |hit|
        ror = hit['id']
        entry = { 'id' => ror, 'name' => get_display_name(hit), 'type' => [type], 'score' => score, 'match' => 'false', 'uri' => ror }
        results[key]['result'].push(entry)
        score -= 1
      end
    end
  end
  results.to_json
end

get '/flyout/?', provides: [:json] do
  callback = params.delete('callback')
  ror = params['id']
  ror_record = get_ror(ror)

  display_name = get_display_name(ror_record)
  country_name = get_country_name(ror_record)

  html = File.open('views/flyout.erb').read
  template = ERB.new(html)
  b = binding
  b.local_variable_set(:ror_record, ror_record)
  b.local_variable_set(:ror_api, ENV['ROR_API'])
  b.local_variable_set(:display_name, display_name)
  b.local_variable_set(:country_name, country_name)
  json = { 'id' => ror, 'html' => template.result(b) }.to_json
  if callback
    content_type :js
    response = "#{callback}(#{json})"
  else
    content_type :json
    response = json
  end
  response
end

get '/suggest/?', provides: [:json] do
  callback = params.delete('callback')
  prefix = params['prefix']
  prefix += '*'
  puts '-->' + prefix

  results = []
  hits = search_ror(prefix)
  score = hits['items'].length

  hits['items'][0, MAX_RESULTS].each do |hit|
    ror = hit['id']
    entry = { 'id' => ror, 'name' => get_display_name(hit), 'score' => score, 'description' => get_country_name(hit) }
    results.push(entry)
    score -= 1
  end

  json = { 'result' => results }.to_json

  if callback
    content_type :js
    response = "#{callback}(#{json})"
  else
    content_type :json
    response = json
  end

  response
end

get '/reconcile/?', provides: %i[html json] do
  callback = params.delete('callback')

  default_types = [

    { 'id' => '/ror/organization', 'name' => 'Organization' }
  ]
  view =  { 'url' => "#{ENV['ROR_RECON']}/reconcile" }

  preview = {
    'width' => 400,
    'height' => 100,
    'url' => "#{ENV['ROR_RECON']}/preview/{{id}}"
  }

  entity = { 'flyout_service_path' => '/flyout?id=${id}', 'service_path' => '/suggest', 'service_url' => ENV['ROR_RECON'] }

  suggest = { 'entity' => entity }

  json = { 'name' => 'ROR Reconciliation Service',
           'identifierSpace' => 'http://ror.org/organization',
           'schemaSpace' => 'http://ror.org/ns/type.object.id',
           'defaultTypes' => default_types,
           'view' => view,
           'suggest' => suggest,
           'preview' => preview }.to_json

  if callback
    content_type :js
    response = "#{callback}(#{json})"
  else
    content_type :json
    response = json
  end

  response
end

get '/', provides: [:html] do
  redirect 'https://www.ror.org', 'Roar!'
end