PrettyPrinter.h

Go to the documentation of this file.

#ifndef BENCODING_PRETTYPRINTER_H
#define BENCODING_PRETTYPRINTER_H

#include <memory>
#include <string>

#include "BItemVisitor.h"

namespace bencoding {

class BItem;

class PrettyPrinter: private BItemVisitor {
public:
    static std::unique_ptr<PrettyPrinter> create();

    std::string getPrettyRepr(std::shared_ptr<BItem> data,
        const std::string &indent = "    ");

private:
    PrettyPrinter();

    virtual void visit(BDictionary *bDictionary) override;
    virtual void visit(BInteger *bInteger) override;
    virtual void visit(BList *bList) override;
    virtual void visit(BString *bString) override;

    void storeCurrentIndent();
    void increaseIndentLevel();
    void decreaseIndentLevel();

private:
    std::string prettyRepr = "";

    std::string indentLevel = "    ";

    std::string currentIndent = "";
};

std::string getPrettyRepr(std::shared_ptr<BItem> data,
    const std::string &indent = "    ");

} // namespace bencoding

#endif