60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
#include <nntpchan/storage.hpp>
|
|
#include <nntpchan/sanitize.hpp>
|
|
#include <cassert>
|
|
#include <sstream>
|
|
|
|
namespace nntpchan
|
|
{
|
|
ArticleStorage::ArticleStorage()
|
|
{
|
|
}
|
|
|
|
ArticleStorage::ArticleStorage(const fs::path & fpath) {
|
|
SetPath(fpath);
|
|
}
|
|
|
|
ArticleStorage::~ArticleStorage()
|
|
{
|
|
}
|
|
|
|
void ArticleStorage::SetPath(const fs::path & fpath)
|
|
{
|
|
basedir = fpath;
|
|
fs::create_directories(basedir);
|
|
assert(init_skiplist("posts_skiplist"));
|
|
}
|
|
|
|
bool ArticleStorage::init_skiplist(const std::string & subdir) const
|
|
{
|
|
fs::path skiplist = basedir / fs::path(subdir);
|
|
fs::create_directories(skiplist);
|
|
const auto subdirs = {"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "a", "b", "c", "d", "e", "f"};
|
|
for (const auto & s : subdirs)
|
|
fs::create_directories(skiplist / s);
|
|
return true;
|
|
}
|
|
|
|
bool ArticleStorage::Accept(const std::string& msgid) const
|
|
{
|
|
if (!IsValidMessageID(msgid)) return false;
|
|
auto p = MessagePath(msgid);
|
|
return !fs::exists(p);
|
|
}
|
|
|
|
fs::path ArticleStorage::MessagePath(const std::string & msgid) const
|
|
{
|
|
return basedir / msgid;
|
|
}
|
|
|
|
FileHandle_ptr ArticleStorage::OpenRead(const std::string & msgid) const
|
|
{
|
|
return OpenFile(MessagePath(msgid), eRead);
|
|
}
|
|
|
|
FileHandle_ptr ArticleStorage::OpenWrite(const std::string & msgid) const
|
|
{
|
|
return OpenFile(MessagePath(msgid), eWrite);
|
|
}
|
|
|
|
}
|