From bf24824660497189da1a7d03b524e648271b4d03 Mon Sep 17 00:00:00 2001 From: Bruce Hill Date: Sat, 28 Apr 2018 20:45:17 -0700 Subject: Improving idiomatic nomsu codegen --- nomsu_tree.lua | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'nomsu_tree.lua') diff --git a/nomsu_tree.lua b/nomsu_tree.lua index c2d409e..93588c4 100644 --- a/nomsu_tree.lua +++ b/nomsu_tree.lua @@ -326,7 +326,14 @@ Tree("Action", { nomsu:append(next_space, bit.value) next_space = " " else - local arg_nomsu = bit.type ~= "Block" and bit:as_nomsu(true) + local arg_nomsu + if last_colon == i - 1 and bit.type == "Action" then + arg_nomsu = nil + elseif bit.type == "Block" then + arg_nomsu = nil + else + arg_nomsu = bit:as_nomsu(true) + end if arg_nomsu and #arg_nomsu < MAX_LINE then if bit.type == "Action" then if can_use_colon and i > 1 then -- cgit v1.2.3