# Copyright (c) 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 oslo_utils import uuidutils
import six
import sqlalchemy as sa
from sqlalchemy.dialects.mysql import MEDIUMTEXT

from fenix.db.sqlalchemy import model_base as mb

# Helpers


def _generate_unicode_uuid():
    return six.text_type(uuidutils.generate_uuid())


def MediumText():
    return sa.Text().with_variant(MEDIUMTEXT(), 'mysql')


def _id_column():
    return sa.Column(sa.String(36),
                     primary_key=True,
                     default=_generate_unicode_uuid)


class MaintenanceSession(mb.FenixBase):
    """Maintenance session"""

    __tablename__ = 'sessions'

    session_id = sa.Column(sa.String(36), primary_key=True)
    prev_state = sa.Column(sa.String(length=32), nullable=True)
    state = sa.Column(sa.String(length=32), nullable=True)
    maintenance_at = sa.Column(sa.DateTime(), nullable=True)
    meta = sa.Column(MediumText(), nullable=False)
    workflow = sa.Column(sa.String(length=255), nullable=True)

    def to_dict(self):
        return super(MaintenanceSession, self).to_dict()


class MaintenanceActionPlugin(mb.FenixBase):
    """Maintenance action plugin"""

    __tablename__ = 'action_plugins'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    plugin = sa.Column(sa.String(length=255), nullable=False)
    type = sa.Column(sa.String(length=32), nullable=True)
    meta = sa.Column(MediumText(), nullable=False)

    def to_dict(self):
        return super(MaintenanceActionPlugin, self).to_dict()


class MaintenanceActionPluginInstance(mb.FenixBase):
    """Maintenance action instance"""

    __tablename__ = 'action_plugin_instances'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    hostname = sa.Column(sa.String(255), nullable=False)
    plugin = sa.Column(sa.String(255), nullable=False)
    state = sa.Column(sa.String(length=32), nullable=True)

    def to_dict(self):
        return super(MaintenanceActionPluginInstance, self).to_dict()


class MaintenanceHost(mb.FenixBase):
    """Maintenance host"""

    __tablename__ = 'hosts'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    hostname = sa.Column(sa.String(length=255), primary_key=True)
    type = sa.Column(sa.String(length=32), nullable=True)
    maintained = sa.Column(sa.Boolean, default=False)
    disabled = sa.Column(sa.Boolean, default=False)
    details = sa.Column(sa.String(length=255), nullable=True)
    plugin = sa.Column(sa.String(length=255), nullable=True)
    plugin_state = sa.Column(sa.String(length=32), nullable=True)

    def to_dict(self):
        return super(MaintenanceHost, self).to_dict()


class MaintenanceProject(mb.FenixBase):
    """Maintenance project"""

    __tablename__ = 'projects'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    project_id = sa.Column(sa.String(length=36), primary_key=True)
    state = sa.Column(sa.String(length=37), nullable=True)

    def to_dict(self):
        return super(MaintenanceProject, self).to_dict()


class MaintenanceInstance(mb.FenixBase):
    """Maintenance instance"""

    __tablename__ = 'instances'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    instance_id = sa.Column(sa.String(length=36), primary_key=True)
    state = sa.Column(sa.String(length=16), nullable=True)
    action = sa.Column(sa.String(32), nullable=True)
    project_id = sa.Column(sa.String(36), nullable=True)
    project_state = sa.Column(sa.String(length=37), nullable=True)
    instance_name = sa.Column(sa.String(length=255), nullable=False)
    action_done = sa.Column(sa.Boolean, default=False)
    host = sa.Column(sa.String(length=255), nullable=False)
    details = sa.Column(sa.String(length=255), nullable=True)

    def to_dict(self):
        return super(MaintenanceInstance, self).to_dict()


class MaintenanceDownload(mb.FenixBase):
    """Maintenance project"""

    __tablename__ = 'downloads'

    id = _id_column()
    session_id = sa.Column(sa.String(36), sa.ForeignKey('sessions.session_id'),
                           nullable=False)
    local_file = sa.Column(sa.String(length=160), primary_key=True)

    def to_dict(self):
        return super(MaintenanceDownload, self).to_dict()


class ProjectInstance(mb.FenixBase):
    """Project instances"""

    __tablename__ = 'project_instances'

    instance_id = sa.Column(sa.String(36), primary_key=True)
    project_id = sa.Column(sa.String(36), nullable=True)
    group_id = sa.Column(sa.String(36), nullable=False)
    instance_name = sa.Column(sa.String(length=255), nullable=False)
    max_interruption_time = sa.Column(sa.Integer, nullable=False)
    migration_type = sa.Column(sa.String(36), nullable=False)
    resource_mitigation = sa.Column(sa.Boolean, default=False)
    lead_time = sa.Column(sa.Integer, nullable=False)

    def to_dict(self):
        return super(ProjectInstance, self).to_dict()


class InstanceGroup(mb.FenixBase):
    """Instances groups"""

    __tablename__ = 'instance_groups'

    group_id = sa.Column(sa.String(36), primary_key=True)
    project_id = sa.Column(sa.String(36), nullable=True)
    group_name = sa.Column(sa.String(length=255), nullable=False)
    anti_affinity_group = sa.Column(sa.Boolean, default=False)
    max_instances_per_host = sa.Column(sa.Integer, nullable=True)
    max_impacted_members = sa.Column(sa.Integer, nullable=False)
    recovery_time = sa.Column(sa.Integer, nullable=False)
    resource_mitigation = sa.Column(sa.Boolean, default=False)

    def to_dict(self):
        return super(InstanceGroup, self).to_dict()