From 73323c9ee72a3c18b81335e25597471d83c82a0f Mon Sep 17 00:00:00 2001
From: King_DuckZ <king_duckz@gmx.com>
Date: Thu, 27 Apr 2017 09:33:47 +0100
Subject: [PATCH] Add plain text mode

---
 html/index.html.mstch   |  2 +-
 src/pastie_response.cpp | 36 ++++++++++++++++++++++++++----------
 src/pastie_response.hpp |  1 +
 3 files changed, 28 insertions(+), 11 deletions(-)

diff --git a/html/index.html.mstch b/html/index.html.mstch
index b010c6f..dc717c3 100644
--- a/html/index.html.mstch
+++ b/html/index.html.mstch
@@ -11,7 +11,7 @@
 	<div>
 		<p class="title">Syntax Highlighting:</p>
 			<select name="lang" class="selectBox">
-				<option value="" selected="selected">None  </option>
+				<option value="plaintext" selected="selected">None  </option>
 				{{#languages}}
 				<option value="{{language_name}}">{{language_name}}</option>
 				{{/languages}}
diff --git a/src/pastie_response.cpp b/src/pastie_response.cpp
index 6e7768a..565b7e2 100644
--- a/src/pastie_response.cpp
+++ b/src/pastie_response.cpp
@@ -18,30 +18,40 @@
 #include "pastie_response.hpp"
 #include "incredis/incredis.hpp"
 #include "settings_bag.hpp"
+#include "escapist.hpp"
 #include <ciso646>
 #include <srchilite/sourcehighlight.h>
 #include <srchilite/langmap.h>
 #include <sstream>
 
 namespace tawashi {
+	namespace {
+		const char g_nolang_token[] = "plaintext";
+	} //unnamed namespace
+
 	PastieResponse::PastieResponse (const Kakoune::SafePtr<SettingsBag>& parSettings) :
 		Response(Response::ContentType, "text/html", "text", parSettings, true),
 		m_langmap_dir(parSettings->as<std::string>("langmap_dir")),
-		m_plain_text(false)
+		m_plain_text(false),
+		m_syntax_highlight(true)
 	{
 	}
 
 	void PastieResponse::on_process() {
 		auto env = cgi_env().query_string_split();
-		if (env["m"] == "plain" or cgi_env().query_string().empty()) {
+		const std::string& query_str(cgi_env().query_string());
+		if (env["m"] == "plain" or query_str.empty()) {
 			this->change_type(Response::ContentType, "text/plain; charset=utf-8");
 			m_plain_text = true;
 		}
+		else if (query_str == g_nolang_token) {
+			m_syntax_highlight = false;
+		}
 		else {
 			srchilite::LangMap lang_map(m_langmap_dir, "lang.map");
 			lang_map.open();
-			if (not cgi_env().query_string().empty())
-				m_lang_file = lang_map.getFileName(cgi_env().query_string());
+			if (not query_str.empty())
+				m_lang_file = lang_map.getFileName(query_str);
 			else
 				m_lang_file = "default.lang";
 		}
@@ -61,18 +71,24 @@ namespace tawashi {
 		highlighter.setGenerateEntireDoc(false);
 		highlighter.setGenerateLineNumbers(true);
 		const auto lang = m_lang_file;
-		//Escapist houdini;
-		//std::istringstream iss(houdini.escape_html(*pastie));
 
-		if (m_plain_text) {
-			parContext["pastie"] = *pastie;
+		std::string processed_pastie;
+		if (m_syntax_highlight) {
+			processed_pastie = std::move(*pastie);
 		}
 		else {
-			std::istringstream iss(*pastie);
+			Escapist houdini;
+			processed_pastie = houdini.escape_html(*pastie);
+		}
+
+		if (not m_plain_text and m_syntax_highlight) {
+			std::istringstream iss(std::move(processed_pastie));
 			std::ostringstream oss;
 			highlighter.highlight(iss, oss, lang);
-			parContext["pastie"] = oss.str();
+			processed_pastie = oss.str();
 		}
+
+		parContext["pastie"] = std::move(processed_pastie);
 	}
 
 	std::string PastieResponse::on_mustache_retrieve() {
diff --git a/src/pastie_response.hpp b/src/pastie_response.hpp
index 0a3da11..e6ae4ac 100644
--- a/src/pastie_response.hpp
+++ b/src/pastie_response.hpp
@@ -34,5 +34,6 @@ namespace tawashi {
 		std::string m_lang_file;
 		std::string m_langmap_dir;
 		bool m_plain_text;
+		bool m_syntax_highlight;
 	};
 } //namespace tawashi