98 lines
2.6 KiB
Python
Executable File
98 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import argparse
|
|
import io
|
|
import yaml
|
|
|
|
|
|
def migrate(values):
|
|
"""Actual changes here"""
|
|
|
|
# migrate ClusterName to clusterName
|
|
if "ClusterName" in values:
|
|
values["clusterName"] = values["ClusterName"]
|
|
values.pop("ClusterName")
|
|
|
|
# Remove HighAvailableControlplane
|
|
try:
|
|
values["global"]["highAvailable"] = values["HighAvailableControlplane"]
|
|
values.pop("HighAvailableControlplane")
|
|
except KeyError:
|
|
pass
|
|
|
|
# Create new clusterwide cloudprovider data if possible
|
|
# IamArn: arn:aws:iam::<ACCOUNT_ID>:role/<REGION>.<CLUSTERNAME>.cert-manager
|
|
try:
|
|
if values["cert-manager"]["IamArn"]:
|
|
account_id = values["cert-manager"]["IamArn"].split(":")[4]
|
|
region = values["cert-manager"]["IamArn"].split(":")[5].split('.')[0].split('/')[1]
|
|
if "global" not in values:
|
|
values["global"] = {}
|
|
if "aws" not in values["global"]:
|
|
values["global"]["aws"] = {}
|
|
|
|
values["global"]["aws"]["region"] = region
|
|
values["global"]["aws"]["accountId"] = account_id
|
|
|
|
except KeyError:
|
|
pass
|
|
|
|
return values
|
|
|
|
|
|
class MyDumper(yaml.Dumper):
|
|
"""
|
|
Required to add additional indent for arrays to match yq behaviour to reduce noise in diffs
|
|
"""
|
|
def increase_indent(self, flow=False, indentless=False):
|
|
return super(MyDumper, self).increase_indent(flow, False)
|
|
|
|
def str_presenter(dumper, data):
|
|
if len(data.splitlines()) > 1: # check for multiline string
|
|
return dumper.represent_scalar("tag:yaml.org,2002:str", data, style="|")
|
|
return dumper.represent_scalar("tag:yaml.org,2002:str", data)
|
|
|
|
|
|
def rec_sort(d):
|
|
if isinstance(d, dict):
|
|
res = dict()
|
|
|
|
# Always have "enabled" first if present
|
|
if "enabled" in d.keys():
|
|
res["enabled"] = rec_sort(d["enabled"])
|
|
d.pop("enabled")
|
|
|
|
# next is "name" if present
|
|
if "name" in d.keys():
|
|
res["name"] = rec_sort(d["name"])
|
|
d.pop("name")
|
|
|
|
for k in sorted(d.keys()):
|
|
res[k] = rec_sort(d[k])
|
|
return res
|
|
if isinstance(d, list):
|
|
for idx, elem in enumerate(d):
|
|
d[idx] = rec_sort(elem)
|
|
|
|
return d
|
|
|
|
|
|
yaml.add_representer(str, str_presenter)
|
|
|
|
# to use with safe_dump:
|
|
yaml.representer.SafeRepresenter.add_representer(str, str_presenter)
|
|
|
|
# Read values
|
|
values = yaml.safe_load(sys.stdin)
|
|
|
|
# Output new values
|
|
buffer = io.StringIO()
|
|
yaml.dump(
|
|
rec_sort(migrate(values)),
|
|
sys.stdout,
|
|
default_flow_style=False,
|
|
indent=2,
|
|
sort_keys=False,
|
|
Dumper=MyDumper
|
|
)
|