#pragma once

#include <QPoint>
#include <QRect>
#include <QString>

#include <boost/noncopyable.hpp>
#include <climits>

#include "messages/messagecolor.hpp"
#include "singletons/fontmanager.hpp"

class QPainter;

namespace chatterino {
namespace messages {
class MessageElement;
class Image;

namespace layouts {

class MessageLayoutElement : boost::noncopyable
{
public:
    MessageLayoutElement(MessageElement &creator, const QSize &size);

    const QRect &getRect() const;
    MessageElement &getCreator() const;
    void setPosition(QPoint point);
    bool hasTrailingSpace() const;

    MessageLayoutElement *setTrailingSpace(bool value);

    virtual void addCopyTextToString(QString &str, int from = 0, int to = INT_MAX) const = 0;
    virtual int getSelectionIndexCount() = 0;
    virtual void paint(QPainter &painter) = 0;
    virtual void paintAnimated(QPainter &painter) = 0;
    virtual int getMouseOverIndex(const QPoint &abs) = 0;

protected:
    bool trailingSpace = true;

private:
    QRect rect;
    //    bool isInNewLine;
    MessageElement &creator;
};

// IMAGE
class ImageLayoutElement : public MessageLayoutElement
{
public:
    ImageLayoutElement(MessageElement &creator, Image &image, QSize size);

protected:
    virtual void addCopyTextToString(QString &str, int from = 0, int to = INT_MAX) const override;
    virtual int getSelectionIndexCount() override;
    virtual void paint(QPainter &painter) override;
    virtual void paintAnimated(QPainter &painter) override;
    virtual int getMouseOverIndex(const QPoint &abs) override;

private:
    Image &image;
};

// TEXT
class TextLayoutElement : public MessageLayoutElement
{
public:
    TextLayoutElement(MessageElement &creator, QString &text, QSize size, QColor color,
                      FontStyle style, float scale);

protected:
    virtual void addCopyTextToString(QString &str, int from = 0, int to = INT_MAX) const override;
    virtual int getSelectionIndexCount() override;
    virtual void paint(QPainter &painter) override;
    virtual void paintAnimated(QPainter &painter) override;
    virtual int getMouseOverIndex(const QPoint &abs) override;

private:
    QString text;
    QColor color;
    FontStyle style;
    float scale;
};
}  // namespace layouts
}  // namespace messages
}  // namespace chatterino