From 3e89092833a6d407e711fe4ae5f44474ff34cf64 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Tue, 1 Jan 2019 17:15:51 -0800
Subject: Some changes to the error API, a fix for statement block parsing, and
 replacing ((foo 1 baz)'s meaning) with $(foo 1 baz).

---
 nomsu_decompiler.lua | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'nomsu_decompiler.lua')

diff --git a/nomsu_decompiler.lua b/nomsu_decompiler.lua
index bf04f4a..02f2426 100644
--- a/nomsu_decompiler.lua
+++ b/nomsu_decompiler.lua
@@ -506,7 +506,10 @@ tree_to_nomsu = function(tree)
         else
           nomsu:add("\\")
           local interp_nomsu = recurse(bit)
-          if not (interp_nomsu:is_multiline()) then
+          if interp_nomsu:is_multiline() then
+            local curr_indent = nomsu:text():match("\n( *)[^\n]*$") or nomsu:text():match("^( *)")
+            interp_nomsu = NomsuCode((interp_nomsu:text():gsub("\n", "\n" .. curr_indent)))
+          else
             local space = max_line - nomsu:trailing_line_len()
             if bit.type == "Var" then
               local next_str = tree[i + 1]
@@ -524,6 +527,8 @@ tree_to_nomsu = function(tree)
                 interp_nomsu2 = tree_to_nomsu(bit)
               end
               if #interp_nomsu2:text():lines() > 3 or #interp_nomsu2:text() >= MAX_LINE * GOLDEN_RATIO then
+                local curr_indent = nomsu:text():match("\n( *)[^\n]*$") or nomsu:text():match("^( *)")
+                interp_nomsu2 = NomsuCode((interp_nomsu2:text():gsub("\n", "\n" .. curr_indent)))
                 interp_nomsu = interp_nomsu2
               else
                 nomsu:add("\n..\\")
-- 
cgit v1.2.3