From 3c4cbff4d88892f153b70ccbbc69d1087ecbcc49 Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Wed, 13 Dec 2023 17:32:53 +0800
Subject: add yue.format() function. fix yue ast to code function issues.

---
 spec/inputs/test/format_spec.yue | 126 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 126 insertions(+)
 create mode 100644 spec/inputs/test/format_spec.yue

(limited to 'spec/inputs')

diff --git a/spec/inputs/test/format_spec.yue b/spec/inputs/test/format_spec.yue
new file mode 100644
index 0000000..5076445
--- /dev/null
+++ b/spec/inputs/test/format_spec.yue
@@ -0,0 +1,126 @@
+files = [
+	"spec/inputs/macro_export.yue"
+	"spec/inputs/attrib.yue"
+	"spec/inputs/macro.yue"
+	"spec/inputs/using.yue"
+	"spec/inputs/whitespace.yue"
+	"spec/inputs/nil_coalescing.yue"
+	"spec/inputs/stub.yue"
+	"spec/inputs/pipe.yue"
+	"spec/inputs/teal_lang.yue"
+	"spec/inputs/string.yue"
+	"spec/inputs/local.yue"
+	"spec/inputs/tables.yue"
+	"spec/inputs/operators.yue"
+	"spec/inputs/lists.yue"
+	"spec/inputs/compile_doc.yue"
+	"spec/inputs/switch.yue"
+	"spec/inputs/multiline_chain.yue"
+	"spec/inputs/existential.yue"
+	"spec/inputs/export_default.yue"
+	"spec/inputs/assign.yue"
+	"spec/inputs/literals.yue"
+	"spec/inputs/luarocks_upload.yue"
+	"spec/inputs/ambiguous.yue"
+	"spec/inputs/bubbling.yue"
+	"spec/inputs/try_catch.yue"
+	"spec/inputs/funcs.yue"
+	"spec/inputs/do.yue"
+	"spec/inputs/with.yue"
+	"spec/inputs/export.yue"
+	"spec/inputs/macro_todo.yue"
+	"spec/inputs/backcall.yue"
+	"spec/inputs/cond.yue"
+	"spec/inputs/in_expression.yue"
+	"spec/inputs/comprehension.yue"
+	"spec/inputs/macro_teal.yue"
+	"spec/inputs/import.yue"
+	"spec/inputs/unless_else.yue"
+	"spec/inputs/destructure.yue"
+	"spec/inputs/return.yue"
+	"spec/inputs/loops.yue"
+	"spec/inputs/class.yue"
+	"spec/inputs/vararg.yue"
+	"spec/inputs/goto.yue"
+	"spec/inputs/metatable.yue"
+	"spec/inputs/syntax.yue"
+	"spec/inputs/global.yue"
+	"spec/inputs/plus.yue"
+	"spec/inputs/test/class_spec.yue"
+	"spec/inputs/test/table_spreading_spec.yue"
+	"spec/inputs/test/loops_spec.yue"
+	"spec/inputs/test/format_spec.yue"
+	"spec/inputs/unicode/macro_export.yue"
+	"spec/inputs/unicode/attrib.yue"
+	"spec/inputs/unicode/macro.yue"
+	"spec/inputs/unicode/using.yue"
+	"spec/inputs/unicode/whitespace.yue"
+	"spec/inputs/unicode/nil_coalescing.yue"
+	"spec/inputs/unicode/stub.yue"
+	"spec/inputs/unicode/pipe.yue"
+	"spec/inputs/unicode/string.yue"
+	"spec/inputs/unicode/local.yue"
+	"spec/inputs/unicode/tables.yue"
+	"spec/inputs/unicode/operators.yue"
+	"spec/inputs/unicode/lists.yue"
+	"spec/inputs/unicode/switch.yue"
+	"spec/inputs/unicode/multiline_chain.yue"
+	"spec/inputs/unicode/existential.yue"
+	"spec/inputs/unicode/export_default.yue"
+	"spec/inputs/unicode/assign.yue"
+	"spec/inputs/unicode/literals.yue"
+	"spec/inputs/unicode/ambiguous.yue"
+	"spec/inputs/unicode/bubbling.yue"
+	"spec/inputs/unicode/try_catch.yue"
+	"spec/inputs/unicode/funcs.yue"
+	"spec/inputs/unicode/do.yue"
+	"spec/inputs/unicode/with.yue"
+	"spec/inputs/unicode/export.yue"
+	"spec/inputs/unicode/macro_todo.yue"
+	"spec/inputs/unicode/backcall.yue"
+	"spec/inputs/unicode/cond.yue"
+	"spec/inputs/unicode/in_expression.yue"
+	"spec/inputs/unicode/comprehension.yue"
+	"spec/inputs/unicode/import.yue"
+	"spec/inputs/unicode/unless_else.yue"
+	"spec/inputs/unicode/destructure.yue"
+	"spec/inputs/unicode/return.yue"
+	"spec/inputs/unicode/loops.yue"
+	"spec/inputs/unicode/class.yue"
+	"spec/inputs/unicode/vararg.yue"
+	"spec/inputs/unicode/goto.yue"
+	"spec/inputs/unicode/metatable.yue"
+	"spec/inputs/unicode/syntax.yue"
+	"spec/inputs/unicode/global.yue"
+	"spec/inputs/unicode/plus.yue"
+]
+
+import "yue"
+
+rewriteLineCol = (item)->
+	item[2] = 0
+	item[3] = 0
+	for i = 4, #item
+		switch type item[i] when "table"
+			if item[i][1] == "comment"
+				table.remove item, i
+				rewriteLineCol item
+				return
+			rewriteLineCol item[i]
+
+<- describe "format"
+for file in *files
+	<- it file
+	f = io.open file
+	code = f\read "a*"
+	f\close!
+
+	original_ast = yue.to_ast code
+	assert.is_not_nil original_ast
+	rewriteLineCol original_ast
+	formated = yue.format code
+	ast = yue.to_ast formated
+	assert.is_not_nil ast
+	rewriteLineCol ast
+	assert.same original_ast, ast
+
-- 
cgit v1.2.3-55-g6feb