♻️ rework field accessors

Author: ianardee

docs/code_samples/default_v2.txt

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'mindee'
 
 input_path = '/path/to/the/file.ext'
@@ -22,7 +24,7 @@ inference_params = Mindee::Input::InferenceParameters.new(
     polygon: nil,
     # Boost the precision and accuracy of all extractions.
     # Calculate confidence scores for all fields.
-    confidence: nil,
+    confidence: nil
 )
 
 # Load a file from disk

lib/mindee/parsing/v2/field/inference_fields.rb

@@ -6,7 +6,8 @@ module Mindee
   module Parsing
     module V2
       module Field
-        # Collection of inference fields that extends Hash functionality.
+        # Represents a hash-like collection of inference fields, providing methods for
+        # retrieval and string representation.
         class InferenceFields < Hash
           # @return [Integer] Level of indentation for rst display.
           attr_reader :indent_level
@@ -29,23 +30,37 @@ def get(key)
             self[key]
           end
 
-          # Allow dot notation access to fields.
-          # @param method_name [Symbol] The method name (field key).
-          # @return [BaseField, nil] The field or nil if not found.
-          def method_missing(method_name, *args, &block)
-            key = method_name.to_s
-            if key?(key)
-              self[key]
-            else
-              super
-            end
+          # Get a field by key and ensure it is a SimpleField.
+          # @param key [String] Field key to retrieve.
+          # @return [SimpleField] The SimpleField.
+          # @raise [TypeError] If the field is not a SimpleField.
+          def get_simple_field(key)
+            field = self[key]
+            raise TypeError, "Field #{key} is not a SimpleField" unless field.is_a?(SimpleField)
+
+            field
           end
 
-          # Check if method_missing should handle the method.
-          # @param method_name [Symbol] The method name.
-          # @return [Boolean] `true` if the method should be handled.
-          def respond_to_missing?(method_name, include_private = false)
-            key?(method_name.to_s) || super
+          # Get a field by key and ensure it is a ListField.
+          # @param key [String] Field key to retrieve.
+          # @return [ListField] The ListField.
+          # @raise [TypeError] If the field is not a ListField.
+          def get_list_field(key)
+            field = self[key]
+            raise TypeError, "Field #{key} is not a ListField" unless field.is_a?(ListField)
+
+            field
+          end
+
+          # Get a field by key and ensure it is an ObjectField.
+          # @param key [String] Field key to retrieve.
+          # @return [ObjectField] The ObjectField.
+          # @raise [TypeError] If the field is not an ObjectField.
+          def get_object_field(key)
+            field = self[key]
+            raise TypeError, "Field #{key} is not a ObjectField" unless field.is_a?(ObjectField)
+
+            field
           end
 
           # rubocop:disable Metrics/CyclomaticComplexity
@@ -64,13 +79,7 @@ def to_s(indent = 0)
               line = "#{padding}:#{field_key}:"
 
               case (field_value.class.name || '').split('::').last
-              when 'ListField'
-                # Check if ListField has items and they're not empty
-                list_f = field_value # @type var list_f: ListField
-                if defined?(list_f.items) && list_f.items && !list_f.items.empty?
-                  line += list_f.to_s
-                end
-              when 'ObjectField'
+              when 'ListField', 'ObjectField'
                 line += field_value.to_s
               when 'SimpleField'
                 # Check if SimpleField has a non-empty value

lib/mindee/parsing/v2/field/list_field.rb

@@ -29,10 +29,30 @@ def initialize(server_response, indent_level = 0)
             end
           end
 
+          def simple_items
+            fields = []
+            @items.each do |item|
+              raise TypeError, "Invalid field type detected: #{item.class}" unless item.is_a?(SimpleField)
+
+              fields << item
+            end
+            fields
+          end
+
+          def object_items
+            fields = []
+            @items.each do |item|
+              raise TypeError, "Invalid field type detected: #{item.class}" unless item.is_a?(ObjectField)
+
+              fields << item
+            end
+            fields
+          end
+
           # String representation of the list field.
           # @return [String] Formatted string with bullet points for each item.
           def to_s
-            return "\n" if @items.empty?
+            return '' unless @items && !@items.empty?
 
             parts = ['']
             @items.each do |item|
@@ -47,40 +67,6 @@ def to_s
 
             parts.join("\n  * ")
           end
-
-          # Check if the list is empty.
-          # @return [Boolean] `true` if the list has no items.
-          def empty?
-            @items.empty?
-          end
-
-          # Get the number of items in the list.
-          # @return [Integer] Number of items.
-          def size
-            @items.size
-          end
-
-          # Get the number of items in the list (alias for size).
-          # @return [Integer] Number of items.
-          def length
-            @items.length
-          end
-
-          # Get an item by index.
-          # @param index [Integer] The index of the item to retrieve.
-          # @return [BaseField, nil] The item at the given index.
-          def [](index)
-            @items[index]
-          end
-
-          # Iterator for Enumerator inheritance.
-          # NOTE: Untyped due to incomplete support in current supported version of RBS.
-          def each(&block)
-            return to_enum(:each) unless block_given?
-
-            @items.each(&block)
-            self
-          end
         end
       end
     end

lib/mindee/parsing/v2/field/object_field.rb

@@ -78,7 +78,7 @@ def multi_str
             out_str
           end
 
-          # Allow dot notation access to nested fields.
+          # Allow direct access to nested fields.
           # @param method_name [Symbol] The method name (field key).
           # @return [ObjectField, nil] The field or nil if not found.
           def method_missing(method_name, ...)

lib/mindee/parsing/v2/field/simple_field.rb

@@ -15,7 +15,12 @@ class SimpleField < BaseField
           # @param indent_level [Integer] Level of indentation for rst display.
           def initialize(server_response, indent_level = 0)
             super
-            @value = server_response.key?('value') ? server_response['value'] : nil
+            value = server_response['value']
+            @value = if value.is_a?(Integer)
+                       value.to_f
+                     else
+                       value
+                     end
           end
 
           # String representation of the field value.

sig/mindee/parsing/v2/field/inference_fields.rbs

@@ -6,11 +6,12 @@ module Mindee
         class InferenceFields < Hash[String, ListField | ObjectField | SimpleField?]
           attr_reader indent_level: Integer
 
+          def get_list_field: (String) -> ListField
+          def get_simple_field: (String) -> SimpleField
+          def get_object_field: (String) -> ObjectField
           def logger: () -> Logger
           def initialize: (Hash[String | Symbol, untyped], ?Integer) -> void
           def get: (String) -> (ListField | ObjectField | SimpleField?)
-          def method_missing: (Symbol, *untyped) -> (ListField | ObjectField | SimpleField?)
-          def respond_to_missing?: (Symbol, ?bool) -> bool
           def to_s: (?Integer) -> String
         end
       end

sig/mindee/parsing/v2/field/list_field.rbs

@@ -6,8 +6,10 @@ module Mindee
         class ListField < BaseField
           include Enumerable[BaseField]
 
-          attr_reader items: Array[BaseField]
+          attr_reader items: Array[SimpleField | ObjectField | ListField]
           def initialize: (Hash[String | Symbol, untyped], ?Integer) -> void
+          def object_items: -> Array[ObjectField]
+          def simple_items: -> Array[SimpleField]
           def to_s: -> String
           def empty?: -> bool
           def size: -> Integer

sig/mindee/parsing/v2/field/simple_field.rbs

@@ -4,7 +4,7 @@ module Mindee
     module V2
       module Field
         class SimpleField < BaseField
-          attr_reader value: String | Integer | Float | bool?
+          attr_reader value: String | Integer | Float | bool | nil
 
           def initialize: (Hash[String | Symbol, untyped], ?Integer) -> void
           def to_s: -> String