1 files changed, 75 insertions, 0 deletions
diff --git a/helm/plugin/workflows.py b/helm/plugin/workflows.py
new file mode 100644
index 0000000..f916eac
--- /dev/null
+++ b/helm/plugin/workflows.py
@@ -0,0 +1,75 @@
+# ============LICENSE_START==========================================
+# ===================================================================
+# Copyright (c) 2018-2020 AT&T
+# Copyright (c) 2020 Pantheon.tech. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END============================================
+
+from cloudify.decorators import workflow
+from cloudify.workflows import ctx
+from cloudify.exceptions import NonRecoverableError
+import json
+import yaml
+import base64
+
+
+@workflow
+def upgrade(node_instance_id, config_set, config, config_url, config_format,
+            chart_version, chart_repo_url, repo_user, repo_user_password, **kwargs):
+    node_instance = ctx.get_node_instance(node_instance_id)
+
+    if not node_instance_id:
+        raise NonRecoverableError(
+            'No such node_instance_id in deployment: {0}.'.format(
+                node_instance_id))
+
+    kwargs = {}
+    kwargs['config'] = ''
+    kwargs['chart_version'] = str(chart_version)
+    kwargs['chart_repo'] = str(chart_repo_url)
+    kwargs['config_set'] = str(config_set)
+    kwargs['config_json'] = str(config)
+    kwargs['config_url'] = str(config_url)
+    kwargs['config_format'] = str(config_format)
+    kwargs['repo_user'] = str(repo_user)
+    kwargs['repo_user_passwd'] = str(repo_user_password)
+    operation_args = {'operation': 'upgrade', }
+    operation_args['kwargs'] = kwargs
+    node_instance.execute_operation(**operation_args)
+
+
+@workflow
+def rollback(node_instance_id, revision, **kwargs):
+    node_instance = ctx.get_node_instance(node_instance_id)
+
+    if not node_instance_id:
+        raise NonRecoverableError(
+            'No such node_instance_id in deployment: {0}.'.format(
+                node_instance_id))
+
+    kwargs = {}
+    kwargs['revision'] = str(revision)
+    operation_args = {'operation': 'rollback', }
+    operation_args['kwargs'] = kwargs
+    node_instance.execute_operation(**operation_args)
+
+@workflow
+def status(**kwargs):
+
+    for node in ctx.nodes:
+        for node_instance in node.instances:
+            kwargs = {}
+            operation_args = {'operation': 'status', }
+            operation_args['kwargs'] = kwargs
+            node_instance.execute_operation(**operation_args)