ingress-annotator.star

#   Copyright IBM Corporation 2021
#
#   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.

# transform creates a new artifact of type "KubernetesYamls" which
# has an annotation for ingress-class added to every ingress resource yaml created during
# transformation
def transform(new_artifacts, old_artifacts):
    pathMappings = []
    artifacts = []

    for a in new_artifacts:
        yamlsPath = a["paths"]["KubernetesYamls"][0]
        serviceName = a["name"]
        artifacts.append(a)

        fileList = fs.read_dir(yamlsPath)
        yamlsBasePath = yamlsPath.split("/")[-1]
        # Create a custom path template for the service, whose values gets filled and can be used in other pathmappings
        pathTemplateName = serviceName + yamlsBasePath
        pathTemplateName = pathTemplateName.replace("-", "")
        tplPathData = {'PathTemplateName': pathTemplateName}
        pathMappings.append({'type': 'PathTemplate', \
                            'sourcePath': "{{ OutputRel \"" + yamlsPath + "\" }}", \
                            'templateConfig': tplPathData})
        for f in fileList:
            filePath = fs.path_join(yamlsPath, f)
            s = fs.read_as_string(filePath)
            yamlData = yaml.loads(s)
            if yamlData['kind'] != 'Ingress':
                continue
            if 'annotations' not in yamlData['metadata']:
                yamlData['metadata']['annotations'] = {'kubernetes.io/ingress.class': 'haproxy'}
            else:
                yamlData['metadata']['annotations']['kubernetes.io/ingress.class'] = 'haproxy'
            s = yaml.dumps(yamlData)
            fs.write(filePath, s)
            pathMappings.append({'type': 'Default', \
                    'sourcePath': yamlsPath, \
                    'destinationPath': "{{ ." + pathTemplateName + " }}"})
        
    return {'pathMappings': pathMappings, 'artifacts': artifacts}