mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-14 13:00:10 -06:00
fa4c03da65
Change how xargs.partition computes the command length (including arguments) depending on the plataform. More specifically, 'win32' uses the amount of characters while posix system uses the byte count.
93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
|
|
from pre_commit import parse_shebang
|
|
from pre_commit.util import cmd_output
|
|
|
|
|
|
# TODO: properly compute max_length value
|
|
def _get_platform_max_length():
|
|
# posix minimum
|
|
return 4 * 1024
|
|
|
|
|
|
def _get_command_length(command, arg):
|
|
parts = command + (arg,)
|
|
full_cmd = ' '.join(parts)
|
|
|
|
# win32 uses the amount of characters, more details at:
|
|
# https://blogs.msdn.microsoft.com/oldnewthing/20031210-00/?p=41553/
|
|
if sys.platform == 'win32':
|
|
return len(full_cmd)
|
|
|
|
return len(full_cmd.encode(sys.getdefaultencoding()))
|
|
|
|
|
|
class ArgumentTooLongError(RuntimeError):
|
|
pass
|
|
|
|
|
|
def partition(cmd, varargs, _max_length=None):
|
|
_max_length = _max_length or _get_platform_max_length()
|
|
cmd = tuple(cmd)
|
|
ret = []
|
|
|
|
ret_cmd = []
|
|
# Reversed so arguments are in order
|
|
varargs = list(reversed(varargs))
|
|
|
|
while varargs:
|
|
arg = varargs.pop()
|
|
|
|
if _get_command_length(cmd + tuple(ret_cmd), arg) <= _max_length:
|
|
ret_cmd.append(arg)
|
|
elif not ret_cmd:
|
|
raise ArgumentTooLongError(arg)
|
|
else:
|
|
# We've exceeded the length, yield a command
|
|
ret.append(cmd + tuple(ret_cmd))
|
|
ret_cmd = []
|
|
varargs.append(arg)
|
|
|
|
ret.append(cmd + tuple(ret_cmd))
|
|
|
|
return tuple(ret)
|
|
|
|
|
|
def xargs(cmd, varargs, **kwargs):
|
|
"""A simplified implementation of xargs.
|
|
|
|
negate: Make nonzero successful and zero a failure
|
|
"""
|
|
negate = kwargs.pop('negate', False)
|
|
retcode = 0
|
|
stdout = b''
|
|
stderr = b''
|
|
|
|
try:
|
|
parse_shebang.normexe(cmd[0])
|
|
except parse_shebang.ExecutableNotFoundError as e:
|
|
return e.to_output()
|
|
|
|
for run_cmd in partition(cmd, varargs, **kwargs):
|
|
proc_retcode, proc_out, proc_err = cmd_output(
|
|
*run_cmd, encoding=None, retcode=None
|
|
)
|
|
# This is *slightly* too clever so I'll explain it.
|
|
# First the xor boolean table:
|
|
# T | F |
|
|
# +-------+
|
|
# T | F | T |
|
|
# --+-------+
|
|
# F | T | F |
|
|
# --+-------+
|
|
# When negate is True, it has the effect of flipping the return code
|
|
# Otherwise, the retuncode is unchanged
|
|
retcode |= bool(proc_retcode) ^ negate
|
|
stdout += proc_out
|
|
stderr += proc_err
|
|
|
|
return retcode, stdout, stderr
|