# mq.py - patch queues for mercurial

#

# Copyright 2005, 2006 Chris Mason <mason@suse.com>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

'''manage a stack of patches

This extension lets you work with a stack of patches in a Mercurial

repository. It manages two stacks of patches - all known patches, and

applied patches (subset of known patches).

Known patches are represented as patch files in the .hg/patches

directory. Applied patches are both patch files and changesets.

Common tasks (use :hg:`help command` for more details)::

  create new patch                          qnew

  import existing patch                     qimport

  print patch series                        qseries

  print applied patches                     qapplied

  add known patch to applied stack          qpush

  remove patch from applied stack           qpop

  refresh contents of top applied patch     qrefresh

By default, mq will automatically use git patches when required to

avoid losing file mode changes, copy records, binary files or empty

files creations or deletions. This behaviour can be configured with::

  [mq]

  git = auto/keep/yes/no

If set to 'keep', mq will obey the [diff] section configuration while

preserving existing git patches upon qrefresh. If set to 'yes' or

'no', mq will override the [diff] section and always generate git or

regular patches, possibly losing data in the second case.

You will by default be managing a patch queue named "patches". You can

create other, independent patch queues with the :hg:`qqueue` command.

'''

from mercurial.i18n import _

from mercurial.node import bin, hex, short, nullid, nullrev

from mercurial.lock import release

from mercurial import commands, cmdutil, hg, scmutil, util, revset

from mercurial import repair, extensions, url, error, phases

from mercurial import patch as patchmod

import os, re, errno, shutil

commands.norepo += " qclone"

seriesopts = [('s', 'summary', None, _('print first line of patch header'))]

cmdtable = {}

command = cmdutil.command(cmdtable)

# Patch names looks like unix-file names.

# They must be joinable with queue directory and result in the patch path.

normname = util.normpath

class statusentry(object):

    def __init__(self, node, name):

        self.node, self.name = node, name

    def __repr__(self):

        return hex(self.node) + ':' + self.name

class patchheader(object):

    def __init__(self, pf, plainmode=False):

        def eatdiff(lines):

            while lines:

                l = lines[-1]

                if (l.startswith("diff -") or

                    l.startswith("Index:") or

                    l.startswith("===========")):

                    del lines[-1]

                else:

                    break

        def eatempty(lines):

            while lines:

                if not lines[-1].strip():

                    del lines[-1]

                else:

                    break

        message = []

        comments = []

        user = None

        date = None

        parent = None

        format = None

        subject = None

        branch = None

        nodeid = None

        diffstart = 0

        for line in file(pf):

            line = line.rstrip()

            if (line.startswith('diff --git')

                or (diffstart and line.startswith('+++ '))):

                diffstart = 2

                break

            diffstart = 0 # reset

            if line.startswith("--- "):

                diffstart = 1

                continue

            elif format == "hgpatch":

                # parse values when importing the result of an hg export

                if line.startswith("# User "):

                    user = line[7:]

                elif line.startswith("# Date "):

                    date = line[7:]

                elif line.startswith("# Parent "):

                    parent = line[9:].lstrip()

                elif line.startswith("# Branch "):

                    branch = line[9:]

                elif line.startswith("# Node ID "):

                    nodeid = line[10:]

                elif not line.startswith("# ") and line:

                    message.append(line)

                    format = None

            elif line == '# HG changeset patch':

                message = []

                format = "hgpatch"

            elif (format != "tagdone" and (line.startswith("Subject: ") or

                                           line.startswith("subject: "))):

                subject = line[9:]

                format = "tag"

            elif (format != "tagdone" and (line.startswith("From: ") or

                                           line.startswith("from: "))):

                user = line[6:]

                format = "tag"

            elif (format != "tagdone" and (line.startswith("Date: ") or

                                           line.startswith("date: "))):

                date = line[6:]

                format = "tag"

            elif format == "tag" and line == "":

                # when looking for tags (subject: from: etc) they

                # end once you find a blank line in the source

                format = "tagdone"

            elif message or line:

                message.append(line)

            comments.append(line)

        eatdiff(message)

        eatdiff(comments)

        # Remember the exact starting line of the patch diffs before consuming

        # empty lines, for external use by TortoiseHg and others

        self.diffstartline = len(comments)

        eatempty(message)

        eatempty(comments)

        # make sure message isn't empty

        if format and format.startswith("tag") and subject:

            message.insert(0, "")

            message.insert(0, subject)

        self.message = message

        self.comments = comments

        self.user = user

        self.date = date

        self.parent = parent

        # nodeid and branch are for external use by TortoiseHg and others

        self.nodeid = nodeid

        self.branch = branch

        self.haspatch = diffstart > 1

        self.plainmode = plainmode

    def setuser(self, user):

        if not self.updateheader(['From: ', '# User '], user):

            try:

                patchheaderat = self.comments.index('# HG changeset patch')

                self.comments.insert(patchheaderat + 1, '# User ' + user)

            except ValueError:

                if self.plainmode or self._hasheader(['Date: ']):

                    self.comments = ['From: ' + user] + self.comments

                else:

                    tmp = ['# HG changeset patch', '# User ' + user, '']

                    self.comments = tmp + self.comments

        self.user = user

    def setdate(self, date):

        if not self.updateheader(['Date: ', '# Date '], date):

            try:

                patchheaderat = self.comments.index('# HG changeset patch')

                self.comments.insert(patchheaderat + 1, '# Date ' + date)

            except ValueError:

                if self.plainmode or self._hasheader(['From: ']):

                    self.comments = ['Date: ' + date] + self.comments

                else:

                    tmp = ['# HG changeset patch', '# Date ' + date, '']

                    self.comments = tmp + self.comments

        self.date = date

    def setparent(self, parent):

        if not self.updateheader(['# Parent '], parent):

            try:

                patchheaderat = self.comments.index('# HG changeset patch')

                self.comments.insert(patchheaderat + 1, '# Parent ' + parent)

            except ValueError:

                pass

        self.parent = parent

    def setmessage(self, message):

        if self.comments:

            self._delmsg()

        self.message = [message]

        self.comments += self.message

    def updateheader(self, prefixes, new):

        '''Update all references to a field in the patch header.

        Return whether the field is present.'''

        res = False

        for prefix in prefixes:

            for i in xrange(len(self.comments)):

                if self.comments[i].startswith(prefix):

                    self.comments[i] = prefix + new

                    res = True

                    break

        return res

    def _hasheader(self, prefixes):

        '''Check if a header starts with any of the given prefixes.'''

        for prefix in prefixes:

            for comment in self.comments:

                if comment.startswith(prefix):

                    return True

        return False

    def __str__(self):

        if not self.comments:

            return ''

        return '\n'.join(self.comments) + '\n\n'

    def _delmsg(self):

        '''Remove existing message, keeping the rest of the comments fields.

        If comments contains 'subject: ', message will prepend

        the field and a blank line.'''

        if self.message:

            subj = 'subject: ' + self.message[0].lower()

            for i in xrange(len(self.comments)):

                if subj == self.comments[i].lower():

                    del self.comments[i]

                    self.message = self.message[2:]

                    break

        ci = 0

        for mi in self.message:

            while mi != self.comments[ci]:

                ci += 1

            del self.comments[ci]

class queue(object):

    def __init__(self, ui, path, patchdir=None):

        self.basepath = path

        try:

            fh = open(os.path.join(path, 'patches.queue'))

            cur = fh.read().rstrip()

            fh.close()

            if not cur:

                curpath = os.path.join(path, 'patches')

            else:

                curpath = os.path.join(path, 'patches-' + cur)

        except IOError:

            curpath = os.path.join(path, 'patches')

        self.path = patchdir or curpath

        self.opener = scmutil.opener(self.path)

        self.ui = ui

        self.applieddirty = False

        self.seriesdirty = False

        self.added = []

        self.seriespath = "series"

        self.statuspath = "status"

        self.guardspath = "guards"

        self.activeguards = None

        self.guardsdirty = False

        # Handle mq.git as a bool with extended values

        try:

            gitmode = ui.configbool('mq', 'git', None)

            if gitmode is None:

                raise error.ConfigError()

            self.gitmode = gitmode and 'yes' or 'no'

        except error.ConfigError:

            self.gitmode = ui.config('mq', 'git', 'auto').lower()

        self.plainmode = ui.configbool('mq', 'plain', False)

    @util.propertycache

    def applied(self):

        def parselines(lines):

            for l in lines:

                entry = l.split(':', 1)

                if len(entry) > 1:

                    n, name = entry

                    yield statusentry(bin(n), name)

                elif l.strip():

                    self.ui.warn(_('malformated mq status line: %s\n') % entry)

                # else we ignore empty lines

        try:

            lines = self.opener.read(self.statuspath).splitlines()

            return list(parselines(lines))

        except IOError, e:

            if e.errno == errno.ENOENT:

                return []

            raise

    @util.propertycache

    def fullseries(self):

        try:

            return self.opener.read(self.seriespath).splitlines()

        except IOError, e:

            if e.errno == errno.ENOENT:

                return []

            raise

    @util.propertycache

    def series(self):

        self.parseseries()

        return self.series

    @util.propertycache

    def seriesguards(self):

        self.parseseries()

        return self.seriesguards

    def invalidate(self):

        for a in 'applied fullseries series seriesguards'.split():

            if a in self.__dict__:

                delattr(self, a)

        self.applieddirty = False

        self.seriesdirty = False

        self.guardsdirty = False

        self.activeguards = None

    def diffopts(self, opts={}, patchfn=None):

        diffopts = patchmod.diffopts(self.ui, opts)

        if self.gitmode == 'auto':

            diffopts.upgrade = True

        elif self.gitmode == 'keep':

            pass

        elif self.gitmode in ('yes', 'no'):

            diffopts.git = self.gitmode == 'yes'

        else:

            raise util.Abort(_('mq.git option can be auto/keep/yes/no'

                               ' got %s') % self.gitmode)

        if patchfn:

            diffopts = self.patchopts(diffopts, patchfn)

        return diffopts

    def patchopts(self, diffopts, *patches):

        """Return a copy of input diff options with git set to true if

        referenced patch is a git patch and should be preserved as such.

        """

        diffopts = diffopts.copy()

        if not diffopts.git and self.gitmode == 'keep':

            for patchfn in patches:

                patchf = self.opener(patchfn, 'r')

                # if the patch was a git patch, refresh it as a git patch

                for line in patchf:

                    if line.startswith('diff --git'):

                        diffopts.git = True

                        break

                patchf.close()

        return diffopts

    def join(self, *p):

        return os.path.join(self.path, *p)

    def findseries(self, patch):

        def matchpatch(l):

            l = l.split('#', 1)[0]

            return l.strip() == patch