#!/usr/bin/env python

#
# Copyright (c) 2007 Kristinn B. Gylfason <citeulike@askur.org>
# All rights reserved.
#
# This code is derived from software contributed to CiteULike.org
# by
#    Diwaker Gupta
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 3. All advertising materials mentioning features or use of this software
#    must display the following acknowledgement:
#        This product includes software developed by
#		 CiteULike <http://www.citeulike.org> and its
#		 contributors.
# 4. Neither the name of CiteULike nor the names of its
#    contributors may be used to endorse or promote products derived
#    from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY CITEULIKE.ORG AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

import re, sys, urllib2




RIS_SERVER_ROOT = 'http://www.iop.org/EJ/sview/'
RIS_SERVER_POST_STR = 'submit=1&format=refmgr'

# regexp to screen scrape the DOI and get the article info from it

DOI_REGEXP = r'<meta name="citation_doi" content="([^"]+)" />'
DOI_REGEXP_FLAGS = re.IGNORECASE

# error messages
ERR_STR_PREFIX = 'status\terr\t'
ERR_STR_FETCH = 'Unable to fetch the page: '
ERR_STR_TRY_AGAIN = 'The server may be down.  Please try later.'
ERR_STR_NO_DOI = 'No valid document object identifier (DOI) found on the page: '
ERR_STR_NO_RIS = 'No RIS entry found for the DOI: '
ERR_STR_REPORT = 'Please report the error to plugins@citeulike.org.'

# read url from std input
url = sys.stdin.readline()
# get rid of the newline at the end
url = url.strip()

# if this is a (restricted) URL, try to redirect to a readable one
match  = re.search(r'http://iopscience.iop.org/([^?]*)', url)
if match:
	print "status\tredirect\thttp://www.iop.org/EJ/abstract/%s" % match.group(1)
	sys.exit(0)


# fetch the page the user is viewing and exit gracefully in case of trouble
try:
	f = urllib2.urlopen(url)
except:
	print ERR_STR_PREFIX + ERR_STR_FETCH + url + '.  ' + ERR_STR_TRY_AGAIN
	sys.exit(1)

content = f.read()

# screen scrape the DOI from the page and exit gracefully if none is found
doi_match  = re.search(DOI_REGEXP, content, DOI_REGEXP_FLAGS)
if not doi_match:
	print ERR_STR_PREFIX + ERR_STR_NO_DOI + url + '.  ' + ERR_STR_REPORT
	sys.exit(1)

doi = doi_match.group(1)

url_match = re.search(r'<meta name="citation_abstract_html_url" content="http://www.iop.org/EJ/abstract/([^"]+)" />', content, DOI_REGEXP_FLAGS)
url_suffix = url_match.group(1)


# fetch the RIS entry for the DOI and exit gracefully in case of trouble
req = urllib2.Request(RIS_SERVER_ROOT + url_suffix)
req.add_data(RIS_SERVER_POST_STR)
try:
	f = urllib2.urlopen(req)
except:
	print ERR_STR_PREFIX + ERR_STR_NO_RIS + doi + '.  ' + ERR_STR_REPORT
	sys.exit(1)

ris_entry = f.read()
# get rid of the extra newline at the end
ris_entry = ris_entry.strip()

# Tidy up author name formatting to help our parser out
auth_exp = re.compile(r'A1  - (.+?),(.+?)$', re.M)
ris_entry = re.sub(auth_exp, r'A1  - \1, \2', ris_entry)


# print the results
print "begin_tsv"
print "linkout\tDOI\t\t%s\t\t" % (doi)
print "type\tJOUR"
print "doi\t" + doi
print "end_tsv"
print "begin_ris"
print ris_entry
print "end_ris"
print "status\tok"