Emacs lua-mode indentation woes

expected.lua

local foo = function(
    bar,
    baz
  )
  print
  {
    "A", "B";
    "C";
  }
  local bar = 5
  print("FOO")
  print(
      foo(),
      bar(
          baz(),
        )
    )
  for i = 1, 10 do
    print(i)
  end
  if
    true
    and false
  then
    print("HUH")
  end
end

return
{
  foo = foo;
}

input.lua

local foo = function(
bar,
baz
)
print
{
"A", "B";
"C";
}
local bar = 5
print("FOO")
print(
foo(),
bar(
baz(),
)
)
for i = 1, 10 do
print(i)
end
if
true
and false
then
print("HUH")
end
end

return
{
foo = foo;
}

output.6563ae.lua

local foo = function(
bar,
baz
                    )
print
{
"A", "B";
"C";
}
local bar = 5
print("FOO")
print(
   foo(),
bar(
   baz(),
   )
     )
for i = 1, 10 do
   print(i)
end
if
true
and false
then
   print("HUH")
end
end

return
{
foo = foo;
}