From 8beb845f550c58bdd39931d759cfc8d78f20b90d Mon Sep 17 00:00:00 2001
From: Johan Walles <johan.walles@gmail.com>
Date: Thu, 9 Mar 2023 06:46:40 +0100
Subject: [PATCH] Make the tests pass

---
 src/refiner.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/refiner.rs b/src/refiner.rs
index 1aa88a4..344cca6 100644
--- a/src/refiner.rs
+++ b/src/refiner.rs
@@ -105,8 +105,15 @@ fn format_split(old_text: &str, new_text: &str) -> Option<(Vec<String>, Vec<Stri
     let mut new_collector = TokenCollector::create(StyledToken::new("+".to_string(), Style::New));
 
     // Tokenize adds and removes before diffing them
-    let tokenized_old = tokenizer::tokenize(old_text);
-    let tokenized_new = tokenizer::tokenize(new_text);
+    let mut tokenized_old = tokenizer::tokenize(old_text);
+    let mut tokenized_new = tokenizer::tokenize(new_text);
+
+    // Help visualize what actually happens in "No newline at end of file" diffs
+    if old_text.ends_with('\n') && !new_text.ends_with('\n') {
+        tokenized_old.insert(tokenized_old.len() - 1, "⏎");
+    } else if new_text.ends_with('\n') && !old_text.ends_with('\n') {
+        tokenized_new.insert(tokenized_new.len() - 1, "⏎");
+    }
 
     let diff = tokenized_old.diff(&tokenized_new);
     match diff {