From 65fe0a00ba7a26288f3661d29ee6fd016dfe5a60 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jes=C3=BAs=20P=C3=A9rez=20Lorenzo?=
 <jesusperezlorenzo@rigzin.local>
Date: Mon, 20 Jan 2025 05:29:39 +0000
Subject: [PATCH] chore: add collect script to run heml charts

---
 collect.py | 70 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 collect.py

diff --git a/collect.py b/collect.py
new file mode 100644
index 0000000..64b53a6
--- /dev/null
+++ b/collect.py
@@ -0,0 +1,70 @@
+import json
+import requests
+import os
+import time
+from subprocess import Popen, PIPE
+
+#5 for now
+url = "https://artifacthub.io/api/v1/packages/search?offset=0&limit=60&facets=false&kind=0&deprecated=false&sort=stars"
+
+if os.path.exists("helm-charts.json"):
+    with open("helm-charts.json", "r") as f:
+        templates = json.loads(f.read())
+else:
+    r = requests.get(url)
+    templates = json.loads(r.text)
+
+    # Save to file
+    with open("helm-charts.json", "w") as f:
+        json.dump(templates, f)
+
+blacklist = ['cilium', 'longhorn']
+uninstall = True
+install = True
+
+if install:
+    p = Popen("helm repo update", shell=True)
+    p.communicate()
+
+time.sleep(1)
+
+if install:
+    for package in templates['packages']:
+        repo_name = package['repository']['name']
+        repo_url = package['repository']['url']
+        chart_name = package['name']
+        version = package['version']
+
+        # Check blacklist, these repos have issues
+        if chart_name in blacklist:
+            continue
+        
+        # command = "helm repo add {repo_name} {repo_url}".format(repo_name=repo_name, repo_url=repo_url)
+        # print("######" + command) 
+        # p = Popen(command, stdout=PIPE, stderr=PIPE, shell=True)
+        # p.communicate()
+
+        # time.sleep(2)
+
+        command = "helm install my-{chart_name} {repo_name}/{chart_name2} --namespace {chart_name3} --create-namespace --version {version} --timeout 45s".format(chart_name=chart_name, repo_name=repo_name, chart_name2=chart_name, chart_name3=chart_name, version=version)
+        print("######" + command)
+        p = Popen(command, stdout=PIPE, stderr=PIPE, shell=True)
+        p.communicate()
+
+        time.sleep(10)
+
+time.sleep(5)
+
+if uninstall:
+    for package in templates['packages']:
+        repo_name = package['repository']['name']
+        repo_url = package['repository']['url']
+        chart_name = package['name']
+        version = package['version']
+
+        command = "helm uninstall --namespace {chart_name3} my-{chart_name} --timeout 45s".format(chart_name3=chart_name, chart_name=chart_name)
+        print("######" + command)
+        p = Popen(command, stdout=PIPE, stderr=PIPE, shell=True)
+        p.communicate()
+
+        time.sleep(6)