Formatter.hpp

Go to the documentation of this file.

#pragma once

#include <chrono>
#include <memory>
#include <vector>
#include <string>
#include <lug/System/Export.hpp>

namespace lug {
namespace System {
namespace Logger {

namespace priv {
class Message;
} // priv
class Formatter;

using FlagHandlerPointer = std::string (Formatter::*)(const std::tm* now);

namespace priv {

class Formattable {
public:
    virtual std::string format(Message* = nullptr) const = 0;
};

class UserChars : public Formattable {
public:
    void addChar(char c);
    virtual std::string format(Message* = nullptr) const;

private:
    std::string _chars;
};

class LevelFlag : public Formattable {
public:
    virtual std::string format(Message* = nullptr) const;
};

class MessageFlag : public Formattable {
public:
    virtual std::string format(Message* = nullptr) const;
};

struct Token {
    Token() = default;
    Token(FlagHandlerPointer flagHandle) : basic(flagHandle) {}
    Token(std::unique_ptr<Formattable> formattable) : advanced(std::move(formattable)) {}

    FlagHandlerPointer basic{nullptr};
    std::unique_ptr<Formattable> advanced{nullptr};
};

} // priv
class LUG_SYSTEM_API Formatter {
public:
    Formatter(const std::string& pattern);

    Formatter(const Formatter&) = delete;
    Formatter(Formatter&&) = delete;

    Formatter& operator=(const Formatter&) = delete;
    Formatter& operator=(Formatter&&) = delete;

    virtual ~Formatter ();

    virtual void format(priv::Message& msg);
    virtual void format(priv::Message& msg, const std::tm* now);

private:
    void handleFlag(char c);
    void compilePattern(const std::string& pattern);

    std::string handleFlagy(const std::tm* now); // Year 2 digits
    std::string handleFlagY(const std::tm* now); // Year 4 digits
    std::string handleFlagm(const std::tm* now); // Month from 1 to 12
    std::string handleFlagd(const std::tm* now); // Day from 1 to 31
    std::string handleFlagH(const std::tm* now); // Hour from 0 to 24
    std::string handleFlagM(const std::tm* now); // Minute from 0 to 59
    std::string handleFlagS(const std::tm* now); // Sec from 0 to 61

    std::vector<priv::Token> _formatChain;
};

} // Logger
} // System
} // lug