diff --git a/src/ast.hpp b/src/ast.hpp index c1f8189..666e1f3 100644 --- a/src/ast.hpp +++ b/src/ast.hpp @@ -384,14 +384,21 @@ namespace almo { std::string code; // Highlight.js によって正確にハイライトするために、言語を指定する必要がある。 - // ```python -> python を持っておき、 `to_html` する際に として出力する。 + // ```python -> python を持っておき、 `to_html` する際に として出力する。 std::string language; public: CodeBlock(std::string code, std::string language, std::string uuid) : code(code), language(language), uuid(uuid) { } - // Highlight.js によって正確にハイライトするために、 にクラスを付与する。 std::string to_html() const override { - return "
" + code + "
"; + std::string code_class; + + if (language == "") { + code_class = "language-plaintext"; + } else { + code_class = "language-" + language; + } + + return "
" + code + "
"; }