Update INSTALL.sh

INSTALL.sh

@@ -51,6 +51,7 @@ SETUP_SWAP_ANYWAY=false
 SWAP_FILE="1"
 SELFHOSTED_SCREENSHOTS=false
 SEND_EMAIL_AFTER_INSTALL=false
+SET_PREMIUM=false
 
 # Paths
 LOG_FILE="openpanel_install.log"
@@ -244,7 +245,7 @@ FUNCTIONS=(
     temp_fix_for_skeleton
     setup_swap
     configure_mysql
-
+    set_premium_features
     start_services
     set_system_cronjob
     cleanup
@@ -361,10 +362,13 @@ parse_args() {
     }
 
 
-    for arg in "$@"; do
-        case $arg in
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        --key=*)
+            SET_PREMIUM=true
+            license_key="${1#*=}"
+            ;;
         --hostname=*)
-                # Extract domain after "--hostname="
             SET_HOSTNAME_NOW=true
             new_hostname="${1#*=}"
             ;;
@@ -416,25 +420,20 @@ parse_args() {
             INSTALL_MAIL=true
             ;;
         --post_install=*)
-                # Extract path after "--post_install="
             post_install_path="${1#*=}"
             ;;
         --screenshots=*)
-                # Extract path after "--screenshots="
             SCREENSHOTS_API_URL="${1#*=}"
             ;;
         --version=*)
-                # Extract path after "--version="
             CUSTOM_VERSION=true
             version="${1#*=}"
             ;;
         --swap=*)
-                # Extract path after "--swap="
             SETUP_SWAP_ANYWAY=true
             SWAP="${1#*=}"
             ;;
         --email=*)
-                # Extract path after "--email="
             SEND_EMAIL_AFTER_INSTALL=true
             EMAIL="${1#*=}"
             ;;
@@ -443,12 +442,14 @@ parse_args() {
             exit 0
             ;;
         *)
-                echo "Unknown option: $arg"
+            echo "Unknown option: $1"
             show_help
             exit 1
             ;;
     esac
-    done
+    shift
+done
+
 }
 
 detect_installed_panels() {
@@ -1208,6 +1209,15 @@ download_and_import_docker_images() {
 }
 
 
+set_premium_features(){
+ if [ "$SET_HOSTNAME_NOW" = true ]; then
+    echo "Setting OpenPanel enterprise version license key $license_key"
+    opencli config update key "$license_key"
+ fi
+}
+
+
+
 set_custom_hostname(){
         if [ "$SET_HOSTNAME_NOW" = true ]; then
             # Check if the provided hostname is a valid FQDN