fix: Merge pull request #23 from abolfazl8131/fix/helm

fix helm prompt for secret generation

Author: abolfazl8131

app/__pycache__/prompt_generators.cpython-311.pyc

@@ -1,28 +1,21 @@
 import os
 
 # Define project structure
-project_name = 'app/media/MyHelm'
-directories = ['charts/', 'templates/']
-templates_dir = 'templates/web'
-files = ['Chart.yaml', 'values.yaml']
-template_subdirectories = ['web']
-template_files = ['service.yaml', 'helpers.tpl']
-
-# Chart.yaml content
-chart_yaml_content = """apiVersion: v2
+project_name = "app/media/MyHelm"
+directories = ["charts", "templates/web"]
+files = ["Chart.yaml", "values.yaml"]
+template_files = ["service.yaml", "secret.yaml"]
+chart_content = """apiVersion: v2
 name: my-helm-chart
 description: A Helm chart for Kubernetes
 version: 0.1.0
 """
-
-# values.yaml content
-values_yaml_content = """web:
+values_content = """web:
   image: nginx
   replicas: 1
   service:
-    targetPort: 80
+    port: 80
   persistence:
-    enabled: true
     size: 1Gi
     accessModes:
       - ReadWriteOnce
@@ -34,37 +27,41 @@
     host: www.example.com
 """
 
-# Create project structure
+# Create project directories and files
 os.makedirs(project_name, exist_ok=True)
 for directory in directories:
     os.makedirs(os.path.join(project_name, directory), exist_ok=True)
 
-# Create templates/web directory
-os.makedirs(os.path.join(project_name, templates_dir), exist_ok=True)
-
-# Create files
-with open(os.path.join(project_name, 'Chart.yaml'), 'w') as chart_file:
-    chart_file.write(chart_yaml_content)
+for file in files:
+    with open(os.path.join(project_name, file), 'w') as f:
+        if file == "Chart.yaml":
+            f.write(chart_content)
+        elif file == "values.yaml":
+            f.write(values_content)
 
-with open(os.path.join(project_name, 'values.yaml'), 'w') as values_file:
-    values_file.write(values_yaml_content)
-
-# Create template files and directories
-for template in template_subdirectories:
-    template_path = os.path.join(project_name, 'templates', template)
-    os.makedirs(template_path, exist_ok=True)
+# Create template files
+for template in directories[1:]:
     for template_file in template_files:
-        with open(os.path.join(template_path, template_file), 'w') as file:
-            file.write(f"# {template_file} for {template}\n")
-
-# Create secret.yaml if env variable is considered
-with open(os.path.join(template_path, 'secret.yaml'), 'w') as secret_file:
-    secret_file.write("# secret.yaml for environment variables\n")
-
-# Create helpers.tpl
-with open(os.path.join(template_path, 'helpers.tpl'), 'w') as helpers_file:
-    helpers_file.write(f"""{{- define "{template}.labels" -}}
-name: {{ .Release.Name }}
-{{- end -}}
-// Add additional helper functions if needed
+        with open(os.path.join(project_name, template, template_file), 'w') as f:
+            if template_file == "service.yaml":
+                f.write("""apiVersion: v1
+kind: Service
+metadata:
+  name: {{ include "{{ .Chart.Name }}.fullname" . }}
+spec:
+  type: ClusterIP
+  ports:
+    - port: {{ .Values.web.service.port }}
+      targetPort: {{ .Values.web.service.port }}
+  selector:
+    app: {{ include "{{ .Chart.Name }}.fullname" . }}
+""")
+            elif template_file == "secret.yaml":
+                f.write("""apiVersion: v1
+kind: Secret
+metadata:
+  name: {{ include "{{ .Chart.Name }}.fullname" . }}-secret
+type: Opaque
+data:
+  ENV1: {{ .Values.web.env | toJson | b64enc | quote }}
 """)

app/directory_generators/helm_generator.py

@@ -1 +1,7 @@
-# secret.yaml for environment variables
+apiVersion: v1
+kind: Secret
+metadata:
+  name: {{ include "{{ .Chart.Name }}.fullname" . }}-secret
+type: Opaque
+data:
+  ENV1: {{ .Values.web.env | toJson | b64enc | quote }}

app/media/MyHelm/templates/web/helpers.tpl

@@ -1 +1,11 @@
-# service.yaml for web
+apiVersion: v1
+kind: Service
+metadata:
+  name: {{ include "{{ .Chart.Name }}.fullname" . }}
+spec:
+  type: ClusterIP
+  ports:
+    - port: {{ .Values.web.service.port }}
+      targetPort: {{ .Values.web.service.port }}
+  selector:
+    app: {{ include "{{ .Chart.Name }}.fullname" . }}

app/media/MyHelm/templates/web/secret.yaml

@@ -2,9 +2,8 @@ web:
   image: nginx
   replicas: 1
   service:
-    targetPort: 80
+    port: 80
   persistence:
-    enabled: true
     size: 1Gi
     accessModes:
       - ReadWriteOnce

app/media/MyHelm/templates/web/service.yaml

@@ -82,14 +82,18 @@ def helm_template_generator(input : HelmTemplateGeneration) -> str:
             for each template, initialize this file => service.yaml.
             set replicas of pods following this dict format : {replicas_}.
             set persistance (pvc) of pods following this dict fomrat : {persistance}
-            set environment variables of pods like this dict format : {envs}
+            set environment variables of pods following this dict format : {envs} based on helm standard environment setting.(
+                for example something like that:
+                    env:
+                     name=value
+            )
             initialize ingress with a default host for pod if the pod ingress is true in here {ingress_}.
             
             if environment variable is considered for pod, then create secret.yaml in the related template.
+            creating secret.yaml based on environemt variables in the {envs} in the template is very important.
 
             please set a something default in chart.yaml and values.yaml based on the requirement.
-            please generate helpers.tpl for each template and initialize resources and set a label for reach resource.
-
+           
             just Generate a python code without any additional notes or ```python3 entry
             """
     return prompt