# Copyright 2015 VMware, Inc.  All rights reserved.
#
#    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.

import logging

from tabulate import tabulate

from oslo_config import cfg
from oslo_serialization import jsonutils

from vmware_nsx._i18n import _LI

LOG = logging.getLogger(__name__)


def output_formatter(resource_name, resources_list, attrs):
    """Method to format the output response from NSX/Neutron.

    Depending on the --fmt cli option we format the output as
    JSON or as a table.
    """
    LOG.info(_LI('%(resource_name)s'), {'resource_name': resource_name})
    if not resources_list:
        LOG.info(_LI('No resources found'))
        return ''

    fmt = cfg.CONF.fmt
    if fmt == 'psql':
        resource_attr_values = []
        for resource in resources_list:
            resource_list = []
            for attr in attrs:
                resource_list.append(resource.get(attr))
            resource_attr_values.append(resource_list)
        return tabulate(resource_attr_values, attrs, tablefmt=fmt)

    elif fmt == 'json':
        js_output = {}
        js_output[resource_name] = []
        for resource in resources_list:
            result = {}
            for attr in attrs:
                result[attr] = resource[attr]
            js_output[resource_name].append(result)
        return jsonutils.dumps(js_output, sort_keys=True, indent=4)