From 72cc9bd0ae4bf847c4519b0c508c300db8141eb3 Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Wed, 1 Feb 2023 15:22:54 +0800
Subject: fix yue.to_ast().

---
 src/yuescript/yuescript.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/yuescript/yuescript.cpp b/src/yuescript/yuescript.cpp
index 20fa258..ed09aaa 100644
--- a/src/yuescript/yuescript.cpp
+++ b/src/yuescript/yuescript.cpp
@@ -161,7 +161,7 @@ struct yue_stack {
 static int yuetoast(lua_State* L) {
 	size_t size = 0;
 	const char* input = luaL_checklstring(L, 1, &size);
-	int flattenLevel = 2;
+	int flattenLevel = 0;
 	if (lua_isnoneornil(L, 2) == 0) {
 		flattenLevel = static_cast<int>(luaL_checkinteger(L, 2));
 		flattenLevel = std::max(std::min(2, flattenLevel), 0);
@@ -270,8 +270,10 @@ static int yuetoast(lua_State* L) {
 							lua_pushinteger(L, node->m_begin.m_col);
 							lua_rawseti(L, -2, 3);
 							for (int i = count, j = 4; i >= 1; i--, j++) {
-								lua_rawgeti(L, tableIndex, len);
+								lua_rawgeti(L, tableIndex, len - i + 1);
 								lua_rawseti(L, -2, j);
+							}
+							for (int i = 1; i <= count; i++) {
 								lua_pushnil(L);
 								lua_rawseti(L, tableIndex, len);
 								len--;
-- 
cgit v1.2.3-55-g6feb