From 39a0121856f8230332bcef1b6a7108696f2a765d Mon Sep 17 00:00:00 2001
From: Bruce Hill <bitbucket@bruce-hill.com>
Date: Tue, 17 Jul 2018 15:00:57 -0700
Subject: Improvements to nomsu codegen.

---
 code_obj.lua | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'code_obj.lua')

diff --git a/code_obj.lua b/code_obj.lua
index a861bf2..0fcd34d 100644
--- a/code_obj.lua
+++ b/code_obj.lua
@@ -103,9 +103,18 @@ do
             if trailing_text then
               self.current_indent = #spaces
               self.trailing_line_len = #trailing_text
+            else
+              self.trailing_line_len = self.trailing_line_len + #b
             end
           else
-            self.trailing_line_len = math.min(self.trailing_line_len + #tostring(b), b.trailing_line_len)
+            do
+              local trailing_text = match(tostring(b), "\n([^\n]*)$")
+              if trailing_text then
+                self.trailing_line_len = #trailing_text + self.current_indent
+              else
+                self.trailing_line_len = self.trailing_line_len + #tostring(b)
+              end
+            end
             if self.current_indent ~= 0 then
               indents[#bits] = self.current_indent
             end
-- 
cgit v1.2.3