Fix cast to capped varchar for JSON (#24396)

Summary:

https://www.internalfb.com/tasks/?t=211442303

There was an error in running query on Prestissimo not Presto - "Scalar function presto.default.substr not registered with arguments: (JSON, BIGINT, BIGINT)".

This is not due to missing function, as the function signature does not exist in Presto. It occurs when attempting to cast JSON as varchar of capped length.
Related Diff: https://www.internalfb.com/diff/D59531026
Note: Exception is still raised for try_cast() behavior. Alignment is out of scope for this PR

Differential Revision: D68353517

Author: natashasehgal

presto-native-execution/presto_cpp/main/types/PrestoToVeloxExpr.cpp

@@ -181,8 +181,15 @@ std::optional<TypedExprPtr> convertCastToVarcharWithMaxLength(
   VELOX_DCHECK(end == returnType.data() + returnType.size() - 1);
 
   VELOX_DCHECK_EQ(args.size(), 1);
-  const auto arg = args[0];
 
+  auto arg = args[0];
+  // If the argument is of JSON type, convert it to VARCHAR before applying substr.
+  if (velox::isJsonType(arg->type())) {
+    arg = std::make_shared<CallTypedExpr>(
+        velox::VARCHAR(),
+        std::vector<TypedExprPtr>{arg},
+        "presto.default.json_format");
+  }
   return std::make_shared<CallTypedExpr>(
       arg->type(),
       std::vector<TypedExprPtr>{

presto-native-execution/presto_cpp/main/types/tests/RowExpressionTest.cpp

@@ -18,13 +18,17 @@
 #include "presto_cpp/presto_protocol/core/presto_protocol_core.h"
 #include "velox/core/Expressions.h"
 #include "velox/type/Type.h"
+#include "velox/functions/prestosql/types/JsonType.h"
 
 using namespace facebook::presto;
 using namespace facebook::velox;
 using namespace facebook::velox::core;
 
 class RowExpressionTest : public ::testing::Test {
  public:
+  RowExpressionTest(){
+    registerJsonType();
+  }
   static void SetUpTestCase() {
     memory::MemoryManager::testingSetInstance({});
   }
@@ -626,6 +630,20 @@ TEST_F(RowExpressionTest, castToVarchar) {
     ASSERT_TRUE(returnExpr->nullOnFailure());
     ASSERT_EQ(returnExpr->type()->toString(), "VARCHAR");
   }
+  // CAST(json AS varchar(3))
+  {
+    std::shared_ptr<protocol::CallExpression> p =
+        json::parse(makeCastToVarchar(false, "json", "varchar(3)"));
+    auto expr = converter_->toVeloxExpr(p);
+    auto returnExpr = std::dynamic_pointer_cast<const CallTypedExpr>(expr);
+    ASSERT_NE(returnExpr, nullptr);
+    ASSERT_EQ(returnExpr->name(), "presto.default.substr");
+
+    auto returnArg = std::dynamic_pointer_cast<const ConstantTypedExpr>(
+        returnExpr->inputs()[2]);
+    ASSERT_EQ(returnArg->type()->toString(), "BIGINT");
+    ASSERT_EQ(returnArg->value().toJson(returnArg->type()), "3");
+  }
 }
 
 TEST_F(RowExpressionTest, special) {