Add kubevirt_cdi_upload module (#52990)

1 files changed, 177 insertions, 0 deletions

diff --git a/lib/ansible/modules/cloud/kubevirt/kubevirt_cdi_upload.py b/lib/ansible/modules/cloud/kubevirt/kubevirt_cdi_upload.py
new file mode 100644
index 0000000000..79edb92776
--- /dev/null
+++ b/lib/ansible/modules/cloud/kubevirt/kubevirt_cdi_upload.py
@@ -0,0 +1,177 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# Copyright: (c) 2019, Ansible Project
+# 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
+
+
+ANSIBLE_METADATA = {'metadata_version': '1.1',
+                    'status': ['preview'],
+                    'supported_by': 'community'}
+
+DOCUMENTATION = '''
+
+module: kubevirt_cdi_upload
+
+short_description: Upload local VM images to CDI Upload Proxy.
+
+version_added: "2.8"
+
+author: KubeVirt Team (@kubevirt)
+
+
+description:
+  - Use Openshift Python SDK to create UploadTokenRequest objects.
+  - Transfer contents of local files to the CDI Upload Proxy.
+
+options:
+  pvc_name:
+    description:
+      - Use to specify the name of the target PersistentVolumeClaim.
+    required: true
+  pvc_namespace:
+    description:
+      - Use to specify the namespace of the target PersistentVolumeClaim.
+    required: true
+  upload_host:
+    description:
+      - URL containing the host and port on which the CDI Upload Proxy is available.
+      - "More info: U(https://github.com/kubevirt/containerized-data-importer/blob/master/doc/upload.md#expose-cdi-uploadproxy-service)"
+  upload_host_verify_ssl:
+    description:
+      - Whether or not to verify the CDI Upload Proxy's SSL certificates against your system's CA trust store.
+    default: true
+    type: bool
+  path:
+    description:
+      - Path of local image file to transfer.
+  merge_type:
+    description:
+      - Whether to override the default patch merge approach with a specific type. By default, the strategic
+        merge will typically be used.
+    type: list
+    choices: [ json, merge, strategic-merge ]
+
+extends_documentation_fragment:
+  - k8s_auth_options
+
+requirements:
+  - python >= 2.7
+  - openshift >= 0.8.2
+  - requests >= 2.0.0
+'''
+
+EXAMPLES = '''
+- name: Upload local image to pvc-vm1
+  kubevirt_cdi_upload:
+    pvc_namespace: default
+    pvc_name: pvc-vm1
+    upload_host: https://localhost:8443
+    upload_host_verify_ssl: false
+    path: /tmp/cirros-0.4.0-x86_64-disk.img
+'''
+
+RETURN = '''# '''
+
+import copy
+import traceback
+
+from collections import defaultdict
+
+from ansible.module_utils.k8s.common import AUTH_ARG_SPEC
+from ansible.module_utils.k8s.raw import KubernetesRawModule
+
+# 3rd party imports
+try:
+    import requests
+    HAS_REQUESTS = True
+except ImportError:
+    HAS_REQUESTS = False
+
+
+SERVICE_ARG_SPEC = {
+    'pvc_name': {'required': True},
+    'pvc_namespace': {'required': True},
+    'upload_host': {'required': True},
+    'upload_host_verify_ssl': {
+        'type': 'bool',
+        'default': True
+    },
+    'path': {'required': True},
+    'merge_type': {
+        'type': 'list',
+        'choices': ['json', 'merge', 'strategic-merge']
+    },
+}
+
+
+class KubeVirtCDIUpload(KubernetesRawModule):
+    def __init__(self, *args, **kwargs):
+        super(KubeVirtCDIUpload, self).__init__(*args, k8s_kind='UploadTokenRequest', **kwargs)
+
+        if not HAS_REQUESTS:
+            self.fail("This module requires the python 'requests' package. Try `pip install requests`.")
+
+    @property
+    def argspec(self):
+        """ argspec property builder """
+        argument_spec = copy.deepcopy(AUTH_ARG_SPEC)
+        argument_spec.update(SERVICE_ARG_SPEC)
+        return argument_spec
+
+    def execute_module(self):
+        """ Module execution """
+
+        API = 'v1alpha1'
+        KIND = 'UploadTokenRequest'
+
+        self.client = self.get_api_client()
+
+        api_version = 'upload.cdi.kubevirt.io/{0}'.format(API)
+        pvc_name = self.params.get('pvc_name')
+        pvc_namespace = self.params.get('pvc_namespace')
+        upload_host = self.params.get('upload_host')
+        upload_host_verify_ssl = self.params.get('upload_host_verify_ssl')
+        path = self.params.get('path')
+
+        definition = defaultdict(defaultdict)
+
+        definition['kind'] = KIND
+        definition['apiVersion'] = api_version
+
+        def_meta = definition['metadata']
+        def_meta['name'] = pvc_name
+        def_meta['namespace'] = pvc_namespace
+
+        def_spec = definition['spec']
+        def_spec['pvcName'] = pvc_name
+
+        # Let's check the file's there before we do anything else
+        imgfile = open(path, 'rb')
+
+        resource = self.find_resource(KIND, api_version, fail=True)
+        definition = self.set_defaults(resource, definition)
+        result = self.perform_action(resource, definition)
+
+        headers = {'Authorization': "Bearer {0}".format(result['result']['status']['token'])}
+        files = {'file': imgfile}
+        url = "{0}/{1}/upload".format(upload_host, API)
+        requests.post(url, files=files, headers=headers, verify=upload_host_verify_ssl)
+
+        self.exit_json(changed=True)
+
+
+def main():
+    module = KubeVirtCDIUpload()
+    try:
+        module.execute_module()
+    except Exception as e:
+        module.fail_json(msg=str(e), exception=traceback.format_exc())
+
+
+if __name__ == '__main__':
+    main()