From e242057583774d24bafc3dd3e574d6f950bef4a8 Mon Sep 17 00:00:00 2001
From: Tixx <83774803+WiserTixx@users.noreply.github.com>
Date: Sun, 22 Sep 2024 23:58:32 +0200
Subject: [PATCH] Improve port cli flag

---
 src/Options.cpp | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/Options.cpp b/src/Options.cpp
index 64a631c..9cd9b7c 100644
--- a/src/Options.cpp
+++ b/src/Options.cpp
@@ -11,16 +11,23 @@ void InitOptions(int argc, char *argv[], Options &options) {
     while (i < argc) {
         std::string argument(argv[i]);
         if (argument == "-p" || argument == "--port") {
-            if (argc > i + 1) {
+            if (argc > i) {
                 std::string error_message =
                     "No port specified, resorting to default (";
-                error_message += options.port;
+                error_message += std::to_string(options.port);
                 error_message += ")";
                 error(error_message);
+                i++;
                 continue;
             }
 
-            int port = atoi(argv[i + 1]);
+            int port = options.port;
+
+            try {
+                port = std::stoi(argv[i + 1]);
+            } catch (std::exception& e) { 
+                error("Invalid port specified: " + std::string(argv[i + 1]) + " " + std::string(e.what()));
+            }
 
             if (port <= 0) {
                 std::string error_message =
@@ -29,6 +36,7 @@ void InitOptions(int argc, char *argv[], Options &options) {
                 error_message += options.port;
                 error_message += ")";
                 error(error_message);
+                i++;
                 continue;
             }