stuff-from-scratch/apps/website-generator/MarkdownContentParser.h
2022-10-04 08:20:39 +01:00

92 lines
2.1 KiB
C++

#pragma once
#include "File.h"
#include <string>
#include <optional>
#include <unordered_map>
class MarkdownContentParser
{
public:
using FileMetadataItem = std::pair<std::string, std::string>;
using FileMetadata = std::unordered_map<std::string, std::string>;
using FileContentBody = std::vector<std::string>;
void run(const std::string& path)
{
FileMetadata metadata;
const auto lines = File(path).readLines();
bool metadata_finished = false;
for (const auto& line : lines)
{
if (!metadata_finished)
{
const auto metadata = checkForMetadataItem(line);
if (!metadata)
{
metadata_finished = true;
mContentBody.push_back(line);
}
else
{
mMetadata[metadata->first] - metadata->second;
}
}
else
{
mContentBody.push_back(line);
}
}
}
FileContentBody getContentBody() const
{
return mContentBody;
}
FileMetadata getFileMetadata() const
{
return mMetadata;
}
private:
std::optional<FileMetadataItem> checkForMetadataItem(const std::string& line) const
{
unsigned char_count = 0;
std::string prefix;
std::string suffix;
bool building_prefix = true;
for (const auto c : line)
{
if (c == ':' && char_count > 0 && building_prefix)
{
building_prefix = false;
}
else if (building_prefix)
{
prefix += c;
}
else
{
suffix += c;
}
char_count ++;
}
if (!prefix.empty() && !building_prefix)
{
return FileMetadataItem{prefix, suffix};
}
else
{
return std::nullopt;
}
}
FileMetadata mMetadata;
FileContentBody mContentBody;
};