add file.gzip_compress_write

This commit is contained in:
Xertis 2024-06-06 12:17:37 +03:00
parent ad62b759fb
commit bfeb32841f

View File

@ -3,6 +3,7 @@
#include "LuaState.hpp"
#include "../scripting.hpp"
#include "../../../engine.hpp"
#include "../../../coders/gzip.hpp"
#include "../../../files/files.hpp"
#include "../../../files/engine_paths.hpp"
#include "../../../util/stringutil.hpp"
@ -129,7 +130,7 @@ static int l_file_read_bytes(lua_State* L) {
int newTable = lua_gettop(L);
for(size_t i = 0; i < length; i++) {
lua_pushnumber(L, bytes[i]);
lua_pushinteger(L, bytes[i]);
lua_rawseti(L, newTable, i+1);
}
return 1;
@ -206,6 +207,24 @@ static int l_file_list(lua_State* L) {
return 1;
}
static int l_file_gzip_compress(lua_State* L) {
fs::path path = resolve_path(state->requireString(1));
if (fs::is_regular_file(path)) {
size_t length = static_cast<size_t>(fs::file_size(path));
auto bytesPtr = files::read_bytes(path, length);
std::vector<ubyte> bytes (bytesPtr.get(), bytesPtr.get() + length);
std::vector<ubyte> compressed_bytes;
compressed_bytes = gzip::compress(bytes.data(), bytes.size());
lua_pushboolean(L, files::write_bytes(path, compressed_bytes.data(), compressed_bytes.size()));
return 1;
}
throw std::runtime_error("file does not exist " + util::quote(path.u8string()));
}
const luaL_Reg filelib [] = {
{"exists", lua_wrap_errors<l_file_exists>},
{"find", lua_wrap_errors<l_file_find>},
@ -222,5 +241,6 @@ const luaL_Reg filelib [] = {
{"resolve", lua_wrap_errors<l_file_resolve>},
{"write_bytes", lua_wrap_errors<l_file_write_bytes>},
{"write", lua_wrap_errors<l_file_write>},
{"gzip_compress_write", lua_wrap_errors<l_file_gzip_compress>},
{NULL, NULL}
};