diff --git a/src/cowtool.cpp b/src/cowtool.cpp index 366dff1..abb8c63 100644 --- a/src/cowtool.cpp +++ b/src/cowtool.cpp @@ -72,9 +72,9 @@ int main(int argc, char* argv[]) { if (!std::filesystem::exists("/etc/cowtool/disabled/install_"+name+".sh")) { - std::cout << "First time installation! Watching you! Type 'manure' to complete the script!" << std::endl; + std::cout << "First time installation! Watching you! Type 'manure' to cancel, 'exit' to save!" << std::endl; chdir("/sources/cowtool/buildspace/"); - std::cout << "Entered buildspace - don't for get to clean up before exiting!" << std::endl; + std::cout << "Entered buildspace - don't forget to clean up before exiting!" << std::endl; std::ofstream out; out.open("/etc/cowtool/disabled/install_"+name+".sh", std::ios::app); @@ -96,9 +96,14 @@ int main(int argc, char* argv[]) { out.close(); } - - system((editor+" /etc/cowtool/disabled/install_"+name+".sh").c_str()); - std::cout << "Remember to `milk` to update `"+ name << "`!" << std::endl; + if (input=="manure") { + // Cancel! + system(("rm /etc/cowtool/disabled/install_"+name+".sh").c_str()); + std::cout << "Cancelled `"+ name << "`!" << std::endl; + } else { + system((editor+" /etc/cowtool/disabled/install_"+name+".sh").c_str()); + std::cout << "Remember to `milk` to update `"+ name << "`!" << std::endl; + } } else if (firstArg == "milk") {