91 lines
3.1 KiB
C++
91 lines
3.1 KiB
C++
|
#include "file.h"
|
||
|
|
#include "../../defs/defs.h"
|
||
|
|
#include "../../error/error.h"
|
||
|
|
#include <vector>
|
||
|
|
#include <fstream>
|
||
|
|
|
||
|
|
Value modules::file(std::vector<Value> args) {
|
||
|
|
if (args.size() < 2) {
|
||
|
|
error("Not enough args for file module");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
if (args[1].valtype != ValueType::String) {
|
||
|
|
error("Expecting a string to identify file to open");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
if (args[0].valtype == ValueType::Identifier) {
|
||
|
|
if (args[0].string_val == "read") {
|
||
|
|
std::ifstream file(args[1].string_val);
|
||
|
|
if (!file) {
|
||
|
|
error("File " + args[1].string_val + " does not exist (in file module)");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
std::string buf;
|
||
|
|
std::string retval;
|
||
|
|
bool notfirst = false;
|
||
|
|
while (getline(file, buf)) {
|
||
|
|
if (notfirst) {
|
||
|
|
buf += "\n";
|
||
|
|
} else {
|
||
|
|
notfirst = true;
|
||
|
|
}
|
||
|
|
retval += buf;
|
||
|
|
}
|
||
|
|
file.close();
|
||
|
|
return Value(retval);
|
||
|
|
} else if (args[0].string_val == "readlines") {
|
||
|
|
std::ifstream file(args[1].string_val);
|
||
|
|
if (!file) {
|
||
|
|
error("File " + args[1].string_val + " does not exist (in file module)");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
std::string buf;
|
||
|
|
std::vector<Value> retval;
|
||
|
|
while (getline(file, buf)) {
|
||
|
|
retval.push_back(Value(buf));
|
||
|
|
}
|
||
|
|
file.close();
|
||
|
|
return Value(retval);
|
||
|
|
} else if (args[0].string_val == "exists") {
|
||
|
|
if (std::ifstream(args[1].string_val)) {
|
||
|
|
return Value("true");
|
||
|
|
} else {
|
||
|
|
return Value("false");
|
||
|
|
}
|
||
|
|
} else if (args[0].string_val == "write") {
|
||
|
|
if (args.size() < 3) {
|
||
|
|
error("Not enough args for write submodule of file module");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
if (args[2].valtype != ValueType::String) {
|
||
|
|
error("Expecting a string of content to write to file");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
std::ofstream file(args[1].string_val, std::ofstream::trunc);
|
||
|
|
file << args[2].string_val;
|
||
|
|
file.close();
|
||
|
|
return Value("true");
|
||
|
|
} else if (args[0].string_val == "append") {
|
||
|
|
if (args.size() < 3) {
|
||
|
|
error("Not enough args for write submodule of file module");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
if (args[2].valtype != ValueType::String) {
|
||
|
|
error("Expecting a string of content to write to file");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
std::ofstream file(args[1].string_val, std::iostream::app);
|
||
|
|
file << args[2].string_val;
|
||
|
|
file.close();
|
||
|
|
|
||
|
|
} else {
|
||
|
|
error("Unknown submodule of file");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
error("Expecting identifier of submodule in file module");
|
||
|
|
return Value("Error");
|
||
|
|
}
|
||
|
|
return Value("Error");
|
||
|
|
}
|