From 1e66aaece6ea7cd3c705ca56ce5558e8f87681b8 Mon Sep 17 00:00:00 2001 From: Charlie Stanton Date: Sun, 21 Apr 2024 17:16:01 +0100 Subject: Add substitute next commands --- main/main.go | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) (limited to 'main/main.go') diff --git a/main/main.go b/main/main.go index 3fb1cbf..47bcddb 100644 --- a/main/main.go +++ b/main/main.go @@ -11,11 +11,29 @@ import ( type ProgramState struct { value, xreg, yreg, zreg []walk.Value start, prevStart, end, nextEnd bool + // TODO: This will only ever have 0 or 1 values, it is a slice out of laziness + peekStack []walk.WalkItem in walk.StredReader out walk.StredWriter program []Command pc int } +func (state *ProgramState) Read() (walk.WalkItem, error) { + if len(state.peekStack) > 0 { + item := state.peekStack[len(state.peekStack) - 1] + state.peekStack = state.peekStack[:len(state.peekStack) - 1] + return item, nil + } + return state.in.Read() +} +func (state *ProgramState) Peek() (walk.WalkItem, error) { + item, err := state.Read() + if err != nil { + return walk.WalkItem{}, err + } + state.peekStack = append(state.peekStack, item) + return item, nil +} type config struct { quiet bool @@ -38,7 +56,7 @@ func run(config config) { } for { - walkItem, err := state.in.Read() + walkItem, err := state.Read() if err != nil { break } -- cgit v1.2.3