From 47cf7ca86d38621879792a102ebeeb7e6a58bbf1 Mon Sep 17 00:00:00 2001
From: The_Spider <nhyatt@smoothnet.org>
Date: Thu, 27 Jan 2022 11:41:02 -0600
Subject: [PATCH] podyaml now allows custom pod definations

---
 vars/functions.groovy | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/vars/functions.groovy b/vars/functions.groovy
index 7c99da0..b435899 100644
--- a/vars/functions.groovy
+++ b/vars/functions.groovy
@@ -1,6 +1,6 @@
 #!groovy
 
-def podYaml (Map config) {
+def podYaml (Map config, List customImages = [[:]]) {
     def output = """
 ---
 apiVersion: v1
@@ -8,8 +8,7 @@ kind: Pod
 metadata:
   name: ${config.templateName}
 spec:
-  containers:
-"""
+  containers:"""
     if (config.kaniko == true) {
         output += """
   - name: kaniko
@@ -17,24 +16,35 @@ spec:
     image: ${config.repo}/library/kaniko:latest
     tty: true
     command:
-    - /busybox/sh
-"""
+    - /busybox/sh"""
     }
     if (config.alpine == true) {
         output += """
   - name: alpine
     imagePullPolicy: Always
     image: ${config.repo}/library/alpine:latest
-    tty: true
-"""
+    tty: true"""
     }
     if (config.fedora == true) {
         output += """
   - name: fedora
     imagePullPolicy: Always
     image: ${config.repo}/dockerhub/library/fedora:latest
-    tty: true
-"""
+    tty: true"""
+    }
+    for (image in customImages) {
+        if (image.name && image.name != "" && image.path && image.path != "") {
+            output += """
+  - name: ${image.name}
+    imagePullPolicy: Always
+    image: ${image.path}
+    tty: true"""
+            if (image.command && image.command != "") {
+                output += """
+    command:
+    - ${image.command}"""
+            }
+        }
     }
     return output
 }