From 4d3ae67617c2c1301a1a90601e20e15614d20319 Mon Sep 17 00:00:00 2001
From: Dmitriy Pleshevskiy <dmitriy@pleshevski.ru>
Date: Mon, 20 May 2024 17:26:50 +0300
Subject: [PATCH] fixup do statement

---
 lib.nix      | 2 +-
 lib.test.nix | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib.nix b/lib.nix
index 5d736f6..2abd2b1 100644
--- a/lib.nix
+++ b/lib.nix
@@ -172,7 +172,7 @@ let
     ]));
   if' = condition: trueBody: ifelse condition trueBody [ ];
 
-  do = body: spaceBetween [ kw_do ] ++ body ++ [ kw_end ];
+  do = body: spaceBetween ([ kw_do ] ++ body ++ [ kw_end ]);
 
   isLuaNil = expr: getType expr == "nil";
   LuaNil = { _type = "nil"; };
diff --git a/lib.test.nix b/lib.test.nix
index 9b815c6..249c9a7 100644
--- a/lib.test.nix
+++ b/lib.test.nix
@@ -258,4 +258,8 @@ with nix2lua; pkgs.lib.runTests {
     expr = toLua (requireTo "ls" "luasnip");
     expected = ''local ls = require("luasnip")'';
   };
+  "test returns a do-block" = {
+    expr = toLua (do [ (requireTo "ls" "luasnip") ]);
+    expected = ''do local ls = require("luasnip") end'';
+  };
 }