1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import os
from ansible.errors import AnsibleError, AnsibleAction, AnsibleActionFail, AnsibleActionSkip, AnsibleConnectionFailure
from ansible.module_utils._text import to_native, to_bytes
from ansible.plugins.action import ActionBase
from ansible.utils.cmd_functions import run_cmd
class ActionModule(ActionBase):
TRANSFERS_FILES = False
DECOMPRESSORS = {
'application/gzip': 'gunzip',
'application/zstd': 'unzstd',
'application/x-bzip2': 'bunzip2',
'application/x-xz': 'unxz',
}
def _get_decompressor_for_file(self, file):
try:
rc, out, err = run_cmd('file --mime-type --mime-encoding %s' % (file))
if rc != 0:
raise Exception('unable to read mime-type of file: %s' % (file))
out = to_native(out, errors='surrogate_or_strict')
mimetype = out.rsplit(':', 1)[1].split(';')[0]
return self.DECOMPRESSORS.get(mimetype.strip(), 'cat')
except Exception as e:
raise AnsibleActionFail(to_native(e))
def run(self, tmp=None, task_vars=None):
if task_vars is None:
task_vars = dict()
result = super(ActionModule, self).run(tmp, task_vars)
del tmp # tmp no longer has any effect
try:
if self._play_context.check_mode:
raise AnsibleActionSkip('Check mode is not supported for this task.')
result['changed'] = True
source = to_native(self._task.args.get('src', ''), errors='surrogate_or_strict')
destination = to_native(self._task.args.get('dest', ''), errors='surrogate_or_strict')
decompressor = self._get_decompressor_for_file(source)
try:
source = self._loader.get_real_file(self._find_needle('files', source), decrypt=False)
with open(to_bytes(source, errors='surrogate_or_strict'), 'rb') as f:
source = to_bytes(f.read(), nonstring='passthru')
except AnsibleError as e:
raise AnsibleActionFail(to_native(e))
result.update(self._low_level_execute_command('%s > %s' % (decompressor, destination), in_data=source))
# workaround for gzip trailing garbage handling
if 'rc' in result and result['rc'] == 2:
if 'stderr' in result and 'decompression OK' in result['stderr']:
result['rc'] = 0
self._fixup_perms2((destination), execute=False)
except AnsibleAction as e:
result.update(e.result)
return result
|
