minad-mimemagic 0.1.1

data/script/generate-mime.rb

@@ -0,0 +1,97 @@
+#!/usr/bin/ruby
+
+require 'rexml/document'
+
+def str2int(s)
+  return s.to_i(16) if s[0..1].downcase == '0x'
+  return s.to_i(8) if s[0..0].downcase == '0'
+  s.to_i(10)
+end
+
+def get_matches(parent)
+  parent.get_elements('match').map {|match|
+    if match.attributes['mask']
+      nil
+    else
+      type = match.attributes['type']
+      value = match.attributes['value']
+      offset = match.attributes['offset'].split(':').map {|x| x.to_i }
+      offset = offset.size == 2 ? offset[0]..offset[1] : offset[0]
+      case type
+      when 'string'
+        value.gsub!(/\\(x[\dA-Fa-f]{1,2}|0\d{1,3}|\d{1,3}|.)/) { eval("\"\\#{$1}\"") }
+      when 'big16'
+        value = str2int(value)
+        value = ((value >> 8).chr + (value & 0xFF).chr)
+      when 'big32'
+        value = str2int(value)
+        value = (((value >> 24) & 0xFF).chr + ((value >> 16) & 0xFF).chr + ((value >> 8) & 0xFF).chr + (value & 0xFF).chr)
+      when 'little16'
+        value = str2int(value)
+        value = ((value & 0xFF).chr + (value >> 8).chr)
+      when 'little32'
+        value = str2int(value)
+        value = ((value & 0xFF).chr + ((value >> 8) & 0xFF).chr + ((value >> 16) & 0xFF).chr + ((value >> 24) & 0xFF).chr)
+      when 'host16' # use little endian
+        value = str2int(value)
+        value = ((value & 0xFF).chr + (value >> 8).chr)
+      when 'host32' # use little endian
+        value = str2int(value)
+        value = ((value & 0xFF).chr + ((value >> 8) & 0xFF).chr + ((value >> 16) & 0xFF).chr + ((value >> 24) & 0xFF).chr)
+      when 'byte'
+        value = str2int(value)
+        value = value.chr
+      end
+      children = get_matches(match)
+      children.empty? ? [offset, value] : [offset, value, children]
+    end
+  }.compact
+end
+
+FILE = ARGV[0] || '/usr/share/mime/packages/freedesktop.org.xml'
+file = File.new(FILE)
+doc = REXML::Document.new(file)
+extensions = {}
+types = {}
+magics = []
+doc.each_element('mime-info/mime-type') do |mime|
+  type = mime.attributes['type']
+  subclass = mime.get_elements('sub-class-of').map{|x| x.attributes['type']}
+  exts = mime.get_elements('glob').map{|x| x.attributes['pattern'] =~ /^\*\.([^\[\]]+)$/ ? $1.downcase : nil }.compact
+  mime.get_elements('magic').each do |magic|
+    priority = magic.attributes['priority'].to_i
+    matches = get_matches(magic)
+    magics << [priority, type, matches]
+  end
+  if !exts.empty?
+    exts.each{|x|
+      extensions[x] = type if !extensions.include?(x)
+    }
+    types[type] = [exts,subclass]
+  end
+end
+
+magics = magics.sort {|a,b| b[0] <=> a[0] }.map {|x| [x[1], x[2]] }
+
+puts "# Generated from #{FILE}"
+puts "class MimeMagic"
+puts "  private"
+puts "  EXTENSIONS = {"
+extensions.keys.sort.each do |key|
+  puts "    '#{key}' => '#{extensions[key]}',"
+end
+puts "  }"
+puts "  TYPES = {"
+types.keys.sort.each do |key|
+  exts = types[key][0].sort.inspect
+  parents = types[key][1].sort.inspect
+  puts "    '#{key}' => [#{exts}, #{parents}],"
+end
+puts "  }"
+puts "  MAGIC = ["
+magics.each do |type, matches|
+  puts "    ['#{type}', #{matches.inspect}],"
+end
+puts "  ]"
+puts "end"
+

data/test/test_mimemagic.rb

@@ -0,0 +1,31 @@
+require 'mimemagic'
+
+class TC_MimeMagic < Test::Unit::TestCase
+  def test_text?
+    assert MimeMagic.new('text/plain').text?
+    assert MimeMagic.new('text/html').text?
+    assert !MimeMagic.new('application/octet-stream').text?
+    assert !MimeMagic.new('image/png').text?
+  end
+
+  def test_child_of?
+    assert MimeMagic.new('text/html').child_of?('text/plain')
+    assert MimeMagic.new('text/x-java').child_of?('text/plain')
+  end
+
+  def test_extensions
+    assert_equal %w(htm html), MimeMagic.new('text/html').extensions
+  end
+
+  def test_by_extension
+    assert_equal 'text/html', MimeMagic.by_extension('html').to_s
+    assert_equal 'application/x-ruby', MimeMagic.by_extension('rb').to_s
+    assert_nil MimeMagic.by_extension('crazy')
+    assert_nil MimeMagic.by_extension('')
+  end
+
+  def test_by_magic
+    assert_equal 'application/x-executable', MimeMagic.by_magic(File.open('/bin/ls')).to_s
+    assert_equal 'application/x-sharedlib', MimeMagic.by_magic(File.open('/lib/libc.so.6')).to_s
+  end
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: minad-mimemagic
+version: !ruby/object:Gem::Version 
+  version: 0.1.1
+platform: ruby
+authors: 
+- Daniel Mendler
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-05-09 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.8.3
+    version: 
+description: 
+email: 
+- mail@daniel-mendler.de
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/mimemagic.rb
+- lib/mimemagic_tables.rb
+- test/test_mimemagic.rb
+- script/freedesktop.org.xml
+- script/generate-mime.rb
+- Rakefile
+has_rdoc: true
+homepage: 
+post_install_message: 
+rdoc_options: 
+- --main
+- README.txt
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: mimemagic
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Mime detection by extension or content
+test_files: 
+- test/test_mimemagic.rb