from rest_framework import serializers
from rest_polymorphic.serializers import PolymorphicSerializer

import seminar.models as m
from seminar import treelib

DEFAULT_NODE_DEPTH = 2

class TextSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.Text
		fields = '__all__'

class ProblemSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.Problem
		fields = '__all__'

class UlohaVzorakNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.UlohaVzorakNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH
	
class UlohaZadaniNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.UlohaZadaniNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class RocnikNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.RocnikNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class CisloNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.CisloNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class MezicisloNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.MezicisloNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class TemaVCisleNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.TemaVCisleNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class OrgTextNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.OrgTextNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class PohadkaNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.PohadkaNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class TextNodeSerializer(serializers.ModelSerializer):
	text = TextSerializer()

	class Meta:
		model = m.TextNode
		fields = ('id','text','polymorphic_ctype')
		depth = DEFAULT_NODE_DEPTH

class TextNodeWriteSerializer(serializers.ModelSerializer):
	text = TextSerializer()

	def update(self,node,validated_data):
		node.text.na_web = validated_data.get('text').get('na_web')
		node.text.save()
		return node

	class Meta:
		model = m.TextNode
		fields = ('id','text')
		depth = DEFAULT_NODE_DEPTH

class TextNodeCreateSerializer(serializers.ModelSerializer):
	text = TextSerializer()
	refnode = serializers.CharField()
	where = serializers.CharField()

	def create(self,validated_data):
		temp_text = validated_data.pop('text')
		where = validated_data.pop('where')
		refnode_id = validated_data.pop('refnode')
		refnode = m.TreeNode.objects.get(pk=refnode_id)
		text = m.Text.objects.create(**temp_text)
		if where == 'syn':
			node = treelib.create_child(refnode,m.TextNode,text=text)
		elif where == 'za':
			node = treelib.create_node_after(refnode,m.TextNode,text=text)
		elif where == 'pred':
			node = treelib.create_node_before(refnode,m.TextNode,text=text)
		node.where = None
		node.refnode = None
		return node

	class Meta:
		model = m.TextNode
		fields = ('text','where','refnode')
		depth = DEFAULT_NODE_DEPTH

class CastNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.CastNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH

class CastNodeCreateSerializer(serializers.ModelSerializer):
	refnode = serializers.CharField()
	where = serializers.CharField()

	def create(self,validated_data):
		temp_nadpis = validated_data.pop('nadpis')
		where = validated_data.pop('where')
		refnode_id = validated_data.pop('refnode')
		refnode = m.TreeNode.objects.get(pk=refnode_id)
		if where == 'syn':
			node = treelib.create_child(refnode,m.CastNode,nadpis=temp_nadpis)
		elif where == 'za':
			node = treelib.create_node_after(refnode,m.CastNode,nadpis=temp_nadpis)
		elif where == 'pred':
			node = treelib.create_node_before(refnode,m.CastNode,nadpis=temp_nadpis)
		node.where = None
		node.refnode = None
		return node

	class Meta:
		model = m.CastNode
		fields = ('nadpis','where','refnode')
		depth = DEFAULT_NODE_DEPTH

class ReseniNodeSerializer(serializers.ModelSerializer):
	class Meta:
		model = m.ReseniNode
		fields = '__all__'
		depth = DEFAULT_NODE_DEPTH


class TreeNodeSerializer(PolymorphicSerializer):
	model_serializer_mapping = {
		m.RocnikNode: RocnikNodeSerializer,
		m.CisloNode: CisloNodeSerializer,
		m.MezicisloNode: MezicisloNodeSerializer,
		m.TemaVCisleNode: TemaVCisleNodeSerializer,
		m.OrgTextNode: OrgTextNodeSerializer,
		m.UlohaZadaniNode: UlohaZadaniNodeSerializer,
		m.UlohaVzorakNode: UlohaVzorakNodeSerializer,
		m.PohadkaNode: PohadkaNodeSerializer,
		m.TextNode: TextNodeSerializer,
		m.CastNode: CastNodeSerializer,
		m.ReseniNode: ReseniNodeSerializer,
	}