Implement Prism -> Sorbet conversion for if node

main/pipeline/pipeline.cc

@@ -157,7 +157,7 @@ core::LocOffsets locOffset(pm_location_t *loc, pm_parser_t *parser) {
     return core::LocOffsets{locStart, locEnd};
 }
 
-const unique_ptr<parser::Node> convertPrismToSorbet(pm_node_t *node, pm_parser_t *parser, core::GlobalState &gs) {
+unique_ptr<parser::Node> convertPrismToSorbet(pm_node_t *node, pm_parser_t *parser, core::GlobalState &gs) {
     switch (PM_NODE_TYPE(node)) {
         case PM_FALSE_NODE: {
             auto falseNode = reinterpret_cast<pm_false_node *>(node);
@@ -171,6 +171,23 @@ const unique_ptr<parser::Node> convertPrismToSorbet(pm_node_t *node, pm_parser_t
 
             return make_unique<parser::Float>(locOffset(loc, parser), std::to_string(floatNode->value));
         }
+        case PM_IF_NODE: {
+            auto ifNode = reinterpret_cast<pm_if_node *>(node);
+            auto *loc = &ifNode->base.location;
+
+            auto condition = convertPrismToSorbet(ifNode->predicate, parser, gs);
+
+            std::unique_ptr<parser::Node> ifTrue;
+
+            if (ifNode->statements == nullptr) {
+                ifTrue = nullptr;
+            } else {
+                ifTrue = convertPrismToSorbet(reinterpret_cast<pm_node *>(ifNode->statements), parser, gs);
+            }
+
+            // TODO: handle elsif/else clause
+            return make_unique<parser::If>(locOffset(loc, parser), std::move(condition), std::move(ifTrue), nullptr);
+        }
         case PM_INTEGER_NODE: {
             auto intNode = reinterpret_cast<pm_integer_node *>(node);
             pm_location_t *loc = &intNode->base.location;
@@ -297,7 +314,6 @@ const unique_ptr<parser::Node> convertPrismToSorbet(pm_node_t *node, pm_parser_t
         case PM_GLOBAL_VARIABLE_WRITE_NODE:
         case PM_HASH_NODE:
         case PM_HASH_PATTERN_NODE:
-        case PM_IF_NODE:
         case PM_IMAGINARY_NODE:
         case PM_IMPLICIT_NODE:
         case PM_IMPLICIT_REST_NODE:

test/prism_regression/if.parse-tree.exp

@@ -0,0 +1,6 @@
+If {
+  condition = True {
+  }
+  then_ = NULL
+  else_ = NULL
+}

test/prism_regression/if.rb

@@ -0,0 +1,3 @@
+# typed: strict
+
+if true; end

test/prism_regression/if_with_stmt.parse-tree.exp

@@ -0,0 +1,8 @@
+If {
+  condition = False {
+  }
+  then_ = Integer {
+    val = "5"
+  }
+  else_ = NULL
+}

test/prism_regression/if_with_stmt.rb

@@ -0,0 +1,5 @@
+# typed: static
+
+if false
+  5
+end

test/prism_regression/if_with_stmts.parse-tree.exp

@@ -0,0 +1,17 @@
+If {
+  condition = True {
+  }
+  then_ = Begin {
+    stmts = [
+      True {
+      }
+      Integer {
+        val = "4"
+      }
+      String {
+        val = <U string>
+      }
+    ]
+  }
+  else_ = NULL
+}

test/prism_regression/if_with_stmts.rb

@@ -0,0 +1,7 @@
+# typed: strict
+
+if true
+  true
+  4
+  "string"
+end