From 87f89841ac74b0f7d255820c8ab27d4efce1b726 Mon Sep 17 00:00:00 2001
From: Dnomd343 <i@343.re>
Date: Fri, 28 Oct 2022 16:59:48 +0800
Subject: [PATCH] feat: load options from env

---
 kms.php | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/kms.php b/kms.php
index c74befa..2082206 100755
--- a/kms.php
+++ b/kms.php
@@ -75,17 +75,21 @@ function get_param(string $field, string $default): string {
 
 function load_params(): void {
     global $argv;
-    if (in_array('--debug', $argv)) { // enter debug mode
-        logging::$logLevel = logging::DEBUG;
+    if (strtolower(getenv('DEBUG')) === 'true' || in_array('--debug', $argv)) {
+        logging::$logLevel = logging::DEBUG; // enter debug mode
     }
 
     global $ENABLE_HTTP;
-    if (in_array('--disable-http', $argv)) { // disable http service
+    if (strtolower(getenv('DISABLE_HTTP')) === 'true' || in_array('--disable-http', $argv)) {
         logging::warning('Disable http service');
-        $ENABLE_HTTP = false;
+        $ENABLE_HTTP = false; // disable http service
     }
 
     global $KMS_PORT;
+    if (getenv('KMS_PORT')) {
+        $KMS_PORT = intval(getenv('KMS_PORT'));
+        logging::debug('Get KMS_PORT from env -> ' . $KMS_PORT);
+    }
     $KMS_PORT = intval(get_param('--kms-port', strval($KMS_PORT)));
     if ($KMS_PORT < 1 || $KMS_PORT > 65535) { // 1 ~ 65535
         logging::critical('Illegal KMS Port -> ' . $KMS_PORT);
@@ -98,6 +102,10 @@ function load_params(): void {
     }
 
     global $HTTP_PORT;
+    if (getenv('HTTP_PORT')) {
+        $HTTP_PORT = intval(getenv('HTTP_PORT'));
+        logging::debug('Get HTTP_PORT from env -> ' . $HTTP_PORT);
+    }
     $HTTP_PORT = intval(get_param('--http-port', strval($HTTP_PORT)));
     if ($HTTP_PORT < 1 || $HTTP_PORT > 65535) { // 1 ~ 65535
         logging::critical('Illegal HTTP Port -> ' . $HTTP_PORT);