From 38c05a3b8a019621c8e01e0dd19dac306a99431b Mon Sep 17 00:00:00 2001
From: Luis Lopes <josluis.lopes@gmail.com>
Date: Thu, 3 Nov 2022 03:34:29 +0000
Subject: [PATCH] Improve portainer settings. Allow enterprise edition to be
 set (#781)

---
 public/v4/apps/portainer.yml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/public/v4/apps/portainer.yml b/public/v4/apps/portainer.yml
index 89f3b9c..976e07b 100644
--- a/public/v4/apps/portainer.yml
+++ b/public/v4/apps/portainer.yml
@@ -1,7 +1,7 @@
 captainVersion: 4
 services:
     $$cap_appname:
-        image: portainer/portainer-ce:$$cap_portainer_version
+        image: portainer/portainer-$$cap_portainer_edition:$$cap_portainer_version
         volumes:
             - /var/run/docker.sock:/var/run/docker.sock
             - $$cap_appname-data:/data
@@ -15,6 +15,11 @@ caproverOneClickApp:
           defaultValue: 2.0.0
           description: Check out their Docker page for the valid tags https://hub.docker.com/r/portainer/portainer-ce/tags
           validRegex: /^([^\s^\/])+$/
+        - id: $$cap_portainer_edition
+          label: Portainer Edition
+          defaultValue: 'ce'
+          description: 'Community edition (ce) or Enterprise edition (ee)'
+          validRegex: /^(?:ce|ee)$/
     instructions:
         start: >-
             Portainer is a lightweight management UI which allows you to easily manage your Docker host or Swarm cluster.