Util: Add file class
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
/*
|
||||
* Copyright (C) 2021 Riyyi
|
||||
*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*/
|
||||
|
||||
#include <cassert> // assert
|
||||
#include <cstdint> // int32_t
|
||||
#include <fstream> // ifstream, ios, ofstream
|
||||
#include <memory> // make_unique
|
||||
#include <string>
|
||||
|
||||
#include "util/file.h"
|
||||
|
||||
namespace Util {
|
||||
|
||||
File::File(const std::string& path)
|
||||
: m_path(path)
|
||||
{
|
||||
// Create input stream object and open file
|
||||
std::ifstream file(path, std::ios::in);
|
||||
assert(file.is_open());
|
||||
|
||||
// Get length of the file
|
||||
file.seekg(0, std::ios::end);
|
||||
int32_t size = file.tellg();
|
||||
file.seekg(0, std::ios::beg);
|
||||
assert(size != -1);
|
||||
|
||||
// Allocate memory filled with zeros
|
||||
auto buffer = std::make_unique<char[]>(size);
|
||||
|
||||
// Fill buffer with file contents
|
||||
file.read(buffer.get(), size);
|
||||
file.close();
|
||||
|
||||
m_data = std::string(buffer.get(), size);
|
||||
}
|
||||
|
||||
File::~File()
|
||||
{
|
||||
}
|
||||
|
||||
// -----------------------------------------
|
||||
|
||||
void File::clear()
|
||||
{
|
||||
m_data.clear();
|
||||
}
|
||||
|
||||
File& File::append(std::string data)
|
||||
{
|
||||
m_data.append(data);
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
File& File::flush()
|
||||
{
|
||||
// Create output stream object and open file
|
||||
std::ofstream file(m_path, std::ios::out | std::ios::trunc);
|
||||
assert(file.is_open());
|
||||
|
||||
// Write data to disk
|
||||
file.write(m_data.c_str(), m_data.size());
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
} // namespace Util
|
||||
@@ -0,0 +1,35 @@
|
||||
/*
|
||||
* Copyright (C) 2021 Riyyi
|
||||
*
|
||||
* SPDX-License-Identifier: MIT
|
||||
*/
|
||||
|
||||
#ifndef FILE_H
|
||||
#define FILE_H
|
||||
|
||||
#include <string>
|
||||
|
||||
namespace Util {
|
||||
|
||||
class File {
|
||||
public:
|
||||
File(const std::string& path);
|
||||
virtual ~File();
|
||||
|
||||
void clear();
|
||||
File& append(std::string data);
|
||||
File& flush();
|
||||
|
||||
const char* c_str() const { return m_data.c_str(); }
|
||||
const std::string& data() const { return m_data; }
|
||||
const std::string& path() const { return m_path; }
|
||||
|
||||
private:
|
||||
std::string m_path;
|
||||
std::string m_data;
|
||||
};
|
||||
|
||||
} // namespace Util
|
||||
|
||||
|
||||
#endif // FILE_H
|
||||
Reference in New Issue
Block a user