# This is a combination of 2 commits.

# This is the 1st commit message:

WIP

# This is the commit message #2:

More WIP, still does not work

Author: halostatue

lib/mime/type.rb

@@ -184,11 +184,13 @@ def like?(other)
 
   # Compares the +other+ MIME::Type against the exact content type or the
   # simplified type (the simplified type will be used if comparing against
-  # something that can be treated as a String with #to_s). In comparisons,
-  # this is done against the lowercase version of the MIME::Type.
+  # something that can be treated as a String with #to_s). In comparisons, this
+  # is done against the lowercase version of the MIME::Type.
   #
   # Note that this implementation of #<=> is deprecated and will be changed
   # in the next major version to be the same as #priority_compare.
+  #
+  # Note that MIME::Types no longer compare against nil.
   def <=>(other)
     return priority_compare(other) if other.is_a?(MIME::Type)
     simplified <=> other
@@ -201,7 +203,7 @@ def <=>(other)
   # consumers of mime-types. For the next major version of MIME::Types, this
   # method will become #<=> and #priority_compare will be removed.
   def priority_compare(other)
-    if (cmp = __sort_priority <=> other.__sort_priority).zero?
+    if (cmp = __sort_priority <=> other.__sort_priority) == 0
       simplified <=> other.simplified
     else
       cmp
@@ -243,7 +245,8 @@ def hash
   # The computed sort priority value. This is _not_ intended to be used by most
   # callers.
   def __sort_priority # :nodoc:
-    @__sort_priority || update_sort_priority
+    update_sort_priority if !instance_variable_defined?(:@__sort_priority) || @__sort_priority.nil?
+    @__sort_priority
   end
 
   # Returns the whole MIME content-type string.
@@ -330,7 +333,7 @@ def preferred_extension=(value) # :nodoc:
     if value
       add_extensions(value)
       set_preferred_extension_priority(value)
-    else
+    elsif instance_variable_defined?(:@preferred_extension)
       clear_extension_priority(@preferred_extension)
     end
     @preferred_extension = value
@@ -579,7 +582,7 @@ def encode_with(coder)
     coder["registered"] = registered?
     coder["provisional"] = provisional? if provisional?
     coder["signature"] = signature? if signature?
-    coder["sort-priority"] = __sort_priority
+    coder["sort-priority"] = __sort_priority || 0b11111111
     coder["extension-priorities"] = __extension_priorities unless __extension_priorities.empty?
     coder
   end
@@ -659,12 +662,12 @@ def simplify_matchdata(matchdata, remove_x = false, joiner: "/")
     end
   end
 
-  private
-
-  def __extension_priorities
+  def __extension_priorities # :nodoc:
     @extension_priorities ||= {}
   end
 
+  private
+
   def clear_extension_priority(ext)
     __extension_priorities.delete(ext) if ext
   end
@@ -699,14 +702,15 @@ def clear_sort_priority
   # 16, to a minimum of 0.
   def update_sort_priority
     extension_count = @extensions.length
-    obsolete = @obsolete ? 1 << 7 : 0
-    provisional = @provisional ? 1 << 6 : 0
-    registered = @registered ? 0 : 1 << 5
+    obsolete = (instance_variable_defined?(:@obsolete) && @obsolete) ? 1 << 7 : 0
+    provisional = (instance_variable_defined?(:@provisional) && @provisional) ? 1 << 6 : 0
+    registered = (instance_variable_defined?(:@registered) && @registered) ? 0 : 1 << 5
     complete = extension_count.nonzero? ? 0 : 1 << 4
     extension_count = [0, 16 - extension_count].max
 
     @__sort_priority = obsolete | registered | provisional | complete | extension_count
-    @__priority_penalty = (@obsolete ? 3 : 0) + (@registered ? 0 : 2)
+    @__priority_penalty = ((instance_variable_defined?(:@obsolete) && @obsolete) ? 3 : 0) +
+      ((instance_variable_defined?(:@registered) && @registered) ? 0 : 2)
   end
 
   def __priority_penalty

lib/mime/type/columnar.rb

@@ -15,8 +15,10 @@
 class MIME::Type::Columnar < MIME::Type
   def initialize(container, content_type, extensions) # :nodoc:
     @container = container
+    @__priority_penalty = nil
     self.content_type = content_type
-    self.extensions = extensions
+    @extensions = Set[*Array(extensions).flatten.compact].freeze
+    clear_sort_priority
   end
 
   def self.column(*methods, file: nil) # :nodoc:
@@ -60,7 +62,7 @@ def update_sort_priority
       obsolete = (@__sort_priority & (1 << 7)) != 0
       registered = (@__sort_priority & (1 << 5)) == 0
 
-      @__priority_penalty = (@obsolete ? 3 : 0) + (@registered ? 0 : 2)
+      @__priority_penalty = (obsolete ? 3 : 0) + (registered ? 0 : 2)
     end
   end
 

lib/mime/types.rb

@@ -148,12 +148,10 @@ def [](type_id, complete: false, registered: false)
   #   puts MIME::Types.type_for(%w(citydesk.xml citydesk.gif))
   #     => [application/xml, image/gif, text/xml]
   def type_for(filename)
-    extensions = Array(filename).map { |fn| fn.chomp.downcase[/\.?([^.]*?)$/, 1] }
-
-    extensions.flat_map { |ext|
-      @extension_index[ext]
+    Array(filename).flat_map { |fn|
+      @extension_index[fn.chomp.downcase[/\.?([^.]*?)\z/m, 1]]
     }.compact.inject(Set.new, :+).sort { |a, b|
-      by_ext = a.extension_priority(*extensions) <=> b.extension_priority(*extensions)
+      by_ext = a.extension_priority(*a.extensions) <=> b.extension_priority(*b.extensions)
 
       if by_ext.zero?
         a.priority_compare(b)

lib/mime/types/_columnar.rb

@@ -30,7 +30,6 @@ def load_base_data(path) # :nodoc:
       line = line.split
       content_type = line.shift
       extensions = line
-      # content_type, *extensions = line.split
 
       type = MIME::Type::Columnar.new(self, content_type, extensions)
       @__mime_data__ << type
@@ -162,9 +161,9 @@ def dict(line, transform: nil)
   def dict_extension_priority(h, k, v)
     return if v.nil?
 
-    v = v.to_i if v.kind_of?(String)
-    v = v.trunc if v.kind_of?(Float)
-    v =  [[-20, v].max, 20].min
+    v = v.to_i if v.is_a?(String)
+    v = v.trunc if v.is_a?(Float)
+    v = [[-20, v].max, 20].min
 
     return if v.zero?
 

test/test_mime_type.rb

@@ -175,11 +175,6 @@ def mime_type(content_type)
       refute_equal text_plain, "text/html"
       assert_operator text_plain, :>, "text/html"
     end
-
-    it "correctly compares against nil" do
-      refute_equal text_html, nil
-      assert_operator text_plain, :<, nil
-    end
   end
 
   describe "#ascii?" do
@@ -326,16 +321,20 @@ def mime_type(content_type)
   end
 
   describe "#priority_compare" do
+    def priority(type)
+      "#{type} (#{("%08b" % type.__sort_priority).chars.join(" ")})"
+    end
+
     def assert_priority_less(left, right)
-      assert_equal(-1, left.priority_compare(right))
+      assert_equal(-1, left.priority_compare(right), "#{priority(left)} is not less than #{priority(right)}")
     end
 
     def assert_priority_same(left, right)
-      assert_equal 0, left.priority_compare(right)
+      assert_equal 0, left.priority_compare(right), "#{priority(left)} is not equal to #{priority(right)}"
     end
 
     def assert_priority_more(left, right)
-      assert_equal 1, left.priority_compare(right)
+      assert_equal 1, left.priority_compare(right), "#{priority(left)} is not more than #{priority(right)}"
     end
 
     def assert_priority(left, middle, right)
@@ -348,19 +347,24 @@ def assert_priority(left, middle, right)
     let(:text_1p) { mime_type("content-type" => "text/1") }
     let(:text_2) { mime_type("content-type" => "text/2") }
 
-    it "sorts (1) based on the simplified type" do
+    it "sorts based on the simplified type when the sort priorities are the same" do
       assert_priority text_1, text_1p, text_2
     end
 
-    it "sorts (2) based on extensions" do
-      text_1.extensions = ["foo", "bar"]
-      text_2.extensions = ["foo"]
+    it "sorts obsolete types higher than non-obsolete types" do
+      text_1.obsolete = text_1p.obsolete = false
+      text_1b = mime_type(text_1) { |t| t.obsolete = true }
+
+      assert_priority_less text_1, text_1b
 
-      assert_priority_same text_1, text_2
+      assert_priority text_1, text_1p, text_1b
+    end
 
-      text_2.registered = true
+    it "sorts provisional types higher than non-provisional types" do
+      text_1.provisional = text_1p.provisional = true
+      text_1b = mime_type(text_1) { |t| t.provisional = false }
 
-      assert_priority_more text_1, text_2
+      assert_priority text_1, text_1p, text_1b
     end
 
     it "sorts (3) based on the registration state" do
@@ -377,13 +381,6 @@ def assert_priority(left, middle, right)
       assert_priority text_1, text_1p, text_1b
     end
 
-    it "sorts (5) based on obsolete status" do
-      text_1.obsolete = text_1p.obsolete = false
-      text_1b = mime_type(text_1) { |t| t.obsolete = true }
-
-      assert_priority text_1, text_1p, text_1b
-    end
-
     it "sorts (5) based on the use-instead value" do
       text_1.obsolete = text_1p.obsolete = true
       text_1.use_instead = text_1p.use_instead = "abc/xyz"
@@ -395,6 +392,13 @@ def assert_priority(left, middle, right)
 
       assert_priority text_1, text_1p, text_1b
     end
+
+    it "sorts based on extensions (more extensions sort lower)" do
+      text_1.extensions = ["foo", "bar"]
+      text_2.extensions = ["foo"]
+
+      assert_priority_less text_1, text_2
+    end
   end
 
   describe "#raw_media_type" do
@@ -502,10 +506,10 @@ def assert_has_keys(wanted_keys, actual, msg = nil)
 
   describe "#to_json" do
     let(:expected_1) {
-      '{"content-type":"a/b","encoding":"base64","registered":false}'
+      '{"content-type":"a/b","encoding":"base64","registered":false,"sort-priority":48}'
     }
     let(:expected_2) {
-      '{"content-type":"a/b","encoding":"base64","registered":true,"provisional":true}'
+      '{"content-type":"a/b","encoding":"base64","registered":true,"provisional":true,"sort-priority":80}'
     }
 
     it "converts to JSON when requested" do

test/test_mime_types_class.rb

@@ -47,7 +47,7 @@ def setup
       }
       # This is this way because of a new type ending with gzip that only
       # appears in some data files.
-      assert_equal %w[application/gzip application/x-gzip multipart/x-gzip], types
+      assert_equal %w[application/gzip multipart/x-gzip application/x-gzip], types
       assert_equal 3, types.size
     end
 
@@ -86,9 +86,8 @@ def setup
       assert_equal %w[image/jpeg], MIME::Types.of(["foo.jpeg", "bar.jpeg"])
     end
 
-    it "finds multiple extensions" do
-      assert_equal %w[image/jpeg text/plain],
-        MIME::Types.type_for(%w[foo.txt foo.jpeg])
+    it "finds multiple extensions ordered by the filename list" do
+      assert_equal %w[text/plain image/jpeg], MIME::Types.type_for(%w[foo.txt foo.jpeg])
     end
 
     it "does not find unknown extensions" do