fix panic on message escape

n-hass · Jun 6, 2024 · ce85965 · ce85965
1 parent e7c8a7f
commit ce85965
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 5 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -1,4 +1,4 @@
 {
-	"nixEnvSelector.nixFile": "${workspaceRoot}/shell.nix",
+	"nixEnvSelector.nixFile": "${workspaceFolder}/shell.nix",
 	"terminal.integrated.defaultProfile.linux": "zsh with nix-shell",
 }
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "telelog"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2021"
 
 [dependencies]

diff --git a/src/helpers.rs b/src/helpers.rs
@@ -78,6 +78,7 @@ pub fn colour_translate(priority: u8) -> String {
 
 pub fn escape_message(message: &str) -> String {
 	let mut message = message.to_string();
+
 	if message.starts_with("<code>") {
 		message = message.strip_prefix("<code>").unwrap_or("").to_string();
 	}
@@ -86,10 +87,15 @@ pub fn escape_message(message: &str) -> String {
 		message = message.strip_suffix("</code>").unwrap_or("").to_string();
 	}
 
-	message
+	message = message
 		.replace("&", "&amp;")
 		.replace("<", "&lt;")
 		.replace(">", "&gt;")
 		.replace("\"", "&quot;")
-		.replace("'", "&#39;")
+		.replace("'", "&#39;");
+
+
+	message = "<code>".to_string() + &message + "</code>";
+
+	message
 }