# -----------------------------------------------------------------------------
# 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.
# -----------------------------------------------------------------------------

import warnings
from json import dumps
from copy import deepcopy

from tornado.web import authenticated, HTTPError

from qiita_core.util import execute_as_transaction
from qiita_db.study import Study
from qiita_db.portal import Portal
from qiita_db.exceptions import QiitaDBError
from .base_handlers import BaseHandler


class PortalEditBase(BaseHandler):
    study_cols = ['study_id', 'study_title', 'study_alias']

    def check_admin(self):
        if self.current_user.level != "admin":
            raise HTTPError(403, reason="%s does not have access to portal "
                            "editing!" % self.current_user.id)

    @execute_as_transaction
    def get_info(self, portal="QIITA"):
        # Add the portals and, optionally, checkbox to the information
        studies = [s.id for s in Portal(portal).get_studies()]
        if not studies:
            return []

        study_info = Study.get_info(studies, info_cols=self.study_cols)
        info = []
        for s in study_info:
            # Make sure in correct order
            hold = dict(s)
            hold['portals'] = ', '.join(sorted(Study(s['study_id'])._portals))
            info.append(hold)
        return info


class StudyPortalHandler(PortalEditBase):
    @authenticated
    @execute_as_transaction
    def get(self):
        self.check_admin()
        info = self.get_info()
        portals = Portal.list_portals()
        headers = deepcopy(self.study_cols)
        headers.insert(0, "portals")
        self.render('portals_edit.html', headers=headers, info=info,
                    portals=portals, submit_url="/admin/portals/studies/")

    @authenticated
    @execute_as_transaction
    def post(self):
        self.check_admin()
        portal = self.get_argument('portal')
        studies = map(int, self.get_arguments('selected'))
        action = self.get_argument('action')

        try:
            portal = Portal(portal)
        except Exception:
            raise HTTPError(400, reason="Not valid portal: %s" % portal)
        try:
            with warnings.catch_warnings(record=True) as warns:
                if action == "Add":
                    portal.add_studies(studies)
                elif action == "Remove":
                    portal.remove_studies(studies)
                else:
                    raise HTTPError(400, reason="Unknown action: %s" % action)
        except QiitaDBError as e:
            self.write(action.upper() + " ERROR:<br/>" + str(e))
            return

        msg = '; '.join([str(w.message) for w in warns])
        self.write(action + " completed successfully<br/>" + msg)


class StudyPortalAJAXHandler(PortalEditBase):
    @authenticated
    @execute_as_transaction
    def get(self):
        self.check_admin()
        portal = self.get_argument('view-portal')
        echo = self.get_argument('sEcho')
        info = self.get_info(portal=portal)
        # build the table json
        results = {
            "sEcho": echo,
            "iTotalRecords": len(info),
            "iTotalDisplayRecords": len(info),
            "aaData": info
        }

        # return the json in compact form to save transmit size
        self.write(dumps(results, separators=(',', ':')))