import subprocess
import sys

class Chart(object):
    def __setattr__(self, name, value):
        if name == 'status':
            value = value.casefold()
        if 'update_available' in name:
            value = True if value.casefold() == "true" else False
        super(Chart, self).__setattr__(name, value)

    def new_attr(self, attr):
        setattr(self, attr, attr)

INSTALLED_CHARTS = []
CATALOG = "ALL"
SEMVER = "minor"

def parse_headers(charts: str):
    for line in charts.split("\n"):
        if line.startswith("+-"):
            continue
        if "name" in line.casefold():
            return [col.strip() for col in line.casefold().strip().split("|") if col and col != ""]

def parse_charts(charts: str):
    headers = parse_headers(charts)
    table_data = charts.split("\n")[3:-2:]  # Skip the header part of the table
    for row in table_data:
        row = [section.strip() for section in row.split("|") if section and section != ""]
        chart = Chart()
        for item in zip(headers, row):
            setattr(chart, item[0], item[1])
        INSTALLED_CHARTS.append(chart)

def check_semver(current: str, latest: str):
    split_current_semver = current.split(".", 3)
    split_latest_semver = latest.split(".", 3)
    if split_current_semver[0] != split_latest_semver[0]:
      type="major"
      if SEMVER == "major":
        return True
    if split_current_semver[1] != split_latest_semver[1]:
      type="minor"
      if SEMVER != "patch":
        return True
    if split_current_semver[2] != split_latest_semver[2]:
      type="patch"
      return True
    return False
    

def execute_upgrades():
    if CATALOG == "ALL":
      filtered = filter(lambda a: a.update_available and a.status == "active", INSTALLED_CHARTS)
    else:
      filtered = filter(lambda a: a.update_available and a.status == "active" and a.catalog == CATALOG, INSTALLED_CHARTS)
    for chart in filtered:
      pre_update_ver = chart.human_version
      post_update_ver = chart.human_latest_version
      split_current_version = chart.human_version.split("_", 1)
      current_version = split_current_version[1]
      split_latest = chart.human_latest_version.split("_", 1)
      latest_version = split_latest[1]
      if check_semver(current_version, latest_version):
        pre_update_ver = chart.human_version
        result = subprocess.run(['cli', '-c', f'app chart_release upgrade release_name={chart.name}'], capture_output=True)
        post_update_ver = chart.human_latest_version
        if "Upgrade complete" not in result.stdout.decode('utf-8'):
            print(f"{chart.name} failed to upgrade. \n{result.stdout.decode('utf-8')}")
        else:
            print(f"{chart.name} upgraded ({pre_update_ver} --> {post_update_ver})")

def fetch_charts():
  rawcharts = subprocess.run(["cli", "-c", "app chart_release query"], stdout=subprocess.PIPE)
  charts = rawcharts.stdout.decode('utf-8')
  return(charts)
  
def run()
    get_args()
    charts = fetch_charts()
    parse_charts(charts)
    execute_upgrades()
  
def get_args():
  if len(sys.argv) == 3:
    SEMVER = sys.argv[2] or "minor"
  if len(sys.argv) == 2:
    CATALOG = sys.argv[1] or "ALL"

if __name__ == '__main__':
    run()
    exit(0)