fix: belongs_to broken for some model names (#203)

* fix: belongs_to broken for some model names

* refactor: use the klass first if available

lib/avo/app/fields/belongs_to.rb

@@ -22,7 +22,8 @@ def hydrate_field(fields, model, resource, view)
         fields[:searchable] = @searchable
         fields[:is_relation] = true
         fields[:database_id] = foreign_key model
-        target_resource = App.get_resources.find { |r| r.class == "Avo::Resources::#{name}".safe_constantize }
+
+        target_resource = get_target_resource model
 
         relation_model = model.public_send(@relation_method)
 
@@ -61,6 +62,16 @@ def foreign_key(model)
           model.class.reflections[@relation_method].foreign_key
         end
       end
+
+      def get_target_resource(model)
+        if model._reflections[id.to_s].klass.present?
+          App.get_resource_by_model_name model._reflections[id.to_s].klass.to_s
+        elsif model._reflections[id.to_s].options[:class_name].present?
+          App.get_resource_by_model_name model._reflections[id.to_s].options[:class_name]
+        else
+          App.get_resource_by_name class_name.to_s
+        end
+      end
     end
   end
 end