# Copyright (c) 2009 Google Inc. All rights reserved.

# Copyright (c) 2009 Apple Inc. All rights reserved.

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions are

# met:

# 

#     * Redistributions of source code must retain the above copyright

# notice, this list of conditions and the following disclaimer.

#     * Redistributions in binary form must reproduce the above

# copyright notice, this list of conditions and the following disclaimer

# in the documentation and/or other materials provided with the

# distribution.

#     * Neither the name of Google Inc. nor the names of its

# contributors may be used to endorse or promote products derived from

# this software without specific prior written permission.

# 

# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import os

from optparse import make_option

import webkitpy.tool.steps as steps

from webkitpy.common.checkout.changelog import ChangeLog, view_source_url

from webkitpy.common.system.executive import ScriptError

from webkitpy.tool.commands.abstractsequencedcommand import AbstractSequencedCommand

from webkitpy.tool.commands.stepsequence import StepSequence

from webkitpy.tool.comments import bug_comment_from_commit_text

from webkitpy.tool.grammar import pluralize

from webkitpy.tool.multicommandtool import AbstractDeclarativeCommand

from webkitpy.common.system.deprecated_logging import error, log

class Update(AbstractSequencedCommand):

    name = "update"

    help_text = "Update working copy (used internally)"

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

    ]

class Build(AbstractSequencedCommand):

    name = "build"

    help_text = "Update working copy and build"

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.Build,

    ]

class BuildAndTest(AbstractSequencedCommand):

    name = "build-and-test"

    help_text = "Update working copy, build, and run the tests"

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.Build,

        steps.RunTests,

    ]

class Land(AbstractSequencedCommand):

    name = "land"

    help_text = "Land the current working directory diff and updates the associated bug if any"

    argument_names = "[BUGID]"

    show_in_main_help = True

    steps = [

        steps.EnsureBuildersAreGreen,

        steps.UpdateChangeLogsWithReviewer,

        steps.ValidateReviewer,

        steps.Build,

        steps.RunTests,

        steps.Commit,

        steps.CloseBugForLandDiff,

    ]

    long_help = """land commits the current working copy diff (just as svn or git commit would).

land will NOT build and run the tests before committing, but you can use the --build option for that.

If a bug id is provided, or one can be found in the ChangeLog land will update the bug after committing."""

    def _prepare_state(self, options, args, tool):

        return {

            "bug_id": (args and args[0]) or tool.checkout().bug_id_for_this_commit(options.git_commit),

        }

class LandCowboy(AbstractSequencedCommand):

    name = "land-cowboy"

    help_text = "Prepares a ChangeLog and lands the current working directory diff."

    steps = [

        steps.PrepareChangeLog,

        steps.EditChangeLog,

        steps.ConfirmDiff,

        steps.Build,

        steps.RunTests,

        steps.Commit,

    ]

class AbstractPatchProcessingCommand(AbstractDeclarativeCommand):

    # Subclasses must implement the methods below.  We don't declare them here

    # because we want to be able to implement them with mix-ins.

    #

    # def _fetch_list_of_patches_to_process(self, options, args, tool):

    # def _prepare_to_process(self, options, args, tool):

    @staticmethod

    def _collect_patches_by_bug(patches):

        bugs_to_patches = {}

        for patch in patches:

            bugs_to_patches[patch.bug_id()] = bugs_to_patches.get(patch.bug_id(), []) + [patch]

        return bugs_to_patches

    def execute(self, options, args, tool):

        self._prepare_to_process(options, args, tool)

        patches = self._fetch_list_of_patches_to_process(options, args, tool)

        # It's nice to print out total statistics.

        bugs_to_patches = self._collect_patches_by_bug(patches)

        log("Processing %s from %s." % (pluralize("patch", len(patches)), pluralize("bug", len(bugs_to_patches))))

        for patch in patches:

            self._process_patch(patch, options, args, tool)

class AbstractPatchSequencingCommand(AbstractPatchProcessingCommand):

    prepare_steps = None

    main_steps = None

    def __init__(self):

        options = []

        self._prepare_sequence = StepSequence(self.prepare_steps)

        self._main_sequence = StepSequence(self.main_steps)

        options = sorted(set(self._prepare_sequence.options() + self._main_sequence.options()))

        AbstractPatchProcessingCommand.__init__(self, options)

    def _prepare_to_process(self, options, args, tool):

        self._prepare_sequence.run_and_handle_errors(tool, options)

    def _process_patch(self, patch, options, args, tool):

        state = { "patch" : patch }

        self._main_sequence.run_and_handle_errors(tool, options, state)

class ProcessAttachmentsMixin(object):

    def _fetch_list_of_patches_to_process(self, options, args, tool):

        return map(lambda patch_id: tool.bugs.fetch_attachment(patch_id), args)

class ProcessBugsMixin(object):

    def _fetch_list_of_patches_to_process(self, options, args, tool):

        all_patches = []

        for bug_id in args:

            patches = tool.bugs.fetch_bug(bug_id).reviewed_patches()

            log("%s found on bug %s." % (pluralize("reviewed patch", len(patches)), bug_id))

            all_patches += patches

        return all_patches

class CheckStyle(AbstractPatchSequencingCommand, ProcessAttachmentsMixin):

    name = "check-style"

    help_text = "Run check-webkit-style on the specified attachments"

    argument_names = "ATTACHMENT_ID [ATTACHMENT_IDS]"

    main_steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.ApplyPatch,

        steps.CheckStyle,

    ]

class BuildAttachment(AbstractPatchSequencingCommand, ProcessAttachmentsMixin):

    name = "build-attachment"

    help_text = "Apply and build patches from bugzilla"

    argument_names = "ATTACHMENT_ID [ATTACHMENT_IDS]"

    main_steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.ApplyPatch,

        steps.Build,

    ]

class PostAttachmentToRietveld(AbstractPatchSequencingCommand, ProcessAttachmentsMixin):

    name = "post-attachment-to-rietveld"

    help_text = "Uploads a bugzilla attachment to rietveld"

    arguments_names = "ATTACHMENTID"

    main_steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.ApplyPatch,

        steps.PostCodeReview,

    ]

class AbstractPatchApplyingCommand(AbstractPatchSequencingCommand):

    prepare_steps = [

        steps.EnsureLocalCommitIfNeeded,

        steps.CleanWorkingDirectoryWithLocalCommits,

        steps.Update,

    ]

    main_steps = [

        steps.ApplyPatchWithLocalCommit,

    ]

    long_help = """Updates the working copy.

Downloads and applies the patches, creating local commits if necessary."""

class ApplyAttachment(AbstractPatchApplyingCommand, ProcessAttachmentsMixin):

    name = "apply-attachment"

    help_text = "Apply an attachment to the local working directory"

    argument_names = "ATTACHMENT_ID [ATTACHMENT_IDS]"

    show_in_main_help = True

class ApplyFromBug(AbstractPatchApplyingCommand, ProcessBugsMixin):

    name = "apply-from-bug"

    help_text = "Apply reviewed patches from provided bugs to the local working directory"

    argument_names = "BUGID [BUGIDS]"

    show_in_main_help = True

class AbstractPatchLandingCommand(AbstractPatchSequencingCommand):

    prepare_steps = [

        steps.EnsureBuildersAreGreen,

    ]

    main_steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.ApplyPatch,

        steps.ValidateReviewer,

        steps.Build,

        steps.RunTests,

        steps.Commit,

        steps.ClosePatch,

        steps.CloseBug,

    ]

    long_help = """Checks to make sure builders are green.

Updates the working copy.

Applies the patch.

Builds.

Runs the layout tests.

Commits the patch.

Clears the flags on the patch.

Closes the bug if no patches are marked for review."""

class LandAttachment(AbstractPatchLandingCommand, ProcessAttachmentsMixin):

    name = "land-attachment"

    help_text = "Land patches from bugzilla, optionally building and testing them first"

    argument_names = "ATTACHMENT_ID [ATTACHMENT_IDS]"

    show_in_main_help = True

class LandFromBug(AbstractPatchLandingCommand, ProcessBugsMixin):

    name = "land-from-bug"

    help_text = "Land all patches on the given bugs, optionally building and testing them first"

    argument_names = "BUGID [BUGIDS]"

    show_in_main_help = True

class AbstractRolloutPrepCommand(AbstractSequencedCommand):

    argument_names = "REVISION REASON"

    def _commit_info(self, revision):

        commit_info = self.tool.checkout().commit_info_for_revision(revision)

        if commit_info and commit_info.bug_id():

            # Note: Don't print a bug URL here because it will confuse the

            #       SheriffBot because the SheriffBot just greps the output

            #       of create-rollout for bug URLs.  It should do better

            #       parsing instead.

            log("Preparing rollout for bug %s." % commit_info.bug_id())

        else:

            log("Unable to parse bug number from diff.")

        return commit_info

    def _prepare_state(self, options, args, tool):

        revision = args[0]

        commit_info = self._commit_info(revision)

        cc_list = sorted([party.bugzilla_email()

                          for party in commit_info.responsible_parties()

                          if party.bugzilla_email()])

        return {

            "revision": revision,

            "bug_id": commit_info.bug_id(),

            # FIXME: We should used the list as the canonical representation.

            "bug_cc": ",".join(cc_list),

            "reason": args[1],

        }

class PrepareRollout(AbstractRolloutPrepCommand):

    name = "prepare-rollout"

    help_text = "Revert the given revision in the working copy and prepare ChangeLogs with revert reason"

    long_help = """Updates the working copy.

Applies the inverse diff for the provided revision.

Creates an appropriate rollout ChangeLog, including a trac link and bug link.

"""

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.RevertRevision,

        steps.PrepareChangeLogForRevert,

    ]

class CreateRollout(AbstractRolloutPrepCommand):

    name = "create-rollout"

    help_text = "Creates a bug to track a broken SVN revision and uploads a rollout patch."

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.RevertRevision,

        steps.CreateBug,

        steps.PrepareChangeLogForRevert,

        steps.PostDiffForRevert,

    ]

    def _prepare_state(self, options, args, tool):

        state = AbstractRolloutPrepCommand._prepare_state(self, options, args, tool)

        # Currently, state["bug_id"] points to the bug that caused the

        # regression.  We want to create a new bug that blocks the old bug

        # so we move state["bug_id"] to state["bug_blocked"] and delete the

        # old state["bug_id"] so that steps.CreateBug will actually create

        # the new bug that we want (and subsequently store its bug id into

        # state["bug_id"])

        state["bug_blocked"] = state["bug_id"]

        del state["bug_id"]

        state["bug_title"] = "REGRESSION(r%s): %s" % (state["revision"], state["reason"])

        state["bug_description"] = "%s broke the build:\n%s" % (view_source_url(state["revision"]), state["reason"])

        # FIXME: If we had more context here, we could link to other open bugs

        #        that mention the test that regressed.

        if options.parent_command == "sheriff-bot":

            state["bug_description"] += """

This is an automatic bug report generated by the sheriff-bot. If this bug

report was created because of a flaky test, please file a bug for the flaky

test (if we don't already have one on file) and dup this bug against that bug

so that we can track how often these flaky tests case pain.

"Only you can prevent forest fires." -- Smokey the Bear

"""

        return state

class Rollout(AbstractRolloutPrepCommand):

    name = "rollout"

    show_in_main_help = True

    help_text = "Revert the given revision in the working copy and optionally commit the revert and re-open the original bug"

    long_help = """Updates the working copy.

Applies the inverse diff for the provided revision.

Creates an appropriate rollout ChangeLog, including a trac link and bug link.

Opens the generated ChangeLogs in $EDITOR.

Shows the prepared diff for confirmation.

Commits the revert and updates the bug (including re-opening the bug if necessary)."""

    steps = [

        steps.CleanWorkingDirectory,

        steps.Update,

        steps.RevertRevision,

        steps.PrepareChangeLogForRevert,

        steps.EditChangeLog,

        steps.ConfirmDiff,

        steps.Build,

        steps.Commit,

        steps.ReopenBugAfterRollout,

    ]