[Pratt Parser] Add deep copy/replace functionality to AstFactoryInterface

PiperOrigin-RevId: 944578347

Author: dmitriplotnikov

common/expr_factory.h

@@ -38,6 +38,11 @@ namespace tools {
 class ProtoToPredicateBuilder;
 }
 
+namespace parser_internal {
+template <typename ExprNode>
+class AstFactoryInterface;
+}
+
 class ExprFactory {
  protected:
   // `IsExprLike` determines whether `T` is some `Expr`. Currently that means
@@ -385,6 +390,7 @@ class ExprFactory {
   friend class ParserMacroExprFactory;
   friend class OptimizerExprFactory;
   friend class tools::ProtoToPredicateBuilder;
+  friend class parser_internal::AstFactoryInterface<Expr>;
 
   ExprFactory() : accu_var_(kAccumulatorVariableName) {}
 

parser/internal/BUILD

@@ -13,12 +13,30 @@
 # limitations under the License.
 
 load("@rules_cc//cc:cc_library.bzl", "cc_library")
+load("@rules_cc//cc:cc_test.bzl", "cc_test")
 load("//bazel:antlr.bzl", "antlr_cc_library")
 
 package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])
 
+cc_library(
+    name = "ast_factory_interface",
+    hdrs = ["ast_factory_interface.h"],
+)
+
+cc_library(
+    name = "ast_factory",
+    hdrs = ["ast_factory.h"],
+    deps = [
+        ":ast_factory_interface",
+        "//common:constant",
+        "//common:expr",
+        "//common:expr_factory",
+        "@com_google_absl//absl/strings:string_view",
+    ],
+)
+
 cc_library(
     name = "options",
     hdrs = ["options.h"],
@@ -29,3 +47,133 @@ antlr_cc_library(
     src = "Cel.g4",
     package = "cel_parser_internal",
 )
+
+cc_library(
+    name = "lexer",
+    srcs = ["lexer.cc"],
+    hdrs = ["lexer.h"],
+    deps = [
+        "@com_google_absl//absl/base:core_headers",
+        "@com_google_absl//absl/base:no_destructor",
+        "@com_google_absl//absl/base:nullability",
+        "@com_google_absl//absl/container:flat_hash_map",
+        "@com_google_absl//absl/functional:function_ref",
+        "@com_google_absl//absl/log:absl_check",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/strings:string_view",
+    ],
+)
+
+cc_library(
+    name = "pratt_parser_worker",
+    srcs = ["pratt_parser_worker.cc"],
+    hdrs = ["pratt_parser_worker.h"],
+    deps = [
+        ":ast_factory_interface",
+        ":lexer",
+        "//common:operators",
+        "//common:source",
+        "//internal:lexis",
+        "//internal:strings",
+        "//parser:options",
+        "//parser:parser_interface",
+        "@com_google_absl//absl/base:nullability",
+        "@com_google_absl//absl/container:flat_hash_map",
+        "@com_google_absl//absl/log:absl_check",
+        "@com_google_absl//absl/status:statusor",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/strings:str_format",
+        "@com_google_absl//absl/types:optional",
+    ],
+)
+
+cc_library(
+    name = "pratt_parser",
+    srcs = ["pratt_parser.cc"],
+    hdrs = ["pratt_parser.h"],
+    deps = [
+        ":ast_factory",
+        ":pratt_parser_worker",
+        "//common:ast",
+        "//common:expr",
+        "//common:source",
+        "//internal:status_macros",
+        "//parser:macro",
+        "//parser:macro_registry",
+        "//parser:options",
+        "//parser:parser_interface",
+        "@com_google_absl//absl/base:nullability",
+        "@com_google_absl//absl/cleanup",
+        "@com_google_absl//absl/container:flat_hash_map",
+        "@com_google_absl//absl/container:flat_hash_set",
+        "@com_google_absl//absl/log:absl_check",
+        "@com_google_absl//absl/status",
+        "@com_google_absl//absl/status:statusor",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/strings:str_format",
+        "@com_google_absl//absl/types:span",
+    ],
+)
+
+cc_test(
+    name = "ast_factory_test",
+    srcs = ["ast_factory_test.cc"],
+    deps = [
+        ":ast_factory",
+        "//common:constant",
+        "//common:expr",
+        "//internal:testing",
+        "@com_google_absl//absl/strings:string_view",
+    ],
+)
+
+cc_test(
+    name = "lexer_test",
+    srcs = ["lexer_test.cc"],
+    deps = [
+        ":lexer",
+        "//internal:testing",
+    ],
+)
+
+cc_test(
+    name = "pratt_parser_test",
+    srcs = ["pratt_parser_test.cc"],
+    deps = [
+        ":lexer",
+        ":pratt_parser",
+        ":pratt_parser_worker",
+        "//common:ast",
+        "//common:constant",
+        "//common:expr",
+        "//common:source",
+        "//internal:status_macros",
+        "//internal:testing",
+        "//parser:options",
+        "//parser:parser_interface",
+        "//testutil:expr_printer",
+        "@com_google_absl//absl/status",
+        "@com_google_absl//absl/status:status_matchers",
+        "@com_google_absl//absl/status:statusor",
+        "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/strings:str_format",
+        "@com_google_absl//absl/strings:string_view",
+    ],
+)
+
+cc_test(
+    name = "pratt_parser_benchmark",
+    srcs = ["pratt_parser_benchmark.cc"],
+    tags = ["benchmark"],
+    deps = [
+        ":pratt_parser",
+        "//common:source",
+        "//internal:benchmark",
+        "//internal:testing",
+        "//parser",
+        "//parser:options",
+        "//parser:parser_interface",
+        "@com_google_absl//absl/log:absl_check",
+        "@com_google_absl//absl/strings",
+    ],
+)