#
# Copyright (c) 2023 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#

from osc_lib.command import command

from dcmanagerclient.commands.v1 import base
from dcmanagerclient import exceptions


def association_format(peer_group_association=None):
    columns = (
        'id',
        'peer_group_id',
        'system_peer_id',
        'type',
        'sync_status',
        'peer_group_priority'
    )

    if peer_group_association:
        data = (
            peer_group_association.association_id,
            peer_group_association.peer_group_id,
            peer_group_association.system_peer_id,
            peer_group_association.association_type,
            peer_group_association.sync_status,
            peer_group_association.peer_group_priority
        )

    else:
        data = (tuple('<none>' for _ in range(len(columns))),)

    return columns, data


def detail_association_format(peer_group_association=None):
    # Include all the fields in detail_association_format
    # plus some additional fields
    columns = (
        'id',
        'peer_group_id',
        'system_peer_id',
        'association_type',
        'sync_status',
        'peer_group_priority',
        'sync_message',
        'created_at',
        'updated_at',
    )

    if peer_group_association:
        data = (
            peer_group_association.association_id,
            peer_group_association.peer_group_id,
            peer_group_association.system_peer_id,
            peer_group_association.association_type,
            peer_group_association.sync_status,
            peer_group_association.peer_group_priority,
            peer_group_association.sync_message,
            peer_group_association.created_at,
            peer_group_association.updated_at
        )
    else:
        data = (tuple('<none>' for _ in range(len(columns))),)

    return columns, data


class AddPeerGroupAssociation(base.DCManagerShowOne):
    """Add a new peer group association."""

    def _get_format_function(self):
        return detail_association_format

    def get_parser(self, prog_name):
        parser = super(AddPeerGroupAssociation, self).get_parser(prog_name)

        parser.add_argument(
            '--peer-group-id',
            required=True,
            help='Subcloud peer group ID.'
        )

        parser.add_argument(
            '--system-peer-id',
            required=True,
            help='System Peer ID.'
        )

        parser.add_argument(
            '--peer-group-priority',
            required=True,
            type=int,
            help='Priority of this peer group.'
        )
        return parser

    def _get_resources(self, parsed_args):
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager
        kwargs = {
            'peer_group_id': parsed_args.peer_group_id,
            'system_peer_id': parsed_args.system_peer_id,
            'peer_group_priority': parsed_args.peer_group_priority
        }
        return dcmanager_client.peer_group_association_manager.\
            add_peer_group_association(**kwargs)


class ListPeerGroupAssociation(base.DCManagerLister):
    """List peer group associations."""

    def _get_format_function(self):
        return association_format

    def get_parser(self, prog_name):
        parser = super(ListPeerGroupAssociation, self).get_parser(prog_name)
        return parser

    def _get_resources(self, parsed_args):
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager
        return dcmanager_client.peer_group_association_manager.\
            list_peer_group_associations()


class ShowPeerGroupAssociation(base.DCManagerShowOne):
    """Show the details of a peer group association."""

    def _get_format_function(self):
        return detail_association_format

    def get_parser(self, prog_name):
        parser = super(ShowPeerGroupAssociation, self).get_parser(prog_name)

        parser.add_argument(
            'id',
            help='ID of the peer group association to view the details.'
        )

        return parser

    def _get_resources(self, parsed_args):
        association_ref = parsed_args.id
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager
        return dcmanager_client.peer_group_association_manager.\
            peer_group_association_detail(association_ref)


class SyncPeerGroupAssociation(base.DCManagerShowOne):
    """Sync the subcloud peer group to peer site."""

    def _get_format_function(self):
        return detail_association_format

    def get_parser(self, prog_name):
        parser = super(SyncPeerGroupAssociation, self).get_parser(prog_name)

        parser.add_argument(
            'id',
            help='ID of the peer group association to sync.'
        )

        return parser

    def _get_resources(self, parsed_args):
        association_ref = parsed_args.id
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager
        return dcmanager_client.peer_group_association_manager.\
            sync_peer_group_association(association_ref)


class DeletePeerGroupAssociation(command.Command):
    """Delete peer group association from the database."""

    def get_parser(self, prog_name):
        parser = super(DeletePeerGroupAssociation, self).get_parser(prog_name)

        parser.add_argument(
            'id',
            help='ID of the peer group association to delete.'
        )
        return parser

    def take_action(self, parsed_args):
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager
        try:
            dcmanager_client.peer_group_association_manager.\
                delete_peer_group_association(parsed_args.id)
        except Exception as e:
            print(e)
            msg = "Unable to delete peer group association %s" % (
                parsed_args.id)
            raise exceptions.DCManagerClientException(msg)


class UpdatePeerGroupAssociation(base.DCManagerShowOne):
    """Update attributes of a peer group association."""

    def _get_format_function(self):
        return detail_association_format

    def get_parser(self, prog_name):
        parser = super(UpdatePeerGroupAssociation, self).get_parser(prog_name)

        parser.add_argument(
            'id',
            help='ID of the peer group association to update.'
        )

        parser.add_argument(
            '--peer-group-priority',
            required=True,
            type=int,
            help='Priority of the subcloud peer group in this association.'
        )
        return parser

    def _get_resources(self, parsed_args):
        dcmanager_client = self.app.client_manager.\
            peer_group_association_manager

        kwargs = {'peer_group_priority': parsed_args.peer_group_priority}
        try:
            return dcmanager_client.peer_group_association_manager.\
                update_peer_group_association(parsed_args.id, **kwargs)
        except Exception as e:
            print(e)
            msg = "Unable to update peer group association %s" % (
                parsed_args.id)
            raise exceptions.DCManagerClientException(msg)