Handle more node types in recur lookup

Signed-off-by: James Hamlin <jfhamlin@gmail.com>

Author: jfhamlin

pkg/codegen/analyze.go

@@ -18,11 +18,41 @@ func nodeRecurs(n *ast.Node, loopID string) bool {
 	case ast.OpLet, ast.OpLoop:
 		letNode := n.Sub.(*ast.LetNode)
 		return nodeRecurs(letNode.Body, loopID)
+	case ast.OpLetFn:
+		letFnNode := n.Sub.(*ast.LetFnNode)
+		return nodeRecurs(letFnNode.Body, loopID)
 	case ast.OpIf:
 		ifNode := n.Sub.(*ast.IfNode)
 		return nodeRecurs(ifNode.Then, loopID) || nodeRecurs(ifNode.Else, loopID)
-		// TODO: review all other node types
+	case ast.OpTry:
+		tryNode := n.Sub.(*ast.TryNode)
+		if nodeRecurs(tryNode.Body, loopID) {
+			return true
+		}
+		for _, catch := range tryNode.Catches {
+			if nodeRecurs(catch, loopID) {
+				return true
+			}
+		}
+	case ast.OpCatch:
+		catchNode := n.Sub.(*ast.CatchNode)
+		return nodeRecurs(catchNode.Body, loopID)
+	case ast.OpCase:
+		caseNode := n.Sub.(*ast.CaseNode)
+		if nodeRecurs(caseNode.Default, loopID) {
+			return true
+		}
+		for _, branch := range caseNode.Nodes {
+			if nodeRecurs(branch, loopID) {
+				return true
+			}
+		}
+	case ast.OpCaseNode:
+		caseNode := n.Sub.(*ast.CaseNodeNode)
+		return nodeRecurs(caseNode.Then, loopID)
 	default:
 		return false // can't recur in this node type
 	}
+
+	return false
 }