Add JSON ingestion indexing wrappers

Author: jwils

elasticgraph-json_ingestion/lib/elastic_graph/json_ingestion/schema_definition/indexing/field_reference.rb

@@ -0,0 +1,36 @@
+# Copyright 2024 - 2026 Block, Inc.
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+#
+# frozen_string_literal: true
+
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        # @!parse class FieldReference < ::Data; end
+        FieldReference = ::Data.define(
+          :field_reference,
+          :json_schema_layers,
+          :json_schema_customizations
+        )
+
+        # A JSON-schema-aware wrapper around the core indexing field reference.
+        #
+        # @api private
+        class FieldReference < ::Data
+          # Resolves this field reference, or `nil` if unresolvable.
+          #
+          # @return [ElasticGraph::SchemaDefinition::Indexing::Field, nil]
+          def resolve
+            field_reference.resolve
+          end
+
+          # @dynamic initialize, with, field_reference, json_schema_layers, json_schema_customizations
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/lib/elastic_graph/json_ingestion/schema_definition/indexing/field_type/enum.rb

@@ -0,0 +1,41 @@
+# Copyright 2024 - 2026 Block, Inc.
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+#
+# frozen_string_literal: true
+
+require "elastic_graph/schema_definition/indexing/field_type/enum"
+
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        # Namespace for JSON-ingestion indexing field type wrappers.
+        module FieldType
+          # JSON-ingestion indexing field type wrapper for enums.
+          #
+          # @private
+          class Enum
+            def self.wrap(field_type)
+              new(field_type)
+            end
+
+            def initialize(field_type)
+              @field_type = field_type
+            end
+
+            def enum_value_names
+              @field_type.enum_value_names
+            end
+
+            def to_mapping
+              @field_type.to_mapping
+            end
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/lib/elastic_graph/json_ingestion/schema_definition/indexing/field_type/object.rb

@@ -0,0 +1,56 @@
+# Copyright 2024 - 2026 Block, Inc.
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+#
+# frozen_string_literal: true
+
+require "elastic_graph/schema_definition/indexing/field_type/object"
+
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          # JSON-ingestion indexing field type wrapper for objects.
+          #
+          # @private
+          class Object
+            def self.wrap(field_type, json_schema_options:)
+              new(field_type)
+            end
+
+            def initialize(field_type)
+              @field_type = field_type
+            end
+
+            def schema_def_state
+              @field_type.schema_def_state
+            end
+
+            def type_name
+              @field_type.type_name
+            end
+
+            def subfields
+              @field_type.subfields
+            end
+
+            def mapping_options
+              @field_type.mapping_options
+            end
+
+            def doc_comment
+              @field_type.doc_comment
+            end
+
+            def to_mapping
+              @field_type.to_mapping
+            end
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/lib/elastic_graph/json_ingestion/schema_definition/indexing/field_type/scalar.rb

@@ -0,0 +1,40 @@
+# Copyright 2024 - 2026 Block, Inc.
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+#
+# frozen_string_literal: true
+
+require "elastic_graph/schema_definition/indexing/field_type/scalar"
+
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          # JSON-ingestion indexing field type wrapper for scalars.
+          #
+          # @private
+          class Scalar
+            def self.wrap(field_type)
+              new(field_type)
+            end
+
+            def initialize(field_type)
+              @field_type = field_type
+            end
+
+            def scalar_type
+              @field_type.scalar_type
+            end
+
+            def to_mapping
+              @field_type.to_mapping
+            end
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/lib/elastic_graph/json_ingestion/schema_definition/indexing/field_type/union.rb

@@ -0,0 +1,40 @@
+# Copyright 2024 - 2026 Block, Inc.
+#
+# Use of this source code is governed by an MIT-style
+# license that can be found in the LICENSE file or at
+# https://opensource.org/licenses/MIT.
+#
+# frozen_string_literal: true
+
+require "elastic_graph/schema_definition/indexing/field_type/union"
+
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          # JSON-ingestion indexing field type wrapper for unions.
+          #
+          # @private
+          class Union
+            def self.wrap(field_type)
+              new(field_type)
+            end
+
+            def initialize(field_type)
+              @field_type = field_type
+            end
+
+            def subtypes_by_name
+              @field_type.subtypes_by_name
+            end
+
+            def to_mapping
+              @field_type.to_mapping
+            end
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/sig/elastic_graph/json_ingestion/schema_definition/indexing/field_reference.rbs

@@ -0,0 +1,27 @@
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        class FieldReference
+          attr_reader field_reference: ::ElasticGraph::SchemaDefinition::Indexing::FieldReference
+          attr_reader json_schema_layers: ::ElasticGraph::SchemaDefinition::jsonSchemaLayersArray
+          attr_reader json_schema_customizations: ::Hash[::Symbol, untyped]
+
+          def initialize: (
+            field_reference: ::ElasticGraph::SchemaDefinition::Indexing::FieldReference,
+            json_schema_layers: ::ElasticGraph::SchemaDefinition::jsonSchemaLayersArray,
+            json_schema_customizations: ::Hash[::Symbol, untyped]
+          ) -> void
+
+          def with: (
+            ?field_reference: ::ElasticGraph::SchemaDefinition::Indexing::FieldReference,
+            ?json_schema_layers: ::ElasticGraph::SchemaDefinition::jsonSchemaLayersArray,
+            ?json_schema_customizations: ::Hash[::Symbol, untyped]
+          ) -> FieldReference
+
+          def resolve: () -> ::ElasticGraph::SchemaDefinition::Indexing::Field?
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/sig/elastic_graph/json_ingestion/schema_definition/indexing/field_type.rbs

@@ -0,0 +1,27 @@
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        interface _EnumFieldType
+          def enum_value_names: () -> ::Array[::String]
+        end
+
+        interface _ObjectFieldType
+          def schema_def_state: () -> ::ElasticGraph::SchemaDefinition::State
+          def type_name: () -> ::String
+          def subfields: () -> ::Array[::ElasticGraph::SchemaDefinition::Indexing::Field]
+          def mapping_options: () -> ::ElasticGraph::SchemaDefinition::Mixins::HasTypeInfo::optionsHash
+          def doc_comment: () -> ::String?
+        end
+
+        interface _ScalarFieldType
+          def scalar_type: () -> ::ElasticGraph::SchemaDefinition::SchemaElements::ScalarType
+        end
+
+        interface _UnionFieldType
+          def subtypes_by_name: () -> ::Hash[::String, ::Object]
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/sig/elastic_graph/json_ingestion/schema_definition/indexing/field_type/enum.rbs

@@ -0,0 +1,19 @@
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          class Enum
+            include _EnumFieldType
+
+            @field_type: ::ElasticGraph::SchemaDefinition::Indexing::FieldType::Enum
+
+            def self.wrap: (::ElasticGraph::SchemaDefinition::Indexing::FieldType::Enum) -> Enum
+            def initialize: (::ElasticGraph::SchemaDefinition::Indexing::FieldType::Enum) -> void
+            def to_mapping: () -> ::Hash[::String, untyped]
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/sig/elastic_graph/json_ingestion/schema_definition/indexing/field_type/object.rbs

@@ -0,0 +1,23 @@
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          class Object
+            include _ObjectFieldType
+
+            @field_type: ::ElasticGraph::SchemaDefinition::Indexing::FieldType::Object
+
+            def self.wrap: (
+              ::ElasticGraph::SchemaDefinition::Indexing::FieldType::Object,
+              json_schema_options: ::Hash[::Symbol, untyped]
+            ) -> Object
+
+            def initialize: (::ElasticGraph::SchemaDefinition::Indexing::FieldType::Object) -> void
+            def to_mapping: () -> ::Hash[::String, untyped]
+          end
+        end
+      end
+    end
+  end
+end

elasticgraph-json_ingestion/sig/elastic_graph/json_ingestion/schema_definition/indexing/field_type/scalar.rbs

@@ -0,0 +1,19 @@
+module ElasticGraph
+  module JSONIngestion
+    module SchemaDefinition
+      module Indexing
+        module FieldType
+          class Scalar
+            include _ScalarFieldType
+
+            @field_type: ::ElasticGraph::SchemaDefinition::Indexing::FieldType::Scalar
+
+            def self.wrap: (::ElasticGraph::SchemaDefinition::Indexing::FieldType::Scalar) -> Scalar
+            def initialize: (::ElasticGraph::SchemaDefinition::Indexing::FieldType::Scalar) -> void
+            def to_mapping: () -> ::Hash[::String, untyped]
+          end
+        end
+      end
+    end
+  end
+end