Adding upstream version 7.7.0+dfsg.upstream/7.7.0+dfsg

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 120 insertions, 0 deletions

diff --git a/ansible_collections/sensu/sensu_go/plugins/action/bonsai_asset.py b/ansible_collections/sensu/sensu_go/plugins/action/bonsai_asset.py
new file mode 100644
index 000000000..74473aeb7
--- /dev/null
+++ b/ansible_collections/sensu/sensu_go/plugins/action/bonsai_asset.py
@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+# Copyright: (c) 2019, XLAB Steampunk <steampunk@xlab.si>
+#
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+
+from __future__ import absolute_import, division, print_function
+__metaclass__ = type
+
+from ansible.module_utils.six import text_type
+from ansible.plugins.action import ActionBase
+from ansible.utils.vars import merge_hash
+
+from ..module_utils import bonsai, errors
+
+
+def validate(name, args, required, typ):
+    """
+    Make sure that required values are not None and that if the value is
+    present, it is of the correct type.
+    """
+    value = args.get(name)
+    if required and value is None:
+        raise errors.Error("{0} is required argument".format(name))
+    if value is not None and not isinstance(value, typ):
+        raise errors.Error("{0} should be {1}".format(name, typ))
+
+
+class ActionModule(ActionBase):
+
+    _VALID_ARGS = frozenset((
+        "auth", "name", "version", "namespace", "rename", "labels",
+        "annotations", "on_remote",
+    ))
+
+    def run(self, _tmp=None, task_vars=None):
+        self._supports_check_mode = True
+        self._supports_async = True
+
+        result = super(ActionModule, self).run(task_vars=task_vars)
+
+        wrap_async = (
+            self._task.async_val and not self._connection.has_native_async
+        )
+
+        try:
+            self.validate_arguments(self._task.args)
+            asset = self.download_asset_definition(
+                self._task.args.get("on_remote", False),
+                self._task.args["name"],
+                self._task.args["version"],
+                task_vars,
+            )
+            asset_args = self.build_asset_args(self._task.args, asset)
+            return merge_hash(
+                result,
+                self._execute_module(
+                    module_name="sensu.sensu_go.asset", module_args=asset_args,
+                    task_vars=task_vars, wrap_async=wrap_async,
+                ),
+            )
+        except errors.Error as e:
+            return dict(result, failed=True, msg=str(e))
+        finally:
+            if not wrap_async:
+                self._remove_tmp_path(self._connection._shell.tmpdir)
+
+    @staticmethod
+    def validate_arguments(args):
+        # We only validate arguments that we use. We let the asset module
+        # validate the rest (like auth data).
+
+        # Next three string validations might seem strange at first, but there
+        # is a reason for this strangenes. On python 2, we should consider
+        # string to be instance of str or unicode. On python 3, strings are
+        # always instances of str. In order to avoid having a separate
+        # validate calls for python 2 and python 3, we always pass a pair of
+        # types that just happen to be the same on python 3.
+        validate("name", args, required=True, typ=(str, text_type))
+        validate("version", args, required=True, typ=(str, text_type))
+        validate("rename", args, required=False, typ=(str, text_type))
+        validate("labels", args, required=False, typ=dict)
+        validate("annotations", args, required=False, typ=dict)
+        validate("on_remote", args, required=False, typ=bool)
+
+    def download_asset_definition(self, on_remote, name, version, task_vars):
+        if not on_remote:
+            return bonsai.get_asset_parameters(name, version)
+
+        args = dict(name=name, version=version)
+        result = self._execute_module(
+            module_name="sensu.sensu_go.bonsai_asset", module_args=args,
+            task_vars=task_vars, wrap_async=False,
+        )
+        if result.get("failed", False):
+            raise errors.Error(result["msg"])
+
+        return result["asset"]
+
+    @staticmethod
+    def build_asset_args(args, bonsai_args):
+        asset_args = dict(
+            name=args.get("rename", args["name"]),
+            state="present",
+            builds=bonsai_args["builds"],
+        )
+
+        if "auth" in args:
+            asset_args["auth"] = args["auth"]
+
+        if "namespace" in args:
+            asset_args["namespace"] = args["namespace"]
+
+        # Only add optional parameter if it is present in at least one source.
+        for meta in ("labels", "annotations"):
+            if bonsai_args[meta] or args.get(meta):
+                asset_args[meta] = merge_hash(
+                    bonsai_args[meta] or {}, args.get(meta, {}),
+                )
+
+        return asset_args