diff --git a/Src/blog.cpp b/Src/blog.cpp index 63619fe..b9e02c4 100644 --- a/Src/blog.cpp +++ b/Src/blog.cpp @@ -34,10 +34,8 @@ std::string getArgValue(int argc, char* argv[], std::string argIdString) if (strncmp(argId, argv[i], idValSplit) == 0) return std::string(argv[i]).substr(idValSplit+1, strlen(argv[i])-(idValSplit+1)); } - - std::cout << "Missing argument!"; //the program exists if the value is 0 - std::cout << argIdString; - exit(0); + printf("Missing argument! %s", argIdString); //the program exists if the value is 0 + return std::string("") } bool doesArgExist(int argc, char* argv[], std::string argIdString) @@ -57,6 +55,15 @@ void copyFolderAndContents(std::string inUrl, std::string outUrl) std::filesystem::copy(inUrl, outUrl, opts); } +void checkForEmptyArguments() +{ + // I hate doing it like this + if (source.empty() || output.empty() || stat.empty() || siteWebUrl.empty() || blogTemplate.empty() || atomTemplate.empty()) + { + exit(0); + } +} + /* * -in="" // give source folder * -out="" // give output folder @@ -72,6 +79,7 @@ int main(int argc, char* argv[]) std::string blogTemplate = getArgValue(argc, argv, "-blogTemplate="); std::string atomTemplate = getArgValue(argc, argv, "-atomTemplate="); + checkForEmptyArguments() int flags = OptionFlags::NONE; if (doesArgExist(argc, argv, "-hide")) @@ -82,7 +90,7 @@ int main(int argc, char* argv[]) // pre-written pages and resources i.e. images copyFolderAndContents(stat, output); - std::cout << "Enjoy your website :)"; + printf("Enjoy your website :)"); } // TODO: nav sections :eww: