Add space support to flags

src/tl_app.c

@@ -36,9 +36,16 @@ bool tl_lookup_flag(const char *flag) {
 const char *tl_get_flag(const char *flag) {
     size_t flag_len = strlen(flag);
     for (int i = 1; i < arg_count; i++) {
-        // If the argument starts with the flag and is followed by '=' then
-        if (strncmp(args[i], flag, flag_len) == 0 && args[i][flag_len] == '=') {
-            return args[i] + flag_len + 1; // +1 to skip the '='
+        if (strncmp(args[i], flag, flag_len) != 0) {
+            continue;
+        }
+        // If the argument is followed by '=' then return the value after '='
+        if (args[i][flag_len] == '=') {
+            return args[i] + flag_len + 1;
+        }
+        // If the argument is an exact match and the next argument exists then return it
+        if (args[i][flag_len] == '\0' && i + 1 < arg_count) {
+            return args[i + 1];
         }
     }
     return NULL;

tests/unit/tl_app_test.c

@@ -36,13 +36,28 @@ static void test_tl_get_flag(void) {
     TEST_ASSERT_NULL(tl_get_flag("--nonexistent-key"));
 }
 
+static void test_tl_get_flag_space(void) {
+    char *argv[] = {"program", "--key", "value"};
+    tl_init_app(3, argv);
+    TEST_ASSERT_EQUAL_STRING("value", tl_get_flag("--key"));
+}
+
+static void test_tl_init_app_space(void) {
+    char *argv[] = {"program", "--debug-level", "3"};
+    tl_init_app(3, argv);
+    TEST_ASSERT_TRUE(tl_lookup_flag("--debug-level"));
+    TEST_ASSERT_EQUAL_STRING("3", tl_get_flag("--debug-level"));
+}
+
 int main(void) {
     UNITY_BEGIN();
 
     RUN_TEST(test_tl_init_app);
     RUN_TEST(test_tl_parse_args);
     RUN_TEST(test_tl_lookup_flag);
     RUN_TEST(test_tl_get_flag);
+    RUN_TEST(test_tl_get_flag_space);
+    RUN_TEST(test_tl_init_app_space);
 
     return UNITY_END();
 }