diff --git a/app/models/embed/purl/resource.rb b/app/models/embed/purl/resource.rb index b8909b2ac..e385744b1 100644 --- a/app/models/embed/purl/resource.rb +++ b/app/models/embed/purl/resource.rb @@ -47,7 +47,11 @@ def size # @return [ResourceFile] def primary_file - files.find(&:primary?) + files.find { |file| primary_types.include?(file.mimetype) } + end + + def primary_types + @primary_types ||= Array(Settings.primary_mimetypes[type]) end # @return [ResourceFile]