# (c) Copyright 2014-2016 Hewlett-Packard Development Company, L.P.
#
# 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 datetime
import logging
import pprint

from freezerclient import base
from freezerclient import exceptions
from freezerclient import utils


logging = logging.getLogger(__name__)


def format_backup(backup):
    column = (
        'Backup ID',
        'Project ID',
        'User ID',
        'Status',
        'Metadata'
    )
    data = (
        backup.get('backup_uuid'),
        backup.get('project_id'),
        backup.get('user_id'),
        backup.get('status'),
        pprint.pformat(backup.get('backup_metadata'))
    )
    return column, data


class BackupShow(base.FreezerShowOne):
    """Show the metadata of a single backup"""
    def get_parser(self, prog_name):
        parser = super(BackupShow, self).get_parser(prog_name)
        parser.add_argument(dest='backup_uuid',
                            help='ID of the backup')
        return parser

    def take_action(self, parsed_args):
        backup = self.client.backups.get(parsed_args.backup_uuid)
        if not backup:
            raise exceptions.ApiClientException('Backup not found')

        return format_backup(backup)


class BackupList(base.FreezerLister):
    """List all backups for your user"""
    def get_parser(self, prog_name):
        parser = super(BackupList, self).get_parser(prog_name)

        parser.add_argument(
            '--limit',
            dest='limit',
            default=100,
            help='Specify a limit for search query',
        )

        parser.add_argument(
            '--offset',
            dest='offset',
            default=0,
            help='',
        )

        parser.add_argument(
            '--search',
            dest='search',
            default='',
            help='Define a filter for the query',
        )
        return parser

    def take_action(self, parsed_args):
        search = utils.prepare_search(parsed_args.search)

        backups = self.client.backups.list(limit=parsed_args.limit,
                                           offset=parsed_args.offset,
                                           search=search)

        columns = ('Backup ID', 'Backup UUID', 'Hostname', 'Path',
                   'Created at', 'Status', 'Level')

        # Print empty table if no backups found
        if not backups:
            backups = [{}]

        # sort by the time of backup task is created
        backups_l = sorted(backups,
                           key=lambda x: x.get('backup_metadata',
                                               {}).get('time_stamp', ''))

        data = ((b.get('backup_id', ''),
                 b.get('backup_uuid', ''),
                 b.get('backup_metadata', {}).get('hostname', ''),
                 b.get('backup_metadata', {}).get('path_to_backup', ''),
                 datetime.datetime.fromtimestamp(
                 int(b.get('backup_metadata', {}).get('time_stamp', '')))
                 if b.get('backup_metadata') else '',
                 b.get('status', ''),
                 b.get('backup_metadata', {}).get('curr_backup_level', '')
                 ) for b in backups_l)

        return columns, data


class BackupDelete(base.FreezerCommand):
    """Delete a backup from the api"""
    def get_parser(self, prog_name):
        parser = super(BackupDelete, self).get_parser(prog_name)
        parser.add_argument(dest='backup_uuid',
                            help='UUID of the backup')
        return parser

    def take_action(self, parsed_args):
        self.client.backups.delete(parsed_args.backup_uuid)


class BackupCreate(base.FreezerShowOne):
    """Create an backup from a file"""
    def get_parser(self, prog_name):
        parser = super(BackupCreate, self).get_parser(prog_name)
        parser.add_argument('--file',
                            dest='file',
                            required=True,
                            help='Path to json file with the backup')
        return parser

    def take_action(self, parsed_args):
        backup_metadata = utils.doc_from_json_file(parsed_args.file)
        backup_id = self.client.backups.create(backup_metadata)
        backup = self.client.backups.get(backup_id)
        if not backup:
            raise exceptions.ApiClientException('Backup created but not found')
        return format_backup(backup)
