# -*- coding: utf-8 -*-

# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301  USA
import json
import uuid
from ..operators import Attrib
from .dict import DictionarySerializer

class JSONSerializer(DictionarySerializer):

    def serialize_workflow_spec(self, wf_spec, **kwargs):
        thedict = super(JSONSerializer, self).serialize_workflow_spec(
            wf_spec, **kwargs)
        return self._dumps(thedict)

    def deserialize_workflow_spec(self, s_state, **kwargs):
        thedict = self._loads(s_state)
        return super(JSONSerializer, self).deserialize_workflow_spec(
            thedict, **kwargs)

    def serialize_workflow(self, workflow, **kwargs):
        thedict = super(JSONSerializer, self).serialize_workflow(
            workflow, **kwargs)
        return self._dumps(thedict)

    def deserialize_workflow(self, s_state, **kwargs):
        thedict = self._loads(s_state)
        return super(JSONSerializer, self).deserialize_workflow(
            thedict, **kwargs)

    def _object_hook(self, dct):
        if '__uuid__' in dct:
            return uuid.UUID(dct['__uuid__'])

        if '__bytes__' in dct:
            return dct['__bytes__'].encode('ascii')

        if '__attrib__' in dct:
            return Attrib(dct['__attrib__'])

        return dct

    def _default(self, obj):
        if isinstance(obj, uuid.UUID):
            return {'__uuid__': obj.hex}

        if isinstance(obj, bytes):
            return {'__bytes__': obj.decode('ascii')}

        if isinstance(obj, Attrib):
            return {'__attrib__': obj.name}

        raise TypeError('%r is not JSON serializable' % obj)

    def _loads(self, text):
        return json.loads(text, object_hook=lambda o: self._object_hook(o))

    def _dumps(self, dct):
        return json.dumps(dct, sort_keys=True, default=lambda o:
                self._default(o))