// PythonHighlighter.h - simple Python highlighter with triple-quote state
#pragma once

#include "LanguageHighlighter.h"
#include <unordered_set>

namespace kte {
class PythonHighlighter final : public StatefulHighlighter {
public:
	PythonHighlighter();

	void HighlightLine(const Buffer &buf, int row, std::vector<HighlightSpan> &out) const override;

	LineState HighlightLineStateful(const Buffer &buf, int row, const LineState &prev,
	                                std::vector<HighlightSpan> &out) const override;

private:
	std::unordered_set<std::string> kws_;
};
} // namespace kte
