# Copyright 2012 OpenStack Foundation
# 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 copy
import six

from escalatorclient.common import utils
from escalatorclient.openstack.common.apiclient import base

CREATE_PARAMS = (
    'cluster_id', 'version_id', 'version_patch_id', 'hosts',
    'update_object', 'update_script')

OS_REQ_ID_HDR = 'x-openstack-request-id'


class Update(base.Resource):
    def __repr__(self):
        return "<Update %s>" % self._info

    def update(self, **fields):
        self.manager.update(self, **fields)

    def data(self, **kwargs):
        return self.manager.data(self, **kwargs)


class UpdateManager(base.ManagerWithFind):
    resource_class = Update

    def _Update_meta_to_headers(self, fields):
        headers = {}
        fields_copy = copy.deepcopy(fields)

        # NOTE(flaper87): Convert to str, headers
        # that are not instance of basestring. All
        # headers will be encoded later, before the
        # request is sent.

        for key, value in six.iteritems(fields_copy):
            headers['%s' % key] = utils.to_str(value)
        return headers

    @staticmethod
    def _format_update_meta_for_user(meta):
        for key in ['size', 'min_ram', 'min_disk']:
            if key in meta:
                try:
                    meta[key] = int(meta[key]) if meta[key] else 0
                except ValueError:
                    pass
        return meta

    def list(self, **kwargs):
        pass

    def query_progress(self, **kwargs):
        fields = {}
        for field in kwargs:
            if field in CREATE_PARAMS:
                fields[field] = kwargs[field]
            else:
                msg = 'update() got an unexpected argument \'%s\''
                raise TypeError(msg % field)

            if "cluster" in fields:
                url = '/v1/update/%s' % fields['cluster_id']

        resp, body = self.client.get(url)
        return Update(self, self._format_update_meta_for_user(body))

    def update(self, **kwargs):
        """Update a cluster

        TODO(bcwaldon): document accepted params
        """

        fields = {}
        for field in kwargs:
            if field in CREATE_PARAMS:
                fields[field] = kwargs[field]
            elif field == 'return_req_id':
                continue
            else:
                msg = 'update() got an unexpected argument \'%s\''
                raise TypeError(msg % field)

        if "cluster_id" in fields:
            url = '/v1/update/%s' % fields['cluster_id']

            hdrs = self._Update_meta_to_headers(fields)
            resp, body = self.client.post(url, headers=None, data=hdrs)
            return Update(self, self._format_update_meta_for_user(body))