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

import json

from requests_toolbelt import MultipartEncoder

from dcmanagerclient.api import base
from dcmanagerclient.api.base import get_json


class SubcloudBackupManager(base.ResourceManager):

    resource_class = base.Subcloud

    def json_to_resource(self, json_object):
        return self.resource_class.from_payload(self, json_object)

    def subcloud_backup_create(self, url, files, data):
        fields = {}
        if files:
            for k, v in files.items():
                with open(v, "rb") as file:
                    fields.update({k: (v, file)})
        fields.update(data)
        enc = MultipartEncoder(fields=fields)
        headers = {"content-type": enc.content_type}
        resp = self.http_client.post(url, enc, headers=headers)

        if resp.status_code != 200:
            self._raise_api_exception(resp)
        json_response_key = get_json(resp)
        json_objects = json_response_key["subclouds"]
        resource = []
        for json_object in json_objects:
            resource.append(self.json_to_resource(json_object))
        return resource

    def subcloud_backup_delete(self, url, data):
        fields = {}
        fields.update(data)
        enc = MultipartEncoder(fields=fields)
        headers = {"content-type": enc.content_type}

        resp = self.http_client.patch(url, enc, headers=headers)
        if resp.status_code not in {204, 207}:
            self._raise_api_exception(resp)
        elif resp.status_code == 207:
            return json.loads(resp.content)
        return None

    def subcloud_backup_restore(self, url, files, data):
        fields = {}
        if files:
            for k, v in files.items():
                with open(v, "rb") as file:
                    fields.update({k: (v, file)})
        fields.update(data)
        enc = MultipartEncoder(fields=fields)
        headers = {"content-type": enc.content_type}

        resp = self.http_client.patch(url, enc, headers=headers)

        if resp.status_code != 200:
            self._raise_api_exception(resp)

        json_response_key = get_json(resp)
        json_objects = json_response_key["subclouds"]
        resource = []
        for json_object in json_objects:
            resource.append(self.json_to_resource(json_object))
        return resource

    def backup_subcloud_create(self, **kwargs):
        files = kwargs.get("files")
        data = kwargs.get("data")
        url = "/subcloud-backup/"
        return self.subcloud_backup_create(url, files, data)

    def backup_subcloud_delete(self, release_version, **kwargs):
        data = kwargs.get("data")
        url = f"/subcloud-backup/delete/{release_version}"
        return self.subcloud_backup_delete(url, data)

    def backup_subcloud_restore(self, **kwargs):
        files = kwargs.get("files")
        data = kwargs.get("data")
        url = "/subcloud-backup/restore"
        return self.subcloud_backup_restore(url, files, data)