Ensure _raw_params retain exact spaces

This fixes a few issues related to multi-line YAML strings that eat
whitespace and add whitespaces after newlines (that weren't there).
This commit is contained in:
Dag Wieers 2018-09-19 17:48:59 +02:00
parent 77f73f6d2a
commit b72e989e18

View file

@ -97,7 +97,7 @@ def parse_kv(args, check_raw=False):
# recombine the free-form params, if any were found, and assign
# them to a special option for use later by the shell/command module
if len(raw_params) > 0:
options[u'_raw_params'] = ' '.join(raw_params)
options[u'_raw_params'] = join_args(raw_params)
return options
@ -137,6 +137,20 @@ def _count_jinja2_blocks(token, cur_depth, open_token, close_token):
return cur_depth
def join_args(s):
'''
Join the original cmd based on manipulations by split_args().
This retains the original newlines and whitespaces.
'''
result = ''
for p in s:
if len(result) == 0 or result.endswith('\n'):
result += p
else:
result += ' ' + p
return result
def split_args(args):
'''
Splits args on whitespace, but intelligently reassembles
@ -157,9 +171,8 @@ def split_args(args):
# this is going to be the result value when we are done
params = []
# Initial split on white space
args = args.strip()
items = args.strip().split('\n')
# Initial split on newlines
items = args.split('\n')
# iterate over the tokens, and reassemble any that may have been
# split on a space inside a jinja2 block.
@ -182,11 +195,17 @@ def split_args(args):
# we split on spaces and newlines separately, so that we
# can tell which character we split on for reassembly
# inside quotation characters
tokens = item.strip().split(' ')
tokens = item.split(' ')
line_continuation = False
for (idx, token) in enumerate(tokens):
# Empty entries means we have subsequent spaces
# We want to hold onto them so we can reconstruct them later
if len(token) == 0 and idx != 0:
params[-1] += ' '
continue
# if we hit a line continuation character, but
# we're not inside quotes, ignore it and continue
# on to the next token while setting a flag