Fix bug: IN_LBDX_LAMBDA should end at rightmost colon

hsfzxjy · Feb 20, 2021 · a8c73a9 · a8c73a9
1 parent b3a1bd3
commit a8c73a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lambdex/fmt/core/tkutils/rules/rules.py b/lambdex/fmt/core/tkutils/rules/rules.py
@@ -24,7 +24,6 @@ def r(ctx: Context, token: TokenInfo):
         ctx.pop_state()
 
 
-
 @m(exact_type=tk.NL, last_state='ALL')
 @m(exact_type=tk.NEWLINE, last_state='ALL')
 @m(exact_type=tk.WHITESPACE, last_state='ALL')
@@ -108,6 +107,8 @@ def r(ctx: Context, token: TokenInfo):
 
 @m(exact_type=tk.COLON, last_state=State.IN_LBDX_LAMBDA)
 def r(ctx: Context, token: TokenInfo):
+    if ctx.last_op[1] != State.IN_LBDX_CALL:
+        return
     ctx.pop_state()
     ctx.push_state(State.EXPECT_LBDX_LSQB)