# Copyright (c) 2013 AnsibleWorks, Inc.
#
# This file is part of Ansible Commander.
# 
# Ansible Commander 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, version 3 of the License. 
#
# Ansible Commander 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 Ansible Commander. If not, see <http://www.gnu.org/licenses/>.

from django.contrib.auth.models import User
from lib.main.models import *
from rest_framework import serializers, pagination
from django.core.urlresolvers import reverse
from django.core.serializers import json

class BaseSerializer(serializers.ModelSerializer):
    pass

class OrganizationSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    # make certain fields read only
    creation_date = serializers.DateTimeField(read_only=True) # FIXME: is model Date or DateTime, fix model
    active        = serializers.BooleanField(read_only=True)

    class Meta:
        model = Organization
        fields = ('url', 'id', 'name', 'description', 'creation_date', 'related') # whitelist

    def get_related(self, obj):
        ''' related resource URLs '''

        res = dict(
            audit_trail = reverse('main:organizations_audit_trail_list',    args=(obj.pk,)),
            projects    = reverse('main:organizations_projects_list',       args=(obj.pk,)),
            inventories = reverse('main:organizations_inventories_list',    args=(obj.pk,)),
            users       = reverse('main:organizations_users_list',          args=(obj.pk,)),
            admins      = reverse('main:organizations_admins_list',         args=(obj.pk,)),
            tags        = reverse('main:organizations_tags_list',           args=(obj.pk,)),
            teams       = reverse('main:organizations_teams_list',          args=(obj.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))

        return res

class AuditTrailSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = AuditTrail
        fields = ('url', 'id', 'modified_by', 'delta', 'detail', 'comment')

    def get_related(self, obj):
        res = dict()
        if obj.modified_by:
            res['modified_by']  = reverse('main:users_detail', args=(obj.modified_by.pk,))
        return res

class ProjectSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Project
        fields = ('url', 'id', 'name', 'description', 'creation_date', 'local_path')#, 'default_playbook', 'scm_type')

    def get_related(self, obj):
        res = dict(
            organizations = reverse('main:projects_organizations_list', args=(obj.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res


class InventorySerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Inventory
        fields = ('url', 'id', 'name', 'description', 'creation_date', 'related', 'organization')

    def get_related(self, obj):
        res = dict(
            hosts        = reverse('main:inventory_hosts_list',           args=(obj.pk,)),
            groups       = reverse('main:inventory_groups_list',          args=(obj.pk,)),
            organization = reverse('main:organizations_detail', args=(obj.organization.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class HostSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Host
        fields = ('url', 'id', 'name', 'description', 'creation_date', 'related', 'inventory')

    def get_related(self, obj):
        res = dict(
            variable_data = reverse('main:hosts_variable_detail', args=(obj.pk,)),
            inventory     = reverse('main:inventory_detail',      args=(obj.inventory.pk,)),
        )
        # NICE TO HAVE: possible reverse resource to show what groups the host is in
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class GroupSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Group
        fields = ('url', 'id', 'name', 'description', 'creation_date', 'inventory')

    def get_related(self, obj):
        res = dict(
            variable_data = reverse('main:groups_variable_detail', args=(obj.pk,)),
            hosts         = reverse('main:groups_hosts_list',      args=(obj.pk,)),
            children      = reverse('main:groups_children_list',   args=(obj.pk,)),
            all_hosts     = reverse('main:groups_all_hosts_list',  args=(obj.pk,)),
            inventory     = reverse('main:inventory_detail',       args=(obj.inventory.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class TeamSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Team
        fields = ('url', 'id', 'related', 'name', 'description', 'organization', 'creation_date')

    # FIXME: TODO: include related collections but also related FK urls

    def get_related(self, obj):
        res = dict(
            projects     = reverse('main:teams_projects_list',    args=(obj.pk,)),
            users        = reverse('main:teams_users_list',       args=(obj.pk,)),
            credentials  = reverse('main:teams_credentials_list', args=(obj.pk,)),
            organization = reverse('main:organizations_detail',   args=(obj.organization.pk,)),
            permissions  = reverse('main:teams_permissions_list', args=(obj.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class PermissionSerializer(BaseSerializer):

    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Permission
        fields = ( 'url', 'id', 'user', 'team', 'name', 'description', 'creation_date',
                   'project', 'inventory', 'permission_type' )
         
    def get_related(self, obj):
        res = dict()
        if obj.user:
            res['user']        = reverse('main:users_detail', args=(obj.user.pk,))
        if obj.team:
            res['team']        = reverse('main:teams_detail', args=(obj.team.pk,))
        if self.project:
            res['project']     = reverse('main:projects_detail', args=(obj.project.pk,)) 
        if self.inventory:
            res['inventory']   = reverse('main:inventory_detail', args=(obj.inventory.pk,))
        if self.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))

class CredentialSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    # FIXME: may want to make some of these filtered based on user accessing
    class Meta:
        model = Credential
        fields = (
            'url', 'id', 'related', 'name', 'description', 'creation_date',
            'ssh_username', 'ssh_password', 'ssh_key_data', 'ssh_key_unlock',
            'sudo_username', 'sudo_password', 'user', 'team',
        )

    def get_related(self, obj):
        # FIXME: no related collections, do want to add user and team if defined
        res = dict(
        )
        if obj.user:
            res['user']        = reverse('main:users_detail', args=(obj.user.pk,))
        if obj.team:
            res['team']        = reverse('main:teams_detail', args=(obj.team.pk,))
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

    def validate(self, attrs):
        ''' some fields cannot be changed once written '''
        if self.object is not None:
            # this is an update
            if self.object.user != attrs['user']:
                raise serializers.ValidationError("user cannot be changed")
            if self.object.team != attrs['team']:
                raise serializers.ValidationError("team cannot be changed")
        return attrs

class UserSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.SerializerMethodField('get_absolute_url_override')
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = User
        fields = ('url', 'id', 'username', 'first_name', 'last_name', 'email', 'is_active', 'is_superuser', 'related')

    def get_related(self, obj):
        return dict(
            teams                  = reverse('main:users_teams_list',               args=(obj.pk,)),
            organizations          = reverse('main:users_organizations_list',       args=(obj.pk,)),
            admin_of_organizations = reverse('main:users_admin_organizations_list', args=(obj.pk,)),
            projects               = reverse('main:users_projects_list',            args=(obj.pk,)),
            credentials            = reverse('main:users_credentials_list',         args=(obj.pk,)),
            permissions            = reverse('main:users_permissions_list',         args=(obj.pk,)),
        )

    def get_absolute_url_override(self, obj):
        return reverse('main:users_detail', args=(obj.pk,))


class TagSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Tag
        fields = ('url', 'id', 'name')

    def get_related(self, obj):
        res = dict()
        return res

class VariableDataSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = VariableData
        fields = ('url', 'id', 'data', 'related', 'name', 'description', 'creation_date')

    def get_related(self, obj):
        # FIXME: add host or group if defined
        res = dict(
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class JobTemplateSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = JobTemplate
        fields = ('url', 'id', 'related', 'name', 'description', 'job_type', 'credential', 'project', 'inventory', 'created_by', 'creation_date')

    def get_related(self, obj):
        # FIXME: fill in once further defined.  related resources, credential, project, inventory, etc
        res = dict(
            credential  = reverse('main:credentials_detail', args=(obj.credential.pk,)),
            project     = reverse('main:projects_detail',    args=(obj.project.pk,)),
            inventory   = reverse('main:inventory_detail',   args=(obj.inventory.pk,)),
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res

class JobSerializer(BaseSerializer):

    # add the URL and related resources
    url           = serializers.CharField(source='get_absolute_url', read_only=True)
    related       = serializers.SerializerMethodField('get_related')

    class Meta:
        model = Job
        fields = ('url', 'id', 'related', 'name', 'description', 'job_type', 'credential', 'project', 'inventory', 'created_by', 'creation_date')

    def get_related(self, obj):
        # FIXME: fill in once further defined.  related resources, credential, project, inventory, etc
        res = dict(
        )
        if obj.created_by:
            res['created_by']  = reverse('main:users_detail', args=(obj.created_by.pk,))
        return res