prevent escape from scope

Signed-off-by: Dmitry Stoletov <[email protected]>
iMega · Jan 30, 2021 · 7cb4ea6 · 7cb4ea6
1 parent 7b38bd3
commit 7cb4ea6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/formatter/body.go b/formatter/body.go
@@ -37,7 +37,7 @@ func (body) InnerStatement(prev, cur *element) (bool, statement) {
 	return false, nil
 }
 
-func (body) TypeOf() typeStatement { return tsNone }
+func (body) TypeOf() typeStatement { return tsBody }
 
 func (b *body) IsEnd(prev, cur *element) (bool, bool) {
 	if cur.Token.Type == nEnd {

diff --git a/formatter/chain_statments.go b/formatter/chain_statments.go
@@ -61,6 +61,10 @@ func (cs *chainStatments) ExctractStatement(ts typeStatement) statement {
 		if cs.chain[i].TypeOf() == ts {
 			item = i
 		}
+
+		if cs.chain[i].TypeOf() == tsBody {
+			break
+		}
 	}
 
 	if item == -1 {

diff --git a/formatter/document.go b/formatter/document.go
@@ -44,6 +44,7 @@ const (
 	tsFuncCallStatement
 	tsTable
 	tsUnknow
+	tsBody
 )
 
 // chunk ::= block
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,6 +44,7 @@ const ( @@
     	tsFuncCallStatement
     	tsTable
     	tsUnknow
+    	tsBody
     )
     // chunk ::= block
@@ Expand Down @@