change if main fn

diff --git a/src/main.c b/src/main.c
index 3ffa76484140ad093166d4389fa856c9695ccf06..5f0bb2fb28943817f56fd1e52a3142c37f445d52 100644 (file)
--- a/src/main.c
+++ b/src/main.c
@@ -73,17 +73,22 @@ int main(int argc, char* argv[])
        printf("Version %s\n", VERSION);
        if (argc < 2) {
                printf("ERROR: %s <dir>\n", argv[0]);
-               return 1;
+               return EXIT_FAILURE;
        }
 
-       if (argc == 3 && strcmp(argv[2], "lower") == 0)
-               rename_files(argv[1], rename_lower);
-       else if (argc == 3 && strcmp(argv[2], "point") == 0)
-               rename_files(argv[1], rename_point);
-       else if (argc == 3 && strcmp(argv[2], "revert") == 0)
-               rename_files(argv[1], rename_revert);
-       else
-               rename_files(argv[1], rename_string);
+       const char* dir = argv[1];
+       if (argc == 2)
+               rename_files(dir, rename_string);
 
-       return 0;
+       const char* cmd = argv[2];
+       if (strcmp(cmd, "lower") == 0)
+               rename_files(dir, rename_lower);
+
+       if (strcmp(cmd, "point") == 0)
+               rename_files(dir, rename_point);
+
+       if (strcmp(cmd, "revert") == 0)
+               rename_files(dir, rename_revert);
+
+       return EXIT_SUCCESS;
 }
\ No newline at end of file