WiiSave: Use an enum class for save file type

This commit is contained in:
Léo Lam 2018-06-01 18:58:34 +02:00
parent 94953670f2
commit 8af16fdd28
1 changed files with 16 additions and 7 deletions

View File

@ -116,7 +116,13 @@ class Storage
public: public:
struct SaveFile struct SaveFile
{ {
u8 mode, attributes, type; enum class Type : u8
{
File = 1,
Directory = 2,
};
u8 mode, attributes;
Type type;
std::string path; std::string path;
// Only valid for regular (i.e. non-directory) files. // Only valid for regular (i.e. non-directory) files.
Common::Lazy<std::optional<std::vector<u8>>> data; Common::Lazy<std::optional<std::vector<u8>>> data;
@ -192,7 +198,7 @@ public:
SaveFile save_file; SaveFile save_file;
save_file.mode = 0x3c; save_file.mode = 0x3c;
save_file.attributes = 0; save_file.attributes = 0;
save_file.type = file_info.IsDirectory() ? 2 : 1; save_file.type = file_info.IsDirectory() ? SaveFile::Type::Directory : SaveFile::Type::File;
save_file.path = Common::UnescapeFileName(path.substr(m_wii_title_path.length() + 1)); save_file.path = Common::UnescapeFileName(path.substr(m_wii_title_path.length() + 1));
save_file.data = [path]() -> std::optional<std::vector<u8>> { save_file.data = [path]() -> std::optional<std::vector<u8>> {
File::IOFile file{path, "rb"}; File::IOFile file{path, "rb"};
@ -224,7 +230,7 @@ public:
std::string file_path_full = m_wii_title_path + '/' + file_path; std::string file_path_full = m_wii_title_path + '/' + file_path;
File::CreateFullPath(file_path_full); File::CreateFullPath(file_path_full);
if (file.type == 1) if (file.type == SaveFile::Type::File)
{ {
File::IOFile raw_save_file(file_path_full, "wb"); File::IOFile raw_save_file(file_path_full, "wb");
const std::optional<std::vector<u8>>& data = *file.data; const std::optional<std::vector<u8>>& data = *file.data;
@ -232,7 +238,7 @@ public:
return false; return false;
raw_save_file.WriteBytes(data->data(), data->size()); raw_save_file.WriteBytes(data->data(), data->size());
} }
else if (file.type == 2) else if (file.type == SaveFile::Type::Directory)
{ {
File::CreateDir(file_path_full); File::CreateDir(file_path_full);
if (!File::IsDirectory(file_path_full)) if (!File::IsDirectory(file_path_full))
@ -360,9 +366,12 @@ public:
save_file.mode = file_hdr.permissions; save_file.mode = file_hdr.permissions;
save_file.attributes = file_hdr.attrib; save_file.attributes = file_hdr.attrib;
save_file.type = file_hdr.type; const SaveFile::Type type = static_cast<SaveFile::Type>(file_hdr.type);
if (type != SaveFile::Type::Directory && type != SaveFile::Type::File)
return {};
save_file.type = type;
save_file.path = file_hdr.name.data(); save_file.path = file_hdr.name.data();
if (file_hdr.type == 1) if (type == SaveFile::Type::File)
{ {
const u32 rounded_size = Common::AlignUp<u32>(file_hdr.size, BLOCK_SZ); const u32 rounded_size = Common::AlignUp<u32>(file_hdr.size, BLOCK_SZ);
const u64 pos = m_file.Tell(); const u64 pos = m_file.Tell();
@ -409,7 +418,7 @@ public:
file_hdr.magic = FILE_HDR_MAGIC; file_hdr.magic = FILE_HDR_MAGIC;
file_hdr.permissions = save_file.mode; file_hdr.permissions = save_file.mode;
file_hdr.attrib = save_file.attributes; file_hdr.attrib = save_file.attributes;
file_hdr.type = save_file.type; file_hdr.type = static_cast<u8>(save_file.type);
if (save_file.path.length() > 0x44) if (save_file.path.length() > 0x44)
return false; return false;
std::strncpy(file_hdr.name.data(), save_file.path.data(), file_hdr.name.size()); std::strncpy(file_hdr.name.data(), save_file.path.data(), file_hdr.name.size());