app.rb

require "sinatra/base"
require "sinatra/namespace"
require 'haml'
require "tilt"
require 'sass'
require 'json'
require 'museum_provenance'
require 'elasticsearch'

$stdout.sync = true # for foreman logging

module CMOA
  class App < Sinatra::Base
    register Sinatra::Namespace

    configure do
      set :elasticsearch, Elasticsearch::Client.new(log: false)
    end

    helpers do
      def symbolize_keys(hash)
        hash.inject({}){|result, (key, value)|
          new_key = case key
                    when String then key.to_sym
                    else key
                    end
          new_value = case value
                      when Hash then symbolize_keys(value)
                      else value
                      end
          result[new_key] = new_value
          result
        }
      end

      def maintain_history(data,results)
        results[:id] = data[:id]
        if data[:original_text] != results["original_text"] 
          results[:original_text] = data[:original_text]
          results[:parsable] = (results[:original_text] == results["provenance"])
        end
        results
      end
    end


    namespace '/parsers' do

      post "/provenance_line" do
        content_type :json
        MuseumProvenance::Provenance.extract(params[:str]).to_json
      end

      post '/timestring' do
        content_type :json
        p = MuseumProvenance::Period.new("test ")
        p.parse_time_string(params[:str])      
        hash = {}
        hash[:eotb] = p.eotb ? p.eotb.jd : nil    
        hash[:eote] = p.eote ? p.eote.jd : nil   
        hash[:botb] = p.botb ? p.botb.jd : nil
        hash[:bote] = p.bote ? p.bote.jd : nil      
        hash[:botb_precision] = p.beginning.earliest_raw.precision rescue nil 
        hash[:eotb_precision] = p.beginning.latest_raw.precision rescue nil
        hash[:bote_precision] = p.ending.earliest_raw.precision  rescue nil
        hash[:eote_precision] = p.ending.latest_raw.precision rescue nil
        hash.to_json
      end
    end


    namespace '/api/1' do
      
      # Fake endpoint.   Should not be here.
      delete '/periods/:id' do
        content_type :json
        {}.to_json
      end

      # Get a particular artwork
      get '/artworks/:id' do
        content_type :json
        results = settings.elasticsearch.get index: 'cmoa_provenance', type:'artwork', id: params[:id]
        return nil if results.nil?
        return {artwork: results['_source']}.to_json
      end
    
    end
    # END API ENDSPACE

    get '/search' do
      content_type :json
      body =  {
              from: ((params[:page].to_i-1)*10),
              query: 
                {
                  match_phrase_prefix: {
                    _all: {
                      query:  params[:query],
                      slop: 10
                    }
                  }
                }
              }

      results = settings.elasticsearch.search index: 'cmoa_provenance', body: body
      return results.to_json
    end

    post '/add_party' do
      content_type :json
      if params[:period]
        data = params[:period].collect do |key,val| 
          symbolize_keys(val)
        end
        results = MuseumProvenance::Provenance.from_json({period: data})
      else 
        results = MuseumProvenance::Timeline.new()
      end
      results.insert_earliest(MuseumProvenance::Period.new("Unknown party"));
      
      vals = JSON.parse(results.to_json)

      new_id = 0
      if data
        id_numbers = data.collect do |p| 
          p[:id].split("-").first.to_i
        end.sort
        new_id = (id_numbers.last+1).to_s
      end

      vals["period"] = vals["period"].collect.with_index do |r,i|
        if i == 0
          r[:id] = "#{new_id}-#{params['artwork_id']}"
        else
          r = maintain_history(data[i-1],r)
        end
        r["order"] = i
        r
      end
      vals.to_json
    end

    post '/rebuild_structure' do
      content_type :json
      data = params[:period].collect do |key,val| 
        symbolize_keys(val)
      end
      results = MuseumProvenance::Provenance.from_json({period: data}).to_json
      vals = JSON.parse(results)
      vals["period"] = vals["period"].collect.with_index do |r,i|
        maintain_history(data[i],r)
      end
      vals.to_json
    end

    post '/get_structure' do
      content_type :json
      p = params[:provenance]
      MuseumProvenance::Provenance.extract(p).to_json
    end
    
    get '/*' do
      haml :index
    end
  end
end