Merge pull request #437 from pterm/303-some-line-numbers-are-wrong-wh…

…en-using-withshowlinenumber
pterm · Jan 5, 2023 · fb1d5fe · fb1d5fe
2 parents e392e89 + bb0f79e
commit fb1d5fe
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/prefix_printer.go b/prefix_printer.go
@@ -199,9 +199,8 @@ func (p *PrefixPrinter) Sprint(a ...interface{}) string {
 		}
 	}
 
-	_, fileName, line, _ := runtime.Caller(3 + p.LineNumberOffset)
-
 	if p.ShowLineNumber {
+		_, fileName, line, _ := runtime.Caller(3 + p.LineNumberOffset)
 		ret += FgGray.Sprint("\n└ " + fmt.Sprintf("(%s:%d)\n", fileName, line))
 		newLine = false
 	}
@@ -248,7 +247,9 @@ func (p *PrefixPrinter) Print(a ...interface{}) *TextPrinter {
 	if p.Debugger && !PrintDebugMessages {
 		return &tp
 	}
+	p.LineNumberOffset--
 	Fprint(p.Writer, p.Sprint(a...))
+	p.LineNumberOffset++
 	checkFatal(p)
 	return &tp
 }
@@ -286,7 +287,9 @@ func (p *PrefixPrinter) Printfln(format string, a ...interface{}) *TextPrinter {
 	if p.Debugger && !PrintDebugMessages {
 		return &tp
 	}
+	p.LineNumberOffset++
 	Fprint(p.Writer, p.Sprintfln(format, a...))
+	p.LineNumberOffset--
 	checkFatal(p)
 	return &tp
 }