6 files changed, 52 insertions, 10 deletions

diff --git a/spec/inputs/existential.moon b/spec/inputs/existential.moon
index 7ee3bde..3055705 100644
--- a/spec/inputs/existential.moon
+++ b/spec/inputs/existential.moon
@@ -45,3 +45,7 @@ a = tb1?\end? 123 + tb2?\then 456
 
 b = tb1?\end? or tb2?\then
 
+with? io.open "test.txt", "w"
+        \write "hello"
+        \close!
+
diff --git a/spec/inputs/loops.moon b/spec/inputs/loops.moon
index 35427f6..24d960f 100644
--- a/spec/inputs/loops.moon
+++ b/spec/inputs/loops.moon
@@ -91,6 +91,15 @@ while true
         break if true
         print "no"
 
+a = 1
+repeat
+        a += 1
+        if a == 5
+                continue
+        if a == 6
+                break
+        print a
+until a == 10
 
 for x=1,10
         continue if x > 3 and x < 7
diff --git a/src/MoonP/moon_ast.h b/src/MoonP/moon_ast.h
index 7ef25a9..dd11ac4 100644
--- a/src/MoonP/moon_ast.h
+++ b/src/MoonP/moon_ast.h
@@ -190,14 +190,16 @@ AST_NODE(Return)
         AST_MEMBER(Return, &valueList)
 AST_END(Return)
 
+class existential_op_t;
 class Assign_t;
 class Body_t;
 
 AST_NODE(With)
+        ast_ptr<false, existential_op_t> eop;
         ast_ptr<true, ExpList_t> valueList;
         ast_ptr<false, Assign_t> assigns;
         ast_ptr<true, Body_t> body;
-        AST_MEMBER(With, &valueList, &assigns, &body)
+        AST_MEMBER(With, &eop, &valueList, &assigns, &body)
 AST_END(With)
 
 AST_NODE(SwitchCase)
@@ -238,6 +240,12 @@ AST_NODE(While)
         AST_MEMBER(While, &condition, &body)
 AST_END(While)
 
+AST_NODE(Repeat)
+        ast_ptr<true, Body_t> body;
+        ast_ptr<true, Exp_t> condition;
+        AST_MEMBER(Repeat, &body, &condition)
+AST_END(Repeat)
+
 AST_NODE(for_step_value)
         ast_ptr<true, Exp_t> value;
         AST_MEMBER(for_step_value, &value)
@@ -664,7 +672,7 @@ AST_LEAF(BreakLoop)
 AST_END(BreakLoop)
 
 AST_NODE(Statement)
-        ast_sel<true, Import_t, While_t, For_t, ForEach_t,
+        ast_sel<true, Import_t, While_t, Repeat_t, For_t, ForEach_t,
                 Return_t, Local_t, Global_t, Export_t, Macro_t, BreakLoop_t,
                 Label_t, Goto_t, Backcall_t, ExpListAssign_t> content;
         ast_ptr<false, statement_appendix_t> appendix;
diff --git a/src/MoonP/moon_compiler.cpp b/src/MoonP/moon_compiler.cpp
index c9d9c78..f79df07 100644
--- a/src/MoonP/moon_compiler.cpp
+++ b/src/MoonP/moon_compiler.cpp
@@ -43,7 +43,7 @@ inline std::string s(std::string_view sv) {
 }
 
 const std::string_view version() {
-        return "0.3.10"sv;
+        return "0.3.11"sv;
 }
 
 // name of table stored in lua registry
@@ -741,6 +741,7 @@ private:
                 switch (content->getId()) {
                         case id<Import_t>(): transformImport(static_cast<Import_t*>(content), out); break;
                         case id<While_t>(): transformWhile(static_cast<While_t*>(content), out); break;
+                        case id<Repeat_t>(): transformRepeat(static_cast<Repeat_t*>(content), out); break;
                         case id<For_t>(): transformFor(static_cast<For_t*>(content), out); break;
                         case id<ForEach_t>(): transformForEach(static_cast<ForEach_t*>(content), out); break;
                         case id<Return_t>(): transformReturn(static_cast<Return_t*>(content), out); break;
@@ -4251,7 +4252,7 @@ private:
                                 transformAssignment(assignment, temp);
                         }
                 }
-                if (!scoped && !returnValue) {
+                if (!with->eop && !scoped && !returnValue) {
                         pushScope();
                         scoped = traversal::Stop == with->body->traverse([&](ast_node* node) {
                                 if (auto statement = ast_cast<Statement_t>(node)) {
@@ -4297,7 +4298,14 @@ private:
                         }
                 }
                 _withVars.push(withVar);
-                transformBody(with->body, temp, ExpUsage::Common);
+                if (with->eop) {
+                        auto ifNode = x->new_ptr<If_t>();
+                        ifNode->nodes.push_back(toAst<IfCond_t>(withVar + s("~=nil"sv), x));
+                        ifNode->nodes.push_back(with->body);
+                        transformIf(ifNode, temp, ExpUsage::Common);
+                } else {
+                        transformBody(with->body, temp, ExpUsage::Common);
+                }
                 _withVars.pop();
                 if (assignList) {
                         auto assignment = x->new_ptr<ExpListAssign_t>();
@@ -4914,6 +4922,18 @@ private:
                 out.push_back(clearBuf());
         }
 
+        void transformRepeat(Repeat_t* repeat, str_list& out) {
+                str_list temp;
+                pushScope();
+                transformLoopBody(repeat->body, temp, Empty, ExpUsage::Common);
+                transformExp(repeat->condition, temp, ExpUsage::Closure);
+                popScope();
+                _buf << indent() << "repeat"sv << nll(repeat);
+                _buf << temp.front();
+                _buf << indent() << "until "sv << temp.back() << nlr(repeat);
+                out.push_back(clearBuf());
+        }
+
         void transformSwitch(Switch_t* switchNode, str_list& out, ExpUsage usage, ExpList_t* assignList = nullptr) {
                 str_list temp;
                 if (usage == ExpUsage::Closure) {
diff --git a/src/MoonP/moon_parser.cpp b/src/MoonP/moon_parser.cpp
index 7de2a84..4f12a34 100644
--- a/src/MoonP/moon_parser.cpp
+++ b/src/MoonP/moon_parser.cpp
@@ -209,7 +209,7 @@ MoonParser::MoonParser() {
 
         WithExp = ExpList >> -Assign;
 
-        With = key("with") >> DisableDo >> ensure(WithExp, PopDo) >> -key("do") >> Body;
+        With = key("with") >> -existential_op >> DisableDo >> ensure(WithExp, PopDo) >> -key("do") >> Body;
         SwitchCase = key("when") >> ExpList >> -key("then") >> Body;
         SwitchElse = key("else") >> Body;
 
@@ -231,6 +231,7 @@ MoonParser::MoonParser() {
         Unless = key("unless") >> Seperator >> IfCond >> -key("then") >> Body >> *IfElseIf >> -IfElse;
 
         While = key("while") >> DisableDo >> ensure(Exp, PopDo) >> -key("do") >> Body;
+        Repeat = key("repeat") >> Body >> Break >> *EmptyLine >> CheckIndent >> key("until") >> Exp;
 
         for_step_value = sym(',') >> White >> Exp;
         for_args = Space >> Variable >> sym('=') >> Exp >> sym(',') >> White >> Exp >> -for_step_value;
@@ -538,10 +539,9 @@ MoonParser::MoonParser() {
 
         statement_appendix = (if_line | unless_line | CompInner) >> Space;
         Statement = (
-                Import | While | For | ForEach |
-                Return | Local | Global | Export |
-                Macro | Space >> BreakLoop | Label |
-                Goto | Backcall | ExpListAssign
+                Import | While | Repeat | For | ForEach |
+                Return | Local | Global | Export | Macro |
+                Space >> BreakLoop | Label | Goto | Backcall | ExpListAssign
         ) >> Space >>
         -statement_appendix;
 
diff --git a/src/MoonP/moon_parser.h b/src/MoonP/moon_parser.h
index ec51340..1f9caf8 100644
--- a/src/MoonP/moon_parser.h
+++ b/src/MoonP/moon_parser.h
@@ -216,6 +216,7 @@ private:
         AST_RULE(If)
         AST_RULE(Unless)
         AST_RULE(While)
+        AST_RULE(Repeat)
         AST_RULE(for_step_value)
         AST_RULE(For)
         AST_RULE(ForEach)