#ifndef WEDGEHIGHLIGHTER_H
#define WEDGEHIGHLIGHTER_H

#include <QColor>
#include <QString>
#include <QSyntaxHighlighter>
#include <QTextCharFormat>

class WedgeHighlighter : public QSyntaxHighlighter {
public:
	WedgeHighlighter(QTextDocument *parent, const QColor &focusColor, const QColor &highlightColor, const QColor &textColor, const QString &headerFormat)
	: QSyntaxHighlighter(parent), m_focusColor(focusColor), m_highlightColor(highlightColor), m_textColor(textColor), m_headerFormat(headerFormat) {}

	void setColors(const QColor &focusColor, const QColor &highlightColor, const QColor &textColor) {
		m_focusColor = focusColor;
		m_highlightColor = highlightColor;
		m_textColor = textColor;
		rehighlight();
	}

	void setHeaderFormat(const QString &headerFormat) {
		if (m_headerFormat == headerFormat) return;
		m_headerFormat = headerFormat;
		rehighlight();
	}

protected:
	void highlightBlock(const QString &text) override;

private:
	QColor m_focusColor;
	QColor m_highlightColor;
	QColor m_textColor;
	QString m_headerFormat;
};

#endif