72 lines
1.6 KiB
C++
72 lines
1.6 KiB
C++
#pragma once
|
|
#include "prelude/string.hpp"
|
|
#include <cstdio>
|
|
|
|
class File
|
|
{
|
|
public:
|
|
enum class Mode
|
|
{
|
|
WRITE,
|
|
READ,
|
|
};
|
|
|
|
private:
|
|
File(FILE* handle, Mode mode)
|
|
: m_handle(handle), m_mode(mode) {}
|
|
|
|
public:
|
|
static File Open(const StringView& filename, Mode mode)
|
|
{
|
|
StringBuilder sb;
|
|
switch(mode)
|
|
{
|
|
case Mode::WRITE:
|
|
sb.Append('w');
|
|
break;
|
|
case Mode::READ:
|
|
sb.Append('r');
|
|
break;
|
|
}
|
|
FILE* handle = fopen(filename.c_str(), sb.c_str());
|
|
return File(handle, mode);
|
|
}
|
|
|
|
private:
|
|
inline bool Writable() const { return m_mode == File::Mode::WRITE; }
|
|
inline bool Readable() const { return m_mode == File::Mode::READ; }
|
|
|
|
public:
|
|
bool Write(const StringView& content)
|
|
{
|
|
assert(Writable() && "attempt to write to a file in read only mode");
|
|
if (fwrite(content.c_str(), sizeof(char), content.len(), m_handle) != content.len())
|
|
{
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public:
|
|
bool ReadAll(StringBuilder& sb)
|
|
{
|
|
assert(Readable() && "attempt to read from a file in write only mode");
|
|
if (fseek(m_handle, 0, SEEK_END) != 0) return false;
|
|
|
|
auto size = ftell(m_handle);
|
|
if (size == -1) return false;
|
|
|
|
sb.ensure_extra(size);
|
|
|
|
size_t wrote = fread(sb.data, sizeof(char), size, m_handle);
|
|
if (wrote == 0) return false;
|
|
|
|
sb.size = wrote;
|
|
return true;
|
|
}
|
|
|
|
private:
|
|
FILE* m_handle;
|
|
Mode m_mode;
|
|
};
|