Add check to pass through compiled classes

Author: adriangs1996

lib/mortymer/contract.rb

@@ -24,14 +24,19 @@ def initialize(errors)
     end
 
     def self.__internal_struct_repr__
-      @__internal_struct_repr__ || StructCompiler.new.compile(schema.json_schema)
+      @__internal_struct_repr__ ||= StructCompiler.new.compile(schema.json_schema)
     end
 
     def self.json_schema
       Generator.new.from_validation(self)
     end
 
     def self.structify(params)
+      # If params are already built using the internal struct, then there is
+      # no need to re-validate it
+      return params if params.instance_of?(__internal_struct_repr__)
+
+      params = params.to_h if params.is_a?(Dry::Struct)
       result = new.call(params)
       raise ContractError, result.errors.to_h unless result.errors.empty?
 

lib/mortymer/model.rb

@@ -24,7 +24,13 @@ def self.structify(params)
     end
 
     def [](key)
-      @attributes[key.to_sym]
+      public_send(key)
     end
+
+    def key?(attr)
+      attributes.key?(attr.to_sym)
+    end
+
+    alias has_key? key?
   end
 end

lib/mortymer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Mortymer
-  VERSION = "0.0.12"
+  VERSION = "0.0.13"
 end

spec/lib/mortymer/contract_spec.rb

@@ -224,6 +224,45 @@ class EnumContract < Mortymer::Contract
       end
     end
 
+    context "with already structified input" do
+      it "returns the input directly if it's already an instance of the internal struct" do
+        # First create a valid struct instance
+        params = {
+          name: "John Doe",
+          age: 30,
+          active: true
+        }
+        struct_instance = BasicContract.structify(params)
+
+        # Now pass that instance back to structify
+        result = BasicContract.structify(struct_instance)
+
+        # Should be the exact same object (not just equal)
+        expect(result.object_id).to eq(struct_instance.object_id)
+      end
+
+      it "still validates if input is a different struct type" do
+        # Create a similar but different struct class
+        different_struct = Class.new(Mortymer::Model) do
+          attribute :name, Mortymer::Model::String
+          attribute :age, Mortymer::Model::Integer
+          attribute :active, Mortymer::Model::Bool
+        end
+
+        # Create an instance of the different struct
+        different_instance = different_struct.new(
+          name: "John Doe",
+          age: 30,
+          active: true
+        )
+
+        # Should create a new struct of the correct type
+        result = BasicContract.structify(different_instance)
+        expect(result).to be_an_instance_of(BasicContract.__internal_struct_repr__)
+        expect(result).not_to be_an_instance_of(different_struct)
+      end
+    end
+
     context "with symbol keys and values" do
       it "handles symbol keys in input" do
         params = {