42 lines
1.1 KiB
C++
42 lines
1.1 KiB
C++
|
#include <luna/Base64.h>
|
||
|
|
#include <os/ArgumentParser.h>
|
||
|
|
#include <os/File.h>
|
||
|
|
|
||
|
|
Result<int> luna_main(int argc, char** argv)
|
||
|
|
{
|
||
|
|
StringView path;
|
||
|
|
bool decode { false };
|
||
|
|
bool allow_garbage { false };
|
||
|
|
|
||
|
|
os::ArgumentParser parser;
|
||
|
|
parser.add_description("Encode or decode Base64 data. If not given a file, reads from standard input.");
|
||
|
|
parser.add_positional_argument(path, "file", "-"_sv);
|
||
|
|
parser.add_switch_argument(decode, 'd', "decode", "decode data");
|
||
|
|
parser.add_switch_argument(allow_garbage, ' ', "allow-garbage", "ignore non-base64 characters");
|
||
|
|
parser.parse(argc, argv);
|
||
|
|
|
||
|
|
auto file = TRY(os::File::open_input_file(path));
|
||
|
|
|
||
|
|
auto output = os::File::standard_output();
|
||
|
|
|
||
|
|
if (!decode)
|
||
|
|
{
|
||
|
|
auto data = TRY(file->read_all());
|
||
|
|
|
||
|
|
auto encoded = TRY(Base64::encode(data));
|
||
|
|
|
||
|
|
output->write(encoded.view());
|
||
|
|
output->write("\n"_sv);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
auto data = TRY(file->read_all_as_string());
|
||
|
|
|
||
|
|
auto decoded = TRY(Base64::decode(data.view(), allow_garbage));
|
||
|
|
|
||
|
|
output->write(decoded);
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|