# -*- coding: utf-8 -*-

#    Copyright 2013 Mirantis, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

from datetime import datetime

from nailgun import consts
from nailgun.db.sqlalchemy import models

from nailgun.errors import errors
from nailgun.logger import logger

from nailgun.objects import NailgunCollection
from nailgun.objects import NailgunObject

from nailgun.objects import Task

from nailgun.objects.serializers.notification import NotificationSerializer


class Notification(NailgunObject):

    model = models.Notification
    serializer = NotificationSerializer

    schema = {
        "$schema": "http://json-schema.org/draft-04/schema#",
        "title": "Notification",
        "description": "Serialized Notification object",
        "type": "object",
        "properties": {
            "id": {"type": "number"},
            "cluster_id": {"type": "number"},
            "node_id": {"type": "number"},
            "task_id": {"type": "number"},
            "time": {"type": "string"},
            "date": {"type": "string"},
            "topic": {
                "type": "string",
                "enum": list(consts.NOTIFICATION_TOPICS)
            },
            "message": {"type": "string"},
            "status": {
                "type": "string",
                "enum": list(consts.NOTIFICATION_STATUSES)
            }
        }
    }

    @classmethod
    def create(cls, data):
        topic = data.get("topic")
        node_id = data.get("node_id")
        task_uuid = data.pop("task_uuid", None)
        message = data.get("message")

        if topic == 'discover' and node_id is None:
            raise errors.CannotFindNodeIDForDiscovering(
                "No node id in discover notification"
            )

        if "datetime" not in data:
            data["datetime"] = datetime.now()

        task = None
        exist = None
        if task_uuid:
            task = Task.get_by_uuid(task_uuid)
            if task and node_id:
                exist = NotificationCollection.filter_by(
                    None,
                    node_id=node_id,
                    message=message,
                    task_id=task.id
                ).first()

        if not exist:
            super(Notification, cls).create(data)
            logger.info(
                u"Notification: topic: {0} message: {1}".format(
                    data.get("topic"),
                    data.get("message")
                )
            )

    @classmethod
    def to_dict(cls, instance, fields=None):
        notif_dict = cls.serializer.serialize(instance, fields=fields)
        notif_dict['time'] = instance.datetime.strftime('%H:%M:%S')
        notif_dict['date'] = instance.datetime.strftime('%d-%m-%Y')
        return notif_dict


class NotificationCollection(NailgunCollection):

    single = Notification