Cleanup for trailing line printing

1 files changed, 6 insertions, 7 deletions

diff --git a/bp.c b/bp.c
index 858840e..aed503e 100644
--- a/bp.c
+++ b/bp.c
@@ -371,18 +371,17 @@ static int print_matches(FILE *out, file_t *f, pat_t *pattern)
         fprint_context(out, f, prev, m->start);
         if (print_opts.normal_color) fprintf(out, "%s", print_opts.normal_color);
         fprint_match(out, f->start, m, &print_opts);
-        size_t linenum = get_line_number(f, m->end);
-        if (last_line_num < (int)linenum) {
-            fprint_linenum(out, f, linenum, print_opts.normal_color);
-            fputc('\n', out);
-            last_line_num = (int)linenum;
-        }
         if (print_opts.normal_color) fprintf(out, "%s", print_opts.normal_color);
         prev = m->end;
     }
     // Print trailing context if needed:
-    if (matches > 0)
+    if (matches > 0) {
         fprint_context(out, f, prev, NULL);
+        if (last_line_num < 0) {
+            fprint_linenum(out, f, f->nlines, print_opts.normal_color);
+            fputc('\n', out);
+        }
+    }
 
     printing_file = NULL;
     last_line_num = -1;