Merge pull request #13 from bayashi/fix-beforeafter-lines-bug

Fix beforeafter lines bug

main_test.go

@@ -79,6 +79,23 @@ func TestRunner_OK(t *testing.T) {
 				  5: }
 			`),
 		},
+		"service-c grep 56 with C2. Match 2 consecutive lines": {
+			opt: &options{
+				searchPath:   "service-c",
+				searchGrep:   "56",
+				contextLines: 2,
+			},
+			expect: here.Doc(`
+				testdata/service-c
+				testdata/service-c/main.go
+				  2: 
+				  3: func main() {
+				  4: 	baz := 56
+				  5: 	bag := 56
+				  6: 
+				  7: 	foo()
+			`),
+		},
 	} {
 		t.Run(tname, func(t *testing.T) {
 			var o bytes.Buffer

testdata/service-c/main.go

@@ -2,4 +2,11 @@ package c
 
 func main() {
 	baz := 56
+	bag := 56
+
+	foo()
+}
+
+func foo() {
+	println("Result")
 }

xfg.go

@@ -176,6 +176,7 @@ func (x *xfg) grepFile(scanner *bufio.Scanner, fPath string) ([]line, error) {
 					}
 					matchedContents = append(matchedContents, b)
 				}
+				blines = make([]line, x.options.contextLines)
 			}
 
 			if !x.options.noColor {