mirror of
https://github.com/RGBCube/serenity
synced 2025-05-14 08:04:57 +00:00
d5871f5717
Similar to POSIX read, the basic read and write functions of AK::Stream do not have a lower limit of how much data they read or write (apart from "none at all"). Rename the functions to "read some [data]" and "write some [data]" (with "data" being omitted, since everything here is reading and writing data) to make them sufficiently distinct from the functions that ensure to use the entire buffer (which should be the go-to function for most usages). No functional changes, just a lot of new FIXMEs.
63 lines
2.1 KiB
C++
63 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2020, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibCompress/Gzip.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibMain/Main.h>
|
|
#include <unistd.h>
|
|
|
|
static ErrorOr<void> decompress_file(NonnullOwnPtr<Core::File> input_stream, Stream& output_stream)
|
|
{
|
|
auto gzip_stream = Compress::GzipDecompressor { move(input_stream) };
|
|
|
|
auto buffer = TRY(ByteBuffer::create_uninitialized(4096));
|
|
while (!gzip_stream.is_eof()) {
|
|
auto span = TRY(gzip_stream.read_some(buffer));
|
|
TRY(output_stream.write_entire_buffer(span));
|
|
}
|
|
|
|
return {};
|
|
}
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments args)
|
|
{
|
|
Vector<StringView> filenames;
|
|
bool keep_input_files { false };
|
|
bool write_to_stdout { false };
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_option(keep_input_files, "Keep (don't delete) input files", "keep", 'k');
|
|
args_parser.add_option(write_to_stdout, "Write to stdout, keep original files unchanged", "stdout", 'c');
|
|
args_parser.add_positional_argument(filenames, "File to decompress", "FILE");
|
|
args_parser.parse(args);
|
|
|
|
if (write_to_stdout)
|
|
keep_input_files = true;
|
|
|
|
for (auto filename : filenames) {
|
|
|
|
DeprecatedString input_filename;
|
|
DeprecatedString output_filename;
|
|
if (filename.ends_with(".gz"sv)) {
|
|
input_filename = filename;
|
|
output_filename = filename.substring_view(0, filename.length() - 3);
|
|
} else {
|
|
input_filename = DeprecatedString::formatted("{}.gz", filename);
|
|
output_filename = filename;
|
|
}
|
|
|
|
auto input_stream_result = TRY(Core::File::open(input_filename, Core::File::OpenMode::Read));
|
|
auto output_stream = write_to_stdout ? TRY(Core::File::standard_output()) : TRY(Core::File::open(output_filename, Core::File::OpenMode::Write));
|
|
|
|
TRY(decompress_file(move(input_stream_result), *output_stream));
|
|
|
|
if (!keep_input_files)
|
|
TRY(Core::System::unlink(input_filename));
|
|
}
|
|
return 0;
|
|
}
|