Marginally better bytecode builder line API

1 files changed, 34 insertions, 4 deletions

diff --git a/pkg/lang/vm/code/builder.go b/pkg/lang/vm/code/builder.go
index 3aa6b6b..51820fb 100644
--- a/pkg/lang/vm/code/builder.go
+++ b/pkg/lang/vm/code/builder.go
@@ -8,12 +8,17 @@ import (
 type Builder struct {
 	code      []byte
 	debugInfo DebugInfo
+
+	currentLine int
+	lineStart   int
 }
 
 func NewBuilder() Builder {
 	return Builder{
-		code:      make([]byte, 0, 64),
-		debugInfo: NewDebugInfo("unknown file"),
+		code:        make([]byte, 0, 64),
+		debugInfo:   NewDebugInfo("unknown file"),
+		lineStart:   -1,
+		currentLine: -1,
 	}
 }
 
@@ -39,8 +44,33 @@ func (b *Builder) AppendRaw(raw Raw) {
 	b.code = append(b.code, raw...)
 }
 
-func (b *Builder) AppendLine(line int) {
-	b.debugInfo.AppendLine(len(b.code)-1, line)
+func (b *Builder) StartLine(line int) {
+	if b.lineStart != -1 {
+		panic("line already started")
+	}
+
+	b.currentLine = line
+	b.lineStart = len(b.code)
+}
+
+func (b *Builder) EndLine() {
+	defer func() {
+		b.currentLine = -1
+		b.lineStart = -1
+	}()
+
+	if b.lineStart == -1 {
+		panic("line not started")
+	}
+
+	if b.lineStart == len(b.code) {
+		println("line empty")
+		return
+	}
+
+	println("line start", b.lineStart)
+
+	b.debugInfo.AppendLine(len(b.code)-1, b.currentLine)
 }
 
 func (b *Builder) SetInt(at int, x int64) {