@intend-it/parser 1.1.1 → 1.1.2

package/dist/index.js

@@ -9167,6 +9167,7 @@ var Is = createToken({ name: "Is", pattern: /is/ });
 var Defined = createToken({ name: "Defined", pattern: /defined/ });
 var Entry = createToken({ name: "Entry", pattern: /entry/ });
 var Let = createToken({ name: "Let", pattern: /let/ });
+var Return = createToken({ name: "Return", pattern: /return/ });
 var LCurly = createToken({ name: "LCurly", pattern: /{/ });
 var RCurly = createToken({ name: "RCurly", pattern: /}/ });
 var LParen = createToken({ name: "LParen", pattern: /\(/ });
@@ -9207,6 +9208,7 @@ var allTokens = [
   Defined,
   Entry,
   Let,
+  Return,
   Arrow,
   FatArrow,
   GreaterThanEquals,
@@ -9311,7 +9313,8 @@ class IntendParser extends CstParser {
     this.OR([
       { ALT: () => this.SUBRULE(this.invariantStatement) },
       { ALT: () => this.SUBRULE(this.stepStatement) },
-      { ALT: () => this.SUBRULE(this.ensureStatement) }
+      { ALT: () => this.SUBRULE(this.ensureStatement) },
+      { ALT: () => this.SUBRULE(this.returnStatement) }
     ]);
   });
   invariantStatement = this.RULE("invariantStatement", () => {
@@ -9347,6 +9350,13 @@ class IntendParser extends CstParser {
       ]);
     });
   });
+  returnStatement = this.RULE("returnStatement", () => {
+    this.CONSUME(Return);
+    this.CONSUME(Identifier);
+    this.OPTION(() => {
+      this.CONSUME(Semicolon);
+    });
+  });
 }
 
 // src/ast/visitor.ts
@@ -9410,6 +9420,7 @@ class IntendASTVisitor extends BaseIntendVisitor {
     const invariants = [];
     const steps = [];
     const ensures = [];
+    let returnStmt;
     if (ctx.statement) {
       ctx.statement.forEach((stmt) => {
         const result = this.visit(stmt);
@@ -9419,10 +9430,12 @@ class IntendASTVisitor extends BaseIntendVisitor {
           steps.push(result);
         } else if (result.type === "Ensure") {
           ensures.push(result);
+        } else if (result.type === "Return") {
+          returnStmt = result;
         }
       });
     }
-    return { invariants, steps, ensures };
+    return { invariants, steps, ensures, returnStmt };
   }
   statement(ctx) {
     if (ctx.invariantStatement) {
@@ -9431,6 +9444,8 @@ class IntendASTVisitor extends BaseIntendVisitor {
       return this.visit(ctx.stepStatement);
     } else if (ctx.ensureStatement) {
       return this.visit(ctx.ensureStatement);
+    } else if (ctx.returnStatement) {
+      return this.visit(ctx.returnStatement);
     }
     throw new Error("Unknown statement type");
   }
@@ -9454,6 +9469,12 @@ class IntendASTVisitor extends BaseIntendVisitor {
       variableKind
     };
   }
+  returnStatement(ctx) {
+    return {
+      type: "Return",
+      identifier: ctx.Identifier[0].image
+    };
+  }
   ensureStatement(ctx) {
     const tokens = [
       ...ctx.Identifier || [],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@intend-it/parser",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "description": "Parser for the Intend programming language",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",