Fixed code generating.

Author: pigpigyyy

spec/outputs/codes_from_doc.lua

@@ -1109,17 +1109,12 @@ local data = {
 }
 local flat
 local _accum_0 = { }
-local _len_0 = 1
 for k, v in pairs(data) do
-	do
-		local _obj_0 = _accum_0[_len_0]
-		local _len_1 = #_obj_0 + 1
-		for _index_0 = 1, #v do
-			local _elm_0 = v[_index_0]
-			_obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
-		end
+	local _len_0 = #_accum_0 + 1
+	for _index_0 = 1, #v do
+		local _elm_0 = v[_index_0]
+		_accum_0[_len_0], _len_0 = _elm_0, _len_0 + 1
 	end
-	_len_0 = _len_0 + 1
 end
 flat = _accum_0
 local x_coords = {
@@ -3622,17 +3617,12 @@ local data = {
 }
 local flat
 local _accum_0 = { }
-local _len_0 = 1
 for k, v in pairs(data) do
-	do
-		local _obj_0 = _accum_0[_len_0]
-		local _len_1 = #_obj_0 + 1
-		for _index_0 = 1, #v do
-			local _elm_0 = v[_index_0]
-			_obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
-		end
+	local _len_0 = #_accum_0 + 1
+	for _index_0 = 1, #v do
+		local _elm_0 = v[_index_0]
+		_accum_0[_len_0], _len_0 = _elm_0, _len_0 + 1
 	end
-	_len_0 = _len_0 + 1
 end
 flat = _accum_0
 local x_coords = {

spec/outputs/codes_from_doc_zh.lua

@@ -1103,17 +1103,12 @@ local data = {
 }
 local flat
 local _accum_0 = { }
-local _len_0 = 1
 for k, v in pairs(data) do
-	do
-		local _obj_0 = _accum_0[_len_0]
-		local _len_1 = #_obj_0 + 1
-		for _index_0 = 1, #v do
-			local _elm_0 = v[_index_0]
-			_obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
-		end
+	local _len_0 = #_accum_0 + 1
+	for _index_0 = 1, #v do
+		local _elm_0 = v[_index_0]
+		_accum_0[_len_0], _len_0 = _elm_0, _len_0 + 1
 	end
-	_len_0 = _len_0 + 1
 end
 flat = _accum_0
 local x_coords = {
@@ -3610,17 +3605,12 @@ local data = {
 }
 local flat
 local _accum_0 = { }
-local _len_0 = 1
 for k, v in pairs(data) do
-	do
-		local _obj_0 = _accum_0[_len_0]
-		local _len_1 = #_obj_0 + 1
-		for _index_0 = 1, #v do
-			local _elm_0 = v[_index_0]
-			_obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
-		end
+	local _len_0 = #_accum_0 + 1
+	for _index_0 = 1, #v do
+		local _elm_0 = v[_index_0]
+		_accum_0[_len_0], _len_0 = _elm_0, _len_0 + 1
 	end
-	_len_0 = _len_0 + 1
 end
 flat = _accum_0
 local x_coords = {

spec/outputs/lists.lua

@@ -892,17 +892,12 @@ do
 	}
 	local flat
 	local _accum_0 = { }
-	local _len_0 = 1
 	for k, v in pairs(data) do
-		do
-			local _obj_0 = _accum_0[_len_0]
-			local _len_1 = #_obj_0 + 1
-			for _index_0 = 1, #v do
-				local _elm_0 = v[_index_0]
-				_obj_0[_len_1], _len_1 = _elm_0, _len_1 + 1
-			end
+		local _len_0 = #_accum_0 + 1
+		for _index_0 = 1, #v do
+			local _elm_0 = v[_index_0]
+			_accum_0[_len_0], _len_0 = _elm_0, _len_0 + 1
 		end
-		_len_0 = _len_0 + 1
 	end
 	flat = _accum_0
 end

src/yuescript/yue_compiler.cpp

@@ -8317,9 +8317,12 @@ class YueCompilerImpl {
 		auto compInner = static_cast<CompInner_t*>(comp->items.back());
 		str_list temp;
 		std::string accumVar = getUnusedName("_accum_"sv);
-		std::string lenVar = getUnusedName("_len_"sv);
 		addToScope(accumVar);
-		addToScope(lenVar);
+		std::string lenVar;
+		if (!isSpread) {
+			lenVar = getUnusedName("_len_"sv);
+			addToScope(lenVar);
+		}
 		for (auto item : compInner->items.objects()) {
 			switch (item->get_id()) {
 				case id<CompForEach_t>():
@@ -8337,7 +8340,7 @@ class YueCompilerImpl {
 			}
 		}
 		{
-			auto assignLeft = toAst<ExpList_t>(accumVar + '[' + lenVar + (isSpread ? "][]"s : "]"s), x);
+			auto assignLeft = toAst<ExpList_t>(accumVar + '[' + (isSpread ? "]"s : lenVar + ']'), x);
 			auto assign = x->new_ptr<Assign_t>();
 			assign->values.push_back(value);
 			auto assignment = x->new_ptr<ExpListAssign_t>();
@@ -8351,10 +8354,15 @@ class YueCompilerImpl {
 			popScope();
 		}
 		_buf << indent() << "local "sv << accumVar << " = { }"sv << nll(comp);
-		_buf << indent() << "local "sv << lenVar << " = 1"sv << nll(comp);
-		_buf << join(temp);
-		_buf << assignStr;
-		_buf << indent(int(temp.size())) << lenVar << " = "sv << lenVar << " + 1"sv << nll(comp);
+		if (isSpread) {
+			_buf << join(temp);
+			_buf << assignStr;
+		} else {
+			_buf << indent() << "local "sv << lenVar << " = 1"sv << nll(comp);
+			_buf << join(temp);
+			_buf << assignStr;
+			_buf << indent(int(temp.size())) << lenVar << " = "sv << lenVar << " + 1"sv << nll(comp);
+		}
 		for (int ind = int(temp.size()) - 1; ind > -1; --ind) {
 			_buf << indent(ind) << "end"sv << nll(comp);
 		}