win32text.py
143 lines
| 4.5 KiB
| text/x-python
|
PythonLexer
/ hgext / win32text.py
|
r6481 | # win32text.py - LF <-> CRLF/CR translation utilities for Windows/Mac users | ||
|
r5675 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
# | ||||
# To perform automatic newline conversion, use: | ||||
# | ||||
# [extensions] | ||||
# hgext.win32text = | ||||
# [encode] | ||||
# ** = cleverencode: | ||||
|
r6481 | # # or ** = macencode: | ||
|
r5675 | # [decode] | ||
# ** = cleverdecode: | ||||
|
r6481 | # # or ** = macdecode: | ||
|
r5675 | # | ||
|
r6481 | # If not doing conversion, to make sure you do not commit CRLF/CR by accident: | ||
|
r5675 | # | ||
# [hooks] | ||||
# pretxncommit.crlf = python:hgext.win32text.forbidcrlf | ||||
|
r6481 | # # or pretxncommit.cr = python:hgext.win32text.forbidcr | ||
|
r5675 | # | ||
|
r6481 | # To do the same check on a server to prevent CRLF/CR from being pushed or | ||
# pulled: | ||||
|
r5675 | # | ||
# [hooks] | ||||
# pretxnchangegroup.crlf = python:hgext.win32text.forbidcrlf | ||||
|
r6481 | # # or pretxnchangegroup.cr = python:hgext.win32text.forbidcr | ||
|
r5675 | |||
|
r4859 | from mercurial.i18n import gettext as _ | ||
|
r6211 | from mercurial.node import bin, short | ||
|
r4859 | import re | ||
# regexp for single LF without CR preceding. | ||||
re_single_lf = re.compile('(^|[^\r])\n', re.MULTILINE) | ||||
|
r6481 | newlinestr = {'\r\n': 'CRLF', '\r': 'CR'} | ||
filterstr = {'\r\n': 'clever', '\r': 'mac'} | ||||
def checknewline(s, newline, ui=None, repo=None, filename=None): | ||||
# warn if already has 'newline' in repository. | ||||
|
r4859 | # it might cause unexpected eol conversion. | ||
# see issue 302: | ||||
# http://www.selenic.com/mercurial/bts/issue302 | ||||
|
r6481 | if newline in s and ui and filename and repo: | ||
ui.warn(_('WARNING: %s already has %s line endings\n' | ||||
|
r5967 | 'and does not need EOL conversion by the win32text plugin.\n' | ||
'Before your next commit, please reconsider your ' | ||||
'encode/decode settings in \nMercurial.ini or %s.\n') % | ||||
|
r6481 | (filename, newlinestr[newline], repo.join('hgrc'))) | ||
def dumbdecode(s, cmd, **kwargs): | ||||
checknewline(s, '\r\n', **kwargs) | ||||
|
r4859 | # replace single LF to CRLF | ||
return re_single_lf.sub('\\1\r\n', s) | ||||
def dumbencode(s, cmd): | ||||
return s.replace('\r\n', '\n') | ||||
|
r6481 | def macdumbdecode(s, cmd, **kwargs): | ||
checknewline(s, '\r', **kwargs) | ||||
return s.replace('\n', '\r') | ||||
def macdumbencode(s, cmd): | ||||
return s.replace('\r', '\n') | ||||
|
r5967 | def cleverdecode(s, cmd, **kwargs): | ||
|
r6508 | if not util.binary(s): | ||
|
r6473 | return dumbdecode(s, cmd, **kwargs) | ||
return s | ||||
|
r4859 | |||
def cleverencode(s, cmd): | ||||
|
r6508 | if not util.binary(s): | ||
|
r6473 | return dumbencode(s, cmd) | ||
return s | ||||
|
r4859 | |||
|
r6481 | def macdecode(s, cmd, **kwargs): | ||
|
r6508 | if not util.binary(s): | ||
|
r6484 | return macdumbdecode(s, cmd, **kwargs) | ||
return s | ||||
|
r6481 | |||
def macencode(s, cmd): | ||||
|
r6508 | if not util.binary(s): | ||
|
r6484 | return macdumbencode(s, cmd) | ||
return s | ||||
|
r6481 | |||
|
r5966 | _filters = { | ||
|
r4859 | 'dumbdecode:': dumbdecode, | ||
'dumbencode:': dumbencode, | ||||
'cleverdecode:': cleverdecode, | ||||
'cleverencode:': cleverencode, | ||||
|
r6481 | 'macdumbdecode:': macdumbdecode, | ||
'macdumbencode:': macdumbencode, | ||||
'macdecode:': macdecode, | ||||
'macencode:': macencode, | ||||
|
r5966 | } | ||
|
r5675 | |||
|
r6483 | def forbidnewline(ui, repo, hooktype, node, newline, **kwargs): | ||
|
r5675 | halt = False | ||
for rev in xrange(repo.changelog.rev(bin(node)), repo.changelog.count()): | ||||
c = repo.changectx(rev) | ||||
for f in c.files(): | ||||
if f not in c: | ||||
continue | ||||
data = c[f].data() | ||||
|
r6508 | if not util.binary(data) and newline in data: | ||
|
r5675 | if not halt: | ||
ui.warn(_('Attempt to commit or push text file(s) ' | ||||
|
r6481 | 'using %s line endings\n') % | ||
newlinestr[newline]) | ||||
|
r5675 | ui.warn(_('in %s: %s\n') % (short(c.node()), f)) | ||
halt = True | ||||
if halt and hooktype == 'pretxnchangegroup': | ||||
|
r6481 | crlf = newlinestr[newline].lower() | ||
filter = filterstr[newline] | ||||
|
r5675 | ui.warn(_('\nTo prevent this mistake in your local repository,\n' | ||
'add to Mercurial.ini or .hg/hgrc:\n' | ||||
'\n' | ||||
'[hooks]\n' | ||||
|
r6481 | 'pretxncommit.%s = python:hgext.win32text.forbid%s\n' | ||
|
r5675 | '\n' | ||
'and also consider adding:\n' | ||||
'\n' | ||||
'[extensions]\n' | ||||
'hgext.win32text =\n' | ||||
'[encode]\n' | ||||
|
r6481 | '** = %sencode:\n' | ||
|
r5675 | '[decode]\n' | ||
|
r6481 | '** = %sdecode:\n') % (crlf, crlf, filter, filter)) | ||
|
r5675 | return halt | ||
|
r5966 | |||
|
r6481 | def forbidcrlf(ui, repo, hooktype, node, **kwargs): | ||
|
r6483 | return forbidnewline(ui, repo, hooktype, node, '\r\n', **kwargs) | ||
|
r6481 | |||
def forbidcr(ui, repo, hooktype, node, **kwargs): | ||||
|
r6483 | return forbidnewline(ui, repo, hooktype, node, '\r', **kwargs) | ||
|
r6481 | |||
|
r5966 | def reposetup(ui, repo): | ||
if not repo.local(): | ||||
return | ||||
for name, fn in _filters.iteritems(): | ||||
repo.adddatafilter(name, fn) | ||||