From 12cc294c7ac31e3d08d6b8924dfc6cc427a1f712 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bitbucket@bruce-hill.com>
Date: Wed, 17 Jan 2018 16:45:16 -0800
Subject: Fixed some minor bugs that made generated nomsu uglier.

---
 nomsu.moon | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

(limited to 'nomsu.moon')

diff --git a/nomsu.moon b/nomsu.moon
index 3594246..960d95f 100755
--- a/nomsu.moon
+++ b/nomsu.moon
@@ -427,7 +427,7 @@ class NomsuCompiler
                         else
                             nomsu = inline_expression(bit)
                             return nil unless nomsu
-                            buff ..= if nomsu.type == "Var" or nomsu.type == "List" or nomsu.type == "Dict"
+                            buff ..= if bit.type == "Var" or bit.type == "List" or bit.type == "Dict"
                                 "\\"..nomsu
                             else "\\("..nomsu..")"
                         if #buff > max_line then return nil
@@ -461,8 +461,11 @@ class NomsuCompiler
                     line = "\n    "
                     for bit in *tok.value
                         nomsu = inline_expression bit
+                        if line != "\n    " and #line + #", " + #nomsu > max_line
+                            buff ..= line
+                            line = "\n    "
                         sep = line == "\n    " and "" or ", "
-                        if nomsu and #nomsu + #line < max_line
+                        if nomsu
                             line ..= sep..nomsu
                             if #line >= max_line
                                 buff ..= line
@@ -503,7 +506,7 @@ class NomsuCompiler
                         else
                             nomsu = inline_expression(bit)
                             return nil unless nomsu
-                            buff ..= if nomsu.type == "Var" or nomsu.type == "List" or nomsu.type == "Dict"
+                            buff ..= if bit.type == "Var" or bit.type == "List" or bit.type == "Dict"
                                 "\\"..nomsu
                             else "\\("..nomsu..")"
                     return buff
@@ -525,7 +528,10 @@ class NomsuCompiler
             switch tok.type
                 when "Block"
                     if #tok.value == 1
-                        nomsu = noeol_expression(tok.value[1])
+                        nomsu = if tok.value[1].type == "FunctionCall"
+                            inline_expression(tok.value[1])
+                        else
+                            noeol_expression(tok.value[1])
                         if nomsu and #(nomsu\match("[^\n]*")) < max_line
                             return ": "..nomsu
                     return noeol_expression(tok)
-- 
cgit v1.2.3