"""
Copyright (C) 2023 Adobe.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

# file: presets/manager.py
# brief: Manager for preset related operations
# author Adobe - 3D & Immersive
# copyright 2023 Adobe Inc. All rights reserved.


import xml.etree.ElementTree as ET

import traceback

from ..utils import SUBSTANCE_Utils
from ..common import Code_Response, PRESET_EXTENSION


class SUBSTANCE_PresetManager():
    def __init__(self):
        pass

    def preset_write_file(self, filepath, preset):
        try:
            _dir_path = filepath + preset.name + PRESET_EXTENSION
            _root = ET.Element('sbspresets')
            _root.set('formatversion', '1.1')
            _root.set('count', '1')

            _preset_xml = ET.fromstring(preset.value)

            _root.append(_preset_xml)

            _tree = ET.ElementTree(_root)
            ET.indent(_tree, space=" ", level=0)
            _tree.write(_dir_path)
            return Code_Response.success

        except Exception:
            SUBSTANCE_Utils.log_data("ERROR", "Exception - Writing preset error:")
            SUBSTANCE_Utils.log_traceback(traceback.format_exc())

            return Code_Response.preset_export_error

    def preset_read_file(self, filepath):
        try:
            with open(filepath) as _f:
                _preset_file = _f.read()
            _root = ET.fromstring(_preset_file)

            _obj = {}

            for _child in _root.findall('sbspreset'):
                _preset_value = ET.tostring(_child, encoding='unicode')

                _graph = _child.attrib['pkgurl'].split('?')[0]
                _graph = _graph.replace("pkg:///", "")
                _graph = _graph.replace("pkg://", "")

                if _graph not in _obj:
                    _obj[_graph] = []

                _obj[_graph].append({
                    "label": _child.attrib['label'],
                    "value": _preset_value
                })

            return (Code_Response.success, _obj)

        except Exception:
            SUBSTANCE_Utils.log_data("ERROR", "Exception - Reading preset error:")
            SUBSTANCE_Utils.log_traceback(traceback.format_exc())

            return (Code_Response.preset_import_error, None)