mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +03:00
37116e0862
Summary: Looks for similar files between the removed/missing and added/unknown committed files Test Plan: Tests are done without 'commit' (testmode option) mv + add + rm with several thresholds of similarity changing the threshold of similarity mv alone mv + add + rm but not in committed files Reviewers: #sourcecontrol, rmcelroy Reviewed By: rmcelroy Subscribers: durham, mitrandir, pyd, cdelahousse Differential Revision: https://phabricator.fb.com/D2492693 Tasks: 8501037 Signature: t1:2492693:1445396857:a40d0aeb3c6592b1c8d2a93d089e4fdc46bde711
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
# automv.py
|
|
#
|
|
# Copyright 2013 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
"""
|
|
This extension checks at commit/amend time if any of the committed files
|
|
comes from an unrecorded mv
|
|
"""
|
|
|
|
from mercurial import cmdutil, scmutil
|
|
from mercurial import similar, util, commands
|
|
from mercurial.extensions import wrapcommand
|
|
from mercurial import extensions
|
|
from mercurial.i18n import _
|
|
|
|
def extsetup(ui):
|
|
entry = wrapcommand(commands.table, 'commit', mvcheck)
|
|
entry[1].append(('', 'no-move-detection', None,
|
|
_('disable automatic' + 'file move detection')))
|
|
try:
|
|
module = extensions.find('fbamend')
|
|
entry = wrapcommand(module.cmdtable, 'amend', mvcheck)
|
|
entry[1].append(('', 'no-move-detection', None,
|
|
_('disable automatic' + 'file move detection')))
|
|
except KeyError:
|
|
pass
|
|
|
|
def mvcheck(orig, ui, repo, *pats, **opts):
|
|
if not opts.get('no_move_detection'):
|
|
threshold = float(ui.config('automv', 'similaritythres', '0.75'))
|
|
if threshold > 0:
|
|
match = scmutil.match(repo[None], pats, opts)
|
|
added, removed = _interestingfiles(repo, match)
|
|
renames = _findrenames(repo, match, added, removed, threshold)
|
|
_markchanges(repo, renames)
|
|
if 'no_move_detection' in opts:
|
|
del opts['no_move_detection']
|
|
|
|
if ui.configbool('automv', 'testmode'):
|
|
return
|
|
else:
|
|
return orig(ui, repo, *pats, **opts)
|
|
|
|
def _interestingfiles(repo, matcher):
|
|
added, removed = [], []
|
|
|
|
ctx = repo[None]
|
|
dirstate = repo.dirstate
|
|
files = dirstate.matches(matcher)
|
|
for abs, st in dirstate.iteritems():
|
|
if not abs in files:
|
|
continue
|
|
dstate = dirstate[abs]
|
|
# for finding renames
|
|
if dstate == 'r':
|
|
removed.append(abs)
|
|
elif dstate == 'a':
|
|
added.append(abs)
|
|
return added, removed
|
|
|
|
def _findrenames(repo, matcher, added, removed, similarity):
|
|
"""Find renames from removed files of the current commit/amend files
|
|
to the added ones"""
|
|
renames = {}
|
|
if similarity > 0:
|
|
for src, dst, score in similar.findrenames(repo, added, removed,
|
|
similarity):
|
|
if repo.ui.verbose:
|
|
repo.ui.status(_('detected move of %s as %s (%d%% similar)\n')
|
|
% (matcher.rel(src), matcher.rel(dst), score * 100))
|
|
renames[dst] = src
|
|
n = len(renames)
|
|
if n == 1:
|
|
repo.ui.status(_('detected move of 1 file\n'))
|
|
elif n > 1:
|
|
repo.ui.status(_('detected move of %d files\n') % len(renames))
|
|
return renames
|
|
|
|
def _markchanges(repo, renames):
|
|
"""Marks the files in renames as copied."""
|
|
wctx = repo[None]
|
|
wlock = repo.wlock()
|
|
try:
|
|
for dst, src in renames.iteritems():
|
|
wctx.copy(src, dst)
|
|
finally:
|
|
wlock.release()
|
|
|
|
|