import json
import jsonpatch

from tomviz import operators, modules

from ._pipeline import PipelineStateManager

from ._utils import to_namespaces

op_class_attrs = ['description', 'label', 'script', 'type']


class InvalidStateError(RuntimeError):
    pass


class Base(object):
    def __init__(self, **kwargs):
        for k, v in kwargs.items():
            setattr(self, k, v)


class Mortal(object):
    _dead = False

    def _kill(self):
        self._dead = True

    def __getattribute__(self, item):
        # Make an exception for the attribute we need to check or we will go
        # recursive!
        if item == '_dead':
            return super(Mortal, self).__getattribute__(item)

        if self._dead:
            raise InvalidStateError(
                "'%s' no longer represents a valid pipeline element." % self)
        else:
            return super(Mortal, self).__getattribute__(item)


class OperatorMeta(type):
    def __init__(cls, name, bases, dict):
        attrs = {}

        for k, v in dict.items():
            if k not in op_class_attrs:
                attrs[k] = property(lambda self: getattr(self, k),
                                    lambda self, value: setattr(self, k, value))
            else:
                attrs[k] = v

        super(OperatorMeta, cls).__init__(name, bases, attrs)


class ModuleMeta(type):
    def __new__(meta, name, parents, dct):
        attrs = {}

        attrs = {
            '_props': dct
        }

        return super(ModuleMeta, meta).__new__(meta, name, parents, attrs)


class Pipeline(Mortal):
    def __init__(self, datasource):
        self._datasource = datasource

    @property
    def dataSource(self):
        return self._datasource

    def _ds_path(self):
        from . import _pipeline_index
        pipeline_index = _pipeline_index(self._datasource)
        if pipeline_index < 0:
            raise Exception('Pipeline is not valid.')

        return '/dataSources/%d' % pipeline_index

    def pause(self):
        PipelineStateManager().pause_pipeline(self._ds_path())

    def resume(self):
        PipelineStateManager().resume_pipeline(self._ds_path())

    def execute(self):
        paused = self.paused()
        if paused:
            self.resume()

        PipelineStateManager().execute_pipeline(self._ds_path())

        if paused:
            self.pause()

    def paused(self):
        return PipelineStateManager().pipeline_paused(self._ds_path())


class Reader(Base, Mortal):
    pass


class DataSource(Base, Mortal):
    def __init__(self, **kwargs):
        self.reader = Reader(
            name=kwargs.pop('name', None),
            fileNames=kwargs.pop('fileNames', []),
            subsampleSettings=kwargs.pop('subsampleSettings', {}))

        self.operators = []
        self.modules = [modules.Outline(), modules.Slice()]
        super(DataSource, self).__init__(**kwargs)


class Operator(Base, Mortal):
    pass


class Module(Mortal):
    def __init__(self, **kwargs):
        args = self._props
        args.update(kwargs)

        for k, v in args.items():
            if isinstance(v, dict):
                v = to_namespaces(v)
            setattr(self, k, v)

    def _updates(self):
        from ._schemata import dump_module
        current_state = dump_module(self)
        patch = jsonpatch.JsonPatch.from_diff(self._props, current_state)
        patch = json.loads(patch.to_string())

        return patch


class Tomviz(object):
    def __init__(self, pipelines=None):
        self.pipelines = pipelines if pipelines else []


def module_json_to_classes(module_json):
    for name, info in module_json.items():
        info['type'] = name
        del info['id']
        # Default visibility to true
        info['properties']['visibility'] = True
        if not hasattr(modules, name):
            cls = ModuleMeta(name, (Module,), info)
            setattr(modules, name, cls)


def operator_json_to_classes(operator_json):
    for name, info in operator_json.items():
        del info['id']
        if not hasattr(operators, name):
            cls = OperatorMeta(name, (Operator,), info)
            setattr(operators, name, cls)


def init_modules():
    module_json_str = PipelineStateManager().module_json()
    module_json = json.loads(module_json_str)
    module_json_to_classes(module_json)


def init_operators():
    operator_json_str = PipelineStateManager().operator_json()
    operator_json = json.loads(operator_json_str)
    operator_json_to_classes(operator_json)

import json
import jsonpatch

from tomviz import operators, modules

from ._pipeline import PipelineStateManager

from ._utils import to_namespaces

op_class_attrs = ['description', 'label', 'script', 'type']


class InvalidStateError(RuntimeError):
    pass


class Base(object):
    def __init__(self, **kwargs):
        for k, v in kwargs.items():
            setattr(self, k, v)


class Mortal(object):
    _dead = False

    def _kill(self):
        self._dead = True

    def __getattribute__(self, item):
        # Make an exception for the attribute we need to check or we will go
        # recursive!
        if item == '_dead':
            return super(Mortal, self).__getattribute__(item)

        if self._dead:
            raise InvalidStateError(
                "'%s' no longer represents a valid pipeline element." % self)
        else:
            return super(Mortal, self).__getattribute__(item)


class OperatorMeta(type):
    def __init__(cls, name, bases, dict):
        attrs = {}

        for k, v in dict.items():
            if k not in op_class_attrs:
                attrs[k] = property(lambda self: getattr(self, k),
                                    lambda self, value: setattr(self, k, value))
            else:
                attrs[k] = v

        super(OperatorMeta, cls).__init__(name, bases, attrs)


class ModuleMeta(type):
    def __new__(meta, name, parents, dct):
        attrs = {}

        attrs = {
            '_props': dct
        }

        return super(ModuleMeta, meta).__new__(meta, name, parents, attrs)


class Pipeline(Mortal):
    def __init__(self, datasource):
        self._datasource = datasource

    @property
    def dataSource(self):
        return self._datasource

    def _ds_path(self):
        from . import _pipeline_index
        pipeline_index = _pipeline_index(self._datasource)
        if pipeline_index < 0:
            raise Exception('Pipeline is not valid.')

        return '/dataSources/%d' % pipeline_index

    def pause(self):
        PipelineStateManager().pause_pipeline(self._ds_path())

    def resume(self):
        PipelineStateManager().resume_pipeline(self._ds_path())

    def execute(self):
        paused = self.paused()
        if paused:
            self.resume()

        PipelineStateManager().execute_pipeline(self._ds_path())

        if paused:
            self.pause()

    def paused(self):
        return PipelineStateManager().pipeline_paused(self._ds_path())


class Reader(Base, Mortal):
    pass


class DataSource(Base, Mortal):
    def __init__(self, **kwargs):
        self.reader = Reader(
            name=kwargs.pop('name', None),
            fileNames=kwargs.pop('fileNames', []),
            subsampleSettings=kwargs.pop('subsampleSettings', {}))

        self.operators = []
        self.modules = [modules.Outline(), modules.Slice()]
        super(DataSource, self).__init__(**kwargs)


class Operator(Base, Mortal):
    pass


class Module(Mortal):
    def __init__(self, **kwargs):
        args = self._props
        args.update(kwargs)

        for k, v in args.items():
            if isinstance(v, dict):
                v = to_namespaces(v)
            setattr(self, k, v)

    def _updates(self):
        from ._schemata import dump_module
        current_state = dump_module(self)
        patch = jsonpatch.JsonPatch.from_diff(self._props, current_state)
        patch = json.loads(patch.to_string())

        return patch


class Tomviz(object):
    def __init__(self, pipelines=None):
        self.pipelines = pipelines if pipelines else []


def module_json_to_classes(module_json):
    for name, info in module_json.items():
        info['type'] = name
        del info['id']
        # Default visibility to true
        info['properties']['visibility'] = True
        if not hasattr(modules, name):
            cls = ModuleMeta(name, (Module,), info)
            setattr(modules, name, cls)


def operator_json_to_classes(operator_json):
    for name, info in operator_json.items():
        del info['id']
        if not hasattr(operators, name):
            cls = OperatorMeta(name, (Operator,), info)
            setattr(operators, name, cls)


def init_modules():
    module_json_str = PipelineStateManager().module_json()
    module_json = json.loads(module_json_str)
    module_json_to_classes(module_json)


def init_operators():
    operator_json_str = PipelineStateManager().operator_json()
    operator_json = json.loads(operator_json_str)
    operator_json_to_classes(operator_json)

# Make an exception for the attribute we need to check or we will go # recursive! # Default visibility to true