handler.h

/*
 * Copyright 2025 Dominik Czekai
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
#include <fstream>
#include <iostream>
#include <memory>
 
#include "formatter.h"
#include "logstyle.h"
 
namespace cli::logging
{
 
class AbstractHandler
{
public:
    virtual ~AbstractHandler() = default;
 
    virtual void emit(const LogRecord &record) const = 0;
};
 
class BaseHandler : public AbstractHandler
{
public:
    BaseHandler(std::ostream &outStream, std::ostream &errStream,
                std::shared_ptr<AbstractFormatter> formatter, LogLevel minLevel = LogLevel::DEBUG,
                std::shared_ptr<const LogStyleMap> styles = nullptr)
        : out(outStream), err(errStream), formatterPtr(std::move(formatter)),
          styleMapPtr(std::move(styles)), minLevel(minLevel)
    {
    }
 
    ~BaseHandler() override;
 
    void emit(const LogRecord &record) const override;
 
    void setStylingEnabled(bool enabled) { stylingEnabled = enabled; }
 
    void setStyleMap(std::shared_ptr<const LogStyleMap> styles);
 
protected:
    std::ostream &out; // standard stream
    std::ostream &err; // error stream
private:
    bool stylingEnabled{true};
    std::shared_ptr<AbstractFormatter> formatterPtr;
    std::shared_ptr<const LogStyleMap> styleMapPtr;
    LogLevel minLevel;
};
 
class ConsoleHandler : public BaseHandler
{
public:
    explicit ConsoleHandler(
        std::shared_ptr<AbstractFormatter> formatter, LogLevel minLevel = LogLevel::DEBUG,
        std::shared_ptr<const LogStyleMap> styles = std::make_shared<LogStyleMap>(defaultStyles()))
        : BaseHandler(std::cout, std::cerr, formatter, minLevel, std::move(styles))
    {
    }
};
 
class FileHandler : public BaseHandler
{
public:
    explicit FileHandler(const std::string &filename, std::shared_ptr<AbstractFormatter> formatter,
                         LogLevel minLevel = LogLevel::DEBUG,
                         std::shared_ptr<const LogStyleMap> styles = nullptr);
    ~FileHandler() override;
 
private:
    std::ofstream file;
};
 
} // namespace cli::logging