Generate kube config (#56)

Signed-off-by: costimuraru <[email protected]>

Author: costimuraru

src/ops/cli/helmfile.py

@@ -1,12 +1,12 @@
-#Copyright 2019 Adobe. All rights reserved.
-#This file is licensed to you 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
+# Copyright 2019 Adobe. All rights reserved.
+# This file is licensed to you 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 REPRESENTATIONS
-#OF ANY KIND, either express or implied. See the License for the specific language
-#governing permissions and limitations under the License.
+# 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 REPRESENTATIONS
+# OF ANY KIND, either express or implied. See the License for the specific language
+# governing permissions and limitations under the License.
 
 
 import os
@@ -41,34 +41,62 @@ def get_epilog(self):
 
 
 class HelmfileRunner(CompositionConfigGenerator, object):
-    def __init__(self, ops_config, cluster_config_path):
+    def __init__(self, ops_config, cluster_config_path, execute):
         super(HelmfileRunner, self).__init__(["helmfiles"])
         logging.basicConfig(level=logging.INFO)
         self.ops_config = ops_config
         self.cluster_config_path = cluster_config_path
+        self.execute = execute
 
     def run(self, args):
         config_path_prefix = os.path.join(self.cluster_config_path, '')
-        args.helmfile_path = '../ee-k8s-infra/compositions/helmfiles' if args.helmfile_path is None else os.path.join(args.helmfile_path, '')
+        default_helmfiles = '../ee-k8s-infra/compositions/helmfiles'
+        args.helmfile_path = default_helmfiles if args.helmfile_path is None else os.path.join(args.helmfile_path, '')
 
-        compositions= self.get_sorted_compositions(config_path_prefix)
+        compositions = self.get_sorted_compositions(config_path_prefix)
         if len(compositions) == 0 or compositions[0] != "helmfiles":
             raise Exception("Please provide the full path to composition=helmfiles")
         composition = compositions[0]
         conf_path = self.get_config_path_for_composition(config_path_prefix, composition)
-        self.generate_helmfile_config(conf_path, args)
+        data = self.generate_helmfile_config(conf_path, args)
+        self.setup_kube_config(data)
 
         command = self.get_helmfile_command(args)
         return dict(command=command)
 
+    def setup_kube_config(self, data):
+        if 'cluster' in data and 'fqdn' in data['cluster'] and 'eks' in data['cluster']['fqdn']:
+            cluster_name = data['cluster']['fqdn']
+            aws_profile = data['account']['name']
+            region = data['region']['location']
+            file_location = self.generate_eks_kube_config(cluster_name, aws_profile, region)
+            os.environ['KUBECONFIG'] = file_location
+
+    def generate_eks_kube_config(self, cluster_name, aws_profile, region):
+        file_location = self.get_tmp_file()
+        cmd = "aws eks update-kubeconfig --name {} --profile {} --region {} --kubeconfig {}".format(cluster_name,
+                                                                                                    aws_profile,
+                                                                                                    region,
+                                                                                                    file_location)
+        return_code = self.execute(dict(command=cmd))
+        if return_code != 0:
+            raise Exception("Unable to generate EKS kube config. Exit code was {}".format(return_code))
+        return file_location
+
+    @staticmethod
+    def get_tmp_file():
+        import tempfile
+        with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
+            return tmp_file.name
+
     def generate_helmfile_config(self, path, args):
         output_file = args.helmfile_path + "/hiera-generated.yaml"
         logger.info('Generating helmfiles config %s', output_file)
-        self.generator.process(path=path,
-                               filters=["helm"],
-                               output_format="yaml",
-                               output_file=output_file,
-                               print_data=True)
+        return self.generator.process(path=path,
+                                      filters=["helm", "account", "region", "cluster"],
+                                      output_format="yaml",
+                                      output_file=output_file,
+                                      print_data=True)
 
     def get_helmfile_command(self, args):
         cmd = ' '.join(args.extra_args + [args.subcommand])