Decorator_Classes.h

Go to the documentation of this file.

 
 
#pragma once
#ifndef __DECORATOR_CLASSES_H__
#define __DECORATOR_CLASSES_H__
 
#include <string>
#include <memory>
 
#include "helpers/argumentnull_error.h"
#include "helpers/formatstring.h"
 
namespace DesignPatternExamples_cpp
{
 
    //########################################################################
    //########################################################################
 
 
    struct IRenderElement
    {
        using shared_ptr_t = std::shared_ptr<IRenderElement>;
 
        virtual ~IRenderElement() {}
 
        virtual std::string Render() = 0;
    };
 
 
    //########################################################################
    //########################################################################
 
 
    class Decorator : public IRenderElement
    {
    private:
        IRenderElement::shared_ptr_t _wrappedElement;
    
    public:
        Decorator(IRenderElement::shared_ptr_t element)
            : _wrappedElement(element)
        {
            if (!_wrappedElement)
            {
                throw Helpers::argumentnull_error("element",
                    "The element being decorated cannot be null.");
            }
        }
 
 
        std::string Render()
        {
            return _wrappedElement->Render();
        }
    };
 
    //########################################################################
    //########################################################################
 
 
    class WhiteBackgroundDecorator : public Decorator
    {
    public:
 
        WhiteBackgroundDecorator(IRenderElement::shared_ptr_t element)
            : Decorator(element)
        {
        }
 
        // IRenderElement Members
 
        std::string Render()
        {
            return Helpers::formatstring("\x1b[47m%s\x1b[49m", Decorator::Render().c_str());
        }
    };
 
 
    //########################################################################
    //########################################################################
 
 
    class UnderlineDecorator : public Decorator
    {
    public:
 
        UnderlineDecorator(IRenderElement::shared_ptr_t element)
            : Decorator(element)
        {
        }
 
        // IRenderElement Members
 
        std::string Render()
        {
            return Helpers::formatstring("\x1b[4m%s\x1b[24m", Decorator::Render().c_str());
        }
    };
 
 
    //########################################################################
    //########################################################################
 
 
    class RedForegroundDecorator : public Decorator
    {
    public:
 
        RedForegroundDecorator(IRenderElement::shared_ptr_t element)
            : Decorator(element)
        {
        }
 
        // IRenderElement Members
 
        std::string Render()
        {
            return Helpers::formatstring("\x1b[31m%s\x1b[39m", Decorator::Render().c_str());
        }
    };
 
 
    //########################################################################
    //########################################################################
 
 
    class TextElement : public IRenderElement
    {
    private:
        std::string _elementText;
 
    public:
 
        TextElement(std::string element)
        {
            _elementText = element;
        }
 
        // IRenderElement Members
 
        std::string Render()
        {
            return _elementText;
        }
    };
 
} // end namespace
 
#endif // __DECORATOR_CLASSES_H__