# 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)::

  prepare repository to work with patches   qinit

  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.

'''

from mercurial.i18n import _

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

from mercurial.lock import release

from mercurial import commands, cmdutil, hg, patch, util

from mercurial import repair, extensions, url, error

import os, sys, re, errno

commands.norepo += " qclone"

# 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, rev, name=None):

        if not name:

            fields = rev.split(':', 1)

            if len(fields) == 2:

                self.rev, self.name = fields

            else:

                self.rev, self.name = None, None

        else:

            self.rev, self.name = rev, name

    def __str__(self):

        return self.rev + ':' + self.name

class patchheader(object):

        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:

                l = lines[-1]

                if re.match('\s*$', l):

                    del lines[-1]

                else:

                    break

        message = []

        comments = []

        user = None

        date = None

        format = None

        subject = None

        diffstart = 0

        for line in file(pf):

            line = line.rstrip()

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

                diffstart = 2

                break

            if diffstart:

                if line.startswith('+++ '):

                    diffstart = 2

                break

            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 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 == "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)

        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.haspatch = diffstart > 1

    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:

                    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:

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

                else:

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

                    self.comments = tmp + self.comments

        self.date = date

    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

        self.path = patchdir or os.path.join(path, "patches")

        self.opener = util.opener(self.path)

        self.ui = ui

        self.applied_dirty = 0

        self.series_dirty = 0

        self.series_path = "series"

        self.status_path = "status"

        self.guards_path = "guards"

        self.active_guards = None

        self.guards_dirty = 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()

    @util.propertycache

    def applied(self):

        if os.path.exists(self.join(self.status_path)):

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

            return [statusentry(l) for l in lines]

        return []

    @util.propertycache

    def full_series(self):

        if os.path.exists(self.join(self.series_path)):

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

        return []

    @util.propertycache

    def series(self):

        self.parse_series()

        return self.series

    @util.propertycache

    def series_guards(self):

        self.parse_series()

        return self.series_guards

    def invalidate(self):

        for a in 'applied full_series series series_guards'.split():

            if a in self.__dict__:

                delattr(self, a)

        self.applied_dirty = 0

        self.series_dirty = 0

        self.guards_dirty = False

        self.active_guards = None

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

        diffopts = patch.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 find_series(self, patch):

        pre = re.compile("(\s*)([^#]+)")

        index = 0

        for l in self.full_series:

            m = pre.match(l)

            if m:

                s = m.group(2)

                s = s.rstrip()

                if s == patch:

                    return index

            index += 1

        return None

    guard_re = re.compile(r'\s?#([-+][^-+# \t\r\n\f][^# \t\r\n\f]*)')

    def parse_series(self):

        self.series = []

        self.series_guards = []

        for l in self.full_series:

            h = l.find('#')

            if h == -1:

                patch = l

                comment = ''

            elif h == 0:

                continue

            else:

                patch = l[:h]

                comment = l[h:]

            patch = patch.strip()

            if patch:

                if patch in self.series:

                    raise util.Abort(_('%s appears more than once in %s') %

                                     (patch, self.join(self.series_path)))

                self.series.append(patch)

                self.series_guards.append(self.guard_re.findall(comment))

    def check_guard(self, guard):

        if not guard:

            return _('guard cannot be an empty string')

        bad_chars = '# \t\r\n\f'

        first = guard[0]