Add Y and Z registers

3 files changed, 33 insertions, 1 deletions

diff --git a/main/command.go b/main/command.go
index 9403020..296ad69 100644
--- a/main/command.go
+++ b/main/command.go
@@ -107,6 +107,30 @@ func (cmd AppendXRegCommand) exec(state *ProgramState) {
 	state.xreg = append(state.xreg, state.value...)
 }
 
+type SwapYRegCommand struct {}
+func (cmd SwapYRegCommand) exec(state *ProgramState) {
+	v := state.value
+	state.value = state.yreg
+	state.yreg = v
+}
+
+type AppendYRegCommand struct {}
+func (cmd AppendYRegCommand) exec(state *ProgramState) {
+	state.yreg = append(state.yreg, state.value...)
+}
+
+type SwapZRegCommand struct {}
+func (cmd SwapZRegCommand) exec(state *ProgramState) {
+	v := state.value
+	state.value = state.zreg
+	state.zreg = v
+}
+
+type AppendZRegCommand struct {}
+func (cmd AppendZRegCommand) exec(state *ProgramState) {
+	state.zreg = append(state.zreg, state.value...)
+}
+
 type SwapPathCommand struct {}
 func (cmd SwapPathCommand) exec(state *ProgramState) {
 	v := state.value
diff --git a/main/main.go b/main/main.go
index 0c7324b..2067920 100644
--- a/main/main.go
+++ b/main/main.go
@@ -9,7 +9,7 @@ import (
 type Program []Command
 
 type ProgramState struct {
-	path, value, xreg []walk.Atom
+	path, value, xreg, yreg, zreg []walk.Atom
 	in walk.JSONIn
 	out walk.JSONOut
 	program []Command
diff --git a/main/parse.go b/main/parse.go
index d31687a..1972b66 100644
--- a/main/parse.go
+++ b/main/parse.go
@@ -142,6 +142,14 @@ func (p *parser) parseBasicCommand(commandChar rune) Command {
 			return SwapXRegCommand{}
 		case 'X':
 			return AppendXRegCommand{}
+		case 'y':
+			return SwapYRegCommand{}
+		case 'Y':
+			return AppendYRegCommand{}
+		case 'z':
+			return SwapZRegCommand{}
+		case 'Z':
+			return AppendZRegCommand{}
 		case 'k':
 			return SwapPathCommand{}
 		case 'K':