require 'rubygems'
gem 'opentox-ruby-api-wrapper', '= 1.4.4.4'
require 'opentox-ruby-api-wrapper'
#require 'sinatra/respond_to'
#Sinatra::Application.register Sinatra::RespondTo


LOGGER.progname = File.expand_path(__FILE__)

class Dataset
	include DataMapper::Resource
	property :id, Serial
	property :uri, String, :length => 255
	property :file, String, :length => 255
	property :yaml, Text, :length => 2**32-1 
	property :owl, Text, :length => 2**32-1 
	property :created_at, DateTime

	def to_owl
		data = YAML.load(yaml)
		owl = OpenTox::Owl.create 'Dataset', uri
		['title', 'creator'].each do |method|
			owl.set(method, data.send(method))
		end
		if data.data
			data.data.each do |compound,features|
				owl.add_data_entries compound,features
			end
	  end
		owl.rdf
	end

end

DataMapper.auto_upgrade!

## REST API

get '/?' do
	response['Content-Type'] = 'text/uri-list'
	Dataset.all(params).collect{|d| d.uri}.join("\n") + "\n"
end

get '/:id' do
	accept = request.env['HTTP_ACCEPT']
	accept = 'application/rdf+xml' if accept == '*/*' or accept == '' or accept.nil?
	# workaround for browser links
	case params[:id]
	when /.yaml$/
		params[:id].sub!(/.yaml$/,'')
		accept =  'text/x-yaml'
	when /.rdf$/
		params[:id].sub!(/.rdf$/,'')
		accept =  'application/rdf+xml'
	end
	begin
		dataset = Dataset.get(params[:id])
    halt 404, "Dataset #{params[:id]} not found." unless dataset
	rescue => e
		raise e.message + e.backtrace
		halt 404, "Dataset #{params[:id]} not found."
	end
	case accept
	when /rdf/ # redland sends text/rdf instead of application/rdf+xml
		response['Content-Type'] = 'application/rdf+xml'
		unless dataset.owl # lazy owl creation
			dataset.owl = dataset.to_owl
			dataset.save
		end
		dataset.owl
	when /yaml/
		response['Content-Type'] = 'text/x-yaml'
		dataset.yaml
	else
		halt 400, "Unsupported MIME type '#{accept}'"
	end
end

get '/:id/features/:feature_id/?' do
	OpenTox::Dataset.find(url_for("/#{params[:id]}", :full)).feature(params[:feature_id])
end

get '/:id/features/?' do
	YAML.load(Dataset.get(params[:id]).yaml).features.join("\n") + "\n"
end

get '/:id/compounds/?' do
	YAML.load(Dataset.get(params[:id]).yaml).compounds.join("\n") + "\n"
end

post '/?' do

		dataset = Dataset.new
		dataset.save
		dataset.uri = url_for("/#{dataset.id}", :full)
		content_type = request.content_type
		content_type = "application/rdf+xml" if content_type.nil?
		case request.content_type
		when /yaml/
			dataset.yaml =	request.env["rack.input"].read
		when /csv/
			dataset.yaml =	csv2yaml request.env["rack.input"].read
		when "application/rdf+xml"
			dataset.yaml =	owl2yaml request.env["rack.input"].read
		else
			halt 404, "MIME type \"#{request.content_type}\" not supported."
		end
		begin
			#dataset.owl = d.rdf
      #dataset.uri = uri 
			raise "saving failed: "+dataset.errors.inspect unless dataset.save
		rescue => e
			LOGGER.error e.message
			LOGGER.info e.backtrace
			halt 500, "Could not save dataset #{dataset.uri}."
		end
		LOGGER.debug "#{dataset.uri} saved."
	response['Content-Type'] = 'text/uri-list'
	dataset.uri + "\n"
end

delete '/:id/?' do
	begin
		dataset = Dataset.get(params[:id])
		dataset.destroy!
		response['Content-Type'] = 'text/plain'
		"Dataset #{params[:id]} deleted."
	rescue
		halt 404, "Dataset #{params[:id]} does not exist."
	end
end

delete '/?' do
  
	Dataset.all.each do |d|
		begin
			File.delete d.file 
		rescue
			LOGGER.error "Cannot delete dataset file '#{d.file}'"
		end
	 	#d.destroy!
	end
  Dataset.auto_migrate!
	response['Content-Type'] = 'text/plain'
	"All datasets deleted."
end