fix entity finder

factorialco · Aug 26, 2024 · d4d8dd1 · d4d8dd1
1 parent 6dea8b7
commit d4d8dd1
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 29 deletions.
diff --git a/lib/entity_finder.rb b/lib/entity_finder.rb
diff --git a/lib/public/entity_finder.rb b/lib/public/entity_finder.rb
@@ -0,0 +1,18 @@
+# typed: strict
+
+module ResourceRegistry
+  class EntityFinder
+    extend T::Sig
+
+    sig do
+      params(repository: T.class_of(Repositories::Base)).returns(T.nilable(T.class_of(T::Struct)))
+    end
+    def self.call(repository:)
+      entity = repository.entity
+
+      return nil if entity.is_a?(T::Types::Untyped)
+
+      entity
+    end
+  end
+end
diff --git a/lib/resource_registry.rb b/lib/resource_registry.rb
@@ -8,6 +8,7 @@
 require 'public/resource'
 require 'public/versions'
 require 'public/versions/version'
+require 'public/entity_finder'
 require 'schema_registry/json_schema_mapper'
 require 'schema_registry/maybe'
 require 'runtime_generic'