# Copyright 2013 Thierry Carrez <thierry@openstack.org>
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

from django.contrib.auth.decorators import login_required
from django.views.decorators.http import require_POST
from django.http import HttpResponseRedirect, HttpResponseForbidden
from django.shortcuts import render
from django.contrib.auth.models import User
from projects.models import Project, Milestone, Series
from stories.models import Story, Task, Comment, StoryTag

def dashboard(request):
    return render(request, "stories.dashboard.html")


def view(request, storyid):
    story = Story.objects.get(id=storyid)
    active_series = Series.objects.filter(status__gt=0)
    milestones = Milestone.objects.all()
    return render(request, "stories.view.html", {
                  'story': story,
                  'milestones': milestones,
                  'priorities': Story.STORY_PRIORITIES,
                  'taskstatuses': Task.TASK_STATUSES,
                  'active_series': active_series,
                  })

@login_required
@require_POST
def comment(request, storyid):
    story = Story.objects.get(id=storyid)
    if request.POST.get('comment', False):
        newcomment = Comment(story=story,
                             author=request.user,
                             comment_type="comment",
                             content=request.POST['content'])
        newcomment.save()
    return HttpResponseRedirect('/story/%s' % storyid)

@login_required
@require_POST
def set_priority(request, storyid):
    story = Story.objects.get(id=storyid)
    if 'priority' in request.POST:
        priority = request.POST['priority']
        if int(priority) != story.priority:
            pr = story.get_priority_display()
            story.priority = priority
            story.save()
            # We need to refresh the story to get get_priority_display to work
            story = Story.objects.get(id=storyid)
            msg = "Set priority: %s -> %s" % (pr, story.get_priority_display())
            newcomment = Comment(story=story,
                                 action=msg,
                                 author=request.user,
                                 comment_type="random",
                                 content=request.POST.get('comment', ''))
            newcomment.save()
    return HttpResponseRedirect('/story/%s' % storyid)

@login_required
@require_POST
def add_story(request):
    try:
        newstory = Story(
            title=request.POST['title'],
            description=request.POST['description'],
            creator=request.user,
            priority=0,
            )
        newstory.save()
        proposed_projects = request.POST['projects'].split()
        if proposed_projects:
            series=Series.objects.get(status=2)
            tasks = []
            for project in proposed_projects:
                tasks.append(Task(
                    story=newstory,
                    project=Project.objects.get(name=project),
                    series=series,
                ))
            Task.objects.bulk_create(tasks)
        proposed_tags = set(request.POST['tags'].split())
        if proposed_tags:
            tags = []
            for tag in proposed_tags:
                tags.append(StoryTag(story=newstory, name=tag))
            StoryTag.objects.bulk_create(tags)
        msg = 'Story created (%s)' % newstory.title
        newcomment = Comment(story=newstory,
                             action=msg,
                             author=request.user,
                             comment_type="star-empty",
                             content='')
        newcomment.save()
    except KeyError as e:
        pass
    return HttpResponseRedirect('/story/%s' % newstory.id)

@login_required
@require_POST
def add_task(request, storyid):
    story = Story.objects.get(id=storyid)
    try:
        if request.POST['series']:
            series=Series.objects.get(name=request.POST['series'])
        else:
            series=Series.objects.get(status=2)
        newtask = Task(
            story=story,
            title=request.POST['title'],
            project=Project.objects.get(name=request.POST['project']),
            series=series,
            )
        newtask.save()
        msg = "Added %s/%s task " % (
            newtask.project.name, newtask.series.name)
        newcomment = Comment(story=story,
                             action=msg,
                             author=request.user,
                             comment_type="plus-sign",
                             content=request.POST.get('comment', ''))
        newcomment.save()
    except KeyError as e:
        pass
    return HttpResponseRedirect('/story/%s' % story.id)

@login_required
@require_POST
def edit_task(request, taskid):
    task = Task.objects.get(id=taskid)
    try:
        actions = []
        if (task.title != request.POST['title']):
            actions.append("title")
            task.title = request.POST['title']
        milestone = Milestone.objects.get(id=request.POST['milestone'])
        if (milestone != task.milestone):
            actions.append("milestone -> %s" % milestone.name)
            task.milestone = milestone
        status = request.POST['status']
        if (task.status != status):
            task.status = status
            actions.append("status -> %s" % task.get_status_display())
        assignee = User.objects.get(username=request.POST['assignee'])
        if (assignee != task.assignee):
            actions.append("assignee -> %s" % assignee.username)
            task.assignee = assignee
        if actions:
            msg = "Updated %s/%s task " % (task.project.name, task.series.name)
            msg += ", ".join(actions)
            task.save()
            newcomment = Comment(story=task.story,
                                 action=msg,
                                 author=request.user,
                                 comment_type="tasks",
                                 content=request.POST.get('comment', ''))
            newcomment.save()
    except KeyError as e:
        pass
    return HttpResponseRedirect('/story/%s' % task.story.id)

@login_required
@require_POST
def delete_task(request, taskid):
    task = Task.objects.get(id=taskid)
    task.delete()
    msg = "Deleted %s/%s task" % (task.project.name, task.series.name)
    newcomment = Comment(story=task.story,
                         action=msg,
                         author=request.user,
                         comment_type="remove-sign",
                         content=request.POST.get('comment', ''))
    newcomment.save()
    return HttpResponseRedirect('/story/%s' % task.story.id)

@login_required
@require_POST
def edit_story(request, storyid):
    story = Story.objects.get(id=storyid)
    storytags = set(x.name for x in StoryTag.objects.filter(story=story))
    onlytags = True
    try:
        actions = []
        if (story.title != request.POST['title']):
            onlytags = False
            actions.append("title")
            story.title = request.POST['title']
        if (story.description != request.POST['description']):
            onlytags = False
            actions.append("description")
            story.description = request.POST['description']
        proposed_tags = set(request.POST['tags'].split())
        if proposed_tags != storytags:
            actions.append("tags")
            StoryTag.objects.filter(story=story).delete()
            tags = []
            for tag in proposed_tags:
                tags.append(StoryTag(story=story, name=tag))
            StoryTag.objects.bulk_create(tags)
        if actions:
            msg = "Updated story " + ", ".join(actions)
            story.save()
            if onlytags:
                comment_type = "tags"
            else:
                comment_type = "align-left"
            newcomment = Comment(story=story,
                                 action=msg,
                                 author=request.user,
                                 comment_type=comment_type)
            newcomment.save()
    except KeyError as e:
        print e
    return HttpResponseRedirect('/story/%s' % story.id)