add doc codes check.

1 files changed, 24 insertions, 0 deletions

diff --git a/spec/inputs/compile_doc.yue b/spec/inputs/compile_doc.yue
new file mode 100644
index 0000000..4ce2b13
--- /dev/null
+++ b/spec/inputs/compile_doc.yue
@@ -0,0 +1,24 @@
+outputFolder = ...
+for [compiledFile, docFile] in *[
+                ["codes_from_doc.lua", "doc/docs/doc/README.md"]
+                ["codes_from_doc_zh.lua", "doc/docs/zh/doc/README.md"]
+        ]
+        close input = with? io.open docFile
+                import "yue" as :to_lua
+                text = \read "*a"
+                codes = []
+                for code in text\gmatch "```moonscript(.-)```"
+                        if result, err = to_lua code, implicit_return_root: false, reserve_line_number: false
+                                codes[] = result
+                        elseif not err\match "macro exporting module only accepts macro definition"
+                                print err
+                                os.exit 1
+                for code in text\gmatch "<pre>(.-)</pre>"
+                        if result, err = to_lua code\gsub("&lt;", "<")\gsub("&gt;", ">"), implicit_return_root: false, reserve_line_number: false
+                                codes[] = result
+                        else
+                                print err
+                                os.exit 1
+                close output = with io.open "#{outputFolder}/#{compiledFile}", "w+"
+                        \write table.concat codes
+