nix-super/src/libutil/logging.hh

#pragma once

#include "types.hh"

namespace nix {

typedef enum {
    lvlError = 0,
    lvlInfo,
    lvlTalkative,
    lvlChatty,
    lvlDebug,
    lvlVomit
} Verbosity;

class Activity;

class Logger
{
    friend class Activity;

public:

    virtual ~Logger() { }

    virtual void log(Verbosity lvl, const FormatOrString & fs) = 0;

    void log(const FormatOrString & fs)
    {
        log(lvlInfo, fs);
    }

    virtual void setExpected(const std::string & label, uint64_t value = 1) { }
    virtual void setProgress(const std::string & label, uint64_t value = 1) { }
    virtual void incExpected(const std::string & label, uint64_t value = 1) { }
    virtual void incProgress(const std::string & label, uint64_t value = 1) { }

private:

    virtual void startActivity(Activity & activity, Verbosity lvl, const FormatOrString & fs) = 0;

    virtual void stopActivity(Activity & activity) = 0;

};

class Activity
{
public:
    Logger & logger;

    Activity(Logger & logger, Verbosity lvl, const FormatOrString & fs)
        : logger(logger)
    {
        logger.startActivity(*this, lvl, fs);
    }

    ~Activity()
    {
        logger.stopActivity(*this);
    }
};

extern Logger * logger;

Logger * makeDefaultLogger();

extern Verbosity verbosity; /* suppress msgs > this */

/* Print a message if the current log level is at least the specified
   level. Note that this has to be implemented as a macro to ensure
   that the arguments are evaluated lazily. */
#define printMsg(level, args...) \
    do { \
        if (level <= nix::verbosity) { \
            logger->log(level, fmt(args)); \
        } \
    } while (0)

#define printError(args...) printMsg(lvlError, args)
#define printInfo(args...) printMsg(lvlInfo, args)
#define printTalkative(args...) printMsg(lvlTalkative, args)
#define debug(args...) printMsg(lvlDebug, args)
#define vomit(args...) printMsg(lvlVomit, args)

void warnOnce(bool & haveWarned, const FormatOrString & fs);

void writeToStderr(const string & s);

}