rpn: Fix := and =: operators semantics

1 files changed, 9 insertions, 0 deletions

diff --git a/cmd/gt/main.go b/cmd/gt/main.go
index 15372d8..2e3df47 100644
--- a/cmd/gt/main.go
+++ b/cmd/gt/main.go
@@ -125,6 +125,15 @@ func runREPL() error {
 func runRPN(input string) (string, error) {
 	vars := rpn.NewVariables()
 	rpnCalc := rpn.NewRPN(vars)
+
+	// Strip "rpn " or "calc " prefix if present
+	input = strings.TrimSpace(input)
+	if strings.HasPrefix(input, "rpn ") {
+		input = strings.TrimPrefix(input, "rpn ")
+	} else if strings.HasPrefix(input, "calc ") {
+		input = strings.TrimPrefix(input, "calc ")
+	}
+
 	return rpnCalc.ParseAndEvaluate(input)
 }