# -----------------------------------------------------------------------------
# Copyright (c) 2014--, The Qiita Development Team.
#
# Distributed under the terms of the BSD 3-clause License.
#
# The full license is in the file LICENSE, distributed with this software.
# -----------------------------------------------------------------------------

from tornado.web import HTTPError
from tornado.gen import coroutine

from .base_handlers import BaseHandler
from qiita_db.study import Study
from qiita_db.artifact import Artifact
from qiita_db.util import get_artifacts_information
from qiita_db.exceptions import QiitaDBUnknownIDError
from qiita_core.util import execute_as_transaction
from qiita_pet.util import EBI_LINKIFIER
from qiita_pet.handlers.util import doi_linkifier, pubmed_linkifier


class PublicHandler(BaseHandler):
    @coroutine
    @execute_as_transaction
    def get(self):
        study_id = self.get_argument("study_id",  None)
        artifact_id = self.get_argument("artifact_id",  None)

        if study_id is None and artifact_id is None:
            raise HTTPError(
                422, reason='You need to specify study_id or artifact_id')
            self.finish()
        elif study_id is not None:
            try:
                study = Study(int(study_id))
            except QiitaDBUnknownIDError:
                raise HTTPError(
                    422, reason="Study %s doesn't exist" % study_id)
                self.finish()
            artifact_ids = [a.id for a in study.artifacts()
                            if a.visibility == 'public']
        else:
            try:
                artifact = Artifact(int(artifact_id))
            except QiitaDBUnknownIDError:
                raise HTTPError(
                    422, reason="Artifact %s doesn't exist" % artifact_id)
                self.finish()
            if artifact.visibility != 'public':
                raise HTTPError(
                    422, reason="Artifact %s is not public" % artifact_id)
                self.finish()

            study = artifact.study
            if study is None:
                raise HTTPError(422, reason="Artifact %s doesn't belong to "
                                "a study" % artifact_id)
                self.finish()
            artifact_ids = [artifact.id]

        if study.status != 'public':
            raise HTTPError(
                422, reason='Not a public study')
            self.finish()

        study_info = study.info
        study_info['study_id'] = study.id
        study_info['study_title'] = study.title
        study_info['shared_with'] = [s.id for s in study.shared_with]
        study_info['status'] = study.status
        study_info['ebi_study_accession'] = study.ebi_study_accession
        study_info['ebi_submission_status'] = study.ebi_submission_status

        # Clean up StudyPerson objects to string for display
        email = '<a href="mailto:{email}">{name} ({affiliation})</a>'
        pi = study.info['principal_investigator']
        study_info['principal_investigator'] = email.format(**{
            'name': pi.name,
            'email': pi.email,
            'affiliation': pi.affiliation})

        study_info['owner'] = study.owner.id
        # Add needed info that is not part of the initial info pull
        study_info['publications'] = []
        for pub, is_doi in study.publications:
            if is_doi:
                study_info['publications'].append(pubmed_linkifier([pub]))
            else:
                study_info['publications'].append(doi_linkifier([pub]))
        study_info['publications'] = ', '.join(study_info['publications'])

        if study_info['ebi_study_accession']:
            links = ''.join([EBI_LINKIFIER.format(a) for a in study_info[
                'ebi_study_accession'].split(',')])
            study_info['ebi_study_accession'] = '%s (%s)' % (
                links, study_info['ebi_submission_status'])

        self.render("public.html", study_info=study_info,
                    artifacts_info=get_artifacts_information(
                        artifact_ids, False))