Handle 'orphaned signs' on line deletion for signcolumn >= 2

2 files changed, 37 insertions, 5 deletions

diff --git a/src/nvim/option.c b/src/nvim/option.c
index 914b92618c..666c526a18 100644
--- a/src/nvim/option.c
+++ b/src/nvim/option.c
@@ -7583,9 +7583,19 @@ int csh_like_shell(void)
 /// buffer signs and on user configuration.
 int win_signcol_count(win_T *wp)
 {
+  return win_signcol_configured(wp, NULL);
+}
+
+/// Return the number of requested sign columns, based on user / configuration.
+int win_signcol_configured(win_T *wp, int *is_fixed)
+{
   int minimum = 0, maximum = 1, needed_signcols;
   const char *scl = (const char *)wp->w_p_scl;
 
+  if (is_fixed) {
+    *is_fixed = 1;
+  }
+
   // Note: It checks "no" or "number" in 'signcolumn' option
   if (*scl == 'n'
       && (*(scl + 1) == 'o' || (*(scl + 1) == 'u'
@@ -7603,7 +7613,11 @@ int win_signcol_count(win_T *wp)
     return 1;
   }
 
-  // auto or auto:<NUM>
+  if (is_fixed) {
+    // auto or auto:<NUM>
+    *is_fixed = 0;
+  }
+
   if (!strncmp(scl, "auto:", 5)) {
     // Variable depending on a configuration
     maximum = scl[5] - '0';
diff --git a/src/nvim/sign.c b/src/nvim/sign.c
index c7dc1a5b22..5c7b497a19 100644
--- a/src/nvim/sign.c
+++ b/src/nvim/sign.c
@@ -18,6 +18,7 @@
 #include "nvim/move.h"
 #include "nvim/screen.h"
 #include "nvim/syntax.h"
+#include "nvim/option.h"
 
 /// Struct to hold the sign properties.
 typedef struct sign sign_T;
@@ -726,15 +727,29 @@ void sign_mark_adjust(
     long amount_after
 )
 {
-  sign_entry_T *sign;    // a sign in a b_signlist
-  linenr_T new_lnum;   // new line number to assign to sign
+  sign_entry_T *sign;           // a sign in a b_signlist
+  sign_entry_T *next;           // the next sign in a b_signlist
+  sign_entry_T *last = NULL;    // pointer to pointer to current sign
+  sign_entry_T **lastp = NULL;  // pointer to pointer to current sign
+  linenr_T new_lnum;            // new line number to assign to sign
+  int is_fixed = 0;
+  int signcol = win_signcol_configured(curwin, &is_fixed);
 
   curbuf->b_signcols_max = -1;
+  lastp = &curbuf->b_signlist;
 
-  FOR_ALL_SIGNS_IN_BUF(curbuf, sign) {
+  for (sign = curbuf->b_signlist; sign != NULL; sign = next) {
+    next = sign->se_next;
     new_lnum = sign->se_lnum;
     if (sign->se_lnum >= line1 && sign->se_lnum <= line2) {
-      if (amount != MAXLNUM) {
+      if (amount == MAXLNUM && (!is_fixed || signcol >= 2)) {
+        *lastp = next;
+        if (next) {
+          next->se_prev = last;
+        }
+        xfree(sign);
+        continue;
+      } else {
         new_lnum += amount;
       }
     } else if (sign->se_lnum > line2) {
@@ -746,6 +761,9 @@ void sign_mark_adjust(
     if (sign->se_lnum >= line1 && new_lnum <= curbuf->b_ml.ml_line_count) {
       sign->se_lnum = new_lnum;
     }
+
+    last = sign;
+    lastp = &sign->se_next;
   }
 }