diff --git a/src/main.cpp b/src/main.cpp index 025071c..10cdb40 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -22,6 +22,8 @@ void input(int argc, char *argv[]) command.addCommand("run", runScript); command.addCommand("help", help); command.addCommand("ls", listScripts); + command.addCommand("add", addScript); + command.addCommand("edit", editScript); command.addDefaultCommand(runScript); command.runCommand(argv[1], argc, argv); } @@ -48,6 +50,49 @@ void listScripts(int argc, char *argv[]) } } +// add a script in the autom directory +void addScript(int argc, char *argv[]) +{ + + if (std::filesystem::exists(dir + "/" + argv[1])) + { + std::cout << "Script " << argv[1] <<" already exists" << std::endl; + return; + } + + std::cout << "Adding script: " << argv[1] << std::endl; + std::string script = dir + "/" + argv[1]; + std::ofstream file(script); + +#ifdef _WIN32 + file << "@echo off" << std::endl; + _chmod(script.c_str(), _S_IREAD | _S_IWRITE); +#else + file << "#!/bin/bash" << std::endl; + system(("chmod +x " + script).c_str()); +#endif + + file.close(); + + editScript(argv[1]); +} + +// edit a script in the autom directory +void editScript(int argc, char *argv[]) +{ + editScript(argv[1]); +} + +void editScript(std::string name) +{ + std::string script = dir + "/" + name; +#ifdef _WIN32 + system(("notepad " + script).c_str()); +#else + system(("nano " + script).c_str()); +#endif +} + // help function for showing help message void help(int argc, char *argv[]) { diff --git a/src/main.h b/src/main.h index 946c734..8555de4 100644 --- a/src/main.h +++ b/src/main.h @@ -3,6 +3,7 @@ #include #include #include +#include #include "inputparser.h" #include "command.h" @@ -24,6 +25,11 @@ void input(int argc, char *argv[]); void runScript(int argc, char *argv[]); // list all scripts in the autom directory void listScripts(int argc, char *argv[]); +// add a script in the autom directory +void addScript(int argc, char *argv[]); +// edit a script in the autom directory +void editScript(int argc, char *argv[]); +void editScript(std::string name); // help function for showing help message void help(int argc, char *argv[]); // void create(int argc,char *argv[]);