# Copyright (C) 2014 Andrey Antukh <niwi@niwi.be>
# Copyright (C) 2014 Jesús Espino <jespinog@gmail.com>
# Copyright (C) 2014 David Barragán <bameda@dbarragan.com>
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from taiga.timeline.service import register_timeline_implementation
from . import service

@register_timeline_implementation("projects.project", "create")
@register_timeline_implementation("projects.project", "change")
@register_timeline_implementation("projects.project", "delete")
def project_timeline(instance, extra_data={}):
    result ={
        "project": service.extract_project_info(instance),
    }
    result.update(extra_data)
    return result


@register_timeline_implementation("milestones.milestone", "create")
@register_timeline_implementation("milestones.milestone", "change")
@register_timeline_implementation("milestones.milestone", "delete")
def project_timeline(instance, extra_data={}):
    result ={
        "milestone": service.extract_milestone_info(instance),
        "project": service.extract_project_info(instance.project),
    }
    result.update(extra_data)
    return result


@register_timeline_implementation("userstories.userstory", "create")
@register_timeline_implementation("userstories.userstory", "change")
@register_timeline_implementation("userstories.userstory", "delete")
def userstory_timeline(instance, extra_data={}):
    result ={
        "userstory": service.extract_userstory_info(instance),
        "project": service.extract_project_info(instance.project),
    }

    if instance.milestone is not None:
        result["milestone"] = service.extract_milestone_info(instance.milestone)

    result.update(extra_data)
    return result


@register_timeline_implementation("issues.issue", "create")
@register_timeline_implementation("issues.issue", "change")
@register_timeline_implementation("issues.issue", "delete")
def issue_timeline(instance, extra_data={}):
    result ={
        "issue": service.extract_issue_info(instance),
        "project": service.extract_project_info(instance.project),
    }
    result.update(extra_data)
    return result


@register_timeline_implementation("tasks.task", "create")
@register_timeline_implementation("tasks.task", "change")
@register_timeline_implementation("tasks.task", "delete")
def task_timeline(instance, extra_data={}):
    result ={
        "task": service.extract_task_info(instance),
        "project": service.extract_project_info(instance.project),
    }

    if instance.user_story:
        result["task"]["userstory"] = service.extract_userstory_info(instance.user_story)

    result.update(extra_data)
    return result

@register_timeline_implementation("wiki.wikipage", "create")
@register_timeline_implementation("wiki.wikipage", "change")
@register_timeline_implementation("wiki.wikipage", "delete")
def wiki_page_timeline(instance, extra_data={}):
    result ={
        "wikipage": service.extract_wiki_page_info(instance),
        "project": service.extract_project_info(instance.project),
    }
    result.update(extra_data)
    return result


@register_timeline_implementation("projects.membership", "create")
@register_timeline_implementation("projects.membership", "delete")
def membership_timeline(instance, extra_data={}):
    result =  {
        "user": service.extract_user_info(instance.user),
        "project": service.extract_project_info(instance.project),
        "role": service.extract_role_info(instance.role),
    }
    result.update(extra_data)
    return result

@register_timeline_implementation("users.user", "create")
def user_timeline(instance, extra_data={}):
    result =  {
        "user": service.extract_user_info(instance),
    }
    result.update(extra_data)
    return result